From 4a546dbab70f9adec34868b6cbb395a9154800a0 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Tue, 18 Jun 2024 23:07:43 -0300 Subject: [PATCH] refactor: Update loading.py to ignore Pydantic deprecation warnings The code changes in loading.py add a filter to ignore Pydantic deprecation warnings when building custom components or components. This refactor improves the code's maintainability and ensures compatibility with the latest Pydantic version. --- .../langflow/interface/initialize/loading.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/src/backend/base/langflow/interface/initialize/loading.py b/src/backend/base/langflow/interface/initialize/loading.py index 7c29ddf27..c6b933f51 100644 --- a/src/backend/base/langflow/interface/initialize/loading.py +++ b/src/backend/base/langflow/interface/initialize/loading.py @@ -1,10 +1,12 @@ import inspect import json import os +import warnings from typing import TYPE_CHECKING, Any, Type import orjson from loguru import logger +from pydantic import PydanticDeprecatedSince20 from langflow.custom.eval import eval_custom_component_code from langflow.schema import Data @@ -43,12 +45,14 @@ async def instantiate_class( params_copy = update_params_with_load_from_db_fields( custom_component, params_copy, vertex.load_from_db_fields, fallback_to_env_vars ) - if base_type == "custom_components": - return await build_custom_component(params=params_copy, custom_component=custom_component) - elif base_type == "component": - return await build_component(params=params_copy, custom_component=custom_component, vertex=vertex) - else: - raise ValueError(f"Base type {base_type} not found.") + with warnings.catch_warnings(): + warnings.filterwarnings("ignore", category=PydanticDeprecatedSince20) + if base_type == "custom_components": + return await build_custom_component(params=params_copy, custom_component=custom_component) + elif base_type == "component": + return await build_component(params=params_copy, custom_component=custom_component, vertex=vertex) + else: + raise ValueError(f"Base type {base_type} not found.") def convert_params_to_sets(params):