ComfyUI tool use the new internal enumeration class "VariableKey" (la…

…nggenius#8533)
cuiks · Sep 26, 2024 · d93e714 · d93e714
1 parent 8cc982c
commit d93e714
Showing 1 changed file with 1 addition and 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
@@ -333,7 +333,7 @@ def text2img(
                         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: