cubukcum commited on
Commit
6eaddf0
·
verified ·
1 Parent(s): d08e75c

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
- from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel, OpenAIServerModel
3
  from dotenv import load_dotenv
4
  load_dotenv()
5
 
@@ -67,7 +67,7 @@ class BasicAgent:
67
  )
68
 
69
  self.agent = CodeAgent(
70
- tools=[add,subtract,multiply,divide,DuckDuckGoSearchTool()],
71
  model=llm,
72
  )
73
  def __call__(self, question: str) -> str:
 
1
  import os
2
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel, OpenAIServerModel, SpeechToTextTool, WikipediaSearchTool
3
  from dotenv import load_dotenv
4
  load_dotenv()
5
 
 
67
  )
68
 
69
  self.agent = CodeAgent(
70
+ tools=[add,subtract,multiply,divide,DuckDuckGoSearchTool(),WikipediaSearchTool(),SpeechToTextTool()],
71
  model=llm,
72
  )
73
  def __call__(self, question: str) -> str: