jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor(loading.py): add try-except block to handle KeyError when removing 'model' from params dictionary

Browse source
This commit is contained in:
Gabriel Almeida 2023-05-06 08:29:38 -03:00
commit bb41f5af5d
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/interface/loading.py
View file

@ -74,7 +74,12 @@ def instantiate_class(node_type: str, base_type: str, params: Dict) -> Any:
return loaded_toolkit return loaded_toolkit
elif base_type == "embeddings": elif base_type == "embeddings":
# ? Why remove model from params? # ? Why remove model from params?
try:
params.pop("model") params.pop("model")
except KeyError:
pass
# remove all params that are not in class_object.__fields__ # remove all params that are not in class_object.__fields__
try: try:
return class_object(**params) return class_object(**params)