abliznyuk commited on
Commit
bed91f1
·
1 Parent(s): 3b5401f

replace LLM to OpenAI

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -1,5 +1,5 @@
1
- from smolagents import CodeAgent, InferenceClientModel, WikipediaSearchTool, GoogleSearchTool, VisitWebpageTool, \
2
- PythonInterpreterTool
3
 
4
 
5
  def get_prompt():
@@ -16,7 +16,7 @@ class GAIAAgent:
16
  WikipediaSearchTool(),
17
  PythonInterpreterTool(),
18
  ],
19
- model=InferenceClientModel(),
20
  add_base_tools=False
21
  )
22
  self.prompt = get_prompt()
 
1
+ from smolagents import CodeAgent, OpenAIServerModel
2
+ from smolagents import WikipediaSearchTool, GoogleSearchTool, VisitWebpageTool, PythonInterpreterTool
3
 
4
 
5
  def get_prompt():
 
16
  WikipediaSearchTool(),
17
  PythonInterpreterTool(),
18
  ],
19
+ model=OpenAIServerModel(model_id='gpt-4o-mini', max_tokens=4096, temperature=0),
20
  add_base_tools=False
21
  )
22
  self.prompt = get_prompt()