代码调用 gpt-image-2 生图 | OpenAI SDK 与 HTTP 双写法

你的 Python 代码
      │  POST /v1/images/generations
      │  （OpenAI 兼容格式 + Bearer 鉴权）
      ▼
中转服务 4SAPI（国内可达）
      │  转发 + 鉴权 + 计费
      ▼
gpt-image-2 生图模型
      │  返回图片（URL 或 base64）
      ▼
原路回传 → 你的代码保存/使用

# 先把 Key 放进环境变量（推荐）
# Windows PowerShell:
setx OPENAI_API_KEY "你的4sapi密钥"
# macOS / Linux:
export OPENAI_API_KEY="你的4sapi密钥"

pip install openai

import os
import base64
from openai import OpenAI

# base_url 指向 4SAPI 中转，api_key 从环境变量读取
client = OpenAI(
    api_key=os.getenv("OPENAI_API_KEY"),   # 你的 4SAPI 密钥
    base_url="https://4sapi.com/v1",
)

resp = client.images.generate(
    model="gpt-image-2",
    prompt="a cat sitting on a wooden floor, warm light",
    n=1,
    size="1024x1024",
)

# 响应可能返回 url，也可能返回 base64（b64_json），两种都兼容处理
item = resp.data[0]

if getattr(item, "url", None):
    print("图片 URL：", item.url)
elif getattr(item, "b64_json", None):
    img_bytes = base64.b64decode(item.b64_json)
    with open("gpt_image_sdk.png", "wb") as f:
        f.write(img_bytes)
    print("已保存为 gpt_image_sdk.png")

import http.client
import json

conn = http.client.HTTPSConnection("4sapi.com")

payload = json.dumps({
    "model": "gpt-image-2",
    "prompt": "cat",
    "n": 1
})

headers = {
    'Accept': 'application/json',
    'Authorization': 'Bearer YOUR_4SAPI_KEY',   # 注意 Bearer 前缀，换成你的真实密钥
    'Content-Type': 'application/json'
}

conn.request("POST", "/v1/images/generations", payload, headers)
res = conn.getresponse()
data = res.read()
text = data.decode("utf-8")
print(text)

# 把原始响应存成 json，方便后续解析/排查
with open("gpt_image_response.json", "w", encoding="utf-8") as f:
    f.write(text)

import base64
import urllib.request

result = json.loads(text)
item = result["data"][0]

if "url" in item and item["url"]:
    # 返回的是图片链接，下载下来
    urllib.request.urlretrieve(item["url"], "gpt_image_http.png")
    print("已下载为 gpt_image_http.png")
elif "b64_json" in item and item["b64_json"]:
    # 返回的是 base64，解码写文件
    with open("gpt_image_http.png", "wb") as f:
        f.write(base64.b64decode(item["b64_json"]))
    print("已保存为 gpt_image_http.png")

curl https://4sapi.com/v1/images/generations \
  -H "Authorization: Bearer YOUR_4SAPI_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "gpt-image-2",
    "prompt": "a cat sitting on a wooden floor",
    "n": 1
  }'

import OpenAI from "openai";
import fs from "fs";

const client = new OpenAI({
  apiKey: process.env.OPENAI_API_KEY,   // 你的 4SAPI 密钥
  baseURL: "https://4sapi.com/v1",
});

const resp = await client.images.generate({
  model: "gpt-image-2",
  prompt: "a cat sitting on a wooden floor, warm light",
  n: 1,
  size: "1024x1024",
});

const item = resp.data[0];
if (item.url) {
  console.log("图片 URL：", item.url);
} else if (item.b64_json) {
  fs.writeFileSync("gpt_image_node.png", Buffer.from(item.b64_json, "base64"));
  console.log("已保存为 gpt_image_node.png");
}

package main

import (
	"bytes"
	"fmt"
	"io"
	"net/http"
)

func main() {
	payload := []byte(`{"model":"gpt-image-2","prompt":"a cat","n":1}`)
	req, _ := http.NewRequest("POST",
		"https://4sapi.com/v1/images/generations",
		bytes.NewBuffer(payload))
	req.Header.Set("Authorization", "Bearer YOUR_4SAPI_KEY")
	req.Header.Set("Content-Type", "application/json")

	resp, err := http.DefaultClient.Do(req)
	if err != nil {
		panic(err)
	}
	defer resp.Body.Close()

	body, _ := io.ReadAll(resp.Body)
	fmt.Println(string(body))
}

import os
import time
import base64
import random
from openai import OpenAI

client = OpenAI(
    api_key=os.getenv("OPENAI_API_KEY"),
    base_url="https://4sapi.com/v1",
    timeout=60,          # 单次请求超时（秒）
)

def generate_one(prompt, filename, max_retries=4):
    """生成单张图，带指数退避重试。成功返回 True。"""
    for attempt in range(max_retries):
        try:
            resp = client.images.generate(
                model="gpt-image-2",
                prompt=prompt,
                n=1,
                size="1024x1024",
            )
            item = resp.data[0]
            if getattr(item, "b64_json", None):
                with open(filename, "wb") as f:
                    f.write(base64.b64decode(item.b64_json))
            elif getattr(item, "url", None):
                import urllib.request
                urllib.request.urlretrieve(item.url, filename)
            print(f"[OK] {filename}")
            return True
        except Exception as e:
            # 指数退避 + 随机抖动，避免被限流时一起重试
            wait = (2 ** attempt) + random.random()
            print(f"[重试 {attempt+1}/{max_retries}] {e}，{wait:.1f}s 后再试")
            time.sleep(wait)
    print(f"[FAIL] {filename} 最终失败")
    return False

# 批量任务：提示词 → 文件名
tasks = [
    ("a cat sitting on a wooden floor", "cat_01.png"),
    ("a dog running on grass", "dog_01.png"),
    ("a cup of coffee on a desk", "coffee_01.png"),
]

ok = 0
for prompt, filename in tasks:
    if generate_one(prompt, filename):
        ok += 1
    time.sleep(1)   # 简单限流：每张之间停 1 秒，按额度调整

print(f"完成 {ok}/{len(tasks)} 张")

弱提示词：cat
强提示词：a fluffy orange tabby cat sitting on a light wooden floor,
         photorealistic, warm afternoon light, shallow depth of field,
         no text, no watermark