File size: 1,165 Bytes
102dd4f cf6d08c 102dd4f edb038d dc8cb49 102dd4f fcf2812 | 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 | from app.ai.tool_schemas import get_all_tool_schemas, get_tool_schemas
def test_new_user_tools_are_onboarding_only():
names = {schema["function"]["name"] for schema in get_tool_schemas("new_user")}
assert names == {
"about_falzh",
"create_driver_account",
"switch_to_driver",
"switch_to_passenger",
}
def test_driver_tools_exclude_passenger_booking_tools():
names = {schema["function"]["name"] for schema in get_tool_schemas("driver")}
assert "search_trips" not in names
assert "select_trip" not in names
assert "initiate_trip_action" in names
assert "update_trip_field" in names
assert "switch_to_passenger" in names
def test_passenger_tools_exclude_driver_management_tools():
names = {schema["function"]["name"] for schema in get_tool_schemas("passenger")}
assert "add_trip_by_driver" not in names
assert "check_driver_info" not in names
assert "switch_to_driver" in names
assert "create_driver_account" in names
def test_all_tool_schemas_include_every_tool():
names = {schema["function"]["name"] for schema in get_all_tool_schemas()}
assert len(names) == 12
|