mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-10-06 15:02:20 +02:00
17 lines
1.1 KiB
Python
17 lines
1.1 KiB
Python
import pytest
|
|
from src.clean_agent_preview import unbound_lookup_reference
|
|
|
|
|
|
@pytest.mark.parametrize('prompt', ['can u look it up', 'please find that', 'what about its price?', 'Could you search for this please?'])
|
|
def test_missing_subject_requires_context(prompt):
|
|
history = [{'role': 'system', 'content': 'System'}, {'role': 'user', 'content': prompt}]
|
|
assert unbound_lookup_reference(prompt, history)
|
|
assert not unbound_lookup_reference(prompt, history, supplied_context=True)
|
|
prior = [{'role': 'user', 'content': 'I am considering a Firefox phone.'}, {'role': 'assistant', 'content': 'Which features matter?'}]
|
|
assert not unbound_lookup_reference(prompt, prior + history)
|
|
|
|
|
|
@pytest.mark.parametrize('prompt', ['Find the Python documentation', 'Look up this model: WH-1000XM5', 'What about its price compared to a Pixel 9?', 'Search for this https://example.org', 'What is it like living in Japan?', 'helo'])
|
|
def test_explicit_subjects_and_other_intents_keep_normal_routing(prompt):
|
|
assert not unbound_lookup_reference(prompt, [{'role': 'user', 'content': prompt}])
|