data-query-agent/tests/test_models.py

from data_query_agent.models import QueryAnalysis, QueryDecision, SkillParameters


def test_structured_output_schemas_require_every_property() -> None:
    for model in (SkillParameters, QueryDecision, QueryAnalysis):
        schema = model.model_json_schema()
        assert set(schema["required"]) == set(schema["properties"])
        assert schema["additionalProperties"] is False