From cfad82230f19f73399835715e7bcd36b96addaca Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Sat, 11 Oct 2025 15:09:29 +0800
Subject: [PATCH] fix sht

---
 lang_agent/rag/simple.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/lang_agent/rag/simple.py b/lang_agent/rag/simple.py
index b040d65..d33a507 100644
--- a/lang_agent/rag/simple.py
+++ b/lang_agent/rag/simple.py
@@ -44,6 +44,14 @@ class SimpleRag:
         
     @mcp.tool()
     def retrieve(self, query:str):
+        """检索与给定查询相关的文档，并将其序列化为字符串格式。
+        参数:
+            query (str): 用户输入的查询字符串。
+        返回:
+            Tuple[str, List[Document]]: 
+                - 序列化后的文档内容字符串，每个文档包含来源和内容。
+                - 检索到的 Document 对象列表。
+        该工具用于基于向量存储检索相关文档，适用于问答和知识检索场景。"""
         retrieved_docs:List[Document] = self.vec_store.search(query, search_kwargs={"k":3})
         serialized = "\n\n".join(
             (f"Source: {doc.metadata}\nContent: {doc.page_content}")
@@ -53,6 +61,7 @@ class SimpleRag:
 
 
 if __name__ == "__main__":
-    config = tyro.cli(SimpleRagConfig)
-    rag = SimpleRag(config)
+    # config = tyro.cli(SimpleRagConfig)
+    config = SimpleRagConfig()
+    rag:SimpleRag = config.setup()
     mcp.run(transport="stdio")
\ No newline at end of file