From 17863c6e9347925258e5785ad80700085c2a5ffc Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Sat, 6 May 2023 08:30:25 -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 | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/interface/loading.py b/src/backend/langflow/interface/loading.py index 49cf9f698..2b2d8e2c0 100644 --- a/src/backend/langflow/interface/loading.py +++ b/src/backend/langflow/interface/loading.py @@ -74,7 +74,10 @@ 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)