From bb41f5af5db9a2262bd3df4a51a4354a7dec1e44 Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Sat, 6 May 2023 08:29:38 -0300 Subject: [PATCH] refactor(loading.py): add try-except block to handle KeyError when removing 'model' from params dictionary --- src/backend/langflow/interface/loading.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/interface/loading.py b/src/backend/langflow/interface/loading.py index 49cf9f698..737d06297 100644 --- a/src/backend/langflow/interface/loading.py +++ b/src/backend/langflow/interface/loading.py @@ -74,7 +74,12 @@ def instantiate_class(node_type: str, base_type: str, params: Dict) -> Any: return loaded_toolkit elif base_type == "embeddings": # ? Why remove model from params? - params.pop("model") + + try: + params.pop("model") + except KeyError: + pass + # remove all params that are not in class_object.__fields__ try: return class_object(**params)