fallback.py

"""
AIPower fallback pattern — reliability through multi-provider redundancy.

If OpenAI goes down, fall back to DeepSeek. If DeepSeek fails, try Claude.
"""

import os
from openai import OpenAI

client = OpenAI(
    base_url="https://api.aipower.me/v1",
    api_key=os.environ["AIPOWER_API_KEY"],
)

# Fallback chain: prefer GPT, then Claude, then DeepSeek
MODELS = [
    "openai/gpt-5",
    "anthropic/claude-sonnet",
    "deepseek/deepseek-chat",
]


def chat_with_fallback(messages):
    for model in MODELS:
        try:
            r = client.chat.completions.create(model=model, messages=messages, timeout=30)
            return {"model_used": model, "response": r.choices[0].message.content}
        except Exception as e:
            print(f"[{model}] failed: {e}")
            continue
    raise RuntimeError("All models failed")


if __name__ == "__main__":
    result = chat_with_fallback([{"role": "user", "content": "Hello!"}])
    print(f"Used: {result['model_used']}")
    print(f"Response: {result['response']}")