gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 移除LLMClient依赖，改为使用get_embedding函数获取嵌入

Browse Source
This commit is contained in:
墨梓柒
2025-07-15 15:50:56 +08:00
parent 956232d493
commit 2b76dc2e21
1 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/chat/knowledge/embedding_store.py
View File

@@ -26,7 +26,7 @@ from rich.progress import (
TextColumn, TextColumn,
) )
from src.manager.local_store_manager import local_storage from src.manager.local_store_manager import local_storage
from src.llm_models.utils_model import LLMRequest from src.chat.utils.utils import get_embedding
install(extra_lines=3) install(extra_lines=3)
@@ -89,9 +89,8 @@ class EmbeddingStoreItem:
class EmbeddingStore: class EmbeddingStore:
def __init__(self, llm_client: LLMClient, namespace: str, dir_path: str): def __init__(self, namespace: str, dir_path: str):
self.namespace = namespace self.namespace = namespace
self.llm_client = llm_client
self.dir = dir_path self.dir = dir_path
self.embedding_file_path = f"{dir_path}/{namespace}.parquet" self.embedding_file_path = f"{dir_path}/{namespace}.parquet"
self.index_file_path = f"{dir_path}/{namespace}.index" self.index_file_path = f"{dir_path}/{namespace}.index"
@@ -103,7 +102,7 @@ class EmbeddingStore:
self.idx2hash = None self.idx2hash = None
def _get_embedding(self, s: str) -> List[float]: def _get_embedding(self, s: str) -> List[float]:
return self.llm_client.send_embedding_request(global_config["embedding"]["model"], s) return get_embedding(s)
def get_test_file_path(self): def get_test_file_path(self):
return EMBEDDING_TEST_FILE return EMBEDDING_TEST_FILE
@@ -298,17 +297,14 @@ class EmbeddingStore:
class EmbeddingManager: class EmbeddingManager:
def __init__(self, llm_client: LLMClient): def __init__(self, llm_client: LLMClient):
self.paragraphs_embedding_store = EmbeddingStore( self.paragraphs_embedding_store = EmbeddingStore(
llm_client,
local_storage['pg_namespace'], local_storage['pg_namespace'],
EMBEDDING_DATA_DIR_STR, EMBEDDING_DATA_DIR_STR,
) )
self.entities_embedding_store = EmbeddingStore( self.entities_embedding_store = EmbeddingStore(
llm_client,
local_storage['pg_namespace'], local_storage['pg_namespace'],
EMBEDDING_DATA_DIR_STR, EMBEDDING_DATA_DIR_STR,
) )
self.relation_embedding_store = EmbeddingStore( self.relation_embedding_store = EmbeddingStore(
llm_client,
local_storage['pg_namespace'], local_storage['pg_namespace'],
EMBEDDING_DATA_DIR_STR, EMBEDDING_DATA_DIR_STR,
) )