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

@@ -6,6 +6,7 @@ from unittest.mock import MagicMock, patch
import pytest
from langsmith import Client, traceable
from langsmith.run_helpers import tracing_context
from langsmith.utils import get_env_var
from langchain_core.runnables.base import RunnableLambda, RunnableParallel
from langchain_core.tracers.langchain import LangChainTracer
@@ -31,6 +32,7 @@ def _get_posts(client: Client) -> list:
def test_config_traceable_handoff() -> None:
get_env_var.cache_clear()
mock_session = MagicMock()
mock_client_ = Client(
session=mock_session, api_key="test", auto_batch_tracing=False
@@ -186,6 +188,7 @@ def test_tracing_enable_disable(
def my_func(a: int) -> int:
return a + 1
get_env_var.cache_clear()
env_on = env == "true"
with patch.dict("os.environ", {"LANGSMITH_TRACING": env}):
with tracing_context(enabled=enabled):

View File

@@ -9,6 +9,7 @@ from uuid import UUID
import pytest
from langsmith import Client
from langsmith.run_trees import RunTree
from langsmith.utils import get_env_var, get_tracer_project
from langchain_core.outputs import LLMResult
from langchain_core.tracers.langchain import LangChainTracer
@@ -130,6 +131,8 @@ class LangChainProjectNameTest(unittest.TestCase):
]
for case in cases:
get_env_var.cache_clear()
get_tracer_project.cache_clear()
with self.subTest(msg=case.test_name):
with pytest.MonkeyPatch.context() as mp:
for k, v in case.envvars.items():