pateas commited on
Commit
c4bec66
·
unverified ·
1 Parent(s): 280f38d

set temperature to 0

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
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 = OpenAIServerModel(
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()