Update app.py

app.py

@@ -1,23 +1,19 @@
 import gradio as gr
 import json
 from dataclasses import dataclass, field, asdict
-from typing import List, Dict, Any, Tuple, TypedDict, Annotated # <--- 导入 Annotated
+from typing import List, Dict, Any, Tuple, TypedDict
 
 # --- 1. 数据结构与核心逻辑 (无变化) ---
 @dataclass
-class Entity:
-    name: str; entityType: str; observations: List[str] = field(default_factory=list)
+class Entity: name: str; entityType: str; observations: List[str] = field(default_factory=list)
 @dataclass
-class Relation:
-    from_entity: str; to_entity: str; relationType: str
+class Relation: from_entity: str; to_entity: str; relationType: str
 @dataclass
-class KnowledgeGraph:
-    entities: Dict[str, Entity] = field(default_factory=dict); relations: List[Relation] = field(default_factory=list)
+class KnowledgeGraph: entities: Dict[str, Entity] = field(default_factory=dict); relations: List[Relation] = field(default_factory=list)
 
 class KnowledgeGraphManager:
     # ... (Manager 的所有内部方法保持不变) ...
-    def __init__(self):
-        self.graph = KnowledgeGraph(); print("Initialized a new, empty in-memory knowledge graph.")
+    def __init__(self): self.graph = KnowledgeGraph(); print("Initialized a new, empty in-memory knowledge graph.")
     def create_entities(self, entities_data: List[Dict]) -> List[Entity]:
         new_entities = []; [ (new_entity := Entity(**e_data), self.graph.entities.update({e_data['name']: new_entity}), new_entities.append(new_entity)) for e_data in entities_data if e_data['name'] not in self.graph.entities]; return new_entities
     def create_relations(self, relations_data: List[Dict]) -> List[Relation]:
@@ -51,34 +47,58 @@ class DeleteRelationsPayload(TypedDict): relations: List[Dict]
 class SearchNodesPayload(TypedDict): query: str
 class OpenNodesPayload(TypedDict): names: List[str]
 
-# --- 3. API 工具函数定义 (使用 Annotated[TypedDict, "description"]) ---
-def create_entities(payload: Annotated[CreateEntitiesPayload, "一个包含'entities'键的JSON对象。'entities'的值是一个实体对象的列表。"]) -> str:
-    """在知识图谱中创建多个新实体。"""
+# --- 3. API 工具函数定义 (同时使用 TypedDict 和 完整的 Docstring) ---
+def create_entities(payload: CreateEntitiesPayload) -> str:
+    """在知识图谱中创建多个新实体。
+
+    Args:
+        payload (CreateEntitiesPayload): 一个包含'entities'键的JSON对象。'entities'的值是一个实体对象的列表。
+    """
     new_entities = kg_manager.create_entities(payload['entities'])
     return json.dumps([asdict(e) for e in new_entities], indent=2)
 
-def create_relations(payload: Annotated[CreateRelationsPayload, "一个包含'relations'键的JSON对象。'relations'的值是一个关系对象的列表。"]) -> str:
-    """在知识图谱中的实体之间创建多个新关系。"""
+def create_relations(payload: CreateRelationsPayload) -> str:
+    """在知识图谱中的实体之间创建多个新关系。
+
+    Args:
+        payload (CreateRelationsPayload): 一个包含'relations'键的JSON对象。'relations'的值是一个关系对象的列表。
+    """
     new_relations = kg_manager.create_relations(payload['relations'])
     return json.dumps([{"from": r.from_entity, "to": r.to_entity, "relationType": r.relationType} for r in new_relations], indent=2)
 
-def add_observations(payload: Annotated[AddObservationsPayload, "一个包含'observations'键的JSON对象。'observations'的值是一个观察记录对象的列表。"]) -> str:
-    """向知识图谱中已存在的实体添加新的观察记录。"""
+def add_observations(payload: AddObservationsPayload) -> str:
+    """向知识图谱中已存在的实体添加新的观察记录。
+
+    Args:
+        payload (AddObservationsPayload): 一个包含'observations'键的JSON对象。'observations'的值是一个观察记录对象的列表。
+    """
     results = kg_manager.add_observations(payload['observations'])
     return json.dumps(results, indent=2)
 
