From 00f56aa23ce29a7a9383f3c5f79ec5eaf66a1bb9 Mon Sep 17 00:00:00 2001 From: weiweiw <14335254+weiweiw22@user.noreply.gitee.com> Date: Mon, 3 Mar 2025 17:53:32 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BA=92=E8=81=94=E7=BD=91=E6=9F=A5=E8=AF=A2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../server/agent/tools_factory/search_internet.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) 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)