diff --git a/src/py_avro_schema/_schemas.py b/src/py_avro_schema/_schemas.py index f74299b..f36eedf 100644 --- a/src/py_avro_schema/_schemas.py +++ b/src/py_avro_schema/_schemas.py @@ -872,6 +872,7 @@ class PydanticSchema(RecordSchema): @classmethod def handles_type(cls, py_type: Type) -> bool: """Whether this schema class can represent a given Python class""" + py_type = _type_from_annotated(py_type) return hasattr(py_type, "__fields__") def __init__(self, py_type: Type[pydantic.BaseModel], namespace: Optional[str] = None, options: Option = Option(0)): diff --git a/tests/test_pydantic.py b/tests/test_pydantic.py index 35938a0..d1d06a9 100644 --- a/tests/test_pydantic.py +++ b/tests/test_pydantic.py @@ -38,6 +38,23 @@ class PyType(pydantic.BaseModel): assert_schema(PyType, expected) +def test_class_annotated(): + class PyType(pydantic.BaseModel): + field_a: str + + expected = { + "type": "record", + "name": "PyType", + "fields": [ + { + "name": "field_a", + "type": "string", + } + ], + } + assert_schema(Annotated[PyType, ...], expected) + + def test_string_field_default(): class PyType(pydantic.BaseModel): field_a: str = ""