ChatGPT 广告业务年化收入突破 10 亿美元 推动全球 AI 普及新里程碑

import time
import requests
from typing import Dict, Any

class SmartLLMRouter:
    def __init__(self, api_key: str, base_url: str = "https://api.n1n.ai/v1"):
        self.api_key = api_key
        self.base_url = base_url
        self.headers = {
            "Authorization": f"Bearer {self.api_key}",
            "Content-Type": "application/json"
        }

    def execute_chat_completion(self, prompt: str, max_tokens: int = 1000) -> Dict[str, Any]:
        # 定义路由备份策略：主模型 (gpt-4o) -> 备用模型 (deepseek-v3)
        model_pipeline = [
            {"name": "gpt-4o", "timeout": 6.0},
            {"name": "deepseek-v3", "timeout": 10.0}
        ]
        
        payload = {
            "messages": [{"role": "user", "content": prompt}],
            "max_tokens": max_tokens,
            "temperature": 0.7
        }

        for model_config in model_pipeline:
            current_model = model_config["name"]
            timeout_limit = model_config["timeout"]
            payload["model"] = current_model
            
            try:
                print(f"正在尝试调用模型: {current_model}...")
                start_time = time.time()
                response = requests.post(
                    f"{self.base_url}/chat/completions",
                    json=payload,
                    headers=self.headers,
                    timeout=timeout_limit
                )
                elapsed_time = time.time() - start_time
                
                if response.status_code == 200:
                    data = response.json()
                    data["router_info"] = {
                        "selected_model": current_model,
                        "latency_seconds": round(elapsed_time, 3)
                    }
                    return data
                else:
                    print(f"警告: 模型 {current_model} 响应异常，状态码: {response.status_code}")
            
            except requests.exceptions.Timeout:
                print(f"超时: 模型 {current_model} 未能在 {timeout_limit} 秒内响应。")
            except Exception as error:
                print(f"错误: 无法连接至模型 {current_model}，详情: {str(error)}")
        
        raise RuntimeError("路由表中的所有 AI 模型均无法正常服务，请检查网络或配置。")

# 执行入口
if __name__ == "__main__":
    # 请将此处替换为您的 n1n.ai API 密钥
    MY_API_KEY = "your_n1n_api_key_here"
    llm_router = SmartLLMRouter(api_key=MY_API_KEY)
    
    test_prompt = "请分析广告补贴模式对开源大模型生态的影响。"
    try:
        api_response = llm_router.execute_chat_completion(test_prompt)
        print("\n--- 接口调用成功 ---")
        print(f"实际使用模型: {api_response['router_info']['selected_model']}")
        print(f"响应耗时: {api_response['router_info']['latency_seconds']} 秒")
        print(f"回答内容: {api_response['choices'][0]['message']['content'][:150]}...")
    except Exception as e:
        print(f"\n--- 调用失败 ---: {e}")