Update agent.py

agent.py

@@ -6,13 +6,12 @@ from youtube_tool import youtube_transcript_tool, youtube_transcript_snippet_too
 from multiple_tools import round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool, wikipedia_search_tool
 
 
-async def smart_agent(question : str, api_key : str) -> str:
+async def smart_agent(api_key : str) -> AgentWorkflow:
     llm = HuggingFaceInferenceAPI(
         model_name="deepseek-ai/DeepSeek-R1-0528", 
         token=api_key,
         provider="auto",
-        max_iterations=10,
-        max_execution_time=60
+        max_iterations=10
         )
     
     # Create Alfred, our gala agent, with the guest info tool
@@ -20,6 +19,4 @@ async def smart_agent(question : str, api_key : str) -> str:
         [wikipedia_search_tool, youtube_transcript_tool, youtube_transcript_snippet_tool, round_to_two_decimals_tool, text_inverter_tool, google_web_search_tool],
         llm=llm,
     )
-    # Example query Alfred might receive during the gala
-    response = await llamaindex_agent.run(question)
-    return response
+    return llamaindex_agent