jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
Qun 2024-09-20 17:42:47 +08:00 • committed by GitHub
commit 1583283635
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/tools/provider/builtin/comfyui/tools/comfyui_stable_diffusion.py
View file

@ -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: