3.5 KiB
3.5 KiB
Skills
本目录提供了调用 AI 文生图接口(text-to-image)的 Python 脚本,供其他 Agent 直接调用,无需自己实现 API 逻辑。
可用脚本
| 能力 | 脚本路径 | 模型 | 说明 |
|---|---|---|---|
| GPT 绘图 | gpt-image-t2i/generate_image.py |
gpt-image-2-1K / gpt-image-2-2K / gpt-image-2-4K |
通过 OpenAI 兼容接口生成图片 |
| Grok 绘图 | grok-image-t2i/generate_image.py |
grok-imagine-image-lite |
通过 xAI 兼容接口生成图片 |
两个脚本共用同一个 OpenAI 兼容接口:https://api.slomerex.xyz/v1/images/generations,API Key 已硬编码在各自脚本中,Agent 无需处理鉴权。
GPT 模型说明
| 模型 | 输出尺寸 | 备注 |
|---|---|---|
gpt-image-2-1K |
~1254×1254(1:1) | 推荐,线路稳定,适合头像等正方形用途 |
gpt-image-2-2K |
不固定 | 线路不稳定,可能 503 |
gpt-image-2-4K |
不固定 | 线路不稳定,可能 503 |
用
-m参数切换模型(在脚本中修改MODEL变量),不传则使用脚本默认模型。
输出规范(所有 Agent 必须遵守)
- 输出目录:本技能产生的所有文件(生成的图片等)统一保存到项目根目录的
output/文件夹中,不要输出到其他位置。 - 图片命名格式:每次生成的图片统一按
{调用的模型名}-{时间戳}.{后缀名}命名,例如:gpt-image-2-2K-20260818_143025.pnggrok-imagine-image-lite-20260818_143025.jpg- 其中
{调用的模型名}为实际调用的模型名(见上表),{时间戳}使用%Y%m%d_%H%M%S格式(如20260818_143025),{后缀名}按实际图片类型(.png/.jpg/.webp)。
- 脚本默认已按上述规范输出(
output/{模型名}-{时间戳}.{后缀名});如需自定义,通过-o指定,但仍应保持output/目录与「模型名-时间戳」命名格式。
使用方式
# 使用默认提示词(脚本内 PROMPT 变量)生成图片
python3 gpt-image-t2i/generate_image.py
python3 grok-image-t2i/generate_image.py
# 传入自定义提示词(不传则使用脚本内硬编码的 PROMPT)
python3 grok-image-t2i/generate_image.py "a cute cat sitting on a windowsill, soft morning light"
# 指定输出路径(必须输出到 output/ 文件夹,并按「模型名-时间戳」命名)
python3 grok-image-t2i/generate_image.py -o output/grok-imagine-image-lite-20260818_143025.jpg
# 指定图片尺寸(GPT 支持 1024x1024 / 1536x1024 等)
python3 gpt-image-t2i/generate_image.py --size 1536x1024
命令行参数
prompt:图片提示词(可选位置参数),不传则使用脚本内硬编码的PROMPT--size:图片尺寸,默认1024x1024--output, -o:输出文件路径,应指向output/下的文件,并按{模型名}-{时间戳}.{后缀名}命名(如-o output/gpt-image-2-2K-20260818_143025.png)
自定义提示词
优先通过命令行参数传入(如 python3 grok-image-t2i/generate_image.py "你的描述");如需修改默认提示词,编辑脚本中的 PROMPT 变量即可。建议使用详细、具体的英文描述以获得更好的效果。
注意事项
- 脚本依赖 Python 标准库(
urllib、base64等),无需安装第三方包。 - 请求图片下载时需要携带浏览器
User-Agent,否则接口返回 403(脚本已处理)。 - 生成图片通常需要几十秒,脚本设置了 120 秒超时。