diff --git a/backend/open_webui/config.py b/backend/open_webui/config.py index 84c99841d4..19e55db64e 100644 --- a/backend/open_webui/config.py +++ b/backend/open_webui/config.py @@ -3462,10 +3462,16 @@ COMFYUI_WORKFLOW = PersistentConfig( os.getenv("COMFYUI_WORKFLOW", COMFYUI_DEFAULT_WORKFLOW), ) +comfyui_workflow_nodes = os.getenv("COMFYUI_WORKFLOW_NODES", "") +try: + comfyui_workflow_nodes = json.loads(comfyui_workflow_nodes) +except json.JSONDecodeError: + comfyui_workflow_nodes = [] + COMFYUI_WORKFLOW_NODES = PersistentConfig( - "COMFYUI_WORKFLOW", + "COMFYUI_WORKFLOW_NODES", "image_generation.comfyui.nodes", - [], + comfyui_workflow_nodes, ) IMAGES_OPENAI_API_BASE_URL = PersistentConfig( @@ -3582,10 +3588,16 @@ IMAGES_EDIT_COMFYUI_WORKFLOW = PersistentConfig( os.getenv("IMAGES_EDIT_COMFYUI_WORKFLOW", ""), ) +images_edit_comfyui_workflow_nodes = os.getenv("IMAGES_EDIT_COMFYUI_WORKFLOW_NODES", "") +try: + images_edit_comfyui_workflow_nodes = json.loads(images_edit_comfyui_workflow_nodes) +except json.JSONDecodeError: + images_edit_comfyui_workflow_nodes = [] + IMAGES_EDIT_COMFYUI_WORKFLOW_NODES = PersistentConfig( "IMAGES_EDIT_COMFYUI_WORKFLOW_NODES", "images.edit.comfyui.nodes", - [], + images_edit_comfyui_workflow_nodes, ) ####################################