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

annotate mrjunejune/inference/copilot_sidecar_test.py @ 269:de291f396881

install initial production config Copy the ignored repository config into /etc/mrjunejune on first deployment while preserving existing production configuration on later deploys. Co-authored-by: Copilot <[email protected]>

author	MrJuneJune <me@mrjunejune.com>
date	Fri, 07 Aug 2026 13:08:10 -0700
parents	056790c4fb0d
children

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 @ 269:de291f396881