File size: 1,661 Bytes
5db119a 59ff18d 5db119a 92b0d1a 5db119a 59ff18d 92b0d1a 59ff18d 5db119a 59ff18d 6c7c70b 59ff18d 5db119a 59ff18d 5db119a 92b0d1a 6c7c70b 5db119a 59ff18d 5db119a 59ff18d 5db119a 59ff18d 5db119a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
# 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}"
|