magnustragardh commited on
Commit
8c28f83
·
1 Parent(s): 18f0c1b

Change model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ class BasicAgent:
40
  self._tools.append(FunctionTool.from_defaults(WikipediaToolSpec().search_data))
41
 
42
  #self._llm = GoogleGenAI(api_key=GOOGLE_API_KEY, model="gemini-2.0-flash", max_tokens=1600)
43
- self._llm = OpenAI(model="o3", temperature=0.0)
44
  self._agent = ReActAgent(tools=self._tools, llm=self._llm)
45
  # Modify the react prompt.
46
  self._agent.update_prompts({"react_header": SYSTEM_PROMPT})
 
40
  self._tools.append(FunctionTool.from_defaults(WikipediaToolSpec().search_data))
41
 
42
  #self._llm = GoogleGenAI(api_key=GOOGLE_API_KEY, model="gemini-2.0-flash", max_tokens=1600)
43
+ self._llm = OpenAI(model="o4-mini", temperature=0.0)
44
  self._agent = ReActAgent(tools=self._tools, llm=self._llm)
45
  # Modify the react prompt.
46
  self._agent.update_prompts({"react_header": SYSTEM_PROMPT})