pateas commited on
Commit
67d6957
·
unverified ·
1 Parent(s): e200b5f

Replace LiteLLMModel with OpenAIServerModel in BasicAgent initialization

Browse files
Files changed (1) hide show
  1. agent.py +4 -4
agent.py CHANGED
@@ -7,7 +7,7 @@ from typing import Any
7
  from smolagents import (
8
  CodeAgent,
9
  DuckDuckGoSearchTool,
10
- LiteLLMModel,
11
  VisitWebpageTool,
12
  WikipediaSearchTool,
13
  tool,
@@ -110,10 +110,10 @@ def wikipedia_page(title: str) -> str:
110
 
111
  class BasicAgent:
112
  def __init__(self):
113
- model = LiteLLMModel(
114
- model_id="gpt-4o-mini",
115
  api_key=os.getenv("OPENAI_API_KEY"),
116
- temperature=0.0
117
  )
118
  search = DuckDuckGoSearchTool(max_results=5)
119
  # speech_to_text = SpeechToTextTool()
 
7
  from smolagents import (
8
  CodeAgent,
9
  DuckDuckGoSearchTool,
10
+ OpenAIServerModel,
11
  VisitWebpageTool,
12
  WikipediaSearchTool,
13
  tool,
 
110
 
111
  class BasicAgent:
112
  def __init__(self):
113
+ model = OpenAIServerModel(
114
+ model_id="gpt-4.1",
115
  api_key=os.getenv("OPENAI_API_KEY"),
116
+ temperature=0.0,
117
  )
118
  search = DuckDuckGoSearchTool(max_results=5)
119
  # speech_to_text = SpeechToTextTool()