RomanNum3ral
/
MicroFish
mirror of https://github.com/666ghj/MiroFish.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge fe8949ca1c into 96096ea0ff

This commit is contained in:
Octopus 2026-05-28 17:41:08 -04:00 committed by GitHub
parent 96096ea0ff fe8949ca1c
commit 6c1fee1606
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 24 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
backend/app/services/graph_builder.py
View File

@ -233,8 +233,13 @@ class GraphBuilderService:
annotations = {} annotations = {}
for attr_def in entity_def.get("attributes", []): for attr_def in entity_def.get("attributes", []):
attr_name = safe_attr_name(attr_def["name"]) # 使用安全名称 # LLM may return attributes as strings instead of dicts
attr_desc = attr_def.get("description", attr_name) if isinstance(attr_def, str):
attr_name = safe_attr_name(attr_def)
attr_desc = attr_def
else:
attr_name = safe_attr_name(attr_def["name"]) # 使用安全名称
attr_desc = attr_def.get("description", attr_name)
# Zep API 需要 Field 的 description，这是必需的 # Zep API 需要 Field 的 description，这是必需的
attrs[attr_name] = Field(description=attr_desc, default=None) attrs[attr_name] = Field(description=attr_desc, default=None)
annotations[attr_name] = Optional[EntityText] # 类型注解 annotations[attr_name] = Optional[EntityText] # 类型注解
@ -257,8 +262,13 @@ class GraphBuilderService:
annotations = {} annotations = {}
for attr_def in edge_def.get("attributes", []): for attr_def in edge_def.get("attributes", []):
attr_name = safe_attr_name(attr_def["name"]) # 使用安全名称 # LLM may return attributes as strings instead of dicts
attr_desc = attr_def.get("description", attr_name) if isinstance(attr_def, str):
attr_name = safe_attr_name(attr_def)
attr_desc = attr_def
else:
attr_name = safe_attr_name(attr_def["name"]) # 使用安全名称
attr_desc = attr_def.get("description", attr_name)
# Zep API 需要 Field 的 description，这是必需的 # Zep API 需要 Field 的 description，这是必需的
attrs[attr_name] = Field(description=attr_desc, default=None) attrs[attr_name] = Field(description=attr_desc, default=None)
annotations[attr_name] = Optional[str] # 边属性用str类型 annotations[attr_name] = Optional[str] # 边属性用str类型

10
backend/app/services/ontology_generator.py
View File

@ -298,6 +298,11 @@ class OntologyGenerator:
entity_name_map[original_name] = entity["name"] entity_name_map[original_name] = entity["name"]
if "attributes" not in entity: if "attributes" not in entity:
entity["attributes"] = [] entity["attributes"] = []
# Normalize attributes: LLM may return strings instead of dicts
entity["attributes"] = [
attr if isinstance(attr, dict) else {"name": attr, "type": "text", "description": attr}
for attr in entity["attributes"]
]
if "examples" not in entity: if "examples" not in entity:
entity["examples"] = [] entity["examples"] = []
# 确保description不超过100字符 # 确保description不超过100字符
@ -322,6 +327,11 @@ class OntologyGenerator:
edge["source_targets"] = [] edge["source_targets"] = []
if "attributes" not in edge: if "attributes" not in edge:
edge["attributes"] = [] edge["attributes"] = []
# Normalize attributes: LLM may return strings instead of dicts
edge["attributes"] = [
attr if isinstance(attr, dict) else {"name": attr, "type": "text", "description": attr}
for attr in edge["attributes"]
]
if len(edge.get("description", "")) > 100: if len(edge.get("description", "")) > 100:
edge["description"] = edge["description"][:97] + "..." edge["description"] = edge["description"][:97] + "..."