Commit
·
7c18f52
1
Parent(s):
6f24b2c
Attempt to increase max_tokens
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ class BasicAgent:
|
|
| 37 |
wikipedia_search_tool = FunctionTool.from_defaults(WikipediaToolSpec().search_data)
|
| 38 |
self._tools = [search_tool, wikipedia_load_tool, wikipedia_search_tool]
|
| 39 |
|
| 40 |
-
self._llm = Gemini(api_key=GOOGLE_API_KEY, model="models/gemini-2.0-flash-lite", max_tokens=
|
| 41 |
self._agent = ReActAgent(tools=self._tools, llm=self._llm)
|
| 42 |
# Modify the react prompt.
|
| 43 |
self._agent.update_prompts({"react_header": SYSTEM_PROMPT})
|
|
|
|
| 37 |
wikipedia_search_tool = FunctionTool.from_defaults(WikipediaToolSpec().search_data)
|
| 38 |
self._tools = [search_tool, wikipedia_load_tool, wikipedia_search_tool]
|
| 39 |
|
| 40 |
+
self._llm = Gemini(api_key=GOOGLE_API_KEY, model="models/gemini-2.0-flash-lite", max_tokens=1600)
|
| 41 |
self._agent = ReActAgent(tools=self._tools, llm=self._llm)
|
| 42 |
# Modify the react prompt.
|
| 43 |
self._agent.update_prompts({"react_header": SYSTEM_PROMPT})
|