Update app.py

app.py

@@ -319,26 +319,41 @@ async def run_query(query: str):
             handler = web_agent.run(query, ctx=ctx)
             
             # Stream content
-            full_response = ""
             async for event in handler.stream_events():
+                # Add some debugging info to understand event structure
+                # print(f"Event type: {type(event)}")
+                # print(f"Event attrs: {dir(event)}")
+                
                 if isinstance(event, AgentStream):
                     # This is the text being generated
-                    if event.delta:
+                    if hasattr(event, 'delta') and event.delta:
                         yield event.delta
+                
                 elif isinstance(event, ToolCall):
-                    # A tool is being called
-                    tool_name = event.data.get("name", "unknown tool")
+                    # Handle ToolCall differently - check available attributes
+                    tool_name = "unknown tool"
+                    
+                    # Try different possible attribute locations
+                    if hasattr(event, 'name'):
+                        tool_name = event.name
+                    elif hasattr(event, 'function_name'):
+                        tool_name = event.function_name
+                    elif hasattr(event, 'tool_name'):
+                        tool_name = event.tool_name
+                    
                     yield f"\n\n🔧 Using tool: {tool_name}...\n"
+                
                 elif isinstance(event, ToolCallResult):
-                    # Result from a tool call
-                    # Try to find the tool name in the event data
-                    tool_name = "the tool"
-                    if hasattr(event, "parent_id") and event.parent_id:
-                        # This might help identify the parent tool call
-                        tool_name = f"tool {event.parent_id}"
+                    # Handle ToolCallResult - check available attributes
+                    result = "Result not available"
                     
-                    # Get the result from the event data
-                    result = event.data.get("result", "")
+                    # Try different possible attribute locations
+                    if hasattr(event, 'result'):
+                        result = event.result
+                    elif hasattr(event, 'output'):
+                        result = event.output
+                    elif hasattr(event, 'data') and hasattr(event.data, 'get'):
+                        result = event.data.get("result", "")
                     
                     # Truncate long results for display
                     if isinstance(result, str) and len(result) > 200:
@@ -346,10 +361,13 @@ async def run_query(query: str):
                     else:
                         result_preview = str(result)
                     
-                    yield f"\n📊 Got result from {tool_name}\n"
+                    yield f"\n📊 Got result from tool\n"
             
     except Exception as e:
-        yield f"\n\n❌ Error: {str(e)}"
+        yield f"\n\n❌ Error: {str(e)}\n"
+        # For debugging:
+        import traceback
+        yield f"Traceback: {traceback.format_exc()}"
     finally:
         instrumentor.flush()