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

refactor(loading.py): process params before instantiating class object

Browse source
This commit is contained in:
Gabriel Almeida 2023-05-02 16:39:33 -03:00
commit 7a50129149
1 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -33,7 +33,7 @@ def instantiate_class(node_type: str, base_type: str, params: Dict) -> Any:
if node_type in CUSTOM_AGENTS: if node_type in CUSTOM_AGENTS:
if custom_agent := CUSTOM_AGENTS.get(node_type): if custom_agent := CUSTOM_AGENTS.get(node_type):
return custom_agent.initialize(**params) # type: ignore return custom_agent.initialize(**params) # type: ignore
params = process_params(params)
class_object = import_by_type(_type=base_type, name=node_type) class_object = import_by_type(_type=base_type, name=node_type)
# check if it is a class before using issubclass # check if it is a class before using issubclass
@ -107,6 +107,15 @@ def instantiate_class(node_type: str, base_type: str, params: Dict) -> Any:
return class_object(**params) return class_object(**params)
def process_params(params):
"""Process params"""
if "allowed_special" in params:
params["allowed_special"] = set(params["allowed_special"])
if "disallowed_special" in params:
params["disallowed_special"] = set(params["disallowed_special"])
return params
def load_flow_from_json(path: str, build=True): def load_flow_from_json(path: str, build=True):
# This is done to avoid circular imports # This is done to avoid circular imports
from langflow.graph import Graph from langflow.graph import Graph