From 1eb7c770a9cd0f40a3df2e634604ac0ef38a2003 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 22 Nov 2023 21:09:52 -0300 Subject: [PATCH] Refactor import statements and handle OpenAIEmbedding in instantiate_embedding function --- .../langflow/interface/initialize/loading.py | 20 +++++++------------ 1 file changed, 7 insertions(+), 13 deletions(-) diff --git a/src/backend/langflow/interface/initialize/loading.py b/src/backend/langflow/interface/initialize/loading.py index c3021531e..9f737e133 100644 --- a/src/backend/langflow/interface/initialize/loading.py +++ b/src/backend/langflow/interface/initialize/loading.py @@ -10,21 +10,10 @@ from langchain.chains.base import Chain from langchain.document_loaders.base import BaseLoader from langchain.schema import Document 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.importing.utils import ( - get_function, - get_function_custom, - import_by_type, -) +from langflow.interface.importing.utils import get_function, get_function_custom, import_by_type 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.output_parsers.base import output_parser_creator from langflow.interface.retrievers.base import retriever_creator @@ -32,6 +21,8 @@ from langflow.interface.toolkits.base import toolkits_creator from langflow.interface.utils import load_file_into_dict from langflow.interface.wrappers.base import wrapper_creator from langflow.utils import validate +from loguru import logger +from pydantic import ValidationError if TYPE_CHECKING: from langflow import CustomComponent @@ -277,6 +268,9 @@ def instantiate_embedding(node_type, class_object, params: Dict): if "VertexAI" in node_type: return initialize_vertexai(class_object=class_object, params=params) + if "OpenAIEmbedding" in node_type: + params["disallowed_special"] = () + try: return class_object(**params) except ValidationError: