feat: 切换到川虎助理时可以显示启用了的工具

modules/models/ChuanhuAgent.py

@@ -63,7 +63,23 @@ class ChuanhuAgent_Client(BaseLLMModel):
         self.index_summary = None
         self.index = None
         if "Pro" in self.model_name:
-            self.tools = load_tools(["serpapi", "google-search-results-json", "llm-math", "arxiv", "wikipedia", "wolfram-alpha"], llm=self.llm)
+            tools_to_enable = ["llm-math", "arxiv", "wikipedia"]
+            # if exists GOOGLE_CSE_ID and GOOGLE_API_KEY, enable google-search-results-json
+            if os.environ.get("GOOGLE_CSE_ID", None) is not None and os.environ.get("GOOGLE_API_KEY", None) is not None:
+                tools_to_enable.append("google-search-results-json")
+            else:
+                logging.warning("GOOGLE_CSE_ID and/or GOOGLE_API_KEY not found, google-search-results-json is disabled.")
+            # if exists WOLFRAM_ALPHA_APPID, enable wolfram-alpha
+            if os.environ.get("WOLFRAM_ALPHA_APPID", None) is not None:
+                tools_to_enable.append("wolfram-alpha")
+            else:
+                logging.warning("WOLFRAM_ALPHA_APPID not found, wolfram-alpha is disabled.")
+            # if exists SERPAPI_API_KEY, enable serpapi
+            if os.environ.get("SERPAPI_API_KEY", None) is not None:
+                tools_to_enable.append("serpapi")
+            else:
+                logging.warning("SERPAPI_API_KEY not found, serpapi is disabled.")
+            self.tools = load_tools(tools_to_enable, llm=self.llm)
         else:
             self.tools = load_tools(["ddg-search", "llm-math", "arxiv", "wikipedia"], llm=self.llm)
             self.tools.append(

modules/models/models.py

@@ -665,6 +665,7 @@ def get_model(
         elif model_type == ModelType.ChuanhuAgent:
             from .ChuanhuAgent import ChuanhuAgent_Client
             model = ChuanhuAgent_Client(model_name, access_key, user_name=user_name)
+            msg = i18n("启用的工具：") + ", ".join([i.name for i in model.tools])
         elif model_type == ModelType.GooglePaLM:
             from .Google_PaLM import Google_PaLM_Client
             access_key = os.environ.get("GOOGLE_PALM_API_KEY", access_key)