test_max_completion_tokens_parameter() — langchain Function Reference

Source Code

libs/partners/openai/tests/unit_tests/chat_models/test_azure.py lines 145–159

def test_max_completion_tokens_parameter() -> None:
    """Test that max_completion_tokens can be used as a direct parameter."""
    llm = AzureChatOpenAI(
        azure_deployment="gpt-5",
        api_version="2024-12-01-preview",
        azure_endpoint="my-base-url",
        max_completion_tokens=1500,
    )
    messages = [HumanMessage("Hello")]
    payload = llm._get_request_payload(messages)

    # Should use max_completion_tokens instead of max_tokens
    assert "max_completion_tokens" in payload
    assert payload["max_completion_tokens"] == 1500
    assert "max_tokens" not in payload