From 27b274c7ac85d9b094d1f784b06a609ce47d48f9 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 9 Dec 2023 23:37:44 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(types.py):=20remove=20unus?= =?UTF-8?q?ed=20import=20statement=20for=20logger=20to=20improve=20code=20?= =?UTF-8?q?cleanliness=20=F0=9F=94=A7=20chore(types.py):=20reformat=20impo?= =?UTF-8?q?rt=20statements=20to=20improve=20readability=20and=20adhere=20t?= =?UTF-8?q?o=20PEP8=20guidelines=20=F0=9F=94=A7=20chore(types.py):=20remov?= =?UTF-8?q?e=20unused=20optional=20parameter=20'nameless'=20from=20build?= =?UTF-8?q?=5Ffrontend=5Fnode=20function?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/types.py | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/src/backend/langflow/interface/types.py b/src/backend/langflow/interface/types.py index ab28a052b..c6d19c09d 100644 --- a/src/backend/langflow/interface/types.py +++ b/src/backend/langflow/interface/types.py @@ -8,8 +8,6 @@ from uuid import UUID from cachetools import LRUCache, cached from fastapi import HTTPException -from loguru import logger - from langflow.interface.agents.base import agent_creator from langflow.interface.chains.base import chain_creator from langflow.interface.custom.custom_component import CustomComponent @@ -31,8 +29,10 @@ from langflow.interface.vector_store.base import vectorstore_creator from langflow.interface.wrappers.base import wrapper_creator from langflow.template.field.base import TemplateField from langflow.template.frontend_node.constants import CLASSES_TO_REMOVE -from langflow.template.frontend_node.custom_components import CustomComponentFrontendNode +from langflow.template.frontend_node.custom_components import \ + CustomComponentFrontendNode from langflow.utils.util import get_base_classes +from loguru import logger # Used to get the base_classes list @@ -174,14 +174,10 @@ def extract_type_from_optional(field_type): return match[1] if match else None -def build_frontend_node(custom_component: CustomComponent, nameless: Optional[bool] = False): +def build_frontend_node(custom_component: CustomComponent): """Build a frontend node for a custom component""" try: - return ( - CustomComponentFrontendNode().to_dict_nameless().get(type(custom_component).__name__) - if nameless - else CustomComponentFrontendNode().to_dict().get(type(custom_component).__name__) - ) + return CustomComponentFrontendNode().to_dict().get(type(custom_component).__name__) except Exception as exc: logger.error(f"Error while building base frontend node: {exc}") @@ -344,12 +340,11 @@ def build_custom_component_template( custom_component: CustomComponent, user_id: Optional[Union[str, UUID]] = None, update_field: Optional[str] = None, - nameless: Optional[bool] = False, ) -> Optional[Dict[str, Any]]: """Build a custom component template for the langchain""" try: logger.debug("Building custom component template") - frontend_node = build_frontend_node(custom_component, nameless) + frontend_node = build_frontend_node(custom_component) if frontend_node is None: return None