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

Refactor loading.py: Import changes and

Browse source 
instantiate_custom_component modification
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-23 08:40:14 -03:00
commit d7fdea2a89
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -10,8 +10,11 @@ from langchain.chains.base import Chain
from langchain.document_loaders.base import BaseLoader from langchain.document_loaders.base import BaseLoader
from langchain.schema import Document from langchain.schema import Document
from langchain.vectorstores.base import VectorStore from langchain.vectorstores.base import VectorStore
from loguru import logger
from pydantic import ValidationError
from langflow.interface.custom_lists import CUSTOM_NODES from langflow.interface.custom_lists import CUSTOM_NODES
from langflow.interface.importing.utils import get_function, get_function_custom, import_by_type from langflow.interface.importing.utils import eval_custom_component_code, get_function, import_by_type
from langflow.interface.initialize.llm import initialize_vertexai from langflow.interface.initialize.llm import initialize_vertexai
from langflow.interface.initialize.utils import handle_format_kwargs, handle_node_type, handle_partial_variables from langflow.interface.initialize.utils import handle_format_kwargs, handle_node_type, handle_partial_variables
from langflow.interface.initialize.vector_store import vecstore_initializer from langflow.interface.initialize.vector_store import vecstore_initializer
@ -21,8 +24,6 @@ from langflow.interface.toolkits.base import toolkits_creator
from langflow.interface.utils import load_file_into_dict from langflow.interface.utils import load_file_into_dict
from langflow.interface.wrappers.base import wrapper_creator from langflow.interface.wrappers.base import wrapper_creator
from langflow.utils import validate from langflow.utils import validate
from loguru import logger
from pydantic import ValidationError
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow import CustomComponent from langflow import CustomComponent
@ -119,7 +120,7 @@ def instantiate_custom_component(node_type, class_object, params, user_id):
# we need to make a copy of the params because we will be # we need to make a copy of the params because we will be
# modifying it # modifying it
params_copy = params.copy() params_copy = params.copy()
class_object: "CustomComponent" = get_function_custom(params_copy.pop("code")) class_object: "CustomComponent" = eval_custom_component_code(params_copy.pop("code"))
custom_component = class_object(user_id=user_id) custom_component = class_object(user_id=user_id)
built_object = custom_component.build(**params_copy) built_object = custom_component.build(**params_copy)
return built_object, {"repr": custom_component.custom_repr()} return built_object, {"repr": custom_component.custom_repr()}