Update agent.py

agent.py

@@ -4,12 +4,11 @@ from llama_index.llms.openai import OpenAI
 from llama_index.core.agent.workflow import AgentWorkflow
 from llama_index.core.tools import FunctionTool
 
-# LangChain-based tools
 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)
 
@@ -22,8 +21,9 @@ def run_python(code: str) -> str:
 def get_youtube_transcript(url: str) -> str:
     loader = YoutubeLoader.from_youtube_url(url, add_video_info=False)
     docs = loader.load()
-    return " ".join(d.page_content for d in docs)
+    return " ".join(doc.page_content for doc in docs)
 
+# --- Register tools --- #
 TOOLS = [
     FunctionTool.from_defaults(search_duckduckgo),
     FunctionTool.from_defaults(search_wikipedia),
@@ -31,13 +31,17 @@ TOOLS = [
     FunctionTool.from_defaults(get_youtube_transcript),
 ]
 
+# --- Load LLM --- #
 llm = OpenAI(model="gpt-4")
+
+# --- Build agent --- #
 agent = AgentWorkflow.from_tools_or_functions(
     TOOLS,
     llm=llm,
-    system_prompt="You are a helpful AI agent using external tools."
+    system_prompt="You are a helpful AI agent using reasoning and external tools to solve tasks."
 )
 
-import asyncio
+# --- Async run wrapper --- #
 async def answer_question(question: str) -> str:
-    return await agent.run(question)
+    result = await agent.run(question)
+    return str(result)