wehub-resource-sync
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from datetime import date
|
|
|
|
import pytest
|
|
|
|
from livekit.agents import beta
|
|
|
|
pytestmark = pytest.mark.unit
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_dob_two_digit_year_normalized() -> None:
|
|
# The prompt asks the model to normalize two-digit years, but smaller/faster
|
|
# models often pass the spoken value through literally. The tool layer must not
|
|
# accept "90" as year 90 AD. https://github.com/livekit/agents/issues/6067
|
|
task = beta.workflows.GetDOBTask(require_confirmation=True)
|
|
|
|
await task._update_dob_impl(90, 5, 15, ctx=None) # type: ignore[arg-type]
|
|
assert task._current_dob == date(1990, 5, 15)
|
|
|
|
await task._update_dob_impl(5, 3, 1, ctx=None) # type: ignore[arg-type]
|
|
assert task._current_dob == date(2005, 3, 1)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_dob_four_digit_year_unchanged() -> None:
|
|
task = beta.workflows.GetDOBTask(require_confirmation=True)
|
|
|
|
await task._update_dob_impl(1962, 7, 4, ctx=None) # type: ignore[arg-type]
|
|
assert task._current_dob == date(1962, 7, 4)
|
|
|
|
await task._update_dob_impl(2001, 12, 31, ctx=None) # type: ignore[arg-type]
|
|
assert task._current_dob == date(2001, 12, 31)
|