Update app.py

app.py

@@ -6,20 +6,19 @@ import math
 import requests
 from deep_translator import GoogleTranslator
 
-# Framework 1: LlamaIndex
-from llama_index.core.agent import ReActAgent
+# Framework 1: LlamaIndex (Core components for Agent Construction)
+from llama_index.core.agent import AgentRunner, ReActAgentWorker
 from llama_index.core.tools import FunctionTool
 from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 
-# Framework 2: smolagents
-# FIXED IMPORT: HfApiModel is now InferenceClientModel
+# Framework 2: smolagents (Updated Model Class)
 from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel
 
 # 0. SHARED CONFIG
 HF_TOKEN = os.getenv("HF_TOKEN")
 
 # ==========================================
-# PART 1: LLAMAINDEX AGENT
+# PART 1: LLAMAINDEX AGENT (Fixed Initialization)
 # ==========================================
 li_llm = HuggingFaceInferenceAPI(
     model_name="Qwen/Qwen2.5-7B-Instruct",
@@ -34,12 +33,14 @@ def get_tokyo_time() -> str:
 
 li_tools = [FunctionTool.from_defaults(fn=get_tokyo_time)]
 
-# Initializing via constructor to avoid Pydantic __getattr__ issues
-li_agent = ReActAgent.from_tools(
-    tools=li_tools,
-    llm=li_llm,
+# FIX: Using ReActAgentWorker + AgentRunner instead of ReActAgent.from_tools
+# This bypasses the Pydantic AttributeError issue
+worker = ReActAgentWorker.from_tools(
+    tools=li_tools, 
+    llm=li_llm, 
     verbose=True
 )
+li_agent = AgentRunner(worker)
 
 def chat_llama(message, history):
     try:
@@ -51,7 +52,6 @@ def chat_llama(message, history):
 # ==========================================
 # PART 2: SMOLAGENTS
 # ==========================================
-# Using the corrected model class
 smol_model = InferenceClientModel(
     model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
     token=HF_TOKEN
@@ -68,7 +68,7 @@ def weather_tool(location: str) -> str:
 smol_agent = CodeAgent(
     model=smol_model,
     tools=[weather_tool, DuckDuckGoSearchTool()],
-    additional_authorized_imports=['math', 'requests']
+    additional_authorized_imports=['math', 'requests', 'pytz', 'datetime']
 )
 
 def chat_smol(message, history):
@@ -83,14 +83,12 @@ def chat_smol(message, history):
 # ==========================================
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🤖 Multi-Framework Agent Space")
-    gr.Markdown("Compare how different agentic frameworks handle your requests.")
+    gr.Markdown("Compare the two leading agent frameworks side-by-side.")
     
     with gr.Tab("LlamaIndex (ReAct Agent)"):
-        gr.Markdown("This agent uses the classic **Reasoning + Acting** text loop.")
         gr.ChatInterface(fn=chat_llama)
         
     with gr.Tab("smolagents (Code Agent)"):
-        gr.Markdown("This agent solves tasks by writing and executing **Python code**.")
         gr.ChatInterface(fn=chat_smol)
 
 if __name__ == "__main__":