浏览代码

ComfyUI tool use the new internal enumeration class "VariableKey" (#8533)

Qun 7 月之前
父节点
当前提交
1583283635
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      api/core/tools/provider/builtin/comfyui/tools/comfyui_stable_diffusion.py

+ 1 - 1
api/core/tools/provider/builtin/comfyui/tools/comfyui_stable_diffusion.py

@@ -333,7 +333,7 @@ class ComfyuiStableDiffusionTool(BuiltinTool):
                         break
 
             return self.create_blob_message(
-                blob=image, meta={"mime_type": "image/png"}, save_as=self.VARIABLE_KEY.IMAGE.value
+                blob=image, meta={"mime_type": "image/png"}, save_as=self.VariableKey.IMAGE.value
             )
 
         except Exception as e: