feat(personas): Add persona_id support to Ticket/WorkerContext and ConductorEngine

This commit is contained in:
2026-03-10 11:09:11 -04:00
parent 41a40aaa68
commit 7ae99f2bc3
7 changed files with 103 additions and 40 deletions

28
tests/test_persona_id.py Normal file
View File

@@ -0,0 +1,28 @@
import pytest
from src.models import Ticket, WorkerContext
def test_ticket_persona_id_serialization():
ticket = Ticket(
id="test-1", description="Test task", persona_id="SecuritySpecialist"
)
data = ticket.to_dict()
assert data["persona_id"] == "SecuritySpecialist"
def test_ticket_persona_id_deserialization():
data = {"id": "test-2", "description": "Test task 2", "persona_id": "CodeReviewer"}
ticket = Ticket.from_dict(data)
assert ticket.persona_id == "CodeReviewer"
def test_ticket_persona_id_default():
ticket = Ticket(id="test-3", description="Test")
assert ticket.persona_id is None
def test_worker_context_persona_id():
ctx = WorkerContext(
ticket_id="test-1", model_name="gemini-2.5-flash", persona_id="DebugHelper"
)
assert ctx.persona_id == "DebugHelper"