wuyang
21 lines
817 B
Python
21 lines
817 B
Python
from agent_platform.bootstrap import _model_payload
|
|
|
|
|
|
def test_bootstrap_models_override_provider_ids_and_are_public() -> None:
|
|
models = _model_payload()
|
|
assert len(models) == 8
|
|
assert all(model["base_model_id"] is None for model in models)
|
|
assert all(
|
|
model["access_grants"] == [{"principal_type": "user", "principal_id": "*", "permission": "read"}]
|
|
for model in models
|
|
)
|
|
|
|
chat = next(model for model in models if model["id"] == "chat-medium")
|
|
assert chat["params"]["function_calling"] == "native"
|
|
assert chat["meta"]["toolIds"] == ["server:workspace"]
|
|
assert chat["meta"]["capabilities"]["builtin_tools"] is False
|
|
|
|
work = next(model for model in models if model["id"] == "work-medium")
|
|
assert work["params"] == {}
|
|
assert work["meta"]["toolIds"] == []
|