magnustragardh commited on
Commit
7c18f52
·
1 Parent(s): 6f24b2c

Attempt to increase max_tokens

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=800)
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})