Tuchuanhuhuhu commited on
Commit
afaf18f
·
1 Parent(s): 8addcbc

feat: 川虎助理Pro 支持 serpapi

Browse files
modules/models/ChuanhuAgent.py CHANGED
@@ -63,7 +63,7 @@ class ChuanhuAgent_Client(BaseLLMModel):
63
  self.index_summary = None
64
  self.index = None
65
  if "Pro" in self.model_name:
66
- self.tools = load_tools(["google-search-results-json", "llm-math", "arxiv", "wikipedia", "wolfram-alpha"], llm=self.llm)
67
  else:
68
  self.tools = load_tools(["ddg-search", "llm-math", "arxiv", "wikipedia"], llm=self.llm)
69
  self.tools.append(
 
63
  self.index_summary = None
64
  self.index = None
65
  if "Pro" in self.model_name:
66
+ self.tools = load_tools(["serpapi", "google-search-results-json", "llm-math", "arxiv", "wikipedia", "wolfram-alpha"], llm=self.llm)
67
  else:
68
  self.tools = load_tools(["ddg-search", "llm-math", "arxiv", "wikipedia"], llm=self.llm)
69
  self.tools.append(
requirements.txt CHANGED
@@ -23,3 +23,4 @@ wikipedia
23
  google.generativeai
24
  openai
25
  unstructured
 
 
23
  google.generativeai
24
  openai
25
  unstructured
26
+ google-api-python-client