From e3795fe1ed2def9751bc797e4ab97d390a7f6bb3 Mon Sep 17 00:00:00 2001 From: lanceyq <1982376970@qq.com> Date: Fri, 20 Mar 2026 20:43:29 +0800 Subject: [PATCH] [changes] Remove the unused config_id --- api/app/core/memory/agent/utils/write_tools.py | 1 - .../storage_services/clustering_engine/label_propagation.py | 2 -- api/app/repositories/neo4j/graph_saver.py | 6 ++---- 3 files changed, 2 insertions(+), 7 deletions(-) diff --git a/api/app/core/memory/agent/utils/write_tools.py b/api/app/core/memory/agent/utils/write_tools.py index 02aa1b44..b0c68b19 100644 --- a/api/app/core/memory/agent/utils/write_tools.py +++ b/api/app/core/memory/agent/utils/write_tools.py @@ -174,7 +174,6 @@ async def write( # 写入成功后,异步触发聚类(不阻塞写入响应) schedule_clustering_after_write( all_entity_nodes, - config_id=config_id, llm_model_id=str(memory_config.llm_model_id) if memory_config.llm_model_id else None, embedding_model_id=str(memory_config.embedding_model_id) if memory_config.embedding_model_id else None, ) diff --git a/api/app/core/memory/storage_services/clustering_engine/label_propagation.py b/api/app/core/memory/storage_services/clustering_engine/label_propagation.py index 21257f2e..47df8d56 100644 --- a/api/app/core/memory/storage_services/clustering_engine/label_propagation.py +++ b/api/app/core/memory/storage_services/clustering_engine/label_propagation.py @@ -69,13 +69,11 @@ class LabelPropagationEngine: def __init__( self, connector: Neo4jConnector, - config_id: Optional[str] = None, llm_model_id: Optional[str] = None, embedding_model_id: Optional[str] = None, ): self.connector = connector self.repo = CommunityRepository(connector) - self.config_id = config_id self.llm_model_id = llm_model_id self.embedding_model_id = embedding_model_id diff --git a/api/app/repositories/neo4j/graph_saver.py b/api/app/repositories/neo4j/graph_saver.py index 29e337f1..d5750692 100644 --- a/api/app/repositories/neo4j/graph_saver.py +++ b/api/app/repositories/neo4j/graph_saver.py @@ -304,7 +304,6 @@ async def save_dialog_and_statements_to_neo4j( def schedule_clustering_after_write( entity_nodes: List, - config_id: Optional[str] = None, llm_model_id: Optional[str] = None, embedding_model_id: Optional[str] = None, ) -> None: @@ -325,13 +324,12 @@ def schedule_clustering_after_write( end_user_id = entity_nodes[0].end_user_id new_entity_ids = [e.id for e in entity_nodes] logger.info(f"[Clustering] 准备触发聚类,实体数: {len(new_entity_ids)}, end_user_id: {end_user_id}") - asyncio.create_task(_trigger_clustering(new_entity_ids, end_user_id, config_id=config_id, llm_model_id=llm_model_id, embedding_model_id=embedding_model_id)) + asyncio.create_task(_trigger_clustering(new_entity_ids, end_user_id, llm_model_id=llm_model_id, embedding_model_id=embedding_model_id)) async def _trigger_clustering( new_entity_ids: List[str], end_user_id: str, - config_id: Optional[str] = None, llm_model_id: Optional[str] = None, embedding_model_id: Optional[str] = None, ) -> None: @@ -343,7 +341,7 @@ async def _trigger_clustering( from app.core.memory.storage_services.clustering_engine import LabelPropagationEngine logger.info(f"[Clustering] 开始聚类,end_user_id={end_user_id}, 实体数={len(new_entity_ids)}") connector = Neo4jConnector() - engine = LabelPropagationEngine(connector, config_id=config_id, llm_model_id=llm_model_id, embedding_model_id=embedding_model_id) + engine = LabelPropagationEngine(connector, llm_model_id=llm_model_id, embedding_model_id=embedding_model_id) await engine.run(end_user_id=end_user_id, new_entity_ids=new_entity_ids) logger.info(f"[Clustering] 聚类完成,end_user_id={end_user_id}") except Exception as e: