AntonVoronko commited on
Commit
af2cd3a
·
verified ·
1 Parent(s): cab2a41

tools updated

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -25,8 +25,10 @@ class BasicAgent:
25
  api_key = os.getenv("GROQ_API_KEY")
26
  duck_spec = DuckDuckGoSearchToolSpec()
27
  search_tool = FunctionTool.from_defaults(duck_spec.duckduckgo_full_search)
28
- wiki_tool = WikipediaToolSpec().to_tool_list()
29
- arxiv_tool = ArxivToolSpec().to_tool_list()
 
 
30
  llm = Groq(model="deepseek-r1-distill-llama-70b", api_key=api_key)
31
  self.agent = FunctionAgent(tools=[search_tool, wiki_tool, arxiv_tool], llm=llm, system_prompt='''
32
  You are a general AI assistant.
 
25
  api_key = os.getenv("GROQ_API_KEY")
26
  duck_spec = DuckDuckGoSearchToolSpec()
27
  search_tool = FunctionTool.from_defaults(duck_spec.duckduckgo_full_search)
28
+ wiki_spec = WikipediaToolSpec()
29
+ wiki_tool = FunctionTool.from_defaults(wiki_spec.search_data)
30
+ arxiv_spec = ArxivToolSpec()
31
+ arxiv_tool = FunctionTool.from_defaults(arxiv_spec.arxiv_query)
32
  llm = Groq(model="deepseek-r1-distill-llama-70b", api_key=api_key)
33
  self.agent = FunctionAgent(tools=[search_tool, wiki_tool, arxiv_tool], llm=llm, system_prompt='''
34
  You are a general AI assistant.