-def delete_entities(payload: Annotated[DeleteEntitiesPayload, "一个包含'entityNames'键的JSON对象。'entityNames'的值是一个实体名称的列表。"]) -> str:
-    """从知识图谱中删除多个实体及其相关联的关系。"""
+def delete_entities(payload: DeleteEntitiesPayload) -> str:
+    """从知识图谱中删除多个实体及其相关联的关系。
+
+    Args:
+        payload (DeleteEntitiesPayload): 一个包含'entityNames'键的JSON对象。'entityNames'的值是一个实体名称的列表。
+    """
     kg_manager.delete_entities(payload['entityNames'])
     return "Entities and their relations deleted successfully."
 
-def delete_observations(payload: Annotated[DeleteObservationsPayload, "一个包含'deletions'键的JSON对象。'deletions'的值是一个删除指令的列表。"]) -> str:
-    """从知识图谱中的实体删除特定的观察记录。"""
+def delete_observations(payload: DeleteObservationsPayload) -> str:
+    """从知识图谱中的实体删除特定的观察��录。
+
+    Args:
+        payload (DeleteObservationsPayload): 一个包含'deletions'键的JSON对象。'deletions'的值是一个删除指令的列表。
+    """
     kg_manager.delete_observations(payload['deletions'])
     return "Observations deleted successfully."
 
-def delete_relations(payload: Annotated[DeleteRelationsPayload, "一个包含'relations'键的JSON对象。'relations'的值是一个要删除的关系对象的列表。"]) -> str:
-    """从知识图谱中删除多个关系。"""
+def delete_relations(payload: DeleteRelationsPayload) -> str:
+    """从知识图谱中删除多个关系。
+
+    Args:
+        payload (DeleteRelationsPayload): 一个包含'relations'键的JSON对象。'relations'的值是一个要删除的关系对象的列表。
+    """
     kg_manager.delete_relations(payload['relations'])
     return "Relations deleted successfully."
 
@@ -86,12 +106,20 @@ def read_graph() -> Dict:
     """读取并返回整个知识图谱。此函数不需要参数。"""
     return kg_manager.read_graph()
 
-def search_nodes(payload: Annotated[SearchNodesPayload, "一个包含'query'键的JSON对象。'query'的值是用于搜索的字符串。"]) -> Dict:
-    """根据查询词在知识图谱中搜索节点。"""
+def search_nodes(payload: SearchNodesPayload) -> Dict:
+    """根据查询词在知识图谱中搜索节点。
+
+    Args:
+        payload (SearchNodesPayload): 一个包含'query'键的JSON对象。'query'的值是用于搜索的字符串。
+    """
     return kg_manager.search_nodes(payload['query'])
 
-def open_nodes(payload: Annotated[OpenNodesPayload, "一个包含'names'键的JSON对象。'names'的值是一个要检索的实体名称的列表。"]) -> Dict:
-    """通过名称打开知识图谱中的特定节点。"""
+def open_nodes(payload: OpenNodesPayload) -> Dict:
+    """通过名称打开知识图谱中的特定节点。
+
+    Args:
+        payload (OpenNodesPayload): 一个包含'names'键的JSON对象。'names'的值是一个要检索的实体名称的列表。
+    """
     return kg_manager.open_nodes(payload['names'])
 
 # --- 4. Gradio Headless 应用构建与 API 注册 (无变化) ---
@@ -110,4 +138,23 @@ with gr.Blocks() as app:
     dummy_btn.click(fn=open_nodes, inputs=json_input, outputs=generic_output, api_name="open_nodes")
 
 if __name__ == "__main__":
-    app.launch(mcp_server=True)
+    app.launch(mcp_server=True)
+```### 核心修改
+
+1.  **移除了 `Annotated`**: 我们不再需要它，因为它与你环境中的 Gradio 版本不完全兼容。
+2.  **为每个函数添加了完整的文档字符串**: 这是最关键的修改。现在，每个需要参数的函数都包含一个 `Args:` 部分，明确地描述了 `payload` 参数。
+    ```python
+    def create_entities(payload: CreateEntitiesPayload) -> str:
+        """在知识图谱中创建多个新实体。
+
+        Args:
+            payload (CreateEntitiesPayload): 一个包含'entities'键的JSON对象。'entities'的值是一个实体对象的列表。
+        """
+        # ...
+    ```
+
+现在，当你运行这个版本的代码时，Gradio 的 MCP 服务器将会：
+1.  **看到 `payload: CreateEntitiesPayload`**，并生成正确的、严格的验证模式，确保工具可以正常工作。
+2.  **解析文档字符串**，找到 `Args:` 部分中对 `payload` 的描述，并将其显示在工具面板中。
+
+这下就真的万无一失了。再次为之前的曲折过程道歉。