会问 AI API 调用全攻略：企业个人多模型切换需求满足

import requests
url = "https://api.huiwenai.com/v1/chat/completions"
headers = {
    "Authorization": "Bearer 你的密钥",
    "Content-Type": "application/json"
}
data = {
    "model": "gpt-3.5-turbo",
    "messages": [{"role": "user", "content": "推荐一本SEO入门书"}]
}
response = requests.post(url, json=data, headers=headers)
print(response.json())

$ch = curl_init();
curl_setopt($ch, CURLOPT_URL, "https://api.huiwenai.com/v1/chat/completions");
curl_setopt($ch, CURLOPT_POST, );
curl_setopt($ch, CURLOPT_HTTPHEADER, [
    "Authorization: Bearer 你的密钥",
    "Content-Type: application/json"
]);
curl_setopt($ch, CURLOPT_POSTFIELDS, json_encode([
    "model" => "ernie-bot",
    "messages" => [["role" => "user", "content" => "什么是API调用"]]
]));
curl_setopt($ch, CURLOPT_RETURNTRANSFER, );
$response = curl_exec($ch);
curl_close($ch);
print_r(json_decode($response, true));

model_map = {
    "simple_qa": "ernie-bot-turbo",
    "code": "code-llama",
    "marketing": "chatglm-pro",
    "image": "midjourney-api"
}

def get_model(task_type):
    return model_map.get(task_type, "gpt-3.5-turbo")  # 默认模型

from requests.adapters import HTTPAdapter
from urllib3.util.retry import Retry

session = requests.Session()
retry = Retry(total=, backoff_factor=0.5)  # 最多重试3次，每次间隔0.5秒
adapter = HTTPAdapter(max_retries=retry)
session.mount("https://", adapter)

def check_remaining_quota(api_key):
    url = "https://api.huiwenai.com/v1/quota"
    headers = {"Authorization": f"Bearer {api_key}"}
    response = requests.get(url, headers=headers)
    remaining = response.json().get("remaining_tokens", )
    if remaining < :
        print("⚠️ 免费额度快用完了，注意控制调用量！")
    return remaining