wehub-resource-sync
b4fbd6fe9f
Deploy Site / deploy-vercel (push) Has been skipped
Deploy Site / deploy-docs (push) Has been skipped
Build Skills Index / build-index (push) Has been skipped
CI / Deny unrelated histories (push) Has been skipped
CI / Detect affected areas (push) Successful in 27m35s
CI / OSV scan (push) Failing after 4s
CI / Build&Test Docker image (push) Successful in 9s
CI / Supply-chain scan (push) Has been skipped
CI / Lint Docker scripts (push) Failing after 5m13s
CI / Check contributors (push) Failing after 12m8s
CI / Docs Site (push) Failing after 12m8s
CI / TypeScript (push) Failing after 12m8s
CI / Python lints (push) Failing after 12m9s
CI / Python tests (push) Failing after 12m9s
CI / Check uv.lock (push) Failing after 23m22s
CI / CI timing report (push) Has been cancelled
Build Skills Index / trigger-deploy (push) Has been cancelled
CI / All required checks pass (push) Has been cancelled
64 lines
2.5 KiB
Python
64 lines
2.5 KiB
Python
"""Tests for the 1M-context beta header on AWS Bedrock Claude models.
|
|
|
|
Claude Opus 4.6/4.7 and Sonnet 4.6 support a 1M context window, but on AWS
|
|
Bedrock (and Microsoft Foundry) that window is still gated behind the
|
|
``context-1m-2025-08-07`` beta header as of 2026-04. Without it, Bedrock
|
|
caps these models at 200K even though ``model_metadata.py`` advertises 1M.
|
|
|
|
These tests guard the invariant that the header is always emitted on the
|
|
Bedrock client path, and that it survives the MiniMax bearer-auth strip.
|
|
"""
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
|
|
class TestBedrockContext1MBeta:
|
|
"""``context-1m-2025-08-07`` must reach Bedrock Claude requests."""
|
|
|
|
|
|
|
|
def test_common_betas_strips_1m_for_minimax(self):
|
|
"""MiniMax bearer-auth endpoints host their own models — strip 1M beta."""
|
|
from agent.anthropic_adapter import (
|
|
_common_betas_for_base_url,
|
|
_CONTEXT_1M_BETA,
|
|
)
|
|
|
|
for url in (
|
|
"https://api.minimax.io/anthropic",
|
|
"https://api.minimaxi.com/anthropic",
|
|
):
|
|
betas = _common_betas_for_base_url(url)
|
|
assert _CONTEXT_1M_BETA not in betas, (
|
|
f"1M beta must be stripped for MiniMax bearer endpoint {url}"
|
|
)
|
|
# Other betas still present
|
|
assert "interleaved-thinking-2025-05-14" in betas
|
|
|
|
def test_build_anthropic_bedrock_client_sends_1m_beta(self):
|
|
"""AnthropicBedrock client must carry the 1M beta in default_headers.
|
|
|
|
This is the load-bearing assertion for the reported bug:
|
|
without this header Bedrock serves Opus 4.6/4.7 with a 200K cap.
|
|
"""
|
|
import agent.anthropic_adapter as adapter
|
|
|
|
fake_sdk = MagicMock()
|
|
fake_sdk.AnthropicBedrock = MagicMock()
|
|
|
|
with patch.object(adapter, "_anthropic_sdk", fake_sdk):
|
|
adapter.build_anthropic_bedrock_client(region="us-west-2")
|
|
|
|
call_kwargs = fake_sdk.AnthropicBedrock.call_args.kwargs
|
|
assert call_kwargs["aws_region"] == "us-west-2"
|
|
|
|
default_headers = call_kwargs.get("default_headers") or {}
|
|
beta_header = default_headers.get("anthropic-beta", "")
|
|
assert "context-1m-2025-08-07" in beta_header, (
|
|
"Bedrock client must send context-1m-2025-08-07 or Opus 4.6/4.7 "
|
|
"silently caps at 200K context"
|
|
)
|
|
# Other common betas still present — no regression.
|
|
assert "interleaved-thinking-2025-05-14" in beta_header
|
|
assert "fine-grained-tool-streaming-2025-05-14" in beta_header
|