test_count_tokens_converts_output_format_to_output_config() — anthropic-sdk-python Function Reference

Source Code

tests/api_resources/beta/test_output_format_conversion.py lines 136–158

    def test_count_tokens_converts_output_format_to_output_config(
        self, client: Anthropic, respx_mock: MockRouter
    ) -> None:
        """Verify .count_tokens() converts output_format to output_config.format."""
        respx_mock.post("/v1/messages/count_tokens?beta=true").mock(
            return_value=httpx.Response(200, json={"input_tokens": 10})
        )

        with warnings.catch_warnings(record=True):
            warnings.simplefilter("always")
            client.beta.messages.count_tokens(
                messages=[{"role": "user", "content": "Test"}],
                model="claude-sonnet-4-5",
                output_format={"type": "json_schema", "schema": {"type": "array"}},
            )

        request = respx_mock.calls.last.request
        body = json.loads(request.content)

        assert "output_config" in body
        assert "format" in body["output_config"]
        assert body["output_config"]["format"]["type"] == "json_schema"
        assert "output_format" not in body