文本是《工具配置（共33篇）》专题的第 33 篇。阅读本文前，建议先阅读前面的文章：

doubao-Python示例代码

1. 更多教程及资源请访问一步AI官方社区站

社区站：https://www.yiboot.com
一键直达一步AI官方社区
长按识别下方二维码立即访问

2. 资源准备

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

3. 安装依赖

pip install openai

4. 实现代码

import openai

# 这里填写您在https://yibuapi.com上创建的apikey
api_key = ""
# base_url 只需要填写到 /v1，不要包含具体的端点
base_url = "https://yibuapi.com/v1"
# 这是问题
prompt = "帮我画一张登山的图画"

def get_openai_response(prompt, api_key, base_url):
    try:
        client = openai.OpenAI(api_key=api_key, base_url=base_url)

        # 使用正确的图像生成API
        response = client.images.generate(
            model="doubao-seedream-4-0-250828",
            prompt=prompt,
            n=1,
            size="1024x1024"  # 可选: 1024x1024, 1792x1024, 1024x1792
        )

        # 返回生成的图像URL
        return response.data[0].url
    except Exception as e:
        return f"请求失败: {str(e)}"

if __name__ == "__main__":
    image_url = get_openai_response(prompt, api_key, base_url)
    print(f"生成的图像URL: {image_url}\n")