import asyncio
import websockets
import json
import sys

async def test():
    print("🔌 Connexion WebSocket à ws://localhost:7081/ws...")
    try:
        async with websockets.connect('ws://localhost:7081/ws', open_timeout=10) as ws:
            print("✅ Connecté!")
            
            # Receive welcome message
            try:
                msg = await asyncio.wait_for(ws.recv(), timeout=15)
                print(f"📩 Message de bienvenue: {msg[:300]}")
            except asyncio.TimeoutError:
                print("⚠️  Pas de message de bienvenue (normal si pas d'event handler)")
            
            # Send a question
            print("📤 Envoi: 'Quelle heure est-il ?'")
            await ws.send("Quelle heure est-il ?")
            
            # Get response
            try:
                response = await asyncio.wait_for(ws.recv(), timeout=60)
                print(f"📩 Réponse: {response[:500]}")
            except asyncio.TimeoutError:
                print("❌ Timeout: pas de réponse du LLM en 60s")
                
    except Exception as e:
        print(f"❌ Erreur: {e}")

asyncio.run(test())