Spaces:
Running
Running
File size: 4,237 Bytes
7ee5fa7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 |
#!/usr/bin/env python3
"""
Test script to verify API keys are working correctly
Run this before launching the main app to ensure all APIs are accessible
"""
import os
from dotenv import load_dotenv
import openai
import numpy as np
import soundfile as sf
import io
# Load environment variables
load_dotenv()
openai_api_key = os.environ.get("OPENAI_API_KEY")
def test_mistral():
"""Test Mistral API"""
try:
from mistralai import Mistral
api_key = os.environ.get("MISTRAL_API_KEY")
if not api_key:
print("❌ MISTRAL_API_KEY not found in environment")
return False
client = Mistral(api_key=api_key)
response = client.chat.complete(
model="mistral-large-latest",
messages=[{"role": "user", "content": "Say 'Hello' in French"}]
)
print(f"✅ Mistral API working: {response.choices[0].message.content[:50]}...")
return True
except Exception as e:
print(f"❌ Mistral API error: {str(e)}")
return False
def test_gemini():
"""Test Gemini API"""
try:
import google.generativeai as genai
api_key = os.environ.get("GEMINI_API_KEY")
if not api_key:
print("⚠️ GEMINI_API_KEY not found (optional fallback)")
return False
genai.configure(api_key=api_key)
model = genai.GenerativeModel("models/gemini-1.5-flash-latest")
response = model.generate_content("Say 'Hello' in French")
print(f"✅ Gemini API working: {response.text[:50]}...")
return True
except Exception as e:
print(f"⚠️ Gemini API error (fallback): {str(e)}")
return False
def test_groq():
"""Test Groq API"""
try:
from groq import Groq
client = Groq(api_key=os.environ.get("GROQ_API_KEY"),)
if not client:
print("❌ GROQ_API_KEY not found in environment")
return False
# client = Groq(api_key=api_key)
# Test with a simple completion
response = client.chat.completions.create(
messages=[
{
"role": "user",
"content": "Explain the importance of fast language models",
}
],
model="llama-3.3-70b-versatile",
)
print(f"✅ Groq API working: {response.choices[0].message.content}")
return True
except Exception as e:
print(f"❌ Groq API error: {str(e)}")
return False
def test_openai_whisper():
"""Test OpenAI Whisper API (STT)"""
if not openai_api_key:
print("⚠️ OPENAI_API_KEY not found (OpenAI Whisper fallback not available)")
return False
try:
# Generate a 0.5s dummy silent audio (16kHz mono)
sr = 16000
duration = 0.5
audio = np.zeros(int(sr * duration), dtype=np.float32)
buf = io.BytesIO()
sf.write(buf, audio, sr, format='WAV')
buf.seek(0)
openai.api_key = openai_api_key
response = openai.audio.transcriptions.create(
model="whisper-1",
file=("audio.wav", buf),
language="fr"
)
print(f"✅ OpenAI Whisper API working: {response.text}")
return True
except Exception as e:
print(f"❌ OpenAI Whisper API error: {str(e)}")
return False
def main():
print("🔍 Testing API Keys...\n")
mistral_ok = test_mistral()
gemini_ok = test_gemini()
groq_ok = test_groq()
openai_ok = test_openai_whisper()
print("\n📊 Summary:")
if mistral_ok and (groq_ok or openai_ok):
print("✅ All required APIs are working! You can run the app.")
elif not mistral_ok and gemini_ok and (groq_ok or openai_ok):
print("✅ Gemini fallback and Groq/OpenAI Whisper are working. The app will use Gemini for LLM.")
else:
print("❌ Some required APIs are not working. Please check your API keys.")
if not groq_ok and not openai_ok:
print(" - Groq or OpenAI Whisper is required for speech-to-text")
if not mistral_ok and not gemini_ok:
print(" - Either Mistral or Gemini is required for the language model")
if __name__ == "__main__":
main() |