dsh-comfyui-image:給 DeepSeek Harness 補上本地圖片生成能力

前言

DeepSeek Harness (DSH) 採用“一切皆插件”的架構,原生並未包含圖片生成能力。在開發智能體應用時,Agent 經常需要生成圖片。直接調用商業 API 存在成本和隱私問題,而部署 Stable Diffusion 管道通常比較複雜。

dsh-comfyui-image 插件解決了這個問題。它將本地運行的 ComfyUI 作爲後端,對外只暴露一個 Agent 工具 generate_image。開發者不需要關心 ComfyUI 的內部實現,只需通過這個工具即可實現文生圖、圖生圖、局部重繪和背景移除。

它是什麼

  • 插件名稱:dsh-comfyui-image
  • 維護者:wbb316
  • 核心價值:零第三方運行時依賴(僅使用 Node.js 內置 fetch / FormData),後端連接本地 ComfyUI,Agent 側只增加一個工具。

核心功能

該插件通過一個統一的 Agent 工具 generate_image 提供以下能力:

  • 文生圖:根據文本提示生成圖片。
  • 圖生圖:基於輸入圖片進行重繪。
  • 局部重繪:根據遮罩區域重繪圖片。
  • 背景移除:移除圖片背景。

插件自帶 doctor 自檢腳本和 88 項測試,確保環境正確。生成的圖片會附帶一個 .json 元數據 sidecar 文件,包含完整參數和復現工作流。

安裝與前置條件

在安裝插件前,需要滿足以下環境要求:

  • DSH:版本需支持工具註冊表。
  • Node.js:版本 >= 20(依賴內置的 fetch、FormData 和 Blob)。
  • ComfyUI:本地可訪問的服務,默認地址爲 http://127.0.0.1:8188,且 models/checkpoints 目錄下需有底模文件。

安裝命令如下:

dsh plugin --profile <你的 profile> add dsh-comfyui-image

Agent 用法

啓用插件後,Agent 會自動識別並可以使用 generate_image 工具。開發者只需輸入自然語言,Agent 會自動填充參數。

例如:

幫我生成一張柯基宇航員在月球上的圖,數字藝術風格

工具返回的結果中包含生成的圖片絕對路徑、元數據路徑以及詳細的生成參數。

配置機制

配置優先級從高到低依次爲:

  1. 插件配置(profile 的 cordis.patch.yml)
  2. 配置文件(~/.dsh-comfyui-image/config.json)
  3. 環境變量(如 COMFYUI_URL)
  4. 內置默認值

修改配置文件後無需重啓 DSH 即可生效。

注意事項

  • 背景移除:該功能需要 ComfyUI 安裝第三方去背節點(如 ComfyUI_essentials、ComfyUI-BRIA_AI-RMBG 等),否則無法使用。
  • 文件覆蓋:插件會自動處理文件覆蓋,如果目標文件已存在,會自動添加 -1、-2 等後綴。

結語

dsh-comfyui-image 爲 DSH 生態提供了低成本的本地圖片生成方案。它通過標準化的工具接口,讓 Agent 能夠方便地調用本地 ComfyUI 能力,適合對數據隱私和生成成本有較高要求的場景。

羽毛球分组比赛记分
小程序二维码

欢迎使用《羽毛球分组比赛记分》微信小程序

小夜