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()
|