mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-10-06 15:02:20 +02:00
pewdiepie-archdaemon
2e8413a54a
Snapshot current maintainer-preview application changes and regression fixtures for integration into lab. Excludes local runtime data, evaluation outputs and source backups. Focused Python regression selection: 140 passed; full suite not certified.
39 lines
1.8 KiB
Python
39 lines
1.8 KiB
Python
import pytest
|
|
|
|
from src.clean_agent_preview import compact_schemas, provider_compatible_tool_choice_request
|
|
from src.tool_schemas import FUNCTION_TOOL_SCHEMAS
|
|
|
|
|
|
@pytest.mark.parametrize('model', ['Ajax', 'local/Ajax', 'ajax-test'])
|
|
@pytest.mark.parametrize('choice', ['required', {'type': 'function', 'function': {'name': 'manage_tasks'}}])
|
|
def test_ajax_auto_decoding_preserves_selected_tool_boundary(model, choice):
|
|
from copy import deepcopy
|
|
tools = [s for s in FUNCTION_TOOL_SCHEMAS if s['function']['name'] in {'manage_tasks', 'manage_notes'}]
|
|
request = {'tools': tools, 'tool_choice': choice}
|
|
original = deepcopy(request)
|
|
compatible = provider_compatible_tool_choice_request(request, model)
|
|
assert compatible['tool_choice'] == 'auto'
|
|
names = {s['function']['name'] for s in compatible['tools']}
|
|
assert names == ({'manage_tasks'} if isinstance(choice, dict) else {'manage_tasks', 'manage_notes'})
|
|
assert request == original
|
|
|
|
|
|
def test_ajax_explicit_no_tools_is_preserved():
|
|
request = {'tool_choice': 'none', 'tools': []}
|
|
assert provider_compatible_tool_choice_request(request, 'Ajax') is request
|
|
|
|
|
|
@pytest.mark.parametrize('model', ['Ajax', 'ajax_c375', 'local/Ajax', 'ajax-test'])
|
|
def test_ajax_does_not_offer_ask_user(model):
|
|
tools = [s for s in FUNCTION_TOOL_SCHEMAS
|
|
if s['function']['name'] in {'ask_user', 'web_fetch'}]
|
|
names = {s['function']['name'] for s in compact_schemas(tools, model=model)}
|
|
assert names == {'web_fetch'}
|
|
assert any(s['function']['name'] == 'ask_user' for s in tools)
|
|
|
|
|
|
@pytest.mark.parametrize('model', [None, 'kimi-k3', 'odysseus-qwen3.5', 'not-ajax'])
|
|
def test_other_models_keep_ask_user(model):
|
|
tools = [s for s in FUNCTION_TOOL_SCHEMAS if s['function']['name'] == 'ask_user']
|
|
assert [s['function']['name'] for s in compact_schemas(tools, model=model)] == ['ask_user']
|