from openai import OpenAI

client = OpenAI(
    base_url="https://ai.megallm.io/v1",
    api_key="your-api-key"
)

# Use model ID, not display name
response = client.chat.completions.create(
    model="gpt-5",  # Model ID
    messages=[{"role": "user", "content": "Hello!"}]
)

# Switch to Claude using model ID
response = client.chat.completions.create(
    model="claude-opus-4-1-20250805",  # Model ID
    messages=[{"role": "user", "content": "Hello!"}]
)

# Try Gemini using model ID
response = client.chat.completions.create(
    model="gemini-2.5-pro",  # Model ID
    messages=[{"role": "user", "content": "Hello!"}]
)

// Always use model IDs
const models = ['gpt-5', 'claude-opus-4-1-20250805', 'gemini-2.5-pro'];

for (const modelId of models) {
  const response = await fetch("https://ai.megallm.io/v1/chat/completions", {
    method: "POST",
    headers: {
      "Authorization": `Bearer ${API_KEY}`,
      "Content-Type": "application/json"
    },
    body: JSON.stringify({
      model: modelId,  // Using model ID
      messages: [{ role: "user", content: "Hello!" }]
    })
  });

  console.log(`${modelId} response:`, await response.json());
}

# Test multiple models using their IDs
for model in "gpt-5" "claude-opus-4-1-20250805" "gemini-2.5-pro"; do
  echo "Testing $model..."
  curl https://ai.megallm.io/v1/chat/completions \
    -H "Authorization: Bearer $API_KEY" \
    -H "Content-Type: application/json" \
    -d "{
      \"model\": \"$model\",
      \"messages\": [{\"role\": \"user\", \"content\": \"Hello!\"}]
    }"
done