From bff0f3496ce931dc0c993b6c86ca8eaaed6ba10a Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 28 Nov 2023 23:00:20 -0300 Subject: [PATCH] Refactor langflow.interface.types.py --- src/backend/langflow/interface/types.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/interface/types.py b/src/backend/langflow/interface/types.py index 21432c756..9b9d19596 100644 --- a/src/backend/langflow/interface/types.py +++ b/src/backend/langflow/interface/types.py @@ -30,7 +30,8 @@ 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 @@ -341,7 +342,7 @@ def build_langchain_template_custom_component( custom_component: CustomComponent, user_id: Optional[Union[str, UUID]] = None, update_field: Optional[str] = None, -): +) -> Optional[Dict[str, Any]]: """Build a custom component template for the langchain""" try: logger.debug("Building custom component template")