/home/mrjunejune/zenbu: mrjunejune/inference/copilot_sidecar

annotate mrjunejune/inference/copilot_sidecar_test.py @ 279:b3b547563ec7

Add Google connector service and agent wiki Implement the C/Seobeo Google Drive and Gmail connector with encrypted OAuth storage, Zenbu authentication, browser testing, AI tool discovery, chunked HTTP decoding, and Bazel coverage. Consolidate repository guidance into progressive wiki documentation and enforce arena-first allocation for new first-party C code. Co-authored-by: Copilot <[email protected]> Copilot-Session: 84c338fd-0939-4bb3-b7f3-1062eb213e5d

author	MrJuneJune <me@mrjunejune.com>
date	Mon, 17 Aug 2026 22:22:36 -0700
parents	056790c4fb0d
children	49e9e591c9bb

rev	line source
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	1 import asyncio
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	2 import hashlib
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	3 import types
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	4 import unittest
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	5 import uuid
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	6 from dataclasses import replace
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	7
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	8 from mrjunejune.inference.copilot_sidecar import (
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	9 CompiledProfile,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	10 Sidecar,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	11 SidecarConfig,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	12 _derive_sdk_session_id,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	13 _SDK_SESSION_NAMESPACE,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	14 _validate_history,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	15 _HISTORY_MAX_ENTRIES,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	16 _HISTORY_MAX_BYTES,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	17 )
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	18
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	19
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	20 def event(event_type, **data):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	21 return types.SimpleNamespace(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	22 type=types.SimpleNamespace(value=event_type),
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	23 data=types.SimpleNamespace(**data),
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	24 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	25
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	26
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	27 # ---------------------------------------------------------------------------
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	28 # Fake compiled-profile infrastructure
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	29 # ---------------------------------------------------------------------------
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	30
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	31 def _make_content(profile: str) -> str:
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	32 return f"You are the {profile} assistant. Common rules apply."
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	33
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	34
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	35 _FAKE_PROFILES: dict = {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	36 profile: {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	37 "content": _make_content(profile),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	38 "version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	39 "hash": hashlib.sha256(_make_content(profile).encode()).hexdigest(),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	40 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	41 for profile in ("public_visitor", "invited_friend", "june_admin")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	42 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	43
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	44
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	45 def _fake_compile_fn(profile: str) -> dict:
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	46 if profile not in _FAKE_PROFILES:
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	47 raise ValueError(f"Unknown profile: {profile!r}")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	48 return _FAKE_PROFILES[profile]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	49
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	50
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	51 def _fake_sdk_id(conversation_id: str, profile: str) -> str:
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	52 """Compute the deterministic SDK session ID for use in test assertions."""
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	53 p = _FAKE_PROFILES[profile]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	54 compiled = CompiledProfile(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	55 profile=profile,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	56 content=p["content"],
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	57 prompt_version=1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	58 knowledge_version=p["version"],
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	59 hash=p["hash"],
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	60 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	61 return _derive_sdk_session_id(conversation_id, compiled)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	62
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	63
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	64 class FakeSession:
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	65 def __init__(self, session_id, behavior="complete"):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	66 self.session_id = session_id
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	67 self.behavior = behavior
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	68 self.handlers = []
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	69 self.prompts = []
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	70 self.abort_calls = 0
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	71 self.disconnect_calls = 0
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	72 self.disconnect_error = False
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	73
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	74 def on(self, handler):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	75 self.handlers.append(handler)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	76
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	77 def unsubscribe():
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	78 if handler in self.handlers:
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	79 self.handlers.remove(handler)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	80
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	81 return unsubscribe
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	82
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	83 def emit(self, value):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	84 for handler in list(self.handlers):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	85 handler(value)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	86
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	87 async def send(self, prompt):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	88 self.prompts.append(prompt)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	89 message_id = f"message-{len(self.prompts)}"
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	90 if self.behavior == "complete":
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	91 self.emit(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	92 event(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	93 "assistant.message_delta",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	94 delta_content=f"{prompt}-delta",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	95 message_id=message_id,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	96 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	97 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	98 self.emit(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	99 event(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	100 "assistant.message",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	101 content=f"{prompt}-answer",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	102 message_id=message_id,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	103 model="test-model",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	104 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	105 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	106 self.emit(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	107 event(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	108 "assistant.usage",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	109 model="test-model",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	110 input_tokens=3,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	111 output_tokens=5,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	112 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	113 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	114 self.emit(event("session.idle", aborted=False))
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	115 elif self.behavior == "error":
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	116 self.emit(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	117 event(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	118 "session.error",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	119 error_type="provider",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	120 error_code="upstream_error",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	121 message="upstream failed",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	122 status_code=502,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	123 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	124 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	125 return message_id
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	126
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	127 async def abort(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	128 self.abort_calls += 1
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	129 if self.behavior == "abort_error":
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	130 raise RuntimeError("abort failed")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	131 self.emit(event("session.idle", aborted=True))
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	132
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	133 async def disconnect(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	134 self.disconnect_calls += 1
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	135 if self.disconnect_error:
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	136 raise RuntimeError("disconnect failed")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	137
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	138
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	139 class FakeClient:
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	140 def __init__(self, behavior="complete"):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	141 self.behavior = behavior
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	142 self.sessions = {}
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	143 self.create_calls = []
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	144 self.resume_calls = []
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	145 self.delete_calls = []
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	146 self.started = False
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	147 self.stopped = False
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	148 self.resume_started = None
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	149 self.resume_release = None
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	150
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	151 async def start(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	152 self.started = True
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	153
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	154 async def stop(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	155 self.stopped = True
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	156
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	157 async def resume_session(self, session_id, **kwargs):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	158 self.resume_calls.append((session_id, kwargs))
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	159 if self.resume_started is not None:
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	160 self.resume_started.set()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	161 if self.resume_release is not None:
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	162 await self.resume_release.wait()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	163 if session_id not in self.sessions:
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	164 raise LookupError(session_id)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	165 return self.sessions[session_id]
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	166
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	167 async def create_session(self, session_id, **kwargs):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	168 self.create_calls.append((session_id, kwargs))
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	169 session = FakeSession(session_id, self.behavior)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	170 self.sessions[session_id] = session
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	171 return session
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	172
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	173 async def delete_session(self, session_id):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	174 self.delete_calls.append(session_id)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	175 self.sessions.pop(session_id, None)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	176
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	177
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	178 class SidecarTest(unittest.IsolatedAsyncioTestCase):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	179 async def asyncSetUp(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	180 self.output = []
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	181
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	182 async def capture(payload):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	183 self.output.append(payload)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	184
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	185 self.capture = capture
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	186 self.config = SidecarConfig(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	187 base_url="http://litellm.invalid/v1",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	188 model="test-model",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	189 wire_api="responses",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	190 base_directory="/not-used-by-fake",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	191 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	192
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	193 async def make_sidecar(self, behavior="complete", compile_fn=_fake_compile_fn):
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	194 client = FakeClient(behavior)
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	195 sidecar = Sidecar(client, self.config, self.capture, compile_fn=compile_fn)
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	196 await sidecar.start()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	197 return sidecar, client
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	198
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	199 async def test_health_and_invalid_protocol(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	200 sidecar, _ = await self.make_sidecar()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	201 await sidecar.announce_ready()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	202 await sidecar.dispatch({"command": "health", "request_id": "health-1"})
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	203 await sidecar.dispatch({"command": "unknown", "request_id": "bad-1"})
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	204
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	205 self.assertEqual(self.output[0]["type"], "ready")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	206 self.assertIsNone(self.output[0]["request_id"])
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	207 self.assertEqual(self.output[1]["request_id"], "health-1")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	208 self.assertEqual(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	209 [item["type"] for item in self.output[-2:]],
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	210 ["turn.error", "turn.done"],
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	211 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	212
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	213 async def test_multi_turn_reuses_one_session_and_configures_sdk(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	214 sidecar, client = await self.make_sidecar()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	215 for request_id, prompt in (("r1", "first"), ("r2", "second")):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	216 await sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	217 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	218 "command": "turn.start",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	219 "request_id": request_id,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	220 "conversation_id": "conversation-a",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	221 "prompt": prompt,
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	222 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	223 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	224 "knowledge_version": 1,
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	225 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	226 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	227 await sidecar.drain_events()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	228
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	229 self.assertEqual(len(client.create_calls), 1)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	230 self.assertEqual(len(client.resume_calls), 1)
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	231 sdk_id = _fake_sdk_id("conversation-a", "public_visitor")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	232 session = client.sessions[sdk_id]
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	233 self.assertEqual(session.prompts, ["first", "second"])
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	234 options = client.create_calls[0][1]
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	235 self.assertEqual(options["provider"]["type"], "openai")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	236 self.assertEqual(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	237 options["provider"]["base_url"], "http://litellm.invalid/v1"
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	238 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	239 self.assertEqual(options["provider"]["wire_api"], "responses")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	240 self.assertEqual(options["model"], "test-model")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	241 self.assertEqual(options["available_tools"], [])
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	242 self.assertTrue(options["streaming"])
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	243 self.assertEqual(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	244 [item["request_id"] for item in self.output if item["type"] == "turn.done"],
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	245 ["r1", "r2"],
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	246 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	247
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	248 async def test_concurrent_conversations_keep_correlation(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	249 sidecar, _ = await self.make_sidecar()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	250 await asyncio.gather(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	251 sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	252 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	253 "command": "turn.start",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	254 "request_id": "left-request",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	255 "conversation_id": "left",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	256 "prompt": "left",
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	257 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	258 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	259 "knowledge_version": 1,
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	260 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	261 ),
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	262 sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	263 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	264 "command": "turn.start",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	265 "request_id": "right-request",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	266 "conversation_id": "right",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	267 "prompt": "right",
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	268 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	269 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	270 "knowledge_version": 1,
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	271 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	272 ),
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	273 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	274 await sidecar.drain_events()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	275
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	276 correlated = {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	277 (item["request_id"], item["conversation_id"])
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	278 for item in self.output
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	279 if item["type"] in ("assistant.delta", "assistant.completed")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	280 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	281 self.assertEqual(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	282 correlated,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	283 {("left-request", "left"), ("right-request", "right")},
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	284 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	285
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	286 async def test_abort_finishes_active_turn(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	287 sidecar, client = await self.make_sidecar("pending")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	288 await sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	289 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	290 "command": "turn.start",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	291 "request_id": "turn-request",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	292 "conversation_id": "abort-me",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	293 "prompt": "wait",
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	294 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	295 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	296 "knowledge_version": 1,
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	297 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	298 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	299 await sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	300 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	301 "command": "turn.abort",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	302 "request_id": "abort-request",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	303 "conversation_id": "abort-me",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	304 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	305 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	306 await sidecar.drain_events()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	307
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	308 self.assertEqual(client.sessions[_fake_sdk_id("abort-me", "public_visitor")].abort_calls, 1)
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	309 done = [
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	310 item
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	311 for item in self.output
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	312 if item["type"] == "turn.done"
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	313 and item["request_id"] == "turn-request"
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	314 ]
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	315 self.assertEqual(done[0]["aborted"], True)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	316 abort_accept = [
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	317 item
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	318 for item in self.output
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	319 if item["type"] == "turn.accepted"
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	320 and item["request_id"] == "abort-request"
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	321 ]
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	322 self.assertEqual(abort_accept[0]["target_request_id"], "turn-request")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	323
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	324 async def test_session_error_is_terminal(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	325 sidecar, _ = await self.make_sidecar("error")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	326 await sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	327 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	328 "command": "turn.start",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	329 "request_id": "error-request",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	330 "conversation_id": "error-conversation",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	331 "prompt": "fail",
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	332 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	333 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	334 "knowledge_version": 1,
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	335 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	336 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	337 await sidecar.drain_events()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	338
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	339 errors = [item for item in self.output if item["type"] == "turn.error"]
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	340 done = [item for item in self.output if item["type"] == "turn.done"]
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	341 self.assertEqual(errors[0]["error"]["code"], "upstream_error")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	342 self.assertEqual(errors[0]["error"]["status_code"], 502)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	343 self.assertEqual(len(done), 1)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	344 self.assertTrue(done[0]["failed"])
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	345
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	346 async def test_abort_error_terminates_abort_request(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	347 sidecar, _ = await self.make_sidecar("abort_error")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	348 await sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	349 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	350 "command": "turn.start",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	351 "request_id": "active-request",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	352 "conversation_id": "abort-error",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	353 "prompt": "wait",
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	354 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	355 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	356 "knowledge_version": 1,
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	357 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	358 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	359 await sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	360 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	361 "command": "turn.abort",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	362 "request_id": "failed-abort",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	363 "conversation_id": "abort-error",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	364 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	365 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	366
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	367 abort_events = [
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	368 item for item in self.output if item["request_id"] == "failed-abort"
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	369 ]
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	370 self.assertEqual(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	371 [item["type"] for item in abort_events],
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	372 ["turn.accepted", "turn.error", "turn.done"],
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	373 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	374 self.assertTrue(abort_events[-1]["failed"])
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	375
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	376 async def test_delete_and_shutdown_release_resources(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	377 sidecar, client = await self.make_sidecar("pending")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	378 await sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	379 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	380 "command": "turn.start",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	381 "request_id": "active",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	382 "conversation_id": "delete-me",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	383 "prompt": "wait",
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	384 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	385 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	386 "knowledge_version": 1,
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	387 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	388 )
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	389 _delete_sdk_id = _fake_sdk_id("delete-me", "public_visitor")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	390 session = client.sessions[_delete_sdk_id]
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	391 await sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	392 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	393 "command": "conversation.delete",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	394 "request_id": "delete-request",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	395 "conversation_id": "delete-me",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	396 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	397 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	398 await sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	399 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	400 "command": "shutdown",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	401 "request_id": "shutdown-request",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	402 "conversation_id": None,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	403 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	404 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	405
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	406 self.assertEqual(session.disconnect_calls, 1)
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	407 self.assertEqual(client.delete_calls[0], _delete_sdk_id)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	408 self.assertEqual(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	409 set(client.delete_calls),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	410 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	411 _fake_sdk_id("delete-me", "public_visitor"),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	412 _fake_sdk_id("delete-me", "invited_friend"),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	413 _fake_sdk_id("delete-me", "june_admin"),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	414 },
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	415 )
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	416 self.assertTrue(client.stopped)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	417 self.assertTrue(sidecar.shutting_down)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	418 shutdown = [
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	419 item
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	420 for item in self.output
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	421 if item["request_id"] == "shutdown-request"
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	422 ]
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	423 self.assertEqual(shutdown[0]["type"], "turn.done")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	424
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	425 async def test_idle_and_overflow_sessions_are_evicted(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	426 self.config = replace(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	427 self.config,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	428 idle_timeout_seconds=3600,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	429 max_sessions=1,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	430 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	431 sidecar, client = await self.make_sidecar()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	432 for conversation_id in ("old", "new"):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	433 await sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	434 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	435 "command": "turn.start",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	436 "request_id": f"request-{conversation_id}",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	437 "conversation_id": conversation_id,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	438 "prompt": conversation_id,
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	439 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	440 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	441 "knowledge_version": 1,
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	442 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	443 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	444 await sidecar.drain_events()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	445 await asyncio.sleep(0.01)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	446
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	447 old_session = client.sessions[_fake_sdk_id("old", "public_visitor")]
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	448 old_session.disconnect_error = True
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	449 await sidecar.evict_idle_sessions()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	450 self.assertEqual(old_session.disconnect_calls, 1)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	451 self.assertNotIn("old", sidecar._conversations)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	452 self.assertIn("new", sidecar._conversations)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	453
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	454 sidecar._conversations["new"].last_used -= 4000
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	455 await sidecar.evict_idle_sessions()
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	456 self.assertEqual(client.sessions[_fake_sdk_id("new", "public_visitor")].disconnect_calls, 1)
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	457 self.assertEqual(sidecar._conversations, {})
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	458 await sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	459 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	460 "command": "shutdown",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	461 "request_id": "shutdown-eviction",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	462 "conversation_id": None,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	463 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	464 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	465
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	466 async def test_shutdown_waits_for_inflight_session_creation(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	467 client = FakeClient("pending")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	468 client.resume_started = asyncio.Event()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	469 client.resume_release = asyncio.Event()
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	470 sidecar = Sidecar(client, self.config, self.capture, compile_fn=_fake_compile_fn)
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	471 await sidecar.start()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	472
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	473 start_task = asyncio.create_task(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	474 sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	475 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	476 "command": "turn.start",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	477 "request_id": "starting",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	478 "conversation_id": "race",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	479 "prompt": "wait",
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	480 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	481 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	482 "knowledge_version": 1,
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	483 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	484 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	485 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	486 await client.resume_started.wait()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	487 shutdown_task = asyncio.create_task(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	488 sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	489 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	490 "command": "shutdown",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	491 "request_id": "shutdown",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	492 "conversation_id": None,
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	493 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	494 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	495 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	496 await asyncio.sleep(0)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	497 self.assertTrue(sidecar.shutting_down)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	498 client.resume_release.set()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	499 await asyncio.gather(start_task, shutdown_task)
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	500
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	501 self.assertTrue(client.stopped)
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	502 self.assertEqual(client.sessions[_fake_sdk_id("race", "public_visitor")].disconnect_calls, 1)
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	503 await sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	504 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	505 "command": "turn.start",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	506 "request_id": "too-late",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	507 "conversation_id": "late",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	508 "prompt": "no",
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	509 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	510 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	511 "knowledge_version": 1,
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	512 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	513 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	514 late_error = [
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	515 item
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	516 for item in self.output
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	517 if item["request_id"] == "too-late" and item["type"] == "turn.error"
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	518 ]
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	519 self.assertEqual(late_error[0]["error"]["code"], "shutting_down")
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	520
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	521 async def test_command_gates_are_released(self):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	522 sidecar, _ = await self.make_sidecar()
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	523 for index in range(20):
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	524 await sidecar.dispatch(
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	525 {
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	526 "command": "unknown",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	527 "request_id": f"unknown-{index}",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	528 "conversation_id": f"conversation-{index}",
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	529 }
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	530 )
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	531 self.assertEqual(sidecar._conversation_gates, {})
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	532
265 056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	533 # ------------------------------------------------------------------
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	534 # New: profile-aware tests
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	535 # ------------------------------------------------------------------
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	536
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	537 async def test_three_distinct_profiles_produce_distinct_append_prompts(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	538 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	539 for profile in ("public_visitor", "invited_friend", "june_admin"):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	540 await sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	541 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	542 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	543 "request_id": f"req-{profile}",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	544 "conversation_id": f"conv-{profile}",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	545 "prompt": "hello",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	546 "prompt_profile": profile,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	547 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	548 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	549 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	550 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	551 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	552
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	553 contents = [opts["system_message"]["content"] for _, opts in client.create_calls]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	554 self.assertEqual(len(contents), 3)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	555 self.assertEqual(len(set(contents)), 3, "all three profiles must produce distinct content")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	556 for _, opts in client.create_calls:
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	557 self.assertEqual(opts["system_message"]["mode"], "append")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	558
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	559 async def test_session_options_memory_disabled_and_no_tools(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	560 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	561 await sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	562 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	563 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	564 "request_id": "r-opts",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	565 "conversation_id": "conv-opts",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	566 "prompt": "test",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	567 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	568 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	569 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	570 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	571 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	572 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	573 _, opts = client.create_calls[0]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	574 self.assertEqual(opts["memory"], {"enabled": False})
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	575 self.assertEqual(opts["tools"], [])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	576 self.assertEqual(opts["available_tools"], [])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	577 self.assertEqual(opts["mcp_servers"], {})
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	578 self.assertTrue(opts["enable_session_store"])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	579
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	580 async def test_missing_profile_rejected_before_client_calls(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	581 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	582 await sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	583 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	584 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	585 "request_id": "r-missing",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	586 "conversation_id": "conv-missing",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	587 "prompt": "hello",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	588 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	589 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	590 errors = [e for e in self.output if e["type"] == "turn.error"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	591 self.assertEqual(errors[0]["error"]["code"], "invalid_prompt_profile")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	592 done = [e for e in self.output if e["type"] == "turn.done"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	593 self.assertTrue(done[0]["failed"])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	594 self.assertEqual(len(client.create_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	595 self.assertEqual(len(client.resume_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	596
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	597 async def test_unknown_profile_rejected_before_client_calls(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	598 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	599 await sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	600 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	601 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	602 "request_id": "r-unknown",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	603 "conversation_id": "conv-unknown",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	604 "prompt": "hello",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	605 "prompt_profile": "hacker",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	606 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	607 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	608 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	609 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	610 errors = [e for e in self.output if e["type"] == "turn.error"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	611 self.assertEqual(errors[0]["error"]["code"], "invalid_prompt_profile")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	612 self.assertEqual(len(client.create_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	613
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	614 async def test_stale_prompt_version_rejected_before_client_calls(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	615 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	616 await sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	617 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	618 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	619 "request_id": "r-stale-pv",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	620 "conversation_id": "conv-stale-pv",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	621 "prompt": "hello",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	622 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	623 "prompt_version": 999,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	624 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	625 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	626 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	627 errors = [e for e in self.output if e["type"] == "turn.error"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	628 self.assertEqual(errors[0]["error"]["code"], "invalid_prompt_profile")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	629 self.assertEqual(len(client.create_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	630
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	631 async def test_stale_knowledge_version_rejected_before_client_calls(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	632 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	633 await sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	634 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	635 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	636 "request_id": "r-stale-kv",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	637 "conversation_id": "conv-stale-kv",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	638 "prompt": "hello",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	639 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	640 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	641 "knowledge_version": 999,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	642 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	643 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	644 errors = [e for e in self.output if e["type"] == "turn.error"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	645 self.assertEqual(errors[0]["error"]["code"], "invalid_prompt_profile")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	646 self.assertEqual(len(client.create_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	647
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	648 async def test_bool_prompt_version_rejected_before_client_calls(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	649 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	650 await sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	651 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	652 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	653 "request_id": "r-bool-pv",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	654 "conversation_id": "conv-bool-pv",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	655 "prompt": "hello",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	656 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	657 "prompt_version": True,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	658 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	659 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	660 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	661 errors = [e for e in self.output if e["type"] == "turn.error"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	662 self.assertEqual(errors[0]["error"]["code"], "invalid_prompt_profile")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	663 self.assertEqual(len(client.create_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	664
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	665 async def test_bool_knowledge_version_rejected_before_client_calls(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	666 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	667 await sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	668 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	669 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	670 "request_id": "r-bool-kv",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	671 "conversation_id": "conv-bool-kv",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	672 "prompt": "hello",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	673 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	674 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	675 "knowledge_version": True,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	676 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	677 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	678 errors = [e for e in self.output if e["type"] == "turn.error"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	679 self.assertEqual(errors[0]["error"]["code"], "invalid_prompt_profile")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	680 self.assertEqual(len(client.create_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	681
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	682 async def test_same_profile_reuses_existing_session(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	683 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	684 for i in range(3):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	685 await sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	686 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	687 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	688 "request_id": f"r-reuse-{i}",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	689 "conversation_id": "conv-reuse",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	690 "prompt": f"message {i}",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	691 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	692 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	693 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	694 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	695 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	696 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	697
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	698 self.assertEqual(len(client.create_calls), 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	699 self.assertEqual(len(client.resume_calls), 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	700 session = client.sessions[_fake_sdk_id("conv-reuse", "public_visitor")]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	701 self.assertEqual(session.prompts, ["message 0", "message 1", "message 2"])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	702
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	703 async def test_profile_switch_disconnects_and_resumes_with_new_prompt(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	704 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	705
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	706 await sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	707 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	708 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	709 "request_id": "r-switch-1",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	710 "conversation_id": "conv-switch",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	711 "prompt": "first",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	712 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	713 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	714 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	715 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	716 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	717 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	718 visitor_sdk_id = _fake_sdk_id("conv-switch", "public_visitor")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	719 friend_sdk_id = _fake_sdk_id("conv-switch", "invited_friend")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	720 first_session = client.sessions[visitor_sdk_id]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	721 self.assertEqual(len(client.create_calls), 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	722
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	723 await sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	724 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	725 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	726 "request_id": "r-switch-2",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	727 "conversation_id": "conv-switch",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	728 "prompt": "second",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	729 "prompt_profile": "invited_friend",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	730 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	731 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	732 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	733 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	734 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	735
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	736 # Old session is disconnected and permanently deleted before new one opens.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	737 self.assertEqual(first_session.disconnect_calls, 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	738 self.assertIn(visitor_sdk_id, client.delete_calls)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	739
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	740 # The new session uses the invited_friend derived ID, not the visitor one.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	741 resume_id, resume_opts = client.resume_calls[-1]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	742 self.assertEqual(resume_id, friend_sdk_id)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	743 self.assertNotEqual(friend_sdk_id, visitor_sdk_id)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	744 self.assertEqual(resume_opts["system_message"]["mode"], "append")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	745 expected_content = _FAKE_PROFILES["invited_friend"]["content"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	746 self.assertEqual(resume_opts["system_message"]["content"], expected_content)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	747
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	748 # The newly opened session is a distinct object.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	749 second_session = client.sessions[friend_sdk_id]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	750 self.assertIsNot(second_session, first_session)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	751
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	752 done = [e for e in self.output if e["type"] == "turn.done" and e["request_id"] == "r-switch-2"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	753 self.assertFalse(done[0].get("failed", False))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	754
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	755 async def test_concurrent_conversations_profile_isolation(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	756 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	757 await asyncio.gather(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	758 sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	759 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	760 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	761 "request_id": "req-visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	762 "conversation_id": "conv-visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	763 "prompt": "hello visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	764 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	765 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	766 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	767 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	768 ),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	769 sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	770 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	771 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	772 "request_id": "req-admin",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	773 "conversation_id": "conv-admin",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	774 "prompt": "hello admin",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	775 "prompt_profile": "june_admin",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	776 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	777 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	778 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	779 ),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	780 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	781 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	782
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	783 visitor_sdk_id = _fake_sdk_id("conv-visitor", "public_visitor")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	784 admin_sdk_id = _fake_sdk_id("conv-admin", "june_admin")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	785 options_by_sdk_id = {sdk_id: opts for sdk_id, opts in client.create_calls}
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	786 visitor_content = options_by_sdk_id[visitor_sdk_id]["system_message"]["content"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	787 admin_content = options_by_sdk_id[admin_sdk_id]["system_message"]["content"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	788 self.assertNotEqual(visitor_content, admin_content)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	789 self.assertEqual(visitor_content, _FAKE_PROFILES["public_visitor"]["content"])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	790 self.assertEqual(admin_content, _FAKE_PROFILES["june_admin"]["content"])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	791
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	792 async def test_profile_switch_while_active_is_rejected(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	793 sidecar, client = await self.make_sidecar("pending")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	794
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	795 await sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	796 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	797 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	798 "request_id": "active-req",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	799 "conversation_id": "conv-active-switch",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	800 "prompt": "wait",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	801 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	802 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	803 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	804 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	805 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	806
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	807 await sidecar.dispatch(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	808 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	809 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	810 "request_id": "switch-req",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	811 "conversation_id": "conv-active-switch",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	812 "prompt": "switch",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	813 "prompt_profile": "invited_friend",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	814 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	815 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	816 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	817 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	818
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	819 errors = [e for e in self.output if e["type"] == "turn.error" and e["request_id"] == "switch-req"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	820 self.assertTrue(len(errors) > 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	821 done = [e for e in self.output if e["type"] == "turn.done" and e["request_id"] == "switch-req"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	822 self.assertTrue(done[0]["failed"])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	823 # Active-turn switch must not trigger any cleanup at all.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	824 sdk_id = _fake_sdk_id("conv-active-switch", "public_visitor")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	825 self.assertEqual(client.sessions[sdk_id].disconnect_calls, 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	826 self.assertEqual(client.delete_calls, [])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	827
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	828 async def test_startup_compilation_failure_prevents_readiness(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	829 call_count = {"n": 0}
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	830
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	831 def failing_compile_fn(profile: str) -> dict:
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	832 call_count["n"] += 1
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	833 raise ValueError(f"corrupted assets for {profile!r}")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	834
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	835 client = FakeClient()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	836 sidecar = Sidecar(client, self.config, self.capture, compile_fn=failing_compile_fn)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	837 with self.assertRaises(ValueError) as ctx:
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	838 await sidecar.start()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	839 self.assertIn("corrupted", str(ctx.exception))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	840 self.assertFalse(client.started)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	841 self.assertGreater(call_count["n"], 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	842
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	843 # ------------------------------------------------------------------
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	844 # Derived session ID security tests
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	845 # ------------------------------------------------------------------
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	846
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	847 async def test_distinct_profiles_produce_distinct_sdk_session_ids(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	848 """Different profiles on the same conversation must never share an SDK ID."""
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	849 ids = {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	850 profile: _fake_sdk_id("conv-same", profile)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	851 for profile in ("public_visitor", "invited_friend", "june_admin")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	852 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	853 self.assertEqual(len(set(ids.values())), 3, "each profile needs a unique SDK ID")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	854 for sid in ids.values():
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	855 uuid.UUID(sid) # every value must be a valid UUID
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	856
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	857 async def test_same_profile_produces_same_sdk_id_across_restarts(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	858 """The derived ID is deterministic: a sidecar restart resumes the same session."""
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	859 client = FakeClient()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	860 sidecar1 = Sidecar(client, self.config, self.capture, compile_fn=_fake_compile_fn)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	861 await sidecar1.start()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	862 await sidecar1.dispatch({
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	863 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	864 "request_id": "restart-1",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	865 "conversation_id": "conv-restart",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	866 "prompt": "hello",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	867 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	868 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	869 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	870 })
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	871 await sidecar1.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	872
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	873 # Simulate sidecar restart: new Sidecar instance, same FakeClient (SDK store).
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	874 sidecar2 = Sidecar(client, self.config, self.capture, compile_fn=_fake_compile_fn)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	875 await sidecar2.start()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	876 await sidecar2.dispatch({
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	877 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	878 "request_id": "restart-2",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	879 "conversation_id": "conv-restart",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	880 "prompt": "still here",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	881 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	882 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	883 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	884 })
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	885 await sidecar2.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	886
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	887 # Sidecar1 tries resume (fails - no session yet) then creates.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	888 # Sidecar2 tries resume (succeeds - session persists in SDK store).
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	889 # Therefore exactly one create, two resume attempts, both on the same derived ID.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	890 self.assertEqual(len(client.create_calls), 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	891 self.assertEqual(len(client.resume_calls), 2)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	892 derived = _fake_sdk_id("conv-restart", "public_visitor")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	893 self.assertEqual(client.create_calls[0][0], derived)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	894 self.assertTrue(all(r == derived for r, _ in client.resume_calls))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	895
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	896 async def test_profile_switch_permanently_deletes_old_sdk_session(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	897 """Switching profile must delete the old derived SDK session before creating new."""
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	898 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	899 await sidecar.dispatch({
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	900 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	901 "request_id": "admin-turn",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	902 "conversation_id": "conv-priv",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	903 "prompt": "admin question",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	904 "prompt_profile": "june_admin",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	905 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	906 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	907 })
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	908 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	909
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	910 admin_sdk_id = _fake_sdk_id("conv-priv", "june_admin")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	911 visitor_sdk_id = _fake_sdk_id("conv-priv", "public_visitor")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	912 self.assertNotEqual(admin_sdk_id, visitor_sdk_id)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	913
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	914 await sidecar.dispatch({
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	915 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	916 "request_id": "visitor-turn",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	917 "conversation_id": "conv-priv",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	918 "prompt": "public question",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	919 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	920 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	921 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	922 })
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	923 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	924
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	925 # Admin session must be permanently deleted before visitor session opens.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	926 self.assertIn(admin_sdk_id, client.delete_calls)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	927 # Admin session must not be in the live sessions map.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	928 self.assertNotIn(admin_sdk_id, client.sessions)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	929 # Visitor session is a distinct object.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	930 self.assertIn(visitor_sdk_id, client.sessions)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	931
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	932 async def test_delete_conversation_uses_derived_sdk_id_not_conversation_id(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	933 """conversation.delete must call delete_session with the derived UUID."""
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	934 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	935 await sidecar.dispatch({
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	936 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	937 "request_id": "setup-turn",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	938 "conversation_id": "conv-del-check",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	939 "prompt": "hi",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	940 "prompt_profile": "june_admin",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	941 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	942 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	943 })
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	944 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	945
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	946 admin_sdk_id = _fake_sdk_id("conv-del-check", "june_admin")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	947
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	948 await sidecar.dispatch({
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	949 "command": "conversation.delete",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	950 "request_id": "del-req",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	951 "conversation_id": "conv-del-check",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	952 })
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	953
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	954 self.assertEqual(client.delete_calls[0], admin_sdk_id)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	955 self.assertEqual(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	956 set(client.delete_calls),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	957 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	958 _fake_sdk_id("conv-del-check", "public_visitor"),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	959 _fake_sdk_id("conv-del-check", "invited_friend"),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	960 _fake_sdk_id("conv-del-check", "june_admin"),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	961 },
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	962 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	963 self.assertNotIn("conv-del-check", client.delete_calls)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	964
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	965 async def test_delete_nonexistent_conversation_cleans_known_profile_ids(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	966 """An uncached delete purges every currently known derived session ID."""
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	967 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	968 await sidecar.dispatch({
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	969 "command": "conversation.delete",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	970 "request_id": "del-ghost",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	971 "conversation_id": "ghost-conv",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	972 })
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	973
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	974 self.assertEqual(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	975 set(client.delete_calls),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	976 {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	977 _fake_sdk_id("ghost-conv", "public_visitor"),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	978 _fake_sdk_id("ghost-conv", "invited_friend"),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	979 _fake_sdk_id("ghost-conv", "june_admin"),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	980 },
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	981 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	982 self.assertNotIn("ghost-conv", client.delete_calls)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	983 done = [e for e in self.output if e["type"] == "turn.done" and e["request_id"] == "del-ghost"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	984 self.assertEqual(done[0]["action"], "conversation.delete")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	985
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	986 async def test_admin_session_not_reachable_via_visitor_derived_id(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	987 """A visitor SDK ID must differ from the admin one for the same conversation."""
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	988 admin_id = _fake_sdk_id("shared-conv", "june_admin")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	989 visitor_id = _fake_sdk_id("shared-conv", "public_visitor")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	990 self.assertNotEqual(admin_id, visitor_id)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	991
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	992 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	993 # Establish an admin session.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	994 await sidecar.dispatch({
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	995 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	996 "request_id": "admin-req",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	997 "conversation_id": "shared-conv",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	998 "prompt": "secret",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	999 "prompt_profile": "june_admin",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1000 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1001 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1002 })
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1003 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1004
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1005 # The admin session object must NOT be accessible under the visitor-derived ID.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1006 admin_session = client.sessions.get(admin_id)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1007 self.assertIsNotNone(admin_session)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1008 self.assertIsNone(client.sessions.get(visitor_id),
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1009 "visitor SDK ID must not map to any session object at this point")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1010
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1011
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1012 # ---------------------------------------------------------------------------
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1013 # _validate_history unit tests
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1014 # ---------------------------------------------------------------------------
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1015
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1016 class ValidateHistoryTest(unittest.TestCase):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1017 def test_none_returns_empty_list(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1018 self.assertEqual(_validate_history(None), [])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1019
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1020 def test_empty_list_accepted(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1021 self.assertEqual(_validate_history([]), [])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1022
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1023 def test_valid_two_entries(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1024 hist = [
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1025 {"role": "user", "content": "hello"},
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1026 {"role": "assistant", "content": "hi"},
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1027 ]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1028 result = _validate_history(hist)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1029 self.assertEqual(result, hist)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1030
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1031 def test_not_a_list_raises(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1032 for bad in (42, "string", True, False, {}, object()):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1033 with self.assertRaises(ValueError, msg=f"should reject {bad!r}"):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1034 _validate_history(bad)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1035
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1036 def test_too_many_entries_raises(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1037 entries = [{"role": "user", "content": "x"}] * (_HISTORY_MAX_ENTRIES + 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1038 with self.assertRaises(ValueError):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1039 _validate_history(entries)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1040
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1041 def test_exactly_max_entries_accepted(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1042 entries = [
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1043 {"role": "user" if i % 2 == 0 else "assistant", "content": "x"}
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1044 for i in range(_HISTORY_MAX_ENTRIES)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1045 ]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1046 result = _validate_history(entries)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1047 self.assertEqual(len(result), _HISTORY_MAX_ENTRIES)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1048
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1049 def test_non_object_entry_raises(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1050 for bad_entry in (42, "string", True, None, []):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1051 with self.assertRaises(ValueError):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1052 _validate_history([bad_entry])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1053
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1054 def test_invalid_role_raises(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1055 for bad_role in ("system", "SYSTEM", "User", "ASSISTANT", "", " user"):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1056 with self.assertRaises(ValueError, msg=f"role {bad_role!r} must be rejected"):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1057 _validate_history([{"role": bad_role, "content": "x"}])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1058
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1059 def test_bool_role_raises(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1060 with self.assertRaises(ValueError):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1061 _validate_history([{"role": True, "content": "x"}])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1062
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1063 def test_none_content_raises(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1064 with self.assertRaises(ValueError):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1065 _validate_history([{"role": "user", "content": None}])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1066
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1067 def test_bool_content_raises(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1068 with self.assertRaises(ValueError):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1069 _validate_history([{"role": "user", "content": True}])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1070
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1071 def test_int_content_raises(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1072 with self.assertRaises(ValueError):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1073 _validate_history([{"role": "user", "content": 42}])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1074
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1075 def test_extra_key_raises(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1076 with self.assertRaises(ValueError):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1077 _validate_history([{"role": "user", "content": "hi", "injected": "bad"}])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1078
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1079 def test_oversized_total_raises(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1080 # One entry with content just over the byte limit.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1081 big = "x" * (_HISTORY_MAX_BYTES + 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1082 with self.assertRaises(ValueError):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1083 _validate_history([{"role": "user", "content": big}])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1084
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1085 def test_total_at_limit_accepted(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1086 # Two entries whose combined bytes sit at or under the limit.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1087 chunk_size = _HISTORY_MAX_BYTES // 2 - 10 # under limit
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1088 entries = [
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1089 {"role": "user", "content": "a" * chunk_size},
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1090 {"role": "assistant", "content": "b" * chunk_size},
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1091 ]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1092 result = _validate_history(entries)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1093 self.assertEqual(len(result), 2)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1094
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1095 def test_special_characters_accepted(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1096 entry = {"role": "user", "content": "hello \"world\" \\ \n"}
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1097 result = _validate_history([entry])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1098 self.assertEqual(result[0]["content"], entry["content"])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1099
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1100 def test_empty_content_string_accepted(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1101 result = _validate_history([{"role": "user", "content": ""}])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1102 self.assertEqual(result[0]["content"], "")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1103
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1104
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1105 # ---------------------------------------------------------------------------
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1106 # Sidecar history integration tests
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1107 # ---------------------------------------------------------------------------
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1108
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1109 class SidecarHistoryTest(unittest.IsolatedAsyncioTestCase):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1110 async def asyncSetUp(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1111 self.output = []
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1112
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1113 async def capture(payload):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1114 self.output.append(payload)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1115
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1116 self.capture = capture
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1117 self.config = SidecarConfig(
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1118 base_url="http://litellm.invalid/v1",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1119 model="test-model",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1120 wire_api="responses",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1121 base_directory="/not-used-by-fake",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1122 )
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1123
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1124 async def make_sidecar(self, behavior="complete"):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1125 client = FakeClient(behavior)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1126 sidecar = Sidecar(client, self.config, self.capture, compile_fn=_fake_compile_fn)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1127 await sidecar.start()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1128 return sidecar, client
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1129
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1130 def _turn_start(self, request_id, conversation_id, history=None, **extra):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1131 cmd = {
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1132 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1133 "request_id": request_id,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1134 "conversation_id": conversation_id,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1135 "prompt": "hello",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1136 "prompt_profile": "public_visitor",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1137 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1138 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1139 }
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1140 if history is not None:
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1141 cmd["history"] = history
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1142 cmd.update(extra)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1143 return cmd
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1144
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1145 async def test_invalid_history_not_list_rejected_before_client_calls(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1146 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1147 await sidecar.dispatch(self._turn_start("r1", "c1", history=42))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1148 errors = [e for e in self.output if e["type"] == "turn.error"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1149 self.assertEqual(errors[0]["error"]["code"], "invalid_history")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1150 done = [e for e in self.output if e["type"] == "turn.done"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1151 self.assertTrue(done[0]["failed"])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1152 self.assertEqual(len(client.create_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1153 self.assertEqual(len(client.resume_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1154
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1155 async def test_invalid_history_too_many_entries_rejected(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1156 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1157 hist = [{"role": "user", "content": "x"}] * (_HISTORY_MAX_ENTRIES + 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1158 await sidecar.dispatch(self._turn_start("r2", "c2", history=hist))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1159 errors = [e for e in self.output if e["type"] == "turn.error"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1160 self.assertEqual(errors[0]["error"]["code"], "invalid_history")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1161 self.assertEqual(len(client.create_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1162
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1163 async def test_invalid_history_bad_role_rejected(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1164 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1165 hist = [{"role": "system", "content": "inject"}]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1166 await sidecar.dispatch(self._turn_start("r3", "c3", history=hist))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1167 errors = [e for e in self.output if e["type"] == "turn.error"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1168 self.assertEqual(errors[0]["error"]["code"], "invalid_history")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1169 self.assertEqual(len(client.create_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1170
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1171 async def test_invalid_history_bool_role_rejected(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1172 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1173 hist = [{"role": True, "content": "x"}]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1174 await sidecar.dispatch(self._turn_start("r4", "c4", history=hist))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1175 errors = [e for e in self.output if e["type"] == "turn.error"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1176 self.assertEqual(errors[0]["error"]["code"], "invalid_history")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1177 self.assertEqual(len(client.create_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1178
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1179 async def test_invalid_history_none_content_rejected(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1180 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1181 hist = [{"role": "user", "content": None}]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1182 await sidecar.dispatch(self._turn_start("r5", "c5", history=hist))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1183 errors = [e for e in self.output if e["type"] == "turn.error"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1184 self.assertEqual(errors[0]["error"]["code"], "invalid_history")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1185 self.assertEqual(len(client.create_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1186
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1187 async def test_invalid_history_extra_key_rejected(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1188 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1189 hist = [{"role": "user", "content": "hi", "extra": "bad"}]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1190 await sidecar.dispatch(self._turn_start("r6", "c6", history=hist))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1191 errors = [e for e in self.output if e["type"] == "turn.error"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1192 self.assertEqual(errors[0]["error"]["code"], "invalid_history")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1193 self.assertEqual(len(client.create_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1194
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1195 async def test_invalid_history_oversized_rejected(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1196 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1197 big = "x" * (_HISTORY_MAX_BYTES + 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1198 hist = [{"role": "user", "content": big}]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1199 await sidecar.dispatch(self._turn_start("r7", "c7", history=hist))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1200 errors = [e for e in self.output if e["type"] == "turn.error"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1201 self.assertEqual(errors[0]["error"]["code"], "invalid_history")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1202 self.assertEqual(len(client.create_calls), 0)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1203
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1204 async def test_valid_empty_history_accepted(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1205 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1206 await sidecar.dispatch(self._turn_start("r8", "c8", history=[]))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1207 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1208 done = [e for e in self.output if e["type"] == "turn.done" and e["request_id"] == "r8"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1209 self.assertFalse(done[0].get("failed", False))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1210 self.assertEqual(len(client.create_calls), 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1211
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1212 async def test_fresh_create_receives_history_in_system_message(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1213 """When resume_session fails (no persisted session), create_session must
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1214 include the PRIOR OWNED CONVERSATION TRANSCRIPT in system_message."""
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1215 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1216 hist = [
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1217 {"role": "user", "content": "prior question"},
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1218 {"role": "assistant", "content": "prior answer"},
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1219 ]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1220 await sidecar.dispatch(self._turn_start("r-create", "c-create", history=hist))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1221 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1222
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1223 self.assertEqual(len(client.create_calls), 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1224 _, create_opts = client.create_calls[0]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1225 sys_content = create_opts["system_message"]["content"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1226 self.assertIn("PRIOR OWNED CONVERSATION TRANSCRIPT", sys_content)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1227 self.assertIn("prior question", sys_content)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1228 self.assertIn("prior answer", sys_content)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1229 self.assertIn("untrusted", sys_content.lower())
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1230 self.assertEqual(create_opts["system_message"]["mode"], "append")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1231
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1232 async def test_resume_does_not_inject_history(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1233 """An existing persisted session must not receive history in system_message."""
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1234 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1235 hist = [{"role": "user", "content": "prior"}]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1236
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1237 # First turn: creates the session.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1238 await sidecar.dispatch(self._turn_start("r-resume-1", "c-resume", history=hist))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1239 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1240 self.assertEqual(len(client.create_calls), 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1241 first_content = client.create_calls[0][1]["system_message"]["content"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1242
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1243 # Second turn on same conversation: must resume (SDK session persists).
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1244 await sidecar.dispatch(self._turn_start("r-resume-2", "c-resume", history=hist))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1245 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1246 self.assertEqual(len(client.resume_calls), 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1247 # resume_session does not receive options from create_session call.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1248 resume_opts = client.resume_calls[0][1]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1249 resume_sys = resume_opts["system_message"]["content"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1250 # The resume options (base options) must not contain the transcript block.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1251 self.assertNotIn("PRIOR OWNED CONVERSATION TRANSCRIPT", resume_sys)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1252 # The create call had the transcript; verify only one create happened.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1253 self.assertEqual(len(client.create_calls), 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1254
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1255 async def test_history_with_no_field_uses_base_system_message_on_create(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1256 """Absent history field: create_session uses base system_message without transcript."""
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1257 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1258 await sidecar.dispatch(self._turn_start("r-nofield", "c-nofield"))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1259 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1260 self.assertEqual(len(client.create_calls), 1)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1261 _, opts = client.create_calls[0]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1262 self.assertNotIn("PRIOR OWNED CONVERSATION TRANSCRIPT", opts["system_message"]["content"])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1263
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1264 async def test_profile_switch_create_fallback_gets_history(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1265 """After a profile switch the new derived session is a fresh create;
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1266 the history must be injected into the new session's system_message only."""
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1267 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1268
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1269 # First turn: public_visitor session created; no prior history yet.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1270 await sidecar.dispatch(self._turn_start("r-sw-1", "c-switch"))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1271 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1272 visitor_sdk_id = _fake_sdk_id("c-switch", "public_visitor")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1273 friend_sdk_id = _fake_sdk_id("c-switch", "invited_friend")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1274 self.assertNotEqual(visitor_sdk_id, friend_sdk_id)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1275 first_create_content = client.create_calls[0][1]["system_message"]["content"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1276 # No transcript on initial create (no history provided).
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1277 self.assertNotIn("PRIOR OWNED CONVERSATION TRANSCRIPT", first_create_content)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1278
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1279 # Second turn: invited_friend profile — old session deleted, new fresh create.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1280 # Now we send history representing the prior visitor exchange.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1281 friend_hist = [
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1282 {"role": "user", "content": "visitor msg"},
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1283 {"role": "assistant", "content": "answer"},
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1284 ]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1285 await sidecar.dispatch({
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1286 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1287 "request_id": "r-sw-2",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1288 "conversation_id": "c-switch",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1289 "prompt": "switch question",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1290 "prompt_profile": "invited_friend",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1291 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1292 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1293 "history": friend_hist,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1294 })
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1295 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1296
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1297 # Old visitor session must be deleted.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1298 self.assertIn(visitor_sdk_id, client.delete_calls)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1299 # New session created under invited_friend derived ID.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1300 create_ids = [sid for sid, _ in client.create_calls]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1301 self.assertIn(friend_sdk_id, create_ids)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1302
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1303 # Second create must include the history transcript.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1304 second_create_opts = dict(client.create_calls)[friend_sdk_id]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1305 second_content = second_create_opts["system_message"]["content"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1306 self.assertIn("PRIOR OWNED CONVERSATION TRANSCRIPT", second_content)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1307 self.assertIn("visitor msg", second_content)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1308 self.assertIn("answer", second_content)
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1309 self.assertIn("untrusted", second_content.lower())
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1310
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1311 done = [e for e in self.output if e["type"] == "turn.done" and e["request_id"] == "r-sw-2"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1312 self.assertFalse(done[0].get("failed", False))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1313
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1314 async def test_history_not_exposed_in_events(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1315 """History must not appear in any turn.accepted, assistant.delta, or
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1316 turn.done events emitted to the client."""
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1317 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1318 sensitive = "SENSITIVE_TRANSCRIPT_DATA_XYZ"
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1319 hist = [{"role": "user", "content": sensitive}]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1320 await sidecar.dispatch(self._turn_start("r-safe", "c-safe", history=hist))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1321 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1322
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1323 for ev in self.output:
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1324 for field in ("delta", "content", "prompt"):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1325 val = ev.get(field, "")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1326 if isinstance(val, str):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1327 self.assertNotIn(sensitive, val,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1328 f"history leaked into event[{field}]: {ev}")
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1329
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1330 async def test_existing_conversation_history_compatibility(self):
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1331 """Conversations without history (pre-change) work normally: no
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1332 transcript block is injected when history is absent/empty."""
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1333 sidecar, client = await self.make_sidecar()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1334 # Simulate an old-style command with no history key.
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1335 await sidecar.dispatch({
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1336 "command": "turn.start",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1337 "request_id": "r-compat",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1338 "conversation_id": "c-compat",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1339 "prompt": "legacy",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1340 "prompt_profile": "june_admin",
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1341 "prompt_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1342 "knowledge_version": 1,
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1343 })
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1344 await sidecar.drain_events()
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1345 done = [e for e in self.output if e["type"] == "turn.done" and e["request_id"] == "r-compat"]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1346 self.assertFalse(done[0].get("failed", False))
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1347 _, opts = client.create_calls[0]
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1348 self.assertNotIn("PRIOR OWNED CONVERSATION TRANSCRIPT", opts["system_message"]["content"])
056790c4fb0d add role-aware Epi assistant prompts MrJuneJune <me@mrjunejune.com> parents: 260 diff changeset	1349
260 1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	1350
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	1351 if __name__ == "__main__":
1f9877b637e9 Add Copilot-powered cyberpunk JRPG chat MrJuneJune <mrjunejune@users.noreply.github.com> parents: diff changeset	1352 unittest.main()

Mercurial

annotate mrjunejune/inference/copilot_sidecar_test.py @ 279:b3b547563ec7