infra: Clear cache for env-var checks (#26073)

This commit is contained in:
William FH
2024-09-06 14:29:29 -07:00
committed by GitHub
parent 854f37be87
commit 262e19b15d
7 changed files with 22 additions and 12 deletions

View File

@@ -7,6 +7,7 @@ from langchain_core.callbacks.manager import CallbackManager, trace_as_chain_gro
from langchain_core.outputs import LLMResult
from langchain_core.tracers.langchain import LangChainTracer, wait_for_all_tracers
from langchain_core.utils.pydantic import get_fields
from langsmith import utils as ls_utils
from langchain_community.callbacks import get_openai_callback
from langchain_community.callbacks.manager import get_bedrock_anthropic_callback
@@ -17,6 +18,8 @@ def test_callback_manager_configure_context_vars(
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""Test callback manager configuration."""
ls_utils.get_env_var.cache_clear()
ls_utils.get_tracer_project.cache_clear()
monkeypatch.setenv("LANGCHAIN_TRACING_V2", "true")
monkeypatch.setenv("LANGCHAIN_TRACING", "false")
monkeypatch.setenv("LANGCHAIN_API_KEY", "foo")