Summary-Evaluator / src /api_clients.py
navaneethkrishnan's picture
Upload api_clients.py
9011d2b verified
raw
history blame contribute delete
745 Bytes
import os
from openai import OpenAI
import anthropic
import requests
openai_client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
claude_client = anthropic.Client(api_key=os.getenv("ANTHROPIC_API_KEY"))
deepseek_client = OpenAI(
api_key=os.getenv("DEEPSEEK_API_KEY"),
base_url="https://api.deepseek.com/v1"
)
def get_openrouter_models():
try:
headers = {
"Authorization": f"Bearer {os.getenv('OPENROUTER_API_KEY')}",
}
response = requests.get("https://openrouter.ai/api/v1/models", headers=headers)
data = response.json()
models = [m["id"] for m in data["data"] if m.get("id")]
return sorted(set(models))
except:
return ["openai/gpt-4", "openai/gpt-3.5-turbo"]