OpenAI-image-Python示例代码

1. 资源准备

• API Key：此项配置填写在一步API官网创建API令牌，一键直达API令牌创建页面
• 创建API令牌步骤请参考API Key的获取和使用
• API Host：此项配置填写https://yibuapi.com/v1
• 查看支持的模型请参考这篇教程模型在线查询

2. 安装依赖

pip install openai

3. 实现代码（gpt-image-1）

import os

from openai import OpenAI

# 测试gpt-4o-image-vip API (现在使用OpenAI的gpt-4o-image-vip)
def test_gpt_4o_image_vip():
    try:
        client_openai = OpenAI(
            base_url=os.getenv("OPENAI_API_BASE"),  # 如果使用代理需要设置
            api_key=os.getenv("OPENAI_API_KEY")
        )

        response = client_openai.images.generate(
            model="gpt-4o-image-vip",
            prompt="a white cat",
            n=1,
            size="1024x1024"
        )

        print("\ngpt-4o-image-vip测试成功，收到响应")
        if response.data:
            print(f"收到图片（测试不保存文件）")
        else:
            print("未收到图片数据")
    except Exception as e:
        print(f"gpt-4o-image-vip测试失败: {str(e)}")

if __name__ == "__main__":
    # 从环境变量读取配置
    os.environ["OPENAI_API_BASE"] = "https://yibuapi.com/v1"  # 可选
    os.environ["OPENAI_API_KEY"] = ""

    test_gpt_4o_image_vip()

4. 实现代码（dall-e-3）

import os

from openai import OpenAI

# 测试gpt-4o-image-vip API (现在使用OpenAI的gpt-4o-image-vip)
def test_gpt_4o_image_vip():
    try:
        client_openai = OpenAI(
            base_url=os.getenv("OPENAI_API_BASE"),  # 如果使用代理需要设置
            api_key=os.getenv("OPENAI_API_KEY")
        )

        response = client_openai.images.generate(
            model="dall-e-3",
            prompt="a white cat",
            n=1,
            size="1024x1024"
        )

        print("\ngpt-4o-image-vip测试成功，收到响应")
        if response.data:
            print(f"收到图片（测试不保存文件）")
        else:
            print("未收到图片数据")
    except Exception as e:
        print(f"gpt-4o-image-vip测试失败: {str(e)}")

if __name__ == "__main__":
    # 从环境变量读取配置
    os.environ["OPENAI_API_BASE"] = "https://yibuapi.com/v1"  # 可选
    os.environ["OPENAI_API_KEY"] = ""

    test_gpt_4o_image_vip()

5. 运行效果