import os
from openai import OpenAI

# Initialize client pointing to your One-Router gateway
client = OpenAI(
    base_url="https://api.one-router.ai.one-bit.net/v1",
    api_key="sk-llm-your-api-key"
)

# 1. List Available Models
models = client.models.list()
print("Available Models:", [m.id for m in models.data])

# 2. Chat Completion
response = client.chat.completions.create(
    model="qwen3-32b",
    messages=[
        {"role": "system", "content": "You are a secure local AI."},
        {"role": "user", "content": "Explain zero trust architecture."}
    ],
    temperature=0.7
)

print(response.choices[0].message.content)

# 1. Check Health
curl -s https://api.one-router.ai.one-bit.net/health

# 2. Fetch Models
curl -s -X GET "https://api.one-router.ai.one-bit.net/v1/models" \
  -H "Authorization: Bearer sk-llm-your-key-here"

# 3. Chat Completion
curl -s -X POST "https://api.one-router.ai.one-bit.net/v1/chat/completions" \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer sk-llm-your-key-here" \
  -d '{
    "model": "qwen3-32b",
    "messages": [
      {"role": "user", "content": "Hello!"}
    ]
  }'