mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-04 04:28:58 +00:00
infra: Clear cache for env-var checks (#26073)
This commit is contained in:
@@ -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):
|
||||
|
@@ -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():
|
||||
|
Reference in New Issue
Block a user