Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 import json
 from dataclasses import dataclass, field, asdict
-from typing import List, Dict, Any, Tuple, TypedDict
+from typing import List, Dict, Any, Tuple, TypedDict, Annotated
 
-# --- 1. 数据结构与核心逻辑 (无变化) ---
+# ... (所有数据结构和 Manager 类的代码保持不变) ...
 @dataclass
 class Entity: name: str; entityType: str; observations: List[str] = field(default_factory=list)
 @dataclass
@@ -12,7 +12,6 @@ class Relation: from_entity: str; to_entity: str; relationType: str
 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 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
@@ -37,7 +36,6 @@ class KnowledgeGraphManager:
 
 kg_manager = KnowledgeGraphManager()
 
-# --- 2. 为每个工具的 payload 定义 TypedDict (无变化) ---
 class CreateEntitiesPayload(TypedDict): entities: List[Dict]
 class CreateRelationsPayload(TypedDict): relations: List[Dict]
 class AddObservationsPayload(TypedDict): observations: List[Dict]
@@ -47,82 +45,44 @@ class DeleteRelationsPayload(TypedDict): relations: List[Dict]
 class SearchNodesPayload(TypedDict): query: str
 class OpenNodesPayload(TypedDict): names: List[str]
 
-# --- 3. API 工具函数定义 (同时使用 TypedDict 和 完整的 Docstring) ---
-def create_entities(payload: CreateEntitiesPayload) -> str:
-    """在知识图谱中创建多个新实体。
-
-    Args:
-        payload (CreateEntitiesPayload): 一个包含'entities'键的JSON对象。'entities'的值是一个实体对象的列表。
-    """
+def create_entities(payload: Annotated[CreateEntitiesPayload, "一个包含'entities'键的JSON对象。'entities'的值是一个实体对象的列表。"]) -> str:
+    """在知识图谱中创建多个新实体。"""
     new_entities = kg_manager.create_entities(payload['entities'])
     return json.dumps([asdict(e) for e in new_entities], indent=2)
 
-def create_relations(payload: CreateRelationsPayload) -> str:
-    """在知识图谱中的实体之间创建多个新关系。
-
-    Args:
-        payload (CreateRelationsPayload): 一个包含'relations'键的JSON对象。'relations'的值是一个关系对象的列表。
-    """
+def create_relations(payload: Annotated[CreateRelationsPayload, "一个包含'relations'键的JSON对象。'relations'的值是一个关系对象的列表。"]) -> str:
+    """在知识图谱中的实体之间创建多个新关系。"""
     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: AddObservationsPayload) -> str:
-    """向知识图谱中已存在的实体添加新的观察记录。
-
-    Args:
-        payload (AddObservationsPayload): 一个包含'observations'键的JSON对象。'observations'的值是一个观察记录对象的列表。
-    """
+# ... (所有其他API函数的定义也使用 Annotated[TypedDict, "description"] 格式) ...
+def add_observations(payload: Annotated[AddObservationsPayload, "一个包含'observations'键的JSON对象。'observations'的值是一个观察记录对象的列表。"]) -> str:
+    """向知识图ppu中已存在的实体添加新的观察记录。"""
     results = kg_manager.add_observations(payload['observations'])
     return json.dumps(results, indent=2)
-
-def delete_entities(payload: DeleteEntitiesPayload) -> str:
-    """从知识图谱中删除多个实体及其相关联的关系。
-
-    Args:
-        payload (DeleteEntitiesPayload): 一个包含'entityNames'键的JSON对象。'entityNames'的值是一个实体名称的列表。
-    """
+def delete_entities(payload: Annotated[DeleteEntitiesPayload, "一个包含'entityNames'键的JSON对象。'entityNames'的值是一个实体名称的列表。"]) -> str:
+    """从知识图谱中删除多个实体及其相关联的关系。"""
     kg_manager.delete_entities(payload['entityNames'])
     return "Entities and their relations deleted successfully."
-
-def delete_observations(payload: DeleteObservationsPayload) -> str:
-    """从知识图谱中的实体删除特定的观察记录。
-
-    Args:
-        payload (DeleteObservationsPayload): 一个包含'deletions'键的JSON对象。'deletions'的值是一个删除指令的列表。
-    """
+def delete_observations(payload: Annotated[DeleteObservationsPayload, "一个包含'deletions'键的JSON对象。'deletions'的值是一个删除指令的列表。"]) -> str:
+    """从知识图谱中的实体删除特定的观察记录。"""
     kg_manager.delete_observations(payload['deletions'])
     return "Observations deleted successfully."
-
-def delete_relations(payload: DeleteRelationsPayload) -> str:
-    """从知识图谱中删除多个关系。
-
-    Args:
-        payload (DeleteRelationsPayload): 一个包含'relations'键的JSON对象。'relations'的值是一个要删除的关系对象的列表。
-    """
+def delete_relations(payload: Annotated[DeleteRelationsPayload, "一个包含'relations'键的JSON对象。'relations'的值是一个要删除的关系对象的列表。"]) -> str:
+    """从知识图谱中删除多个关系。"""
     kg_manager.delete_relations(payload['relations'])
     return "Relations deleted successfully."
-
 def read_graph() -> Dict:
     """读取并返回整个知识图谱。此函数不需要参数。"""
     return kg_manager.read_graph()
-
-def search_nodes(payload: SearchNodesPayload) -> Dict:
-    """根据查询词在知识图谱中搜索节点。
-
-    Args:
-        payload (SearchNodesPayload): 一个包含'query'键的JSON对象。'query'的值是用于搜索的字符串。
-    """
+def search_nodes(payload: Annotated[SearchNodesPayload, "一个包含'query'键的JSON对象。'query'的值是用于搜索的字符串。"]) -> Dict:
+    """根据查询词在知识图谱中搜索节点。"""
     return kg_manager.search_nodes(payload['query'])
-
-def open_nodes(payload: OpenNodesPayload) -> Dict:
-    """通过名称打开知识图谱中的特定节点。
-
-    Args:
-        payload (OpenNodesPayload): 一个包含'names'键的JSON对象。'names'的值是一个要检索的实体名称的列表。
-    """
+def open_nodes(payload: Annotated[OpenNodesPayload, "一个包含'names'键的JSON对象。'names'的值是一个要检索的实体名称的列表。"]) -> Dict:
+    """通过名称打开知识图谱中的特定节点。"""
     return kg_manager.open_nodes(payload['names'])
 
-# --- 4. Gradio Headless 应用构建与 API 注册 (无变化) ---
+
 with gr.Blocks() as app:
     gr.Markdown("MCP Server is running. This UI is intentionally blank.", visible=True)
     with gr.Row(visible=False):
@@ -138,23 +98,4 @@ 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)
-```### 核心修改
-
-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` 的描述，并将其显示在工具面板中。
-
-这下就真的万无一失了。再次为之前的曲折过程道歉。
+    app.launch(mcp_server=True)