mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 12:43: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.
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
from langchain_core.messages import AIMessage, HumanMessage
|
|
|
|
from agent.middleware.sanitize_openai_responses import _sanitize_messages
|
|
|
|
|
|
def test_sanitize_messages_drops_stale_reasoning_references() -> None:
|
|
messages = [
|
|
HumanMessage(content="review this"),
|
|
AIMessage(
|
|
content=[
|
|
{
|
|
"type": "reasoning",
|
|
"id": "rs_123",
|
|
"summary": [],
|
|
"content": [],
|
|
},
|
|
{
|
|
"type": "function_call",
|
|
"id": "fc_123",
|
|
"call_id": "call_123",
|
|
"name": "read_file",
|
|
"arguments": "{}",
|
|
},
|
|
],
|
|
tool_calls=[
|
|
{
|
|
"name": "read_file",
|
|
"args": {},
|
|
"id": "call_123",
|
|
}
|
|
],
|
|
),
|
|
]
|
|
|
|
_sanitize_messages(messages)
|
|
|
|
content = messages[1].content
|
|
assert isinstance(content, list)
|
|
assert [block["type"] for block in content] == ["function_call"]
|
|
assert messages[1].tool_calls[0]["id"] == "call_123"
|
|
|
|
|
|
def test_sanitize_messages_preserves_encrypted_reasoning() -> None:
|
|
reasoning = {
|
|
"type": "reasoning",
|
|
"id": "rs_123",
|
|
"encrypted_content": "encrypted",
|
|
"summary": [],
|
|
"content": [],
|
|
}
|
|
messages = [AIMessage(content=[reasoning])]
|
|
|
|
_sanitize_messages(messages)
|
|
|
|
assert messages[0].content == [reasoning]
|