Teknium
52 lines
2.3 KiB
Python
52 lines
2.3 KiB
Python
"""The final auxiliary destination owns message sanitation, not virtual MoA."""
|
|
import asyncio
|
|
import copy
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
from openai import AsyncOpenAI, OpenAI
|
|
|
|
from agent.auxiliary_client import _relay_async_completion, _relay_sync_completion
|
|
|
|
|
|
@pytest.mark.parametrize("async_mode", [False, True])
|
|
def test_auxiliary_chat_wire_sanitizes_without_mutating_history(async_mode):
|
|
history = [{"role": "assistant", "content": "answer", "_db_persisted": True,
|
|
"timestamp": 12, "tool_calls": [], "reasoning": "private"}]
|
|
original = copy.deepcopy(history)
|
|
kwargs = {"model": "fixture-model", "messages": history}
|
|
captured = []
|
|
if async_mode:
|
|
async def run():
|
|
async with AsyncOpenAI(api_key="fixture", base_url="http://127.0.0.1:1/v1") as client:
|
|
async def send(request):
|
|
captured.append(request)
|
|
await _relay_async_completion(client, kwargs, create=send)
|
|
asyncio.run(run())
|
|
else:
|
|
with OpenAI(api_key="fixture", base_url="http://127.0.0.1:1/v1") as client:
|
|
_relay_sync_completion(client, kwargs, create=captured.append)
|
|
assert captured[0]["messages"] == [
|
|
{"role": "assistant", "content": "answer", "reasoning": "private"}
|
|
]
|
|
assert kwargs["messages"] == original
|
|
|
|
|
|
@pytest.mark.parametrize("async_mode", [False, True])
|
|
def test_auxiliary_native_adapters_keep_replay_and_tool_fields(async_mode):
|
|
history = [{"role": "assistant", "content": "", "_db_persisted": True,
|
|
"codex_reasoning_items": [{"type": "reasoning", "id": "rs_fixture"}],
|
|
"thinking_blocks": [{"type": "thinking", "thinking": "native", "signature": "sig"}],
|
|
"tool_calls": [{"id": "call_fixture", "type": "function",
|
|
"function": {"name": "fixture", "arguments": "{}"}}]}]
|
|
kwargs = {"model": "native", "messages": history}
|
|
captured = []
|
|
if async_mode:
|
|
async def send(request):
|
|
captured.append(request)
|
|
asyncio.run(_relay_async_completion(SimpleNamespace(), kwargs, create=send))
|
|
else:
|
|
_relay_sync_completion(SimpleNamespace(), kwargs, create=captured.append)
|
|
assert captured[0] is kwargs
|
|
assert captured[0]["messages"] is history
|