Updated code for smolagent

app.py

@@ -3,9 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
-from llama_index.core.agent.workflow import FunctionAgent
-from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
+from smolagents import CodeAgent, WebSearchTool, InferenceClientModel, FinalAnswerTool, tool
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -13,24 +11,29 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-
+web_search = WebSearchTool()
+final_answer = FinalAnswerTool()
 class BasicAgent:
     def __init__(self):
         print("Srj's Agent initialized.")
-        self.llm = HuggingFaceInferenceAPI(
-            model_name="meta-llama/Llama-3.2-1B-Instruct",
-            temperature=0.7,
-            max_tokens=100,
-            token = os.environ.get("HF_TOKEN"),
-            provider="novita",  # Optional
-            )
-
-        tool_spec = DuckDuckGoSearchToolSpec()
-        self.agent = FunctionAgent(
-            tools=tool_spec.to_tool_list(),
-            llm=self.llm,
-            verbose=False
-            )
+        self.model = InferenceClientModel(max_tokens=2096,
+            temperature=0.5,
+            model_id='Qwen/Qwen2.5-Coder-7B-Instruct',
+            provider="nscale"
+                                
+        )       
+        self.agent = CodeAgent(
+            model=self.model,
+            tools=[final_answer, web_search], # add your tools here (don't remove final_answer)
+            max_steps=6,
+            verbosity_level=1,
+            grammar=None,
+            planning_interval=None,
+            name=None,
+            description=None,
+            prompt_templates=prompt_templates
+        )
+
         #self.model = InferenceClientModel(model= "Qwen/Qwen2.5-7B-Instruct",)
         #self.agent = CodeAgent(tools=[WebSearchTool()], model=self.model, stream_outputs=False, add_base_tools=False)
         
@@ -68,7 +71,7 @@ class BasicAgent:
                     
                     Respond with ONLY the final answer."""
         
-        output = asyncio.run(self.agent.run(prompt))
+        output = self.agent.run(prompt))
         return str(output).strip()