From a248b96dba26e5e078b6535933e3527e949b4374 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Sat, 1 Jun 2024 14:34:48 -0700 Subject: [PATCH] refac --- main.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/main.py b/main.py index dd3381f5..6cec49c1 100644 --- a/main.py +++ b/main.py @@ -144,16 +144,17 @@ async def load_modules_from_directory(directory): if os.path.exists(valves_json_path): with open(valves_json_path, "r") as f: valves_json = json.load(f) - ValvesModel = pipeline.valves.__class__ - # Create a ValvesModel instance using default values and overwrite with valves_json - combined_valves = { - **pipeline.valves.model_dump(), - **valves_json, - } - valves = ValvesModel(**combined_valves) - pipeline.valves = valves + if hasattr(pipeline, "valves"): + ValvesModel = pipeline.valves.__class__ + # Create a ValvesModel instance using default values and overwrite with valves_json + combined_valves = { + **pipeline.valves.model_dump(), + **valves_json, + } + valves = ValvesModel(**combined_valves) + pipeline.valves = valves - logging.info(f"Updated valves for module: {module_name}") + logging.info(f"Updated valves for module: {module_name}") pipeline_id = pipeline.id if hasattr(pipeline, "id") else module_name PIPELINE_MODULES[pipeline_id] = pipeline