AI
AI2API
返回首页

Python 示例

以下是使用 Python 调用 AI2API 的示例代码。

使用 requests 库

import requests

API_KEY = "YOUR_API_KEY"
BASE_URL = "https://api.ai2api.top"

def chat(message):
    headers = {
        "Content-Type": "application/json",
        "X-API-Key": API_KEY
    }
    
    payload = {
        "model": "gpt-3.5-turbo",
        "messages": [{"role": "user", "content": message}]
    }
    
    response = requests.post(
        f"{BASE_URL}/v1/chat/completions",
        headers=headers,
        json=payload
    )
    
    return response.json()["choices"][0]["message"]["content"]

# 使用示例
print(chat("你好，请介绍一下你自己"))

流式响应

import requests
import json

def chat_stream(message):
    headers = {
        "Content-Type": "application/json",
        "X-API-Key": "YOUR_API_KEY"
    }
    
    payload = {
        "model": "gpt-3.5-turbo",
        "messages": [{"role": "user", "content": message}],
        "stream": True
    }
    
    response = requests.post(
        "https://api.ai2api.top/v1/chat/completions",
        headers=headers,
        json=payload,
        stream=True
    )
    
    for line in response.iter_lines():
        if line:
            line = line.decode('utf-8')
            if line.startswith('data: '):
                data = line[6:]
                if data == '[DONE]':
                    break
                parsed = json.loads(data)
                content = parsed.get('choices', [{}])[0].get('delta', {}).get('content', '')
                print(content, end='', flush=True)

使用 OpenAI SDK

from openai import OpenAI

client = OpenAI(
    api_key="YOUR_API_KEY",
    base_url="https://api.ai2api.top/v1"
)

# 基础对话
response = client.chat.completions.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role": "system", "content": "你是一个专业的编程助手"},
        {"role": "user", "content": "如何用 Python 读取 CSV 文件？"}
    ]
)

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

异步调用

import asyncio
import aiohttp

async def chat_async(message):
    async with aiohttp.ClientSession() as session:
        async with session.post(
            "https://api.ai2api.top/v1/chat/completions",
            headers={"X-API-Key": "YOUR_API_KEY"},
            json={
                "model": "gpt-3.5-turbo",
                "messages": [{"role": "user", "content": message}]
            }
        ) as response:
            data = await response.json()
            return data["choices"][0]["message"]["content"]

# 使用示例
async def main():
    result = await chat_async("你好")
    print(result)

asyncio.run(main())