#!/usr/bin/env python
"""Demonstrate SQLCoach session persistence across consecutive turns."""
import json
from unittest.mock import Mock
from app.coach_local import SQLCoach
# Create a stable analysis packet (as would come from MCP server)
def make_packet():
return {
"meta": {"run_id": "run_2025_12_06_001", "limits": {"top_k_skills": 10, "bucket_seconds": 5}},
"run_summary": {
"rows": [["run_2025_12_06_001", 350, 85000, 180.0, 472.2, 52.0]]
},
"top_skills": {
"rows": [
["Fireball", 40000, 300, 180.0, 47.1],
["IceShower", 25000, 180, 180.0, 29.4],
["Shadow Burst", 15000, 95, 180.0, 17.6],
["Heal", 5000, 50, 180.0, 5.9],
]
},
"runs_last_n": {
"rows": [[1, 320, 425.0], [2, 340, 478.3], [3, 350, 472.2]]
},
"timeline": {
"rows": [[0, 50, 12, 24.0, 8000], [60, 55, 14, 25.5, 12000], [120, 65, 18, 27.7, 15000]]
},
"skill_deltas": {
"columns": [
"skill_name", "last_share_pct", "prior_avg_share_pct", "delta_share_pp",
"last_hits", "prior_avg_hits", "delta_hits",
"last_crit_rate_pct", "prior_avg_crit_rate_pct", "delta_crit_pp"
],
"rows": [
["Fireball", 47.1, 50.0, -2.9, 300, 310, -10, 52.0, 50.0, 2.0],
["IceShower", 29.4, 28.0, 1.4, 180, 175, 5, 48.0, 48.0, 0.0],
["Shadow Burst", 17.6, 18.0, -0.4, 95, 98, -3, 55.0, 55.0, 0.0],
["Heal", 5.9, 4.0, 1.9, 50, 45, 5, 0.0, 0.0, 0.0],
]
},
"windows": {
"early_window": {"start_s": 0, "end_s": 60, "damage": 20000, "top_skills": ["Fireball"]},
"late_window": {"start_s": 120, "end_s": 180, "damage": 18000, "top_skills": ["IceShower"]},
"top_damage_windows": [
{"start_s": 120, "end_s": 140, "damage": 15000, "top_skills": ["IceShower", "Shadow Burst"]},
],
},
"actions": {"top_levers": ["Restore Fireball usage"]},
"notes": [],
}
def main():
print("=" * 80)
print("SQLCoach Session Persistence Demo")
print("=" * 80)
print()
# Create coach instance (simulating UI MainWindow creation)
model_manager = Mock()
coach = SQLCoach(model_manager)
session_id = coach._coach_session_id
print(f"Coach Session ID: {session_id}")
print(f"(This ID remains stable across all turns until the coach is recreated)")
print()
# Simulate 5 consecutive user questions
questions = [
("Which skill fell off compared to prior runs?", "SKILL_DELTA"),
("Why did Fireball lose damage share?", "FOLLOWUP_WHY"),
("Where did my damage spike?", "SPIKE_ANALYSIS"),
("Give me the top 3 changes for next run", "ACTION_PLAN"),
("Should I front-load damage?", "FRONTLOAD"),
]
packet = make_packet()
session_ids_from_traces = []
for i, (question, expected_intent) in enumerate(questions, 1):
print(f"\n{'─' * 80}")
print(f"Turn {i}: User Question")
print(f"{'─' * 80}")
print(f"Q: {question}")
print()
# Call coach answer (simulating CoachWorker.run())
answer, tool_trace = coach.answer(
question=question,
payload={},
schema={},
query_callback=lambda x: {},
analysis_callback=lambda: packet,
)
# Extract trace info
if tool_trace:
trace_entry = tool_trace[0]
trace_session_id = trace_entry.get("coach_session_id")
tool_name = trace_entry.get("tool", "query_dps")
session_ids_from_traces.append(trace_session_id)
print(f"Intent Detected: {coach._state.get('last_intent', 'unknown').upper()}")
print(f"Tool Called: {tool_name}")
print(f"Session ID in Trace: {trace_session_id}")
print(f"Session ID Match: {'✓' if trace_session_id == session_id else '✗'}")
print()
# Show a preview of the answer
answer_preview = answer[:150].replace("\n", " ") + "..." if len(answer) > 150 else answer.replace("\n", " ")
print(f"Coach Response Preview:")
print(f"{answer_preview}")
else:
print("(No MCP tool trace for this intent)")
# Verification
print()
print(f"\n{'=' * 80}")
print("Session Persistence Verification")
print(f"{'=' * 80}")
print(f"Initial Coach Session ID: {session_id}")
print(f"Unique session IDs in traces: {len(set(session_ids_from_traces))}")
print(f"All traces use same session ID: {'✓ YES' if all(sid == session_id for sid in session_ids_from_traces) else '✗ NO'}")
print()
print("Conclusion:")
print(f"✓ Coach instance is reused across all {len(questions)} turns")
print(f"✓ Session ID is stable and included in every trace")
print(f"✓ Conversational state persists (FOLLOWUP_WHY triggered by stored SKILL_DELTA)")
print()
if __name__ == "__main__":
main()
