From 80db2b92d2ac4e95a2b1bf0bfebbda84a0769f43 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=8Dtalo=20Johnny?= <italojohnnydosanjos@gmail.com>
Date: Wed, 30 Oct 2024 20:52:46 -0300
Subject: [PATCH] fix: suppress LangChainDeprecationWarning messages (#4319)

* fix: suppress LangChainDeprecationWarning messages

* fix: suppress LangChainDeprecationWarning messages
---
 .../langflow/components/embeddings/util/__init__.py   |  9 ++++++++-
 .../base/langflow/components/tools/__init__.py        | 11 +++++++++--
 src/backend/base/langflow/utils/validate.py           | 10 +++++++---
 3 files changed, 24 insertions(+), 6 deletions(-)

diff --git a/src/backend/base/langflow/components/embeddings/util/__init__.py b/src/backend/base/langflow/components/embeddings/util/__init__.py
index 1f85a6f04..5f630621e 100644
--- a/src/backend/base/langflow/components/embeddings/util/__init__.py
+++ b/src/backend/base/langflow/components/embeddings/util/__init__.py
@@ -1,3 +1,10 @@
-from .aiml import AIMLEmbeddingsImpl
+import warnings
+
+from langchain_core._api.deprecation import LangChainDeprecationWarning
+
+with warnings.catch_warnings():
+    warnings.simplefilter("ignore", LangChainDeprecationWarning)
+    from .aiml import AIMLEmbeddingsImpl
+
 
 __all__ = ["AIMLEmbeddingsImpl"]
diff --git a/src/backend/base/langflow/components/tools/__init__.py b/src/backend/base/langflow/components/tools/__init__.py
index ce623408b..a5b8a0d6c 100644
--- a/src/backend/base/langflow/components/tools/__init__.py
+++ b/src/backend/base/langflow/components/tools/__init__.py
@@ -1,5 +1,7 @@
-from .astradb import AstraDBToolComponent
-from .astradb_cql import AstraDBCQLToolComponent
+import warnings
+
+from langchain_core._api.deprecation import LangChainDeprecationWarning
+
 from .bing_search_api import BingSearchAPIComponent
 from .calculator import CalculatorToolComponent
 from .duck_duck_go_search_run import DuckDuckGoSearchComponent
@@ -17,6 +19,11 @@ from .wikipedia_api import WikipediaAPIComponent
 from .wolfram_alpha_api import WolframAlphaAPIComponent
 from .yahoo_finance import YfinanceToolComponent
 
+with warnings.catch_warnings():
+    warnings.simplefilter("ignore", LangChainDeprecationWarning)
+    from .astradb import AstraDBToolComponent
+    from .astradb_cql import AstraDBCQLToolComponent
+
 __all__ = [
     "AstraDBCQLToolComponent",
     "AstraDBToolComponent",
diff --git a/src/backend/base/langflow/utils/validate.py b/src/backend/base/langflow/utils/validate.py
index 4de5f9884..ab9672a55 100644
--- a/src/backend/base/langflow/utils/validate.py
+++ b/src/backend/base/langflow/utils/validate.py
@@ -1,9 +1,11 @@
 import ast
 import contextlib
 import importlib
+import warnings
 from types import FunctionType
 from typing import Optional, Union
 
+from langchain_core._api.deprecation import LangChainDeprecationWarning
 from loguru import logger
 from pydantic import ValidationError
 
@@ -221,9 +223,11 @@ def prepare_global_scope(code, module):
                     raise ModuleNotFoundError(msg) from e
         elif isinstance(node, ast.ImportFrom) and node.module is not None:
             try:
-                imported_module = importlib.import_module(node.module)
-                for alias in node.names:
-                    exec_globals[alias.name] = getattr(imported_module, alias.name)
+                with warnings.catch_warnings():
+                    warnings.simplefilter("ignore", LangChainDeprecationWarning)
+                    imported_module = importlib.import_module(node.module)
+                    for alias in node.names:
+                        exec_globals[alias.name] = getattr(imported_module, alias.name)
             except ModuleNotFoundError as e:
                 msg = f"Module {node.module} not found. Please install it and try again"
                 raise ModuleNotFoundError(msg) from e