diff --git a/src/chat/knowledge/src/__init__.py b/src/__init__.py similarity index 100% rename from src/chat/knowledge/src/__init__.py rename to src/__init__.py diff --git a/src/chat/knowledge/src/embedding_store.py b/src/chat/knowledge/embedding_store.py similarity index 99% rename from src/chat/knowledge/src/embedding_store.py rename to src/chat/knowledge/embedding_store.py index cf139ad3a..b4d1f9d0f 100644 --- a/src/chat/knowledge/src/embedding_store.py +++ b/src/chat/knowledge/embedding_store.py @@ -12,7 +12,7 @@ import faiss from .llm_client import LLMClient from .lpmmconfig import ENT_NAMESPACE, PG_NAMESPACE, REL_NAMESPACE, global_config -from .utils.hash import get_sha256 +from src.chat.knowledge.utils import get_sha256 from .global_logger import logger from rich.traceback import install from rich.progress import ( diff --git a/src/chat/knowledge/src/global_logger.py b/src/chat/knowledge/global_logger.py similarity index 100% rename from src/chat/knowledge/src/global_logger.py rename to src/chat/knowledge/global_logger.py diff --git a/src/chat/knowledge/src/ie_process.py b/src/chat/knowledge/ie_process.py similarity index 98% rename from src/chat/knowledge/src/ie_process.py rename to src/chat/knowledge/ie_process.py index ddc5eb023..f68a848d2 100644 --- a/src/chat/knowledge/src/ie_process.py +++ b/src/chat/knowledge/ie_process.py @@ -6,7 +6,7 @@ from .global_logger import logger from . import prompt_template from .lpmmconfig import global_config, INVALID_ENTITY from .llm_client import LLMClient -from .utils.json_fix import new_fix_broken_generated_json +from src.chat.knowledge.utils.json_fix import new_fix_broken_generated_json def _entity_extract(llm_client: LLMClient, paragraph: str) -> List[str]: diff --git a/src/chat/knowledge/src/kg_manager.py b/src/chat/knowledge/kg_manager.py similarity index 99% rename from src/chat/knowledge/src/kg_manager.py rename to src/chat/knowledge/kg_manager.py index ad5df0923..f9d600f89 100644 --- a/src/chat/knowledge/src/kg_manager.py +++ b/src/chat/knowledge/kg_manager.py @@ -18,7 +18,7 @@ from rich.progress import ( from quick_algo import di_graph, pagerank -from .utils.hash import get_sha256 +from src.chat.knowledge.utils import get_sha256 from .embedding_store import EmbeddingManager, EmbeddingStoreItem from .lpmmconfig import ( ENT_NAMESPACE, diff --git a/src/chat/knowledge/src/llm_client.py b/src/chat/knowledge/llm_client.py similarity index 100% rename from src/chat/knowledge/src/llm_client.py rename to src/chat/knowledge/llm_client.py diff --git a/src/chat/knowledge/src/lpmmconfig.py b/src/chat/knowledge/lpmmconfig.py similarity index 100% rename from src/chat/knowledge/src/lpmmconfig.py rename to src/chat/knowledge/lpmmconfig.py diff --git a/src/chat/knowledge/src/mem_active_manager.py b/src/chat/knowledge/mem_active_manager.py similarity index 95% rename from src/chat/knowledge/src/mem_active_manager.py rename to src/chat/knowledge/mem_active_manager.py index 3998c0664..9c6e65e9a 100644 --- a/src/chat/knowledge/src/mem_active_manager.py +++ b/src/chat/knowledge/mem_active_manager.py @@ -1,7 +1,7 @@ from .lpmmconfig import global_config from .embedding_store import EmbeddingManager from .llm_client import LLMClient -from .utils.dyn_topk import dyn_select_top_k +from src.chat.knowledge.utils import dyn_select_top_k class MemoryActiveManager: diff --git a/src/chat/knowledge/src/open_ie.py b/src/chat/knowledge/open_ie.py similarity index 100% rename from src/chat/knowledge/src/open_ie.py rename to src/chat/knowledge/open_ie.py diff --git a/src/chat/knowledge/src/prompt_template.py b/src/chat/knowledge/prompt_template.py similarity index 100% rename from src/chat/knowledge/src/prompt_template.py rename to src/chat/knowledge/prompt_template.py diff --git a/src/chat/knowledge/src/qa_manager.py b/src/chat/knowledge/qa_manager.py similarity index 98% rename from src/chat/knowledge/src/qa_manager.py rename to src/chat/knowledge/qa_manager.py index b6bbd1207..e76f29d7b 100644 --- a/src/chat/knowledge/src/qa_manager.py +++ b/src/chat/knowledge/qa_manager.py @@ -8,7 +8,7 @@ from .embedding_store import EmbeddingManager from .llm_client import LLMClient from .kg_manager import KGManager from .lpmmconfig import global_config -from .utils.dyn_topk import dyn_select_top_k +from src.chat.knowledge.utils import dyn_select_top_k MAX_KNOWLEDGE_LENGTH = 10000 # 最大知识长度 diff --git a/src/chat/knowledge/src/raw_processing.py b/src/chat/knowledge/raw_processing.py similarity index 96% rename from src/chat/knowledge/src/raw_processing.py rename to src/chat/knowledge/raw_processing.py index a333ef996..ffdcf814b 100644 --- a/src/chat/knowledge/src/raw_processing.py +++ b/src/chat/knowledge/raw_processing.py @@ -3,7 +3,7 @@ import os from .global_logger import logger from .lpmmconfig import global_config -from .utils.hash import get_sha256 +from src.chat.knowledge.utils import get_sha256 def load_raw_data(path: str = None) -> tuple[list[str], list[str]]: diff --git a/src/chat/knowledge/src/utils/__init__.py b/src/chat/knowledge/utils/__init__.py similarity index 100% rename from src/chat/knowledge/src/utils/__init__.py rename to src/chat/knowledge/utils/__init__.py diff --git a/src/chat/knowledge/src/utils/dyn_topk.py b/src/chat/knowledge/utils/dyn_topk.py similarity index 100% rename from src/chat/knowledge/src/utils/dyn_topk.py rename to src/chat/knowledge/utils/dyn_topk.py diff --git a/src/chat/knowledge/src/utils/hash.py b/src/chat/knowledge/utils/hash.py similarity index 100% rename from src/chat/knowledge/src/utils/hash.py rename to src/chat/knowledge/utils/hash.py diff --git a/src/chat/knowledge/src/utils/json_fix.py b/src/chat/knowledge/utils/json_fix.py similarity index 100% rename from src/chat/knowledge/src/utils/json_fix.py rename to src/chat/knowledge/utils/json_fix.py diff --git a/src/chat/knowledge/src/utils/visualize_graph.py b/src/chat/knowledge/utils/visualize_graph.py similarity index 100% rename from src/chat/knowledge/src/utils/visualize_graph.py rename to src/chat/knowledge/utils/visualize_graph.py