mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 15:03:16 +00:00
Adam Moussa
ae1f883b4c
Applies the plan's C5 step: git mv every test per the domain-reorg
move-map (movemap-m50.txt) into tests/{agent,analyzer,auth,dashboard,
github,middleware,models,reviewer,sandbox,slack,tools,webhooks}/, plus
the 13 fork-only placements from the scoping report §2c (Atlassian
webhook tests -> tests/webhooks/, test_atlassian_connect.py and
test_auth_error_leak.py -> tests/auth/, jira/confluence util tests ->
tests/tools/, test_repo_binding_isolation.py -> tests/sandbox/,
bot-identity/autofix tests -> tests/github/).
Path-only move: the only content edits are parents[1] -> parents[2]
fixes in test_e2b_integration.py and test_daytona_integration.py,
required because their __file__-relative ROOT path gained one more
directory level in the move.
Monkeypatch retargets for these files were already completed in C4;
none remained outstanding here.
103 lines
3.9 KiB
Python
103 lines
3.9 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
from langchain_anthropic import ChatAnthropic
|
|
from langchain_aws import ChatBedrockConverse
|
|
from langchain_core.messages import AIMessage, HumanMessage
|
|
|
|
from agent.middleware.sanitize_thinking_blocks import SanitizeThinkingBlocksMiddleware
|
|
|
|
|
|
def _make_request(messages: list[object], model: object | None = None) -> MagicMock:
|
|
request = MagicMock()
|
|
request.model = model or MagicMock(spec=ChatAnthropic)
|
|
request.messages = messages
|
|
return request
|
|
|
|
|
|
class TestSanitizeThinkingBlocksMiddleware:
|
|
def test_drops_empty_thinking_block_for_anthropic(self) -> None:
|
|
message = AIMessage(
|
|
content=[
|
|
{"type": "thinking", "signature": "abc", "thinking": ""},
|
|
{"type": "text", "text": "ok"},
|
|
]
|
|
)
|
|
request = _make_request([message])
|
|
response = MagicMock()
|
|
|
|
def handler(req: object) -> object:
|
|
assert req is request
|
|
return response
|
|
|
|
result = SanitizeThinkingBlocksMiddleware().wrap_model_call(request, handler)
|
|
|
|
assert result is response
|
|
assert message.content == [{"type": "text", "text": "ok"}]
|
|
|
|
def test_preserves_non_empty_thinking_block_for_anthropic(self) -> None:
|
|
thinking_block = {"type": "thinking", "signature": "abc", "thinking": "reasoning"}
|
|
text_block = {"type": "text", "text": "ok"}
|
|
message = AIMessage(content=[thinking_block, text_block])
|
|
request = _make_request([message])
|
|
|
|
SanitizeThinkingBlocksMiddleware().wrap_model_call(request, lambda req: MagicMock())
|
|
|
|
assert message.content == [thinking_block, text_block]
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_drops_missing_thinking_block_for_anthropic(self) -> None:
|
|
message = AIMessage(
|
|
content=[
|
|
{"type": "thinking", "signature": "abc"},
|
|
{"type": "text", "text": "ok"},
|
|
]
|
|
)
|
|
request = _make_request([HumanMessage(content="hi"), message])
|
|
response = MagicMock()
|
|
|
|
async def handler(req: object) -> object:
|
|
assert req is request
|
|
return response
|
|
|
|
result = await SanitizeThinkingBlocksMiddleware().awrap_model_call(request, handler)
|
|
|
|
assert result is response
|
|
assert message.content == [{"type": "text", "text": "ok"}]
|
|
|
|
def test_drops_empty_reasoning_content_block_for_bedrock(self) -> None:
|
|
message = AIMessage(
|
|
content=[
|
|
{"type": "reasoning_content", "reasoning_content": {"text": "", "signature": ""}},
|
|
{"type": "text", "text": "ok"},
|
|
]
|
|
)
|
|
request = _make_request([message], model=MagicMock(spec=ChatBedrockConverse))
|
|
|
|
SanitizeThinkingBlocksMiddleware().wrap_model_call(request, lambda req: MagicMock())
|
|
|
|
assert message.content == [{"type": "text", "text": "ok"}]
|
|
|
|
def test_preserves_non_empty_reasoning_content_block_for_bedrock(self) -> None:
|
|
reasoning_block = {
|
|
"type": "reasoning_content",
|
|
"reasoning_content": {"text": "because", "signature": "sig"},
|
|
}
|
|
text_block = {"type": "text", "text": "ok"}
|
|
message = AIMessage(content=[reasoning_block, text_block])
|
|
request = _make_request([message], model=MagicMock(spec=ChatBedrockConverse))
|
|
|
|
SanitizeThinkingBlocksMiddleware().wrap_model_call(request, lambda req: MagicMock())
|
|
|
|
assert message.content == [reasoning_block, text_block]
|
|
|
|
def test_ignores_non_anthropic_models(self) -> None:
|
|
thinking_block = {"type": "thinking", "signature": "abc", "thinking": ""}
|
|
message = AIMessage(content=[thinking_block, {"type": "text", "text": "ok"}])
|
|
request = _make_request([message], model=MagicMock())
|
|
|
|
SanitizeThinkingBlocksMiddleware().wrap_model_call(request, lambda req: MagicMock())
|
|
|
|
assert message.content == [thinking_block, {"type": "text", "text": "ok"}]
|