From 8e18c50270f4a04a5e7066586ecc4163b00fe7b6 Mon Sep 17 00:00:00 2001 From: goulustis Date: Mon, 13 Oct 2025 21:11:24 +0800 Subject: [PATCH] comment out unused --- lang_agent/rag/simple.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/lang_agent/rag/simple.py b/lang_agent/rag/simple.py index 8ecb78c..3c2b364 100644 --- a/lang_agent/rag/simple.py +++ b/lang_agent/rag/simple.py @@ -2,8 +2,8 @@ from dataclasses import dataclass, field from typing import Type, List import tyro from mcp.server.fastmcp import FastMCP -import tyro from loguru import logger +import os from langchain_community.vectorstores import FAISS from langchain_core.documents.base import Document @@ -13,7 +13,7 @@ from lang_agent.config import ToolConfig from lang_agent.base import LangToolBase -mcp = FastMCP("Rag") +# mcp = FastMCP("Rag") @tyro.conf.configure(tyro.conf.SuppressFixed) @dataclass @@ -31,7 +31,10 @@ class SimpleRagConfig(ToolConfig): def __post_init__(self): 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") + if key is None: + logger.error(f"no ALI_API_KEY provided for embedding") @@ -48,7 +51,6 @@ class SimpleRag(LangToolBase): # self.retriever = self.vec_store.as_retriever(search_kwargs={"k":3}) - @mcp.tool() def retrieve(self, query:str): """ 检索与给定查询相关的文档，并将其序列化为字符串格式。 @@ -75,8 +77,8 @@ class SimpleRag(LangToolBase): return [self.retrieve] -if __name__ == "__main__": - # config = tyro.cli(SimpleRagConfig) - config = SimpleRagConfig() - rag:SimpleRag = config.setup() - mcp.run(transport="stdio") \ No newline at end of file +# if __name__ == "__main__": +# # config = tyro.cli(SimpleRagConfig) +# config = SimpleRagConfig() +# rag:SimpleRag = config.setup() +# mcp.run(transport="stdio") \ No newline at end of file