Spaces:
Sleeping
Sleeping
File size: 15,117 Bytes
25c2b7e 9652ab4 25c2b7e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 | ---
title: test
colorFrom: purple
colorTo: gray
sdk: docker
app_port: 8890
emoji: 🚀
---
# OpenAI兼容的Sora API服务
这是一个为Sora提供OpenAI兼容接口的API服务。该服务使用cloudscraper绕过Cloudflare验证,支持多key轮询、并发处理和标准的OpenAI接口格式。
## 功能特点
- **OpenAI兼容接口**:完全兼容OpenAI的`/v1/chat/completions`接口
- **CF验证绕过**:使用cloudscraper库成功绕过Cloudflare验证
- **多key轮询**:支持多个Sora认证token,根据权重和速率限制智能选择
- **并发处理**:支持多个并发请求
- **流式响应**:支持SSE格式的流式响应
- **图像处理**:支持文本到图像生成和图像到图像生成(Remix)
- **异步处理**:支持异步生成图像,返回立即响应,防止请求超时
- **状态查询**:提供API端点查询异步任务的状态和结果
- **优化性能**:经过代码优化,提高请求处理速度和资源利用率
- **健康检查**:支持容器健康检查功能
## 环境要求
- Python 3.8+
- FastAPI 0.95.0+
- cloudscraper 1.2.71+
- 其他依赖见requirements.txt
## 快速部署指南
你可以直接运行不指定任何环境变量,所有环境变量都可以在面板里面配置
管理员登录密钥默认:sk-123456
api请求密钥不设置默认和管理员登录密钥相同
docker 一键运行
```bash
docker run -d -p 8890:8890 --name sora-api 1hei1/sora-api:latest
```
### 方法一:直接运行
1. 克隆仓库
```bash
git clone https://github.com/1hei1/sora-api.git
cd sora-api
```
2. 安装依赖
```bash
pip install -r requirements.txt
```
3. 配置API Keys(两种方式)
- **方式1**: 创建api_keys.json文件
```json
[
{"key": "Bearer your-sora-token-1", "weight": 1, "max_rpm": 60},
{"key": "Bearer your-sora-token-2", "weight": 2, "max_rpm": 60}
]
```
- **方式2**: 设置环境变量
```bash
# Linux/macOS
export API_KEYS='[{"key": "Bearer your-sora-token-1", "weight": 1, "max_rpm": 60}, {"key": "Bearer your-sora-token-2", "weight": 2, "max_rpm": 60}]'
# Windows (PowerShell)
$env:API_KEYS='[{"key": "Bearer your-sora-token-1", "weight": 1, "max_rpm": 60}, {"key": "Bearer your-sora-token-2", "weight": 2, "max_rpm": 60}]'
# Windows (CMD)
set API_KEYS=[{"key": "Bearer your-sora-token-1", "weight": 1, "max_rpm": 60}, {"key": "Bearer your-sora-token-2", "weight": 2, "max_rpm": 60}]
```
4. 配置代理(可选,如果需要)
```bash
# Linux/macOS - 基本代理
export PROXY_HOST=127.0.0.1
export PROXY_PORT=7890
# Linux/macOS - 带认证的代理
export PROXY_HOST=127.0.0.1
export PROXY_PORT=7890
export PROXY_USER=username
export PROXY_PASS=password
# Windows (PowerShell) - 基本代理
$env:PROXY_HOST="127.0.0.1"
$env:PROXY_PORT="7890"
# Windows (PowerShell) - 带认证的代理
$env:PROXY_HOST="127.0.0.1"
$env:PROXY_PORT="7890"
$env:PROXY_USER="username"
$env:PROXY_PASS="password"
# Windows (CMD) - 基本代理
set PROXY_HOST=127.0.0.1
set PROXY_PORT=7890
# Windows (CMD) - 带认证的代理
set PROXY_HOST=127.0.0.1
set PROXY_PORT=7890
set PROXY_USER=username
set PROXY_PASS=password
```
5. 启动服务
```bash
python run.py
```
6. 访问服务
- API服务地址: http://localhost:8890
- 后台管理面板: http://localhost:8890/admin
### 方法二:Docker部署
1. 构建Docker镜像
```bash
docker build -t sora-api .
```
2. 运行Docker容器(不同配置选项)
**基本运行方式**:
```bash
docker run -d -p 8890:8890 --name sora-api sora-api
```
**使用预打包镜像**:
```bash
docker run -d -p 8890:8890 --name sora-api 1hei1/sora-api:latest
```
**使用预打包镜像并配置API密钥**:
```bash
docker run -d -p 8890:8890 \
-e API_KEYS='[{"key": "Bearer your-sora-token-1", "weight": 1, "max_rpm": 60}, {"key": "Bearer your-sora-token-2", "weight": 2, "max_rpm": 60}]' \
--name sora-api \
1hei1/sora-api:v0.1
```
**使用预打包镜像并配置代理**:
```bash
docker run -d -p 8890:8890 \
-e API_KEYS='[{"key": "Bearer your-sora-token-1", "weight": 1, "max_rpm": 60}]' \
-e PROXY_HOST=host.docker.internal \
-e PROXY_PORT=7890 \
--name sora-api \
1hei1/sora-api:v0.1
```
**带API密钥配置**:
```bash
docker run -d -p 8890:8890 \
-e API_KEYS='[{"key": "Bearer your-sora-token-1", "weight": 1, "max_rpm": 60}, {"key": "Bearer your-sora-token-2", "weight": 2, "max_rpm": 60}]' \
--name sora-api \
sora-api
```
**带基本代理配置**:
```bash
docker run -d -p 8890:8890 \
-e API_KEYS='[{"key": "Bearer your-sora-token-1", "weight": 1, "max_rpm": 60}]' \
-e PROXY_HOST=host.docker.internal \
-e PROXY_PORT=7890 \
--name sora-api \
sora-api
```
**带认证代理配置**:
```bash
docker run -d -p 8890:8890 \
-e API_KEYS='[{"key": "Bearer your-sora-token-1", "weight": 1, "max_rpm": 60}]' \
-e PROXY_HOST=host.docker.internal \
-e PROXY_PORT=7890 \
-e PROXY_USER=username \
-e PROXY_PASS=password \
--name sora-api \
sora-api
```
**使用外部配置文件**:
```bash
# 首先确保api_keys.json文件已正确配置
docker run -d -p 8890:8890 \
-v $(pwd)/api_keys.json:/app/api_keys.json \
-e PROXY_HOST=host.docker.internal \
-e PROXY_PORT=7890 \
--name sora-api \
sora-api
```
**挂载本地目录保存图片**:
```bash
docker run -d -p 8890:8890 \
-v /your/local/path:/app/src/static/images \
--name sora-api \
sora-api
```
**启用详细日志**:
```bash
docker run -d -p 8890:8890 \
-e API_KEYS='[{"key": "Bearer your-sora-token-1", "weight": 1, "max_rpm": 60}]' \
-e VERBOSE_LOGGING=true \
--name sora-api \
sora-api
```
**注意**: 在Docker中使用宿主机代理时,请使用`host.docker.internal`而不是`127.0.0.1`作为代理主机地址。
3. 检查容器状态
```bash
docker ps
docker logs sora-api
```
4. 停止和移除容器
```bash
docker stop sora-api
docker rm sora-api
```
## 环境变量说明
| 环境变量 | 描述 | 默认值 | 示例 |
|---------|------|--------|------|
| `API_HOST` | API服务监听地址 | `0.0.0.0` | `127.0.0.1` |
| `API_PORT` | API服务端口 | `8890` | `9000` |
| `BASE_URL` | API基础URL(生成图片的时候需要用到) | `http://0.0.0.0:8890` | `https://api.example.com` |
| `PROXY_HOST` | HTTP代理主机 | 空(不使用代理) | `127.0.0.1` |
| `PROXY_PORT` | HTTP代理端口 | 空(不使用代理) | `7890` |
| `PROXY_USER` | HTTP代理用户名 | 空(不使用认证) | `username` |
| `PROXY_PASS` | HTTP代理密码 | 空(不使用认证) | `password` |
| `IMAGE_SAVE_DIR` | 图片保存目录 | `src/static/images` | `/data/images` |
| `IMAGE_LOCALIZATION` | 是否启用图片本地化 | `False` | `True` |
| `ADMIN_KEY` | 管理员API密钥(登录界面输入的密码) | `sk-123456` | `sk-youradminkey` |
| `API_AUTH_TOKEN` | API认证令牌(使用api服务传入的key) | 空 | `your-auth-token` |
| `VERBOSE_LOGGING` | 是否启用详细日志 | `False` | `True` |
## API密钥配置说明
API密钥配置采用JSON格式,每个密钥包含以下属性:
- `key`: Sora认证令牌(必须包含Bearer前缀)
- `weight`: 轮询权重,数字越大被选中概率越高
- `max_rpm`: 每分钟最大请求数(速率限制)
示例:
```json
[
{
"key": "Bearer eyJhbGciOiJSUzI1NiIsImtpZCI6IjE5MzQ0ZTY1LWJiYzktNDRkMS1hOWQwLWY5NTdiMDc5YmQwZSIsInR5cCI6IkpXVCJ9...",
"weight": 1,
"max_rpm": 60
},
{
"key": "Bearer eyJhbGciOiJSUzI1NiIsImtpZCI6IjE5MzQ0ZTY1LWJiYzktNDRkMS1hOWQwLWY5NTdiMDc5YmQwZSIsInR5cCI6IkpXVCJ9...",
"weight": 2,
"max_rpm": 60
}
]
```
## 使用示例
### 使用curl发送请求
```bash
# 文本到图像请求(非流式)
curl -X POST http://localhost:8890/v1/chat/completions \
-H "Content-Type: application/json" \
-H "Authorization: Bearer your-api-key" \
-d '{
"model": "sora-1.0",
"messages": [
{"role": "user", "content": "生成一只在草地上奔跑的金毛犬"}
],
"n": 1,
"stream": false
}'
# 文本到图像请求(流式)
curl -X POST http://localhost:8890/v1/chat/completions \
-H "Content-Type: application/json" \
-H "Authorization: Bearer your-api-key" \
-d '{
"model": "sora-1.0",
"messages": [
{"role": "user", "content": "生成一只在草地上奔跑的金毛犬"}
],
"n": 1,
"stream": true
}'
# 查询异步任务状态
curl -X GET http://localhost:8890/v1/generation/chatcmpl-123456789abcdef \
-H "Authorization: Bearer your-api-key"
```
## 常见问题排查
1. **连接超时或无法连接**
- 检查代理配置是否正确
- 如使用代理认证,确认用户名密码正确
- 确认Sora服务器是否可用
- 检查本地网络连接
2. **API密钥加载失败**
- 确认api_keys.json格式正确
- 检查环境变量API_KEYS是否正确设置
- 查看日志中的错误信息
3. **图片生成失败**
- 确认Sora令牌有效性
- 查看日志中的错误信息
- 检查是否超出账户额度限制
4. **Docker容器启动失败**
- 检查端口是否被占用
- 确认环境变量设置正确
- 查看Docker日志中的错误信息
5. **环境变量API_AUTH_TOKEN和ADMIN_KEY的区别**
- 环境变量 API_AUTH_TOKEN代表你在cheery studio 或者newapi里调用填写的令牌
- 环境变量 ADMIN_KEY 代表你管理面板的登录密码
- 当不设置API_AUTH_TOKE时 其值默认等于ADMIN_KEY的值
6. **环境变量BASE_URL的作用**
- 这个是图片本地化时设置的,比如生成了一张图片,获取到图片原始url,由于有的客户端不能访问sora,所以要本地化,这个base_url就是指定本地化图片的url前缀。
6. **token无效**
- 首次使用的token需要设置用户名,可以使用下面的脚本批量设置用户名:
```python
import random
import string
import logging
import cloudscraper
# Configure logging
tlogging = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
# --- Configuration ---
PROXY = {
"http": "http://127.0.0.1:7890",
"https": "http://127.0.0.1:7890"
}
PROFILE_API = "https://sora.chatgpt.com/backend/me"
TOKENS_FILE = "tokens.txt" # 每行一个 Bearer token
RESULTS_FILE = "update_results.txt" # 保存更新结果
USERNAME_LENGTH = 8 # 随机用户名长度
# --- Utilities ---
def random_username(length: int = USERNAME_LENGTH) -> str:
"""生成全小写随机用户名"""
return ''.join(random.choices(string.ascii_lowercase, k=length))
def sanitize_headers(headers: dict) -> dict:
"""
移除所有非 Latin-1 字符,确保 headers 可以被底层 HTTP 库正确编码。
"""
new = {}
for k, v in headers.items():
if isinstance(v, str):
new[k] = v.encode('latin-1', 'ignore').decode('latin-1')
else:
new[k] = v
return new
class SoraBatchUpdater:
def __init__(self, proxy: dict = None):
self.proxy = proxy or {}
def update_username_for_token(self, token: str) -> tuple[bool, str]:
"""
针对单个 Bearer token,生成随机用户名并发送更新请求。
返回 (success, message)。
"""
scraper = cloudscraper.create_scraper()
if self.proxy:
scraper.proxies = self.proxy
headers = {
"Accept": "*/*",
"Accept-Language": "zh-CN,zh;q=0.9",
"Content-Type": "application/json",
"Authorization": f"Bearer {token}",
"sec-ch-ua": '"Chromium";v="136", "Google Chrome";v="136", "Not.A/Brand";v="99"',
"sec-ch-ua-mobile": "?0",
"sec-ch-ua-platform": '"Windows"',
"sec-fetch-dest": "empty",
"sec-fetch-mode": "cors",
"sec-fetch-site": "same-origin",
}
headers = sanitize_headers(headers)
new_username = random_username()
payload = {"username": new_username}
try:
resp = scraper.post(
PROFILE_API,
headers=headers,
json=payload,
allow_redirects=False,
timeout=15
)
status = resp.status_code
if resp.ok:
msg = f"OK ({new_username})"
logging.info("Token %s: updated to %s", token[:6], new_username)
return True, msg
else:
text = resp.text.replace('\n', '')
msg = f"Failed {status}: {text}" # 简要错误信息
logging.warning("Token %s: %s", token[:6], msg)
return False, msg
except Exception as e:
msg = str(e)
logging.error("Token %s exception: %s", token[:6], msg)
return False, msg
def batch_update(self, tokens: list[str]) -> None:
"""
对一组 Bearer token 批量更新用户名,并将结果写入 RESULTS_FILE。
"""
results = []
for token in tokens:
success, message = self.update_username_for_token(token)
results.append((token, success, message))
# 写入结果文件
with open(RESULTS_FILE, 'w', encoding='utf-8') as f:
for token, success, msg in results:
status = 'SUCCESS' if success else 'ERROR'
f.write(f"{token} ---- {status} ---- {msg}\n")
logging.info("Batch update complete. Results saved to %s", RESULTS_FILE)
def load_tokens(filepath: str) -> list[str]:
"""从文件加载每行一个 token 的列表"""
try:
with open(filepath, 'r', encoding='utf-8') as f:
return [line.strip() for line in f if line.strip()]
except FileNotFoundError:
logging.error("Tokens file not found: %s", filepath)
return []
if __name__ == '__main__':
tokens = load_tokens(TOKENS_FILE)
if not tokens:
logging.error("No tokens to update. Exiting.")
else:
updater = SoraBatchUpdater(proxy=PROXY)
updater.batch_update(tokens)
```
## 性能优化
最新版本包含以下性能优化:
1. **代码重构**:简化了代码结构,提高可读性和可维护性
2. **内存优化**:减少不必要的内存使用,优化大型图像处理
3. **异步处理**:全面使用异步处理提高并发性能
4. **错误处理**:改进了错误处理和日志记录
5. **密钥管理**:优化了密钥轮询算法,提高了可靠性
6. **容器优化**:增强了Docker容器配置,支持健康检查
## 贡献
欢迎提交问题报告和改进建议!
## 许可证
MIT |