mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-09-13 03:32:21 +02:00
24 lines
1.0 KiB
Python
24 lines
1.0 KiB
Python
from src.agent_loop import (
|
|
_classify_agent_request,
|
|
_looks_like_explicit_notes_only_turn,
|
|
_looks_like_implicit_notes_turn,
|
|
)
|
|
|
|
|
|
def test_implicit_saved_item_routes_to_notes_domain():
|
|
prompt = "What did I write down under Heat island caveats? Open the saved item and tell me what it says."
|
|
assert _looks_like_implicit_notes_turn(prompt)
|
|
assert "notes_calendar_tasks" in _classify_agent_request([{"role": "user", "content": prompt}], prompt)["domains"]
|
|
assert _looks_like_explicit_notes_only_turn(prompt)
|
|
|
|
|
|
def test_mixed_implicit_note_request_is_not_notes_only():
|
|
prompt = "Open what I saved as Model benchmark notes, then read Launch Checklist from documents."
|
|
assert "notes_calendar_tasks" in _classify_agent_request([{"role": "user", "content": prompt}], prompt)["domains"]
|
|
assert not _looks_like_explicit_notes_only_turn(prompt)
|
|
|
|
|
|
def test_generic_saved_file_does_not_route_to_notes():
|
|
prompt = "Open the saved file from my workspace."
|
|
assert not _looks_like_implicit_notes_turn(prompt)
|