Update app.py

app.py

@@ -3,8 +3,8 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
-from tools.final_answer import FinalAnswerTool
+from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
+from llama_index.core.agent.workflow import AgentWorkflow, ToolCallResult, AgentStream
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -21,15 +21,24 @@ class BasicAgent:
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer
 
-@tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    return "What magic will you build ?"
+def add(a: int, b: int) -> int:
+    """Add two numbers"""
+    return a + b
+
+
+def subtract(a: int, b: int) -> int:
+    """Subtract two numbers"""
+    return a - b
+
+
+def multiply(a: int, b: int) -> int:
+    """Multiply two numbers"""
+    return a * b
+
+
+def divide(a: int, b: int) -> int:
+    """Divide two numbers"""
+    return a / b
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -52,30 +61,16 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        model = HfApiModel(
-        max_tokens=2096,
-        temperature=0.5,
-        model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-        custom_role_conversions=None,
-        )
-        
-        final_answer = FinalAnswerTool()    
-        search_tool = DuckDuckGoSearchTool()
+        llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
         
-        agent = CodeAgent(
-            model=model,
-            tools=[
-                final_answer,
-                search_tool,
-                my_custom_tool,
-            ],
-            max_steps=6,
-            verbosity_level=1,
-            grammar=None,
-            planning_interval=None,
-            name=None,
-            description=None,
-            prompt_templates=prompt_templates
+        agent = AgentWorkflow.from_tools_or_functions(
+            tools_or_functions=[subtract, multiply, divide, add],
+            llm=llm,
+            system_prompt="You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: \
+            FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. \
+            If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. \
+            If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. \
+            If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.",
         )
     except Exception as e:
         print(f"Error instantiating agent: {e}")