Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool, InferenceClientModel, Tool, tool, VisitWebpageTool, HfApiModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool, InferenceClientModel, Tool, tool, VisitWebpageTool, HfApiModel, LiteLLMModel,
 from groq import Groq
 
 # (Keep Constants as is)
@@ -16,60 +16,24 @@ GROQ_KEY = os.environ['GROQ_KEY']
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-        self.client = Groq(api_key=GROQ_KEY)
+        self.model = LiteLLMModel(
+            "openai/deepseek-r1-distill-llama-70b",
+            api_base="https://api.groq.com/openai/v1",
+            api_key=api_key
+        )
+        self.model.flatten_messages_as_text = True
         
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        message = [  
-        {
-            "role": "user",
-            "content": question
-        }]
-        completion = self.client.chat.completions.create(
-            messages=message,
-            model="compound-beta",
-        )
-        answer = completion.choices[0].message.content
-        print(f"First answer: {answer}")
-        message=[
-            {
-                "role": "system",
-                "content": """
-                You are an expert in identifying and extracting definitive answers. Your sole task is to analyze the provided text, which is an agent's response, and extract only the conclusive final answer to the original user's query.
-
-                Output only this core answer. Do not include any explanations, pleasantries, introductory phrases, or any surrounding text.
-
-                Here The Examples:
-                
-                Input: ... Final answer: 12 ...
-                You should output: 12
-
-                Input: $\\boxed{b,c,e}$
-                Output: b, c, e
-
-                Input: Jan
-                Output: Jan
-
-                Input: The Yankee with the most walks in the 1977 regular season was Reggie Jackson, with 58 walks. He had 357 at bats that season.
-                Output: 357
-
-                Input: broccoli, bell pepper, celery, fresh basil, green beans, lettuce, sweet potatoes, zucchini
-                Output: broccoli, bell pepper, celery, fresh basil, green beans, lettuce, sweet potatoes, zucchini
-                """
-            },
-            {
-                "role": "user",
-                "content": answer
-            }
-        ]
-        completion = self.client.chat.completions.create(
-            messages=message,
-            # Change model to compound-beta to use agentic tooling
-            model="llama-3.3-70b-versatile",
+        agent = CodeAgent(
+            tools=[DuckDuckGoSearchTool(), FinalAnswerTool(), VisitWebpageTool()],
+            model=model,
+            add_base_tools=True,
+            verbosity_level=2
         )
-        answer = completion.choices[0].message.content
-        print(f"Second answer: {answer}")
+        
+        response = agent.run(question)
+        print(response)
         return answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):