Refactor langflow.interface.types.py

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-28 23:00:20 -03:00
commit bff0f3496c

View file

@ -30,7 +30,8 @@ from langflow.interface.vector_store.base import vectorstore_creator
from langflow.interface.wrappers.base import wrapper_creator from langflow.interface.wrappers.base import wrapper_creator
from langflow.template.field.base import TemplateField from langflow.template.field.base import TemplateField
from langflow.template.frontend_node.constants import CLASSES_TO_REMOVE 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 langflow.utils.util import get_base_classes
from loguru import logger from loguru import logger
@ -341,7 +342,7 @@ def build_langchain_template_custom_component(
custom_component: CustomComponent, custom_component: CustomComponent,
user_id: Optional[Union[str, UUID]] = None, user_id: Optional[Union[str, UUID]] = None,
update_field: Optional[str] = None, update_field: Optional[str] = None,
): ) -> Optional[Dict[str, Any]]:
"""Build a custom component template for the langchain""" """Build a custom component template for the langchain"""
try: try:
logger.debug("Building custom component template") logger.debug("Building custom component template")