feat(mma): Implement WorkerContext model
This commit is contained in:
@@ -20,3 +20,12 @@ class Track:
|
|||||||
id: str
|
id: str
|
||||||
description: str
|
description: str
|
||||||
tickets: List[Ticket] = field(default_factory=list)
|
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]
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import pytest
|
import pytest
|
||||||
from models import Ticket, Track
|
from models import Ticket, Track, WorkerContext
|
||||||
|
|
||||||
def test_ticket_instantiation():
|
def test_ticket_instantiation():
|
||||||
"""
|
"""
|
||||||
@@ -67,3 +67,25 @@ def test_track_can_handle_empty_tickets():
|
|||||||
"""
|
"""
|
||||||
track = Track(id="TRACK-2", description="Empty Track", tickets=[])
|
track = Track(id="TRACK-2", description="Empty Track", tickets=[])
|
||||||
assert 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
|
||||||
|
|||||||
Reference in New Issue
Block a user