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

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="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})
 
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="gpt-4.1", 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})