test_tracing_enable_disable() — langchain Function Reference

Architecture documentation for the test_tracing_enable_disable() function in test_tracing_interops.py from the langchain codebase.

Function python LangChainCore Runnables calls 1

Entity Profile

LangChainCore→ Runnables→ test_tracing_enable_disable() — langchain Function Reference

Dependency Diagram

graph TD
  9e0de3e4_2d1e_34b8_3248_f3e7a6f60f32["test_tracing_enable_disable()"]
  0944f1ae_0eea_e3a0_544b_8617adb53dbc["test_tracing_interops.py"]
  9e0de3e4_2d1e_34b8_3248_f3e7a6f60f32 -->|defined in| 0944f1ae_0eea_e3a0_544b_8617adb53dbc
  8c16b015_8380_d9f6_e698_f4a12f12a4f5["_get_posts()"]
  9e0de3e4_2d1e_34b8_3248_f3e7a6f60f32 -->|calls| 8c16b015_8380_d9f6_e698_f4a12f12a4f5
  style 9e0de3e4_2d1e_34b8_3248_f3e7a6f60f32 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/runnables/test_tracing_interops.py lines 230–259

def test_tracing_enable_disable(
    mock_get_client: MagicMock, *, enabled: bool | None, env: str
) -> None:
    mock_session = MagicMock()
    mock_client_ = Client(
        session=mock_session, api_key="test", auto_batch_tracing=False
    )
    mock_get_client.return_value = mock_client_

    def my_func(a: int) -> int:
        return a + 1

    if hasattr(get_env_var, "cache_clear"):
        get_env_var.cache_clear()  # type: ignore[attr-defined]
    env_on = env == "true"
    with (
        patch.dict("os.environ", {"LANGSMITH_TRACING": env}),
        tracing_context(enabled=enabled),
    ):
        RunnableLambda(my_func).invoke(1)

    mock_posts = _get_posts(mock_client_)
    if enabled is True:
        assert len(mock_posts) == 1
    elif enabled is False:
        assert not mock_posts
    elif env_on:
        assert len(mock_posts) == 1
    else:
        assert not mock_posts