From 1084d628917e1d8c90464fba2d9cc9ac600aa695 Mon Sep 17 00:00:00 2001 From: Wenjie Zhang Date: Tue, 8 Apr 2025 00:35:29 +0800 Subject: [PATCH] =?UTF-8?q?=E5=8F=AA=E6=9C=89=E5=BD=93=E5=90=AF=E7=94=A8?= =?UTF-8?q?=E6=90=9C=E7=B4=A2=E7=9A=84=E6=97=B6=E5=80=99=E6=89=8D=E4=BC=9A?= =?UTF-8?q?=E5=8A=A0=E8=BD=BD=E7=BD=91=E7=BB=9C=E6=90=9C=E7=B4=A2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/agents/tools_factory.py | 6 ++++-- src/core/retriever.py | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/agents/tools_factory.py b/src/agents/tools_factory.py index 21fa07cc..a7e9eef8 100644 --- a/src/agents/tools_factory.py +++ b/src/agents/tools_factory.py @@ -8,7 +8,7 @@ from pydantic import BaseModel, Field from langchain_core.tools import tool, BaseTool, StructuredTool from langchain_community.tools.tavily_search import TavilySearchResults -from src import graph_base, knowledge_base +from src import graph_base, knowledge_base, config # refs https://github.com/chatchat-space/LangGraph-Chatchat chatchat-server/chatchat/server/agent/tools_factory/tools_registry.py def regist_tool( @@ -136,6 +136,8 @@ def get_knowledge_graph(query: Annotated[str, "The query to get knowledge graph. _TOOLS_REGISTRY = { "calculator": calculator, - "TavilySearchResults": TavilySearchResults(max_results=10), "get_knowledge_graph": get_knowledge_graph, } + +if config.enable_web_search: + _TOOLS_REGISTRY["TavilySearchResults"] = TavilySearchResults(max_results=10) diff --git a/src/core/retriever.py b/src/core/retriever.py index 871997ed..92477b04 100644 --- a/src/core/retriever.py +++ b/src/core/retriever.py @@ -120,7 +120,7 @@ class Retriever: def query_web(self, query, history, refs): """查询网络""" - if not (refs["meta"].get("use_web") and config.enable_web_search): + if not (refs["meta"].get("use_web") or not config.enable_web_search): return {"results": [], "message": "Web search is disabled"} try: