from typing_extensions import TypedDict

from pydantic import BaseModel, TypeAdapter, ConfigDict, with_config
from pydantic.dataclasses import dataclass

PYDANTIC_CFG = ConfigDict(use_attribute_docstrings=True)


class NativeModel(BaseModel):
    model_config = PYDANTIC_CFG

    x: int
    """my docstring"""


@with_config(PYDANTIC_CFG)
class TypedDictModel(TypedDict):
    x: int
    """my docstring"""


@dataclass(config=PYDANTIC_CFG)
class DataclassModel:
    x: int
    """my docstring"""


print(NativeModel.model_json_schema())
# {'properties': {'x': {'description': 'my docstring', 'title': 'X', 'type': 'integer'}}, 'required': ['x'], 'title': 'NativeModel', 'type': 'object'}

print(TypeAdapter(TypedDictModel).json_schema())
# {'properties': {'x': {'description': 'my docstring', 'title': 'X', 'type': 'integer'}}, 'required': ['x'], 'title': 'TypedDictModel', 'type': 'object'}

print(TypeAdapter(DataclassModel).json_schema())
# {'properties': {'x': {'title': 'X', 'type': 'integer'}}, 'required': ['x'], 'title': 'DataclassModel', 'type': 'object'}