Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bug fix

Browse Source
This commit is contained in:
goulustis
2025-10-13 21:44:04 +08:00
parent 32d8ae9b29
commit c13984e042
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
lang_agent/rag/simple.py
View File

@@ -32,8 +32,8 @@ class SimpleRagConfig(ToolConfig):
def __post_init__(self): def __post_init__(self):
if self.api_key == "wrong-key": if self.api_key == "wrong-key":
# logger.info("wrong embedding key, using simple retrieval method") # logger.info("wrong embedding key, using simple retrieval method")
key = os.environ.get("ALI_API_KEY") self.api_key = os.environ.get("ALI_API_KEY")
if key is None: if self.api_key is None:
logger.error(f"no ALI_API_KEY provided for embedding") logger.error(f"no ALI_API_KEY provided for embedding")
@@ -66,7 +66,8 @@ class SimpleRag(LangToolBase):
1. 用户询问“推荐一些辣味食物”，系统会检索并返回相关的辣味美食推荐文档。 1. 用户询问“推荐一些辣味食物”，系统会检索并返回相关的辣味美食推荐文档。
2. 用户搜索“适合夏天的清爽饮品”，系统会检索并返回相关饮品推荐及其来源信息。 2. 用户搜索“适合夏天的清爽饮品”，系统会检索并返回相关饮品推荐及其来源信息。
""" """
retrieved_docs:List[Document] = self.vec_store.search(query, search_kwargs={"k":3}) retrieved_docs:List[Document] = self.vec_store.similarity_search(query,
k=3)
serialized = "\n\n".join( serialized = "\n\n".join(
(f"Source: {doc.metadata}\nContent: {doc.page_content}") (f"Source: {doc.metadata}\nContent: {doc.page_content}")
for doc in retrieved_docs for doc in retrieved_docs
@@ -77,8 +78,10 @@ class SimpleRag(LangToolBase):
return [self.retrieve] return [self.retrieve]
# if __name__ == "__main__": if __name__ == "__main__":
# # config = tyro.cli(SimpleRagConfig) # # config = tyro.cli(SimpleRagConfig)
# config = SimpleRagConfig() config = SimpleRagConfig()
# rag:SimpleRag = config.setup() rag:SimpleRag = config.setup()
u = rag.retrieve("灯与尘")
print(u)
# mcp.run(transport="stdio") # mcp.run(transport="stdio")