Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/run_agent/test_anthropic_response_header_capture.py

@@ -0,0 +1,55 @@
+"""Anthropic Messages path must capture rate-limit + credits headers.
+
+Portal Claude moved off /chat/completions onto /v1/messages. The OpenAI-wire
+streaming path captured both header families from ``stream.response``; the
+Messages path must do the same via ``on_response`` or /status and the Nous
+429 classifier lose last-known state.
+"""
+
+from __future__ import annotations
+
+from types import SimpleNamespace
+from unittest.mock import MagicMock
+
+from run_agent import AIAgent
+
+
+def test_capture_anthropic_response_headers_forwards_to_both_captures():
+    agent = object.__new__(AIAgent)
+    agent._capture_rate_limits = MagicMock()
+    agent._capture_credits = MagicMock()
+
+    response = SimpleNamespace(headers={"x-ratelimit-remaining-requests": "10"})
+    agent._capture_anthropic_response_headers(response)
+
+    agent._capture_rate_limits.assert_called_once_with(response)
+    agent._capture_credits.assert_called_once_with(response)
+
+
+def test_anthropic_messages_create_passes_combined_header_callback():
+    agent = object.__new__(AIAgent)
+    agent.api_mode = "anthropic_messages"
+    agent._anthropic_client = MagicMock()
+    agent._disable_streaming = False
+    agent.log_prefix = ""
+    agent._try_refresh_anthropic_client_credentials = MagicMock(return_value=False)
+    agent._capture_anthropic_response_headers = MagicMock()
+
+    captured = {}
+
+    def _fake_create(client, api_kwargs, **kwargs):
+        captured.update(kwargs)
+        return SimpleNamespace(content=[], stop_reason="end_turn")
+
+    import agent.anthropic_adapter as adapter
+
+    original = adapter.create_anthropic_message
+    adapter.create_anthropic_message = _fake_create
+    try:
+        agent._anthropic_messages_create({"model": "anthropic/claude-opus-4.8"})
+    finally:
+        adapter.create_anthropic_message = original
+
+    assert (
+        captured.get("on_response") is agent._capture_anthropic_response_headers
+    )