Zenith-AI / test_deepseek.py
Shads229's picture
Upload 14 files
f15f315 verified
#!/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)