File size: 745 Bytes
9011d2b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
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"]