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})
|