Update app.py

app.py

@@ -20,11 +20,11 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         self.model = LiteLLMModel(
-            model_id="groq/llama-3.1-8b-instant",
-            api_key=os.getenv("GROQ_API_KEY")
+            model_id="gemini/gemini-2.0-flash",
+            api_key=os.getenv("GEMINI_API_KEY")
         )
         self.agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool()],
+            tools=[DuckDuckGoSearchTool(max_results=3)],
             model=self.model,
             max_steps=6
         )
@@ -36,15 +36,10 @@ class BasicAgent:
         try:
             raw_answer = self.agent.run(full_question)
         except TypeError as e:
-            # Known smolagents bug: final_answer() called with a non-string
-            # (e.g. an int) can raise "argument of type 'int' is not iterable".
-            # Recover by pulling the last step's output instead of losing the answer.
             print(f"Recovering from TypeError: {e}")
             try:
                 last_step = self.agent.memory.steps[-1]
-                raw_answer = getattr(last_step, "action_output", None)
-                if raw_answer is None:
-                    raw_answer = "Unknown"
+                raw_answer = getattr(last_step, "action_output", None) or "Unknown"
             except Exception as inner_e:
                 print(f"Could not recover answer: {inner_e}")
                 raw_answer = "Unknown"