feat(mma): Implement WorkerContext model
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user