0fce82164a
Move provider adapters (anthropic, bedrock, azure), platform adapters (telegram, slack, discord, feishu, dingtalk, matrix), and terminal backends (modal, daytona) out of core into plugins/ workspace members. Core references them via the plugin registries (get_provider_namespace / get_provider_service / get_tool_provider / get_credential_pool_hook) instead of direct imports. - Provider/platform/terminal adapters relocated under plugins/; pyproject extras reference workspace members; nix variants aggregate per-platform extras. - Anthropic credential discovery + OAuth-masquerade guard live in the plugin's credential_pool_hook; browser-open guarded by _can_open_graphical_browser. - Vercel AI Gateway + Vercel Sandbox removed (upstream deletion); get_bedrock_model_ids removed (replaced by bedrock_model_ids_or_none + discover_bedrock_models). - Terminal backends resolve ModalEnvironment / DaytonaEnvironment lazily from the plugin registry. - uv.lock regenerated against the pluginified workspace. Plugin test suites updated for the relocation: imports point at hermes_agent_<plat>.adapter, caplog logger-name filters and monkeypatch targets use the new module paths, and credential/rollback tests patch registries.get_provider_service rather than the removed agent.*_adapter modules. Verified: zero dead imports of relocated modules in core (import smoke test + rename-map grep); nix develop succeeds; targeted plugin suites green (bedrock, anthropic-auxiliary, matrix, dingtalk, feishu, credential_pool, switch_model_rollback). Remaining full-suite failures are pre-existing on the pre-merge tree (telegram setUpModule __code__) or environmental (voice/media/ PTY/network-dependent), not introduced here.
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_format 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 hermes_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
|