Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import json
 from dataclasses import dataclass, field, asdict
-from typing import List, Dict, Any, Tuple, TypedDict # <--- 导入 TypedDict
+from typing import List, Dict, Any, Tuple, TypedDict, Annotated # <--- 导入 Annotated
 
 # --- 1. 数据结构与核心逻辑 (无变化) ---
 @dataclass
@@ -41,51 +41,43 @@ 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]
-class DeleteEntitiesPayload(TypedDict):
-    entityNames: List[str]
-class DeleteObservationsPayload(TypedDict):
-    deletions: List[Dict]
-class DeleteRelationsPayload(TypedDict):
-    relations: List[Dict]
-class SearchNodesPayload(TypedDict):
-    query: str
-class OpenNodesPayload(TypedDict):
-    names: List[str]
-
-# --- 3. API 工具函数定义 (使用 TypedDict 更新了函数签名) ---
-def create_entities(payload: CreateEntitiesPayload) -> str:
+# --- 2. 为每个工具的 payload 定义 TypedDict (无变化) ---
+class CreateEntitiesPayload(TypedDict): entities: List[Dict]
+class CreateRelationsPayload(TypedDict): relations: List[Dict]
+class AddObservationsPayload(TypedDict): observations: List[Dict]
+class DeleteEntitiesPayload(TypedDict): entityNames: List[str]
+class DeleteObservationsPayload(TypedDict): deletions: List[Dict]
+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:
     """在知识图谱中创建多个新实体。"""
     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:
+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:
+def add_observations(payload: Annotated[AddObservationsPayload, "一个包含'observations'键的JSON对象。'observations'的值是一个观察记录对象的列表。"]) -> str:
     """向知识图谱中已存在的实体添加新的观察记录。"""
     results = kg_manager.add_observations(payload['observations'])
     return json.dumps(results, indent=2)
 
-def delete_entities(payload: DeleteEntitiesPayload) -> str:
+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:
+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:
+def delete_relations(payload: Annotated[DeleteRelationsPayload, "一个包含'relations'键的JSON对象。'relations'的值是一个要删除的关系对象的列表。"]) -> str:
     """从知识图谱中删除多个关系。"""
     kg_manager.delete_relations(payload['relations'])
     return "Relations deleted successfully."
@@ -94,23 +86,19 @@ def read_graph() -> Dict:
     """读取并返回整个知识图谱。此函数不需要参数。"""
     return kg_manager.read_graph()
 
-def search_nodes(payload: SearchNodesPayload) -> Dict:
+def search_nodes(payload: Annotated[SearchNodesPayload, "一个包含'query'键的JSON对象。'query'的值是用于搜索的字符串。"]) -> Dict:
     """根据查询词在知识图谱中搜索节点。"""
     return kg_manager.search_nodes(payload['query'])
 
-def open_nodes(payload: OpenNodesPayload) -> Dict:
+def open_nodes(payload: Annotated[OpenNodesPayload, "一个包含'names'键的JSON对象。'names'的值是一个要检索的实体名称的列表。"]) -> Dict:
     """通过名称打开知识图谱中的特定节点。"""
     return kg_manager.open_nodes(payload['names'])
 
-# --- 4. Gradio Headless 应用构建与 API 注册 ---
+# --- 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):
-        json_input = gr.JSON(value={})
-        generic_output = gr.JSON()
-        dummy_btn = gr.Button()
-
+        json_input = gr.JSON(value={}); generic_output = gr.JSON(); dummy_btn = gr.Button()
     dummy_btn.click(fn=create_entities, inputs=json_input, outputs=generic_output, api_name="create_entities")
     dummy_btn.click(fn=create_relations, inputs=json_input, outputs=generic_output, api_name="create_relations")
     dummy_btn.click(fn=add_observations, inputs=json_input, outputs=generic_output, api_name="add_observations")