Update app.py

app.py

@@ -15,6 +15,7 @@ from pydantic import BaseModel, Field
 import nest_asyncio
 import requests
 import os, re
+import asyncio
 #from agents.extensions.models.litellm_model import LitellmModel
 
 os.getenv("OPENAI_API_KEY")
@@ -133,7 +134,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-    def __call__(self, question: str) -> str:
+    async def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
 
         instructions = """
@@ -194,7 +195,9 @@ When you've found the answer, stop reasoning and return ONLY the answer.
 
         result = Runner.run_streamed(my_agent, input=question, max_turns=25)
 
-        for event in result.stream_events():
+        print("=== Run starting ===")
+    
+        async for event in result.stream_events():
             # We'll ignore the raw responses event deltas
             if event.type == "raw_response_event":
                 continue
@@ -213,6 +216,8 @@ When you've found the answer, stop reasoning and return ONLY the answer.
                 else:
                     pass  # Ignore other event types
 
+
+
         print(f"Agent gave answer (first 50 chars): {result.final_output[:50]}...")
         return result.final_output