Update app.py

app.py

@@ -17,14 +17,14 @@ class BasicAgent:
         print("BasicAgent initialized.")
         groq_key = os.getenv("GROQ_API_KEY")
         model = OpenAIServerModel(
-            model_id="openai/gpt-oss-120b",
+            model_id="llama-3.1-8b-instant",
             api_base="https://api.groq.com/openai/v1",
             api_key=groq_key,
         )
         self.agent = CodeAgent(
             tools=[DuckDuckGoSearchTool()],
             model=model,
-            max_steps=6,
+            max_steps=8,
             additional_authorized_imports=["requests", "pandas", "numpy", "time", "bs4", "re"]
         )
 
@@ -43,13 +43,16 @@ class BasicAgent:
         """
 
         try:
-            with concurrent.futures.ThreadPoolExecutor(max_workers=1) as executor:
-                future = executor.submit(self.agent.run, prompt)
-                result = future.result(timeout=90)
-            return str(result).strip()
-        except concurrent.futures.TimeoutError:
-            print("Agent timed out after 90s")
-            return "ERROR: Timeout"
+            executor = concurrent.futures.ThreadPoolExecutor(max_workers=1)
+            future = executor.submit(self.agent.run, prompt)
+            try:
+                result = future.result(timeout=120)
+                executor.shutdown(wait=False)
+                return str(result).strip()
+            except concurrent.futures.TimeoutError:
+                print("Agent timed out after 120s, moving on")
+                executor.shutdown(wait=False, cancel_futures=True)
+                return "ERROR: Timeout"
         except BaseException as e:
             import traceback
             print(f"Agent failed to answer: {e}")