jianf/TrulyMEM-TrueHumanMEM
1
0
Fork 0
mirror of https://gitcode.com/JianFeeeee/TrulyMEM-TrueHumanMEM.git synced 2026-10-15 12:15:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: recall 查询所有会话历史 + 支持逗号分隔关键词

Browse Source
This commit is contained in:
root
2026-04-09 08:01:24 +08:00
parent 3ac49bb07a
commit 82a71c6dc9
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
openclaw_neo4j_demo.py
View File

@ -163,27 +163,31 @@ class Neo4jGraph:
def recall(self, query_intent: str, seed_entities: list = None, depth: int = 2, def recall(self, query_intent: str, seed_entities: list = None, depth: int = 2,
time_range: dict = None, session_filter: str = None) -> dict: time_range: dict = None, session_filter: str = None) -> dict:
with self.driver.session() as session: with self.driver.session() as session:
keywords = [w for w in query_intent.lower().split() if len(w) > 1] # 支持逗号分隔的多个关键词
keywords = [w.strip() for w in query_intent.replace(',', ' ').split() if len(w.strip()) > 0]
# 如果没有查询条件，返回空结果 # 如果没有查询条件，返回空结果
if not keywords and not seed_entities: if not keywords and not seed_entities:
return {"entities": [], "relations": [], "message": "无查询关键词"} return {"entities": [], "relations": [], "message": "无查询关键词"}
params = {"session_id": CURRENT_SESSION_ID} # 默认查询所有会话的历史（不只是当前会话）
# 只有明确指定 session_filter 才限制查询范围
params = {}
cond_parts = ["r.status = 'active'"] cond_parts = ["r.status = 'active'"]
if session_filter: if session_filter:
cond_parts.append("r.session_id = $target_session") cond_parts.append("r.session_id = $session_id")
params["target_session"] = session_filter params["session_id"] = session_filter
# 搜索多个相关关键词（不只是单个） # 搜索多个相关关键词
if keywords: if keywords:
keyword_conditions = [] keyword_conditions = []
for k in keywords: for k in keywords:
k_lower = k.lower()
# 搜索：实体名称、关系类型、目标实体 # 搜索：实体名称、关系类型、目标实体
keyword_conditions.append(f"toLower(e.name) CONTAINS '{k}'") keyword_conditions.append(f"toLower(e.name) CONTAINS '{k_lower}'")
keyword_conditions.append(f"toLower(t.name) CONTAINS '{k}'") keyword_conditions.append(f"toLower(t.name) CONTAINS '{k_lower}'")
keyword_conditions.append(f"toLower(r.type) CONTAINS '{k}'") keyword_conditions.append(f"toLower(r.type) CONTAINS '{k_lower}'")
cond_parts.append(f"({' OR '.join(keyword_conditions)})") cond_parts.append(f"({' OR '.join(keyword_conditions)})")
if seed_entities: if seed_entities: