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

Fix loading.py to handle custom component keys properly

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-09 09:16:45 -03:00
commit 6b73ad2197
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -11,6 +11,9 @@ from langchain.chains.base import Chain
from langchain.document_loaders.base import BaseLoader from langchain.document_loaders.base import BaseLoader
from langchain_community.vectorstores import VectorStore from langchain_community.vectorstores import VectorStore
from langchain_core.documents import Document from langchain_core.documents import Document
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 eval_custom_component_code, get_function, 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
@ -22,8 +25,7 @@ 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
@ -127,7 +129,8 @@ def update_params_with_load_from_db_fields(custom_component, params, load_from_d
for field in load_from_db_fields: for field in load_from_db_fields:
if field in params: if field in params:
try: try:
params[field] = custom_component.keys(field) key = custom_component.keys(params[field])
params[field] = key if key else params[field]
except Exception as exc: except Exception as exc:
logger.error(f"Failed to get value for {field} from custom component. Error: {exc}") logger.error(f"Failed to get value for {field} from custom component. Error: {exc}")
pass pass