fix: read text block and handle refusals in Anthropic provider

2026-11-01 10:22:08 +00:00 · 2026-10-26 12:07:18 +01:00
parent 85650ca0e8
commit f2ef3bebbd
2 changed files with 104 additions and 24 deletions
@@ -0,0 +1,65 @@
+import sys
+import unittest
+from pathlib import Path
+
+
+SCRIPTS_DIR = Path(__file__).resolve().parents[1]
+if str(SCRIPTS_DIR) not in sys.path:
+    sys.path.insert(0, str(SCRIPTS_DIR))
+
+from ai_providers.anthropic_provider import AnthropicProvider
+from ai_providers.base import AIProviderError
+
+
+class CannedAnthropicProvider(AnthropicProvider):
+    def __init__(self, response):
+        super().__init__(api_key="test-key", model="claude-opus-5-5")
+        self.response = response
+        self.captured_payload = None
+
+    def _make_request(self, url, payload, headers):
+        self.captured_payload = payload
+        return self.response
+
+
+class AnthropicProviderTests(unittest.TestCase):
+    def test_skips_thinking_block_before_text(self):
+        provider = CannedAnthropicProvider({
+            "stop_reason": "end_turn",
+            "content": [
+                {"type": "thinking", "thinking": "", "signature": "x"},
+                {"type": "text", "text": "  Hola  "},
+            ],
+        })
+        self.assertEqual(provider.generate("system", "user"), "Hola")
+
+    def test_thinking_only_response_raises(self):
+        provider = CannedAnthropicProvider({
+            "stop_reason": "max_tokens",
+            "content": [{"type": "thinking", "thinking": "", "signature": "x"}],
+        })
+        with self.assertRaises(AIProviderError):
+            provider.generate("system", "user")
+
+    def test_refusal_with_empty_content_raises(self):
+        provider = CannedAnthropicProvider({
+            "stop_reason": "refusal",
+            "stop_details": {"type": "refusal", "category": "cyber"},
+            "content": [],
+        })
+        with self.assertRaisesRegex(AIProviderError, "cyber"):
+            provider.generate("system", "user")
+
+    def test_small_max_tokens_is_raised_to_floor(self):
+        provider = CannedAnthropicProvider({
+            "stop_reason": "end_turn",
+            "content": [{"type": "text", "text": "CONNECTION_OK"}],
+        })
+        provider.generate("system", "user", max_tokens=50)
+        self.assertEqual(provider.captured_payload["max_tokens"], AnthropicProvider.MIN_MAX_TOKENS)
+        provider.generate("system", "user", max_tokens=4000)
+        self.assertEqual(provider.captured_payload["max_tokens"], 4000)
+
+
+if __name__ == "__main__":
+    unittest.main()