wehub-resource-sync
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
from typing import get_args
|
|
|
|
import agents.realtime as realtime
|
|
from agents.realtime.model_events import RealtimeModelEvent
|
|
from agents.usage import Usage
|
|
|
|
|
|
def test_all_events_have_type() -> None:
|
|
"""Test that all events have a type."""
|
|
events = get_args(RealtimeModelEvent)
|
|
assert len(events) > 0
|
|
for event in events:
|
|
assert event.type is not None
|
|
assert isinstance(event.type, str)
|
|
|
|
|
|
def test_usage_event_types_are_publicly_exported() -> None:
|
|
expected_exports = {
|
|
"RealtimeModelCachedTokensDetails",
|
|
"RealtimeModelInputTokensDetails",
|
|
"RealtimeModelOutputTokensDetails",
|
|
"RealtimeModelUsageEvent",
|
|
}
|
|
|
|
assert expected_exports <= set(realtime.__all__)
|
|
for name in expected_exports:
|
|
assert getattr(realtime, name) is not None
|
|
|
|
|
|
def test_custom_model_can_construct_typed_usage_without_openai_types() -> None:
|
|
event = realtime.RealtimeModelUsageEvent(
|
|
usage=Usage(requests=1, input_tokens=8, output_tokens=5, total_tokens=13),
|
|
input_tokens_details=realtime.RealtimeModelInputTokensDetails(
|
|
text_tokens=2,
|
|
audio_tokens=6,
|
|
cached_tokens=3,
|
|
cached_tokens_details=realtime.RealtimeModelCachedTokensDetails(
|
|
text_tokens=1,
|
|
audio_tokens=2,
|
|
),
|
|
),
|
|
output_tokens_details=realtime.RealtimeModelOutputTokensDetails(
|
|
text_tokens=1,
|
|
audio_tokens=4,
|
|
),
|
|
)
|
|
|
|
assert event.input_tokens_details is not None
|
|
assert event.input_tokens_details.audio_tokens == 6
|
|
assert event.output_tokens_details is not None
|
|
assert event.output_tokens_details.audio_tokens == 4
|