Update app.py

app.py

@@ -4,7 +4,7 @@ import datetime
 import pytz
 import asyncio
 
-# Framework 1: LlamaIndex (Updated to use Workflow for stability)
+# Framework 1: LlamaIndex
 from llama_index.core.agent.workflow import AgentWorkflow
 from llama_index.core.tools import FunctionTool
 from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
@@ -16,7 +16,7 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientMod
 HF_TOKEN = os.getenv("HF_TOKEN")
 
 # ==========================================
-# PART 1: LLAMAINDEX AGENT (Using Workflow)
+# PART 1: LLAMAINDEX AGENT
 # ==========================================
 li_llm = HuggingFaceInferenceAPI(
     model_name="Qwen/Qwen2.5-7B-Instruct",
@@ -30,17 +30,18 @@ def get_tokyo_time() -> str:
 
 li_tools = [FunctionTool.from_defaults(fn=get_tokyo_time)]
 
-# The Workflow approach is much more stable against Pydantic errors
+# FIX: Passed li_tools as a positional argument (no 'tools=' key)
 li_agent = AgentWorkflow.from_tools_or_functions(
-    tools=li_tools,
+    li_tools,
     llm=li_llm,
+    system_prompt="You are a helpful assistant with access to a Tokyo time tool."
 )
 
-def chat_llama(message, history):
+async def chat_llama(message, history):
     try:
-        # AgentWorkflow uses async .run()
-        response = asyncio.run(li_agent.run(input=message))
-        return str(response)
+        # AgentWorkflow.run returns a handler; we await the final result
+        result = await li_agent.run(user_msg=message)
+        return str(result)
     except Exception as e:
         return f"LlamaIndex Error: {str(e)}"
 
@@ -67,21 +68,24 @@ smol_agent = CodeAgent(
 
 def chat_smol(message, history):
     try:
+        # smolagents .run() is synchronous by default
         response = smol_agent.run(message)
         return str(response)
     except Exception as e:
         return f"Smolagents Error: {str(e)}"
 
 # ==========================================
-# PART 3: GRADIO UI
+# PART 3: UNIFIED GRADIO UI
 # ==========================================
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🤖 Dual-Agent Testing Space")
+    gr.Markdown("# 🤖 Dual-Agent Testing Space (2026 Edition)")
     
-    with gr.Tab("LlamaIndex (AgentWorkflow)"):
+    with gr.Tab("LlamaIndex (Workflow)"):
+        gr.Markdown("Uses an **Event-Driven Workflow** for reasoning.")
         gr.ChatInterface(fn=chat_llama)
         
     with gr.Tab("smolagents (CodeAgent)"):
+        gr.Markdown("Uses a **Code-Execution Loop** to solve tasks.")
         gr.ChatInterface(fn=chat_smol)
 
 if __name__ == "__main__":