wechat-robot-skills/skills/text-to-image/SKILL.md

5.1 KiB
Raw Blame History

name description argument-hint
text-to-image AI绘图工具当用户想通过文本生成图像时可以调用该工具。根据用户输入内容提取画图提示词选择合适的模型进行绘图返回生成的图片。 需要 prompt 参数(画图提示词),可选 model模型、negative_prompt反向提示词、ratio宽高比、resolution分辨率

Text To Image Skill

描述

这是一个 AI 文生图技能当用户想通过文本描述生成图像时触发。支持多个绘图模型即梦JiMeng、豆包DouBao、造相Z-Image

从数据库中读取绘图配置API 密钥、Base URL 等),根据用户选择的模型调用对应的绘图 API返回生成的图片 URL。

这个仓库里额外提供了一个可执行脚本 text-to-image/scripts/text_to_image.py,方便宿主机器人直接调用。

触发条件

  • 用户想画图、生成图片
  • 用户说「画一张……」「生成一张……的图片」「帮我画……」
  • 用户提到「文生图」「AI绘图」「AI画图」
  • 用户描述了想要生成的图片内容

参数说明JSON Schema

调用脚本时,需要通过第一个命令行参数传入 JSON 字符串,结构如下:

{
  "type": "object",
  "properties": {
    "prompt": {
      "type": "string",
      "description": "根据用户输入内容,提取出的画图提示词,但是不要对提示词进行总结。"
    },
    "model": {
      "type": "string",
      "description": "画图模型选择可选即梦4.5(jimeng-4.5) / 即梦4.6(jimeng-4.6) / 即梦5.0(jimeng-5.0) / 豆包4.5(doubao-seedream-4.5) / 豆包4.0(doubao-seedream-4.0) / 豆包文生图(doubao-seedream-3.0-t2i) / 豆包图生图(doubao-seededit-3.0-i2i) / 造相基础版(Z-Image) / 造相蒸馏版(Z-Image-Turbo) / 造相图片编辑(Qwen-Image-Edit-2511),默认: 空(none)。",
      "enum": [
        "none",
        "jimeng-4.5",
        "jimeng-4.6",
        "jimeng-5.0",
        "doubao-seedream-4.5",
        "doubao-seedream-4.0",
        "doubao-seedream-3.0-t2i",
        "doubao-seededit-3.0-i2i",
        "Z-Image",
        "Z-Image-Turbo",
        "Qwen-Image-Edit-2511"
      ],
      "default": "none"
    },
    "negative_prompt": {
      "type": "string",
      "description": "用于描述图像中不希望出现的元素或特征的文本,可选。"
    },
    "ratio": {
      "type": "string",
      "description": "图像的宽高比可选默认16:9。",
      "default": "16:9"
    },
    "resolution": {
      "type": "string",
      "description": "图像的分辨率可选默认2k。",
      "default": "2k"
    }
  },
  "required": ["prompt"],
  "additionalProperties": false
}

环境变量

  • ROBOT_CODE:机器人实例编码,用作数据库名称。
  • ROBOT_FROM_WX_ID:微信消息来源(群聊 ID 或好友微信 ID用于判断查询群聊配置还是好友配置。
  • MYSQL_HOSTMySQL 数据库地址。
  • MYSQL_PORTMySQL 数据库端口。
  • MYSQL_USERMySQL 数据库用户名。
  • MYSQL_PASSWORDMySQL 数据库密码。

依赖安装

  • 在执行 text-to-image/scripts/text_to_image.py 之前,必须先安装依赖。
  • 建议优先执行安装脚本:python3 skills/text-to-image/scripts/bootstrap.py
  • 该脚本会自动执行:python3 -m pip install -r skills/text-to-image/scripts/requirements.txt
  • 如果不使用安装脚本,也可以直接执行:python3 -m pip install -r skills/text-to-image/scripts/requirements.txt
  • 如果环境里已经安装过这些依赖,重复安装通常不会报错,只会做已满足检查。

执行步骤

  1. 当用户输入绘图相关内容时触发该技能。
  2. 从用户输入中提取 prompt画图提示词不对提示词做总结或修改。可选提取 model、negative_prompt、ratio、resolution 参数。
  3. 在执行脚本前,先安装依赖:python3 skills/text-to-image/scripts/bootstrap.py,或直接执行 python3 -m pip install -r skills/text-to-image/scripts/requirements.txt
  4. 将参数组装为 JSON 字符串,在仓库根目录下执行本地脚本:python3 text-to-image/scripts/text_to_image.py '<JSON参数>'
  5. 脚本内部执行逻辑:
    • 连接 MySQL 数据库(数据库名 = ROBOT_CODE)。
    • 查询 global_settings 表获取全局绘图配置(image_ai_enabledimage_ai_settings)。
    • 如果 ROBOT_FROM_WX_ID@chatroom 结尾,查询 chat_room_settings 表(WHERE chat_room_id = ?)覆盖全局配置;否则查询 friend_settings 表(WHERE wechat_id = ?)覆盖全局配置。
    • 检查绘图功能是否开启(image_ai_enabled)。
    • 解析 image_ai_settings JSON根据选择的模型提取对应配置JiMeng / DouBao / Z-Image
    • 调用对应的绘图 API 生成图片。
    • 输出图片 URL。
  6. 如果脚本执行失败,回复兜底文案:AI 绘图暂时不可用,请稍后再试。

回复要求

  • 成功时,脚本输出 <wechat-robot-image-url>图片URL</wechat-robot-image-url> 格式,直接发送图片,不要额外追加解释文字。
  • 失败时,使用固定兜底文案回复。