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

🔧 chore(types.py): remove unused import statement for logger to improve code cleanliness

Browse source 
🔧 chore(types.py): reformat import statements to improve readability and adhere to PEP8 guidelines
🔧 chore(types.py): remove unused optional parameter 'nameless' from build_frontend_node function
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-09 23:37:44 -03:00
commit 27b274c7ac
1 changed files with 6 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/backend/langflow/interface/types.py
View file

@ -8,8 +8,6 @@ from uuid import UUID
from cachetools import LRUCache, cached from cachetools import LRUCache, cached
from fastapi import HTTPException from fastapi import HTTPException
from loguru import logger
from langflow.interface.agents.base import agent_creator from langflow.interface.agents.base import agent_creator
from langflow.interface.chains.base import chain_creator from langflow.interface.chains.base import chain_creator
from langflow.interface.custom.custom_component import CustomComponent 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.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
# Used to get the base_classes list # 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 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""" """Build a frontend node for a custom component"""
try: try:
return ( return CustomComponentFrontendNode().to_dict().get(type(custom_component).__name__)
CustomComponentFrontendNode().to_dict_nameless().get(type(custom_component).__name__)
if nameless
else CustomComponentFrontendNode().to_dict().get(type(custom_component).__name__)
)
except Exception as exc: except Exception as exc:
logger.error(f"Error while building base frontend node: {exc}") logger.error(f"Error while building base frontend node: {exc}")
@ -344,12 +340,11 @@ def build_custom_component_template(
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,
nameless: Optional[bool] = False,
) -> Optional[Dict[str, Any]]: ) -> 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")
frontend_node = build_frontend_node(custom_component, nameless) frontend_node = build_frontend_node(custom_component)
if frontend_node is None: if frontend_node is None:
return None return None