Skip to content
Merged
Show file tree
Hide file tree
Changes from 4 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
from typing import Any, Dict, List, Optional, Union

from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, model_validator
from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictFloat, StrictInt, StrictStr, model_validator

from ..data_type import DataType
from .discriminator import Discriminator
Expand Down Expand Up @@ -36,7 +36,7 @@ class Schema(BaseModel):
minProperties: Optional[int] = Field(default=None, ge=0)
required: Optional[List[str]] = Field(default=None, min_length=1)
enum: Union[None, List[Any]] = Field(default=None, min_length=1)
const: Union[None, StrictStr, StrictInt] = None
const: Union[None, StrictStr, StrictInt, StrictFloat, StrictBool] = None
type: Union[DataType, List[DataType], None] = Field(default=None)
allOf: List[Union[Reference, "Schema"]] = Field(default_factory=list)
oneOf: List[Union[Reference, "Schema"]] = Field(default_factory=list)
Expand Down
5 changes: 5 additions & 0 deletions tests/test_schema/test_schema.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,11 @@ def test_nullable_with_allof():
assert schema.allOf == []


def test_constant_bool():
schema = Schema.model_validate_json('{"type":"boolean", "enum":[true], "const":true, "default":true}')
assert schema.const is True


def test_nullable_with_type_list():
schema = Schema.model_validate_json('{"type": ["string", "number"], "nullable": true}')
assert schema.type == [DataType.STRING, DataType.NUMBER, DataType.NULL]
Expand Down