agent.py

# agent.py – ReActAgent version with timeouts

import asyncio
from llama_index.llms.openai import OpenAI
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 ---
def search_duckduckgo(query: str) -> str:
    return DuckDuckGoSearchRun().run(query)

def search_wikipedia(query: str) -> str:
    return WikipediaQueryRun(api_wrapper=None).run(query)

def run_python(code: str) -> str:
    return PythonREPLTool().run(code)

def get_youtube_transcript(url: str) -> str:
    loader = YoutubeLoader.from_youtube_url(url, add_video_info=False)
    docs = loader.load()
    return " ".join(doc.page_content for doc in docs)

# --- Tools ---
TOOLS = [
    FunctionTool.from_defaults(search_duckduckgo),
    FunctionTool.from_defaults(search_wikipedia),
    FunctionTool.from_defaults(run_python),
    FunctionTool.from_defaults(get_youtube_transcript),
]

# --- LLM ---
llm = OpenAI(model="gpt-4")

# --- ReActAgent ---
agent = ReActAgent(
    tools=TOOLS,
    llm=llm,
    system_prompt="You are a helpful assistant that uses tools to answer questions accurately and efficiently."
)

# --- Runner with timeout ---
async def answer_question(question: str) -> str:
    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}"