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

Refactor utils.py: Fix formatting and import statements

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-05 21:47:02 -03:00
commit 57d6a78d4e
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/backend/langflow/interface/custom/utils.py
View file

@ -15,12 +15,15 @@ from langflow.interface.custom.attributes import ATTR_FUNC_MAPPING
from langflow.interface.custom.code_parser.utils import extract_inner_type from langflow.interface.custom.code_parser.utils import extract_inner_type
from langflow.interface.custom.custom_component import CustomComponent from langflow.interface.custom.custom_component import CustomComponent
from langflow.interface.custom.directory_reader.utils import ( from langflow.interface.custom.directory_reader.utils import (
build_custom_component_list_from_path, determine_component_name, build_custom_component_list_from_path,
merge_nested_dicts_with_renaming) determine_component_name,
merge_nested_dicts_with_renaming,
)
from langflow.interface.custom.eval import eval_custom_component_code from langflow.interface.custom.eval import eval_custom_component_code
from langflow.template.field.base import TemplateField from langflow.template.field.base import TemplateField
from langflow.template.frontend_node.custom_components import \ from langflow.template.frontend_node.custom_components import (
CustomComponentFrontendNode CustomComponentFrontendNode,
)
from langflow.utils import validate from langflow.utils import validate
from langflow.utils.util import get_base_classes from langflow.utils.util import get_base_classes
@ -221,7 +224,7 @@ def add_extra_fields(frontend_node, field_config, function_args):
key in function_args_names for key in field_config.keys() key in function_args_names for key in field_config.keys()
): ):
for field_name, field_config in _field_config.copy().items(): for field_name, field_config in _field_config.copy().items():
if "name" not in extra_field: if "name" not in field_config or field_name == "code":
continue continue
config = _field_config.get(field_name, {}) config = _field_config.get(field_name, {})
config = config.model_dump() if isinstance(config, BaseModel) else config config = config.model_dump() if isinstance(config, BaseModel) else config