OpenAI以1.25万亿美元估值完成110亿美元融资

import os
import requests
from typing import Dict, Any

class LLMRouter:
    def __init__(self):
        self.api_key = os.getenv("N1N_API_KEY")
        self.base_url = "https://api.n1n.ai/v1/chat/completions"
        self.headers = {
            "Authorization": f"Bearer {self.api_key}",
            "Content-Type": "application/json"
        }

    def generate_completion(self, prompt: str, primary_model: str = "openai/gpt-4o", fallback_model: str = "anthropic/claude-3-5-sonnet") -> Dict[Any, Any]:
        payload = {
            "model": primary_model,
            "messages": [{"role": "user", "content": prompt}],
            "temperature": 0.7
        }

        try:
            response = requests.post(self.base_url, json=payload, headers=self.headers, timeout=30)
            if response.status_code == 200:
                return response.json()
            else:
                print(f"主模型 {primary_model} 失败，状态码: {response.status_code}。正在触发备用模型...")
                return self._fallback_request(prompt, fallback_model)
        except requests.exceptions.RequestException as e:
            print(f"检测到网络异常: {e}。正在切换至备用模型...")
            return self._fallback_request(prompt, fallback_model)

    def _fallback_request(self, prompt: str, model: str) -> Dict[Any, Any]:
        payload = {
            "model": model,
            "messages": [{"role": "user", "content": prompt}],
            "temperature": 0.7
        }
        response = requests.post(self.base_url, json=payload, headers=self.headers, timeout=30)
        response.raise_for_status()
        return response.json()

# 使用示例
if __name__ == "__main__":
    router = LLMRouter()
    result = router.generate_completion("用通俗易懂的语言解释量子计算。")
    print(result.get('choices', [{}])[0].get('message', {}).get('content', '无响应内容'))