#!/usr/bin/env python3
"""
Script de test pour vérifier la connexion à l'API DeepSeek
"""
import asyncio
import sys
from backend.core.engine import DeepSeekClient, DEEPSEEK_MODEL

async def test_deepseek():
    print("🔍 Test de connexion à DeepSeek API...")
    print(f"📡 URL: https://ds2api-tau-woad.vercel.app/v1/chat/completions")
    print(f"🤖 Modèle: {DEEPSEEK_MODEL}\n")

    client = DeepSeekClient()

    # Message de test simple
    messages = [
        {
            "role": "user",
            "content": "Réponds simplement 'OK' si tu me reçois."
        }
    ]

    print("📤 Envoi du message de test...")

    try:
        response_received = False
        async for chunk in client.stream_content(DEEPSEEK_MODEL, messages, {"temperature": 0.7}):
            if "error" in chunk:
                print(f"❌ Erreur: {chunk['error']}")
                return False

            if "response" in chunk:
                candidates = chunk.get("response", {}).get("candidates", [])
                if candidates:
                    for part in candidates[0].get("content", {}).get("parts", []):
                        text = part.get("text", "")
                        if text:
                            print(f"✅ Réponse reçue: {text}")
                            response_received = True

        if response_received:
            print("\n✅ Test réussi ! L'API DeepSeek fonctionne correctement.")
            return True
        else:
            print("\n⚠️ Aucune réponse reçue de l'API.")
            return False

    except Exception as e:
        print(f"\n❌ Erreur lors du test: {str(e)}")
        return False

if __name__ == "__main__":
    result = asyncio.run(test_deepseek())
    sys.exit(0 if result else 1)