diff --git a/comfy/component_model/folder_path_types.py b/comfy/component_model/folder_path_types.py index dd75080b5..ac8750935 100644 --- a/comfy/component_model/folder_path_types.py +++ b/comfy/component_model/folder_path_types.py @@ -75,6 +75,9 @@ class FolderNames: def keys(self): return self.contents.keys() + def get(self, key, __default=None): + return self.contents.get(key, __default) + class SaveImagePathResponse(NamedTuple): full_output_folder: str