Increase the number of tokens.

app.py

@@ -19,15 +19,6 @@ except ImportError:
 
 
 load_dotenv()
-# See https://discord.com/channels/879548962464493619/1360558713274109982:
-#For anyone who has run out of free tier, you can switch to a Gemini model in Google Colab:
-#go to https://aistudio.google.com/apikey and create a new API key
-#insert it into Google Colab using 'Secrets' tab on the left pane
-#make sure to grant access to the notebook!
-#in the notebook, use:
-#from google.colab import userdata
-#model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=userdata.get('GEMINI_API_KEY'))
-#agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=model)
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -42,7 +33,7 @@ class BasicAgent:
         wikipedia_search_tool = FunctionTool.from_defaults(WikipediaToolSpec().search_data)
         self._tools = [search_tool, wikipedia_load_tool, wikipedia_search_tool]
 
-        self._llm = Gemini(api_key=GOOGLE_API_KEY, model="models/gemini-2.0-flash-lite")
+        self._llm = Gemini(api_key=GOOGLE_API_KEY, model="models/gemini-2.0-flash-lite", max_tokens=800)
         self._agent = ReActAgent(tools=self._tools, llm=self._llm)
         # Modify the react prompt.
         self._agent.update_prompts({"react_header": SYSTEM_PROMPT})