File size: 1,081 Bytes
0366d65 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | from datetime import datetime
from dateutil import parser as dtparser
from server.agent import run_agent, run_agent_stream
from server.schema import ActionPlan
def test_stream_yields_partials_then_plan():
steps = 0
final = None
for acc, plan in run_agent_stream("Alex: lunch 1pm tomorrow?"):
steps += 1
assert isinstance(acc, str)
if plan is not None:
final = plan
assert steps > 1 # streamed in chunks
assert isinstance(final, ActionPlan)
assert len(final.events) == 1
# start/end parse as ISO 8601
dtparser.isoparse(final.events[0].start)
dtparser.isoparse(final.events[0].end)
def test_no_time_thread_has_no_events():
plan = run_agent("lol that meme killed me")
assert isinstance(plan, ActionPlan)
assert plan.events == []
assert plan.reply_draft # still drafts a reply
def test_run_agent_marks_images():
# stub ignores image content but the path should not error
plan = run_agent("dinner 6pm", images=["data:image/png;base64,xx"])
assert isinstance(plan, ActionPlan)
|