test_filter_injected_args_from_callbacks() — langchain Function Reference

Architecture documentation for the test_filter_injected_args_from_callbacks() function in test_tools.py from the langchain codebase.

Function python LangChainCore MessageInterface

Entity Profile

LangChainCore→ MessageInterface→ test_filter_injected_args_from_callbacks() — langchain Function Reference

Dependency Diagram

graph TD
  3045a63f_047f_48f2_1f7c_60916590b106["test_filter_injected_args_from_callbacks()"]
  8e7836ae_e72c_f670_72a5_4ca6d46e3555["test_tools.py"]
  3045a63f_047f_48f2_1f7c_60916590b106 -->|defined in| 8e7836ae_e72c_f670_72a5_4ca6d46e3555
  style 3045a63f_047f_48f2_1f7c_60916590b106 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/test_tools.py lines 2994–3025

def test_filter_injected_args_from_callbacks() -> None:
    """Test that injected tool arguments are filtered from callback inputs."""

    @tool
    def search_tool(
        query: str,
        state: Annotated[dict, InjectedToolArg()],
    ) -> str:
        """Search with injected state.

        Args:
            query: The search query.
            state: Injected state context.
        """
        return f"Results for: {query}"

    handler = CallbackHandlerWithInputCapture(captured_inputs=[])
    result = search_tool.invoke(
        {"query": "test query", "state": {"user_id": 123}},
        config={"callbacks": [handler]},
    )

    assert result == "Results for: test query"
    assert handler.tool_starts == 1
    assert len(handler.captured_inputs) == 1

    # Verify that injected 'state' arg is filtered out
    captured = handler.captured_inputs[0]
    assert captured is not None
    assert "query" in captured
    assert "state" not in captured
    assert captured["query"] == "test query"