Spaces:
Sleeping
Sleeping
Replace LiteLLMModel with OpenAIServerModel in BasicAgent initialization
Browse files
agent.py
CHANGED
@@ -7,7 +7,7 @@ from typing import Any
|
|
7 |
from smolagents import (
|
8 |
CodeAgent,
|
9 |
DuckDuckGoSearchTool,
|
10 |
-
|
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 =
|
114 |
-
model_id="gpt-
|
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()
|