feat(mma): Implement WorkerContext model

This commit is contained in:
2026-02-26 19:59:51 -05:00
parent ccd286132f
commit ee719296c4
2 changed files with 32 additions and 1 deletions

View File

@@ -20,3 +20,12 @@ class Track:
id: str
description: str
tickets: List[Ticket] = field(default_factory=list)
@dataclass
class WorkerContext:
"""
Represents the context provided to a Tier 3 Worker for a specific ticket.
"""
ticket_id: str
model_name: str
messages: List[dict]

View File

@@ -1,5 +1,5 @@
import pytest
from models import Ticket, Track
from models import Ticket, Track, WorkerContext
def test_ticket_instantiation():
"""
@@ -67,3 +67,25 @@ def test_track_can_handle_empty_tickets():
"""
track = Track(id="TRACK-2", description="Empty Track", tickets=[])
assert track.tickets == []
def test_worker_context_instantiation():
"""
Verifies that a WorkerContext can be instantiated with ticket_id,
model_name, and messages.
"""
ticket_id = "T1"
model_name = "gemini-2.0-flash-lite"
messages = [
{"role": "user", "content": "Hello"},
{"role": "assistant", "content": "Hi there!"}
]
context = WorkerContext(
ticket_id=ticket_id,
model_name=model_name,
messages=messages
)
assert context.ticket_id == ticket_id
assert context.model_name == model_name
assert context.messages == messages