Parcourir la source

修改方法名

xueyiming il y a 2 semaines
Parent
commit
00738070a2
1 fichiers modifiés avec 6 ajouts et 6 suppressions
  1. 6 6
      mcp_server/server.py

+ 6 - 6
mcp_server/server.py

@@ -35,8 +35,8 @@ def create_mcp_server() -> Server:
     async def call_tool(name: str, arguments: Dict[str, Any]) -> List[types.TextContent]:
         """处理工具调用"""
         # ctx = app.request_context
-        if name == "chat-detail":
-            data = await chat_detail(arguments["query_text"])
+        if name == "rag-search":
+            data = await rag_search(arguments["query_text"])
             result = json.dumps(data, ensure_ascii=False, indent=2)
         else:
             raise ValueError(f"Unknown tool: {name}")
@@ -46,7 +46,7 @@ def create_mcp_server() -> Server:
     async def list_tools() -> List[types.Tool]:
         return [
             types.Tool(
-                name="chat-detail",
+                name="rag-search",
                 title = 'RAG搜索',
                 description="搜索内容并生成总结",
                 inputSchema={
@@ -66,7 +66,7 @@ def create_mcp_server() -> Server:
     return app
 
 
-async def chat_detail(query_text: str) :
+async def rag_search(query_text: str) :
     dataset_id_strs = "11,12"
     dataset_ids = dataset_id_strs.split(",")
     search_type = "hybrid"
@@ -110,13 +110,13 @@ async def chat_detail(query_text: str) :
     data = {
         "result": chat_res["summary"],
         "status": chat_res["status"],
-        "metaData": res,
+        "score": chat_res["score"],
     }
 
     await chat_result_mapper.insert_chat_result(
         query_text,
         dataset_id_strs,
-        json.dumps(data, ensure_ascii=False),
+        json.dumps(res, ensure_ascii=False),
         chat_res["summary"],
         chat_res["relevance_score"],
         chat_res["status"],