Commit
·
bb20cfb
1
Parent(s):
8c28f83
Change model
Browse files
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="
|
| 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})
|