DazedAnon
ba8b0e3bf6
- Join only content blocks that expose .text so live Anthropic calls no longer crash when thinking precedes the TextBlock - Add unit coverage for thinking-then-text response shapes
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""Tests for Anthropic content-block text extraction."""
|
|
|
|
import unittest
|
|
from types import SimpleNamespace
|
|
|
|
from util.translation import _anthropic_content_text
|
|
|
|
|
|
class AnthropicContentTextTests(unittest.TestCase):
|
|
def test_empty_content(self):
|
|
self.assertEqual(_anthropic_content_text(None), "")
|
|
self.assertEqual(_anthropic_content_text([]), "")
|
|
|
|
def test_text_block_only(self):
|
|
content = [SimpleNamespace(type="text", text='{"0":"Hello"}')]
|
|
self.assertEqual(_anthropic_content_text(content), '{"0":"Hello"}')
|
|
|
|
def test_skips_thinking_block_without_text(self):
|
|
# Mirrors Anthropic ThinkingBlock: has .thinking, no .text.
|
|
thinking = SimpleNamespace(type="thinking", thinking="internal notes")
|
|
text = SimpleNamespace(type="text", text='{"0":"Alice"}')
|
|
self.assertEqual(_anthropic_content_text([thinking, text]), '{"0":"Alice"}')
|
|
|
|
def test_accessing_first_block_text_would_fail(self):
|
|
thinking = SimpleNamespace(type="thinking", thinking="notes")
|
|
text = SimpleNamespace(type="text", text="ok")
|
|
content = [thinking, text]
|
|
with self.assertRaises(AttributeError):
|
|
_ = content[0].text
|
|
self.assertEqual(_anthropic_content_text(content), "ok")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|