From 1583283635612affb76d8a6cc5002be9d7e7c484 Mon Sep 17 00:00:00 2001
From: Qun <51054082+QunBB@users.noreply.github.com>
Date: Fri, 20 Sep 2024 17:42:47 +0800
Subject: [PATCH] ComfyUI tool use the new internal enumeration class
 "VariableKey" (#8533)

---
 .../provider/builtin/comfyui/tools/comfyui_stable_diffusion.py  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/api/core/tools/provider/builtin/comfyui/tools/comfyui_stable_diffusion.py b/api/core/tools/provider/builtin/comfyui/tools/comfyui_stable_diffusion.py
index 81fc8cc985..eaa4b0d027 100644
--- 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: