Spaces:
Sleeping
Sleeping
set temperature to 0
Browse files
agent.py
CHANGED
@@ -7,8 +7,8 @@ from typing import Any
|
|
7 |
from smolagents import (
|
8 |
CodeAgent,
|
9 |
DuckDuckGoSearchTool,
|
|
|
10 |
OpenAIServerModel,
|
11 |
-
SpeechToTextTool,
|
12 |
VisitWebpageTool,
|
13 |
WikipediaSearchTool,
|
14 |
tool,
|
@@ -70,7 +70,6 @@ Report your thoughts, and finish your answer with the following template: FINAL
|
|
70 |
# or "final answer:" in answer.lower()
|
71 |
# )
|
72 |
|
73 |
-
|
74 |
@tool
|
75 |
def wikipedia_suggested_page(query: str) -> str:
|
76 |
"""Search Wikipedia for suggested pages based on the query.
|
@@ -112,9 +111,10 @@ def wikipedia_page(title: str) -> str:
|
|
112 |
|
113 |
class BasicAgent:
|
114 |
def __init__(self):
|
115 |
-
model =
|
116 |
model_id="gpt-4o-mini",
|
117 |
api_key=os.getenv("OPENAI_API_KEY"),
|
|
|
118 |
)
|
119 |
search = DuckDuckGoSearchTool(max_results=5)
|
120 |
# speech_to_text = SpeechToTextTool()
|
|
|
7 |
from smolagents import (
|
8 |
CodeAgent,
|
9 |
DuckDuckGoSearchTool,
|
10 |
+
LiteLLMModel,
|
11 |
OpenAIServerModel,
|
|
|
12 |
VisitWebpageTool,
|
13 |
WikipediaSearchTool,
|
14 |
tool,
|
|
|
70 |
# or "final answer:" in answer.lower()
|
71 |
# )
|
72 |
|
|
|
73 |
@tool
|
74 |
def wikipedia_suggested_page(query: str) -> str:
|
75 |
"""Search Wikipedia for suggested pages based on the query.
|
|
|
111 |
|
112 |
class BasicAgent:
|
113 |
def __init__(self):
|
114 |
+
model = LiteLLMModel(
|
115 |
model_id="gpt-4o-mini",
|
116 |
api_key=os.getenv("OPENAI_API_KEY"),
|
117 |
+
temperature=0.0
|
118 |
)
|
119 |
search = DuckDuckGoSearchTool(max_results=5)
|
120 |
# speech_to_text = SpeechToTextTool()
|