changes

test_mock_mode.sh

@@ -0,0 +1,122 @@
+#!/bin/bash
+#
+# Test Script for Mock Web Tools & WebSocket Reconnection
+#
+# This script tests:
+# 1. Mock web tools (no API calls, fake data)
+# 2. WebSocket timeout/reconnection during long operations
+# 3. Complete logging capture
+#
+# Perfect for development/testing without wasting API credits!
+
+set -e
+
+cd "$(dirname "$0")"
+
+echo "=========================================="
+echo "🧪 Mock Mode Test Script"
+echo "=========================================="
+echo ""
+
+# Check if logging server is running
+if ! curl -s http://localhost:8000/health > /dev/null 2>&1; then
+    echo "⚠️  Logging server not detected!"
+    echo "   Starting logging server in background..."
+    python api_endpoint/logging_server.py &
+    SERVER_PID=$!
+    echo "   Server PID: $SERVER_PID"
+    sleep 3
+else
+    echo "✅ Logging server already running"
+    SERVER_PID=""
+fi
+
+echo ""
+echo "📋 Test Configuration:"
+echo "   - Mock web tools: ENABLED"
+echo "   - Mock delay: 60 seconds (triggers WebSocket timeout)"
+echo "   - WebSocket logging: ENABLED"
+echo "   - Expected behavior: Connection timeout + auto-reconnect"
+echo ""
+echo "🔄 Running agent with mock mode..."
+echo "   (This will take ~60 seconds to test reconnection)"
+echo ""
+
+# Run agent with mock mode
+python run_agent.py \
+  --enabled_toolsets web \
+  --enable_websocket_logging \
+  --mock_web_tools \
+  --mock_delay 60 \
+  --query "Find publicly traded water companies benefiting from AI data centers"
+
+echo ""
+echo "=========================================="
+echo "✅ Test Complete!"
+echo "=========================================="
+echo ""
+
+# Find most recent log file
+LATEST_LOG=$(ls -t api_endpoint/logs/realtime/session_*.json 2>/dev/null | head -1)
+
+if [ -n "$LATEST_LOG" ]; then
+    echo "📊 Log Analysis:"
+    echo "   File: $LATEST_LOG"
+    echo ""
+    
+    # Count events
+    echo "   Event Counts:"
+    python3 -c "
+import json
+import sys
+
+with open('$LATEST_LOG') as f:
+    data = json.load(f)
+    events = data.get('events', [])
+    
+    # Count by type
+    counts = {}
+    for e in events:
+        etype = e.get('type', 'unknown')
+        counts[etype] = counts.get(etype, 0) + 1
+    
+    for etype, count in sorted(counts.items()):
+        print(f'     - {etype}: {count}')
+    
+    # Check completeness
+    has_complete = any(e.get('type') == 'complete' for e in events)
+    print()
+    if has_complete:
+        print('   ✅ Session completed successfully!')
+    else:
+        print('   ⚠️  Session incomplete (may have been interrupted)')
+    
+    # Check for reconnections
+    tool_results = [e for e in events if e.get('type') == 'tool_result']
+    tool_calls = [e for e in events if e.get('type') == 'tool_call']
+    
+    if len(tool_results) == len(tool_calls):
+        print('   ✅ All tool calls have results (no missing events)')
+    else:
+        print(f'   ⚠️  Tool calls: {len(tool_calls)}, Results: {len(tool_results)}')
+"
+else
+    echo "⚠️  No log files found"
+fi
+
+# Cleanup
+if [ -n "$SERVER_PID" ]; then
+    echo ""
+    echo "🛑 Stopping logging server (PID: $SERVER_PID)..."
+    kill $SERVER_PID 2>/dev/null || true
+fi
+
+echo ""
+echo "💡 Key Observations to Look For:"
+echo "   1. '[MOCK]' prefix on tool execution messages"
+echo "   2. '🔄 Reconnecting to logging server' after long tool"
+echo "   3. '✅ Reconnected successfully!' confirmation"
+echo "   4. Complete log file with all events captured"
+echo ""
+echo "🎉 Mock mode test completed!"
+