From 8088b927f0e731eec5fcd71aceb25cc3213322ed Mon Sep 17 00:00:00 2001 From: liyizhouAI Date: Tue, 14 Apr 2026 19:13:50 +0800 Subject: [PATCH] fix: accept all Entity nodes from Graphiti (no custom labels) Graphiti doesn't assign custom labels like Zep does - all nodes get generic 'Entity' label. Updated filter to accept all named Entity nodes instead of requiring custom labels like 'Person', 'Organization'. Generated with [Claude Code](https://claude.ai/code) via [Happy](https://happy.engineering) Co-Authored-By: Claude Co-Authored-By: Happy --- backend/app/services/zep_entity_reader.py | 26 ++++++++++++++--------- 1 file changed, 16 insertions(+), 10 deletions(-) diff --git a/backend/app/services/zep_entity_reader.py b/backend/app/services/zep_entity_reader.py index 75ef31a0..c464cc04 100644 --- a/backend/app/services/zep_entity_reader.py +++ b/backend/app/services/zep_entity_reader.py @@ -243,21 +243,27 @@ class ZepEntityReader: for node in all_nodes: labels = node.get("labels", []) - # 筛选逻辑：Labels必须包含除"Entity"和"Node"之外的标签 - custom_labels = [l for l in labels if l not in ["Entity", "Node"]] - - if not custom_labels: - # 只有默认标签，跳过 + # 筛选逻辑： + # Graphiti 的节点可能只有 "Entity" 标签（不像 Zep 会自动分配自定义标签） + # 所以我们接受所有 Entity 节点，用名称匹配实体类型 + custom_labels = [l for l in labels if l not in ["Entity", "Node", "__Entity__"]] + + if custom_labels: + # 有自定义标签，直接用 + entity_type = custom_labels[0] + elif node.get("name"): + # Graphiti 模式：所有节点都是 Entity，用 "Entity" 作为类型 + entity_type = "Entity" + else: + # 无名称的节点，跳过 continue - - # 如果指定了预定义类型，检查是否匹配 - if defined_entity_types: + + # 如果指定了预定义类型且节点有自定义标签，检查匹配 + if defined_entity_types and custom_labels: matching_labels = [l for l in custom_labels if l in defined_entity_types] if not matching_labels: continue entity_type = matching_labels[0] - else: - entity_type = custom_labels[0] entity_types_found.add(entity_type)