From 71300ba42ec36f78321dddecfb901a31018f2db6 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 14 Nov 2023 22:07:12 -0300 Subject: [PATCH] Refactored code by removing unused imports and adding import statements inside functions. --- src/backend/langflow/interface/initialize/loading.py | 6 ++++-- .../langflow/template/frontend_node/custom_components.py | 5 +++-- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/interface/initialize/loading.py b/src/backend/langflow/interface/initialize/loading.py index bcf27b042..c3021531e 100644 --- a/src/backend/langflow/interface/initialize/loading.py +++ b/src/backend/langflow/interface/initialize/loading.py @@ -13,8 +13,6 @@ from langchain.vectorstores.base import VectorStore from loguru import logger from pydantic import ValidationError -from langflow.interface.agents.base import agent_creator -from langflow.interface.chains.base import chain_creator from langflow.interface.custom_lists import CUSTOM_NODES from langflow.interface.importing.utils import ( get_function, @@ -208,6 +206,8 @@ def instantiate_retriever(node_type, class_object, params): def instantiate_chains(node_type, class_object: Type[Chain], params: Dict): + from langflow.interface.chains.base import chain_creator + if "retriever" in params and hasattr(params["retriever"], "as_retriever"): params["retriever"] = params["retriever"].as_retriever() if node_type in chain_creator.from_method_nodes: @@ -220,6 +220,8 @@ def instantiate_chains(node_type, class_object: Type[Chain], params: Dict): def instantiate_agent(node_type, class_object: Type[agent_module.Agent], params: Dict): + from langflow.interface.agents.base import agent_creator + if node_type in agent_creator.from_method_nodes: method = agent_creator.from_method_nodes[node_type] if class_method := getattr(class_object, method, None): diff --git a/src/backend/langflow/template/frontend_node/custom_components.py b/src/backend/langflow/template/frontend_node/custom_components.py index e239775bc..60c3f0bc0 100644 --- a/src/backend/langflow/template/frontend_node/custom_components.py +++ b/src/backend/langflow/template/frontend_node/custom_components.py @@ -1,8 +1,9 @@ +from typing import Optional + +from langflow.interface.custom.constants import DEFAULT_CUSTOM_COMPONENT_CODE from langflow.template.field.base import TemplateField from langflow.template.frontend_node.base import FrontendNode from langflow.template.template.base import Template -from langflow.interface.custom.constants import DEFAULT_CUSTOM_COMPONENT_CODE -from typing import Optional class CustomComponentFrontendNode(FrontendNode):