77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Quick script to test your LLM connection
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
|
|
def test_openai():
|
|
"""Test OpenAI connection"""
|
|
try:
|
|
from openai import OpenAI
|
|
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
|
|
response = client.chat.completions.create(
|
|
model="gpt-4o-mini",
|
|
messages=[{"role": "user", "content": "Say 'hello'"}]
|
|
)
|
|
print("✅ OpenAI: Connected")
|
|
print(f" Response: {response.choices[0].message.content}")
|
|
return True
|
|
except Exception as e:
|
|
print(f"❌ OpenAI: Failed - {e}")
|
|
return False
|
|
|
|
def test_anthropic():
|
|
"""Test Anthropic connection"""
|
|
try:
|
|
from anthropic import Anthropic
|
|
client = Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY"))
|
|
response = client.messages.create(
|
|
model="claude-3-5-haiku-20241022",
|
|
max_tokens=100,
|
|
messages=[{"role": "user", "content": "Say 'hello'"}]
|
|
)
|
|
print("✅ Anthropic: Connected")
|
|
print(f" Response: {response.content[0].text}")
|
|
return True
|
|
except Exception as e:
|
|
print(f"❌ Anthropic: Failed - {e}")
|
|
return False
|
|
|
|
def test_google():
|
|
"""Test Google Gemini connection"""
|
|
try:
|
|
from google import genai
|
|
client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
|
|
response = client.models.generate_content(
|
|
model='gemini-2.0-flash-exp',
|
|
contents="Say 'hello'"
|
|
)
|
|
print("✅ Google Gemini: Connected")
|
|
print(f" Response: {response.text}")
|
|
return True
|
|
except Exception as e:
|
|
print(f"❌ Google Gemini: Failed - {e}")
|
|
return False
|
|
|
|
if __name__ == "__main__":
|
|
print("🔍 Testing LLM connections...\n")
|
|
|
|
results = {
|
|
"OpenAI": test_openai(),
|
|
"Anthropic": test_anthropic(),
|
|
"Google": test_google()
|
|
}
|
|
|
|
print("\n" + "="*50)
|
|
working = [k for k, v in results.items() if v]
|
|
if working:
|
|
print(f"✅ Working providers: {', '.join(working)}")
|
|
else:
|
|
print("❌ No working providers found")
|
|
print("\nMake sure you've set environment variables:")
|
|
print(" export OPENAI_API_KEY=sk-...")
|
|
print(" export ANTHROPIC_API_KEY=sk-ant-...")
|
|
print(" export GEMINI_API_KEY=...")
|