Update agent.py

agent.py

@@ -1,14 +1,15 @@
-# agent.py
+# agent.py – ReActAgent version with timeouts
 
+import asyncio
 from llama_index.llms.openai import OpenAI
-from llama_index.core.agent.workflow import AgentWorkflow
+from llama_index.core.agent.react import ReActAgent
 from llama_index.core.tools import FunctionTool
 
 from langchain_community.tools import DuckDuckGoSearchRun, WikipediaQueryRun
 from langchain_experimental.tools.python.tool import PythonREPLTool
 from langchain_community.document_loaders import YoutubeLoader
 
-# --- Tool wrappers --- #
+# --- Tool Wrappers ---
 def search_duckduckgo(query: str) -> str:
     return DuckDuckGoSearchRun().run(query)
 
@@ -23,7 +24,7 @@ def get_youtube_transcript(url: str) -> str:
     docs = loader.load()
     return " ".join(doc.page_content for doc in docs)
 
-# --- Register tools --- #
+# --- Tools ---
 TOOLS = [
     FunctionTool.from_defaults(search_duckduckgo),
     FunctionTool.from_defaults(search_wikipedia),
@@ -31,17 +32,22 @@ TOOLS = [
     FunctionTool.from_defaults(get_youtube_transcript),
 ]
 
-# --- Load LLM --- #
+# --- LLM ---
 llm = OpenAI(model="gpt-4")
 
-# --- Build agent --- #
-agent = AgentWorkflow.from_tools_or_functions(
-    TOOLS,
+# --- ReActAgent ---
+agent = ReActAgent(
+    tools=TOOLS,
     llm=llm,
-    system_prompt="You are a helpful AI agent using reasoning and external tools to solve tasks."
+    system_prompt="You are a helpful assistant that uses tools to answer questions accurately and efficiently."
 )
 
-# --- Async run wrapper --- #
+# --- Runner with timeout ---
 async def answer_question(question: str) -> str:
-    result = await agent.run(question)
-    return str(result)
+    try:
+        result = await asyncio.wait_for(agent.run(question), timeout=60)
+        return str(result)
+    except asyncio.TimeoutError:
+        return "[TIMEOUT]"
+    except Exception as e:
+        return f"[ERROR] {e}"