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

2024-09-20 17:42:47 +08:00
parent c87f710d58
commit 1583283635
1 changed files with 1 additions and 1 deletions
--- a/api/core/tools/provider/builtin/comfyui/tools/comfyui_stable_diffusion.py
+++ b/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: