Update app.py

app.py

@@ -2,10 +2,10 @@ import os
 import gradio as gr
 import datetime
 import pytz
-import requests
+import asyncio
 
-# Framework 1: LlamaIndex - Using the most direct, stable imports
-from llama_index.core.agent import ReActAgent
+# Framework 1: LlamaIndex (Updated to use Workflow for stability)
+from llama_index.core.agent.workflow import AgentWorkflow
 from llama_index.core.tools import FunctionTool
 from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 
@@ -16,31 +16,30 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientMod
 HF_TOKEN = os.getenv("HF_TOKEN")
 
 # ==========================================
-# PART 1: LLAMAINDEX AGENT
+# PART 1: LLAMAINDEX AGENT (Using Workflow)
 # ==========================================
 li_llm = HuggingFaceInferenceAPI(
     model_name="Qwen/Qwen2.5-7B-Instruct",
     token=HF_TOKEN,
-    task="conversational"
 )
 
 def get_tokyo_time() -> str:
     """Returns the current time in Tokyo, Japan."""
     tz = pytz.timezone('Asia/Tokyo')
-    return f"The current time is {datetime.datetime.now(tz).strftime('%H:%M:%S')}"
+    return f"The current time in Tokyo is {datetime.datetime.now(tz).strftime('%H:%M:%S')}"
 
 li_tools = [FunctionTool.from_defaults(fn=get_tokyo_time)]
 
-# Using the class-based initialization which is now corrected in latest 0.10+ versions
-li_agent = ReActAgent.from_tools(
+# The Workflow approach is much more stable against Pydantic errors
+li_agent = AgentWorkflow.from_tools_or_functions(
     tools=li_tools,
     llm=li_llm,
-    verbose=True
 )
 
 def chat_llama(message, history):
     try:
-        response = li_agent.chat(message)
+        # AgentWorkflow uses async .run()
+        response = asyncio.run(li_agent.run(input=message))
         return str(response)
     except Exception as e:
         return f"LlamaIndex Error: {str(e)}"
@@ -74,15 +73,15 @@ def chat_smol(message, history):
         return f"Smolagents Error: {str(e)}"
 
 # ==========================================
-# PART 3: UNIFIED GRADIO UI
+# PART 3: GRADIO UI
 # ==========================================
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🤖 Multi-Framework AI Space")
+    gr.Markdown("# 🤖 Dual-Agent Testing Space")
     
-    with gr.Tab("LlamaIndex (ReAct)"):
+    with gr.Tab("LlamaIndex (AgentWorkflow)"):
         gr.ChatInterface(fn=chat_llama)
         
-    with gr.Tab("smolagents (Code)"):
+    with gr.Tab("smolagents (CodeAgent)"):
         gr.ChatInterface(fn=chat_smol)
 
 if __name__ == "__main__":