agent 接入新工具 warntool

src/controller/api.py

@@ -79,8 +79,8 @@ class AgentManager:
             verbose=True
         )
         self.agent = new_rate_agent(self.llm, verbose=True, tool_base_url=tool_base_url)
-        # self.rate_agent = RateAgentV3(self.llm, tool_base_url=tool_base_url,version="v0")
-        self.rate_agent = RateAgentV3(self.llm, tool_base_url=tool_base_url)
+        self.rate_agent = RateAgentV3(self.llm, tool_base_url=tool_base_url,version="v0")
+        # self.rate_agent = RateAgentV3(self.llm, tool_base_url=tool_base_url)
         self.router_llm = new_router_llm(self.llm)
         self.re_rewriter_llm = new_re_rewriter_llm(self.llm)
 

src/server/agent_rate.py

@@ -15,6 +15,7 @@ from langchain import hub
 
 from src.agent.tool_rate import RegionRateTool,RankingRateTool
 from src.agent.tool_monitor import MonitorPointTool
+from src.agent.tool_warn import WarningTool
 from src.server.tool_picker import ToolPicker, ToolRunner
 
 # def create_rate_agent(llm, tools: List[BaseTool],prompt: PromptTemplate = None, 
@@ -174,7 +175,8 @@ class RateAgentV3:
         tools = [
             RegionRateTool(base_url=tool_base_url),
             RankingRateTool(base_url=tool_base_url),
-            MonitorPointTool(base_url=tool_base_url)
+            MonitorPointTool(base_url=tool_base_url),
+            WarningTool(base_url=tool_base_url)
         ]
         self.picker = ToolPicker(llm, tools)
         tools_dict = {}

src/utils/logger.py

@@ -30,8 +30,10 @@ def setup_logging(log_level=logging.INFO, log_dir="logs"):
     )
     
     # 设置第三方库的日志级别
-    # logging.getLogger("httpx").setLevel(logging.WARNING)
-    # logging.getLogger("urllib3").setLevel(logging.WARNING)
+    logging.getLogger("httpx").setLevel(logging.INFO)
+    logging.getLogger("urllib3").setLevel(logging.INFO)
+    logging.getLogger("openai").setLevel(logging.INFO)
+    logging.getLogger("httpcore").setLevel(logging.INFO)
     
     logging.info(f"日志配置完成，日志文件: {log_file}")