Update app.py

app.py

@@ -10,15 +10,11 @@ from llama_index.core.agent import ReActAgent
 from llama_index.core.tools import FunctionTool
 from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 
-# ============ 1. SMOLAGENTS TOOLS ============
+# ============ 1. SMOLAGENTS SETUP (Already Working) ============
 
 @tool
 def time_tool(timezone: str = "UTC") -> str:
-    """
-    Get current time in a specific timezone.
-    Args:
-        timezone: The timezone to check (e.g., 'Asia/Tokyo').
-    """
+    """Get current time in a specific timezone."""
     try:
         tz = pytz.timezone(timezone)
         now = datetime.datetime.now(tz)
@@ -28,12 +24,7 @@ def time_tool(timezone: str = "UTC") -> str:
 
 @tool
 def translator_tool(text: str, target_language: str) -> str:
-    """
-    Translates text into a specified language.
-    Args:
-        text: The text or phrase to translate.
-        target_language: The destination language (e.g., 'french', 'german').
-    """
+    """Translates text into a specified language."""
     try:
         translation = GoogleTranslator(source='auto', target=target_language).translate(text)
         return f"Translated to {target_language.title()}: {translation}"
@@ -42,11 +33,7 @@ def translator_tool(text: str, target_language: str) -> str:
 
 @tool
 def weather_tool(location: str) -> str:
-    """
-    Get the current weather for any location worldwide.
-    Args:
-        location: The name of the city or place (e.g., 'London' or 'Tokyo').
-    """
+    """Get the current weather for any location worldwide."""
     try:
         geo_url = f"https://geocoding-api.open-meteo.com/v1/search?name={location}&count=1&language=en&format=json"
         geo_res = requests.get(geo_url).json()
@@ -58,9 +45,7 @@ def weather_tool(location: str) -> str:
     except Exception as e:
         return f"Error: {str(e)}"
 
-# ============ AGENT CONFIGURATION ============
-
-HF_TOKEN = os.getenv("HF_TOKEN", "")
+HF_TOKEN = os.getenv("HF_TOKEN")
 smol_model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", token=HF_TOKEN)
 
 smol_agent = CodeAgent(
@@ -69,17 +54,34 @@ smol_agent = CodeAgent(
     max_steps=5
 )
 
-# LlamaIndex Setup
+# ============ 2. LLAMAINDEX SETUP (Fixed) ============
+
+# FIX 1: Define real functions with docstrings so the Agent knows what they do
+def get_tokyo_time() -> str:
+    """
+    Returns the current time in Tokyo, Japan.
+    Use this tool when the user asks for the time in Tokyo.
+    """
+    return f"Tokyo: {datetime.datetime.now(pytz.timezone('Asia/Tokyo'))}"
+
+def multiply(a: float, b: float) -> float:
+    """
+    Multiplies two numbers a and b.
+    Use this tool for calculations involving multiplication.
+    """
+    return a * b
+
+# FIX 2: Change task to 'text-generation'
 llama_llm = HuggingFaceInferenceAPI(
     model_name="Qwen/Qwen2.5-7B-Instruct", 
     token=HF_TOKEN,
-    task="conversational"
+    task="text-generation" 
 )
 
 llama_agent = ReActAgent.from_tools(
     [
-        FunctionTool.from_defaults(fn=lambda: f"Tokyo: {datetime.datetime.now(pytz.timezone('Asia/Tokyo'))}", name="get_tokyo_time"),
-        FunctionTool.from_defaults(fn=lambda a, b: a * b, name="multiply")
+        FunctionTool.from_defaults(fn=get_tokyo_time),
+        FunctionTool.from_defaults(fn=multiply)
     ],
     llm=llama_llm,
     verbose=True
@@ -88,17 +90,28 @@ llama_agent = ReActAgent.from_tools(
 # ============ GRADIO UI ============
 
 def chat_smol(message, history):
-    return str(smol_agent.run(message))
+    try:
+        return str(smol_agent.run(message))
+    except Exception as e:
+        return f"SmolAgent Error: {str(e)}"
 
 def chat_llama(message, history):
-    response = llama_agent.chat(message)
-    return str(response)
+    try:
+        # LlamaIndex keeps its own history, so we just pass the message
+        response = llama_agent.chat(message)
+        return str(response)
+    except Exception as e:
+        return f"LlamaIndex Error: {str(e)}"
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🤖 Multi-Framework AI Agent Space")
+    
     with gr.Tab("Smolagents (Alfred)"):
+        gr.Markdown("Uses `Qwen2.5-Coder-32B`")
         gr.ChatInterface(chat_smol)
+        
     with gr.Tab("LlamaIndex (ReAct)"):
+        gr.Markdown("Uses `Qwen2.5-7B`")
         gr.ChatInterface(chat_llama)
 
 if __name__ == "__main__":