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

tests/run_agent/test_anthropic_mid_tool_call_drop.py

@@ -0,0 +1,119 @@
+"""The Anthropic streaming path must not accept a tool_use block whose stream
+died before its input arrived (#80498 sibling).
+
+The chat_completions accumulator flags zero-byte tool-call arguments on a
+clean no-finish_reason stream end and routes them through the
+partial-stream-stub retry path. The Anthropic path had the same gap in a
+different shape: a clean SSE close after ``content_block_start(tool_use)``
+but before any ``input_json_delta``/``message_delta`` yields an SDK
+final-message snapshot whose content is NON-empty (the tool_use block is
+there, ``input={}``) and whose ``stop_reason`` is None — which sailed past
+both empty-stream guards and executed the tool with empty input, no retry.
+
+The fix raises EmptyStreamError for a tool_use-bearing message with no
+stop_reason, riding the same bounded stream-retry the eventless case uses.
+"""
+from types import SimpleNamespace
+from unittest.mock import MagicMock
+
+import pytest
+
+
+def _make_anthropic_agent(**kwargs):
+    from run_agent import AIAgent
+
+    defaults = dict(
+        api_key="test-key",
+        base_url="https://example.com/v1",
+        model="claude-opus-4-7",
+        quiet_mode=True,
+        skip_context_files=True,
+        skip_memory=True,
+    )
+    defaults.update(kwargs)
+    agent = AIAgent(**defaults)
+    agent.api_mode = "anthropic_messages"
+    agent._anthropic_client = MagicMock()
+    agent._anthropic_api_key = "test-anthropic-key"
+    agent._create_request_anthropic_client = lambda *a, **k: agent._anthropic_client
+    return agent
+
+
+def _stream_cm(final_message, events=()):
+    cm = MagicMock()
+    stream = MagicMock()
+    stream.__iter__ = MagicMock(return_value=iter(list(events)))
+    stream.get_final_message = MagicMock(return_value=final_message)
+    cm.__enter__ = MagicMock(return_value=stream)
+    cm.__exit__ = MagicMock(return_value=False)
+    return cm
+
+
+def _tool_use_block(name="write_file", input_obj=None):
+    return SimpleNamespace(
+        type="tool_use",
+        id="toolu_x",
+        name=name,
+        input=input_obj if input_obj is not None else {},
+    )
+
+
+def _tool_use_start_event(name="write_file"):
+    return SimpleNamespace(
+        type="content_block_start",
+        content_block=SimpleNamespace(type="tool_use", name=name),
+    )
+
+
+class TestAnthropicMidToolCallStreamDrop:
+    def test_tool_use_without_stop_reason_raises_empty_stream(self):
+        """The #80498-sibling shape: tool_use block present, stop_reason None."""
+        from agent.chat_completion_helpers import EmptyStreamError
+
+        dropped = MagicMock()
+        dropped.content = [_tool_use_block()]
+        dropped.stop_reason = None
+        dropped.usage = SimpleNamespace(input_tokens=10, output_tokens=2)
+
+        agent = _make_anthropic_agent()
+        agent._anthropic_client.messages.stream = MagicMock(
+            return_value=_stream_cm(dropped, events=[_tool_use_start_event()])
+        )
+
+        with pytest.raises(EmptyStreamError, match="tool_use"):
+            agent._interruptible_streaming_api_call({"model": "claude-opus-4-7"})
+
+    def test_completed_tool_use_with_stop_reason_passes(self):
+        """A legitimate tool_use completion (stop_reason set) is untouched."""
+        done = MagicMock()
+        done.content = [_tool_use_block(input_obj={"path": "a.txt"})]
+        done.stop_reason = "tool_use"
+        done.usage = SimpleNamespace(input_tokens=10, output_tokens=5)
+
+        agent = _make_anthropic_agent()
+        agent._anthropic_client.messages.stream = MagicMock(
+            return_value=_stream_cm(done, events=[_tool_use_start_event()])
+        )
+
+        response = agent._interruptible_streaming_api_call(
+            {"model": "claude-opus-4-7"}
+        )
+        assert response is done
+
+    def test_text_only_message_without_stop_reason_passes(self):
+        """No tool_use block -> the new gate stays out of the way (text-only
+        no-stop_reason handling keeps its pre-existing behavior)."""
+        text_only = MagicMock()
+        text_only.content = [SimpleNamespace(type="text", text="partial answer")]
+        text_only.stop_reason = None
+        text_only.usage = SimpleNamespace(input_tokens=10, output_tokens=5)
+
+        agent = _make_anthropic_agent()
+        agent._anthropic_client.messages.stream = MagicMock(
+            return_value=_stream_cm(text_only)
+        )
+
+        response = agent._interruptible_streaming_api_call(
+            {"model": "claude-opus-4-7"}
+        )
+        assert response is text_only