diff --git a/libs/chatchat-server/chatchat/server/agent/tools_factory/search_internet.py b/libs/chatchat-server/chatchat/server/agent/tools_factory/search_internet.py index 18a9231..d7063f8 100644 --- a/libs/chatchat-server/chatchat/server/agent/tools_factory/search_internet.py +++ b/libs/chatchat-server/chatchat/server/agent/tools_factory/search_internet.py @@ -11,9 +11,10 @@ from strsimpy.normalized_levenshtein import NormalizedLevenshtein from chatchat.settings import Settings from chatchat.server.pydantic_v1 import Field from chatchat.server.utils import get_tool_config +from chatchat.utils import build_logger from .tools_registry import BaseToolOutput, regist_tool, format_context - +logger = build_logger() def searx_search(text ,config, top_k: int): print(f"searx_search: text: {text},config:{config},top_k:{top_k}") @@ -124,5 +125,9 @@ def search_engine(query: str, top_k:int=0, engine_name: str="", config: dict={}) @regist_tool(title="互联网搜索") def search_internet(query: str = Field(description="query for Internet search")): """用这个工具实现获取世界、历史、实时新闻、或除电力系统之外的信息查询""" - print(f"search_internet: query: {query}") - return BaseToolOutput(search_engine(query=query), format=format_context) + try: + print(f"search_internet: query: {query}") + return BaseToolOutput(search_engine(query=query), format=format_context) + except Exception as e: + logger.error(f"未知错误: {str(e)}") + return BaseToolOutput(f"搜索过程中发生未知错误,{str(e)}", format=format_context)