github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-16 08:06:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

IMPROVEMENT: filter global warnings properly (#13754)

Browse Source
This commit is contained in:
Bagatur 2023-11-22 16:26:37 -08:00 committed by GitHub
parent 163bf165ed
commit 72c108b003
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 27 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
libs/core/langchain_core/globals/__init__.py
View File

@ -27,7 +27,7 @@ def set_verbose(value: bool) -> None:
warnings.filterwarnings(
"ignore",
message=(
"Importing verbose from langchain_core root module is no longer supported"
"Importing verbose from langchain root module is no longer supported"
),
)
# N.B.: This is a workaround for an unfortunate quirk of Python's
@ -55,7 +55,7 @@ def get_verbose() -> bool:
warnings.filterwarnings(
"ignore",
message=(
"Importing verbose from langchain_core root module is no longer supported"
".*Importing verbose from langchain root module is no longer supported"
),
)
# N.B.: This is a workaround for an unfortunate quirk of Python's
@ -87,7 +87,7 @@ def set_debug(value: bool) -> None:
with warnings.catch_warnings():
warnings.filterwarnings(
"ignore",
message="Importing debug from langchain_core root module is no longer supported",
message="Importing debug from langchain root module is no longer supported",
)
# N.B.: This is a workaround for an unfortunate quirk of Python's
# module-level `__getattr__()` implementation:
@ -113,7 +113,7 @@ def get_debug() -> bool:
with warnings.catch_warnings():
warnings.filterwarnings(
"ignore",
message="Importing debug from langchain_core root module is no longer supported",
message="Importing debug from langchain root module is no longer supported",
)
# N.B.: This is a workaround for an unfortunate quirk of Python's
# module-level `__getattr__()` implementation:
@ -145,7 +145,7 @@ def set_llm_cache(value: Optional["BaseCache"]) -> None:
warnings.filterwarnings(
"ignore",
message=(
"Importing llm_cache from langchain_core root module is no longer supported"
"Importing llm_cache from langchain root module is no longer supported"
),
)
# N.B.: This is a workaround for an unfortunate quirk of Python's
@ -173,7 +173,7 @@ def get_llm_cache() -> "BaseCache":
warnings.filterwarnings(
"ignore",
message=(
"Importing llm_cache from langchain_core root module is no longer supported"
"Importing llm_cache from langchain root module is no longer supported"
),
)
# N.B.: This is a workaround for an unfortunate quirk of Python's

21
libs/langchain/tests/unit_tests/test_globals.py
View File

@ -1,6 +1,27 @@
import warnings
from langchain_core.globals import get_debug as core_get_debug
from langchain_core.globals import get_verbose as core_get_verbose
from langchain_core.globals import set_debug as core_set_debug
from langchain_core.globals import set_verbose as core_set_verbose
from langchain.globals import get_debug, get_verbose, set_debug, set_verbose
def test_no_warning() -> None:
with warnings.catch_warnings():
warnings.simplefilter("error")
get_debug()
set_debug(False)
get_verbose()
set_verbose(False)
core_get_debug()
core_set_debug(False)
core_get_verbose()
core_set_verbose(False)
def test_debug_is_settable_directly() -> None:
from langchain_core.callbacks.manager import _get_debug