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

Refactored code by removing unused imports and

Browse source 
adding import statements inside functions.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-14 22:07:12 -03:00
commit 71300ba42e
2 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/interface/initialize/loading.py
View file

@ -13,8 +13,6 @@ from langchain.vectorstores.base import VectorStore
from loguru import logger from loguru import logger
from pydantic import ValidationError 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.custom_lists import CUSTOM_NODES
from langflow.interface.importing.utils import ( from langflow.interface.importing.utils import (
get_function, 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): 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"): if "retriever" in params and hasattr(params["retriever"], "as_retriever"):
params["retriever"] = params["retriever"].as_retriever() params["retriever"] = params["retriever"].as_retriever()
if node_type in chain_creator.from_method_nodes: 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): 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: if node_type in agent_creator.from_method_nodes:
method = agent_creator.from_method_nodes[node_type] method = agent_creator.from_method_nodes[node_type]
if class_method := getattr(class_object, method, None): if class_method := getattr(class_object, method, None):

5
src/backend/langflow/template/frontend_node/custom_components.py
View file

@ -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.field.base import TemplateField
from langflow.template.frontend_node.base import FrontendNode from langflow.template.frontend_node.base import FrontendNode
from langflow.template.template.base import Template from langflow.template.template.base import Template
from langflow.interface.custom.constants import DEFAULT_CUSTOM_COMPONENT_CODE
from typing import Optional
class CustomComponentFrontendNode(FrontendNode): class CustomComponentFrontendNode(FrontendNode):