@@ -112,22 +112,46 @@ class ResultConverter:
112112 @overload
113113 def to_schema (
114114 self ,
115- data : ModelOrRowMappingT ,
115+ data : "ModelOrRowMappingT" ,
116+ * ,
117+ schema_type : None = None ,
118+ ) -> "ModelOrRowMappingT" : ...
119+
120+ @overload
121+ def to_schema (
122+ self ,
123+ data : "Union[ModelProtocol, RowMapping]" ,
124+ * ,
125+ schema_type : "type[ModelDTOT]" ,
126+ ) -> "ModelDTOT" : ...
127+
128+ @overload
129+ def to_schema (
130+ self ,
131+ data : "ModelOrRowMappingT" ,
116132 total : "Optional[int]" = None ,
117- filters : "Union[Sequence[Union[StatementFilter, ColumnElement[bool]]], Sequence[StatementFilter], None]" = None ,
118133 * ,
119134 schema_type : None = None ,
120- ) -> ModelOrRowMappingT : ...
135+ ) -> " ModelOrRowMappingT" : ...
121136
122137 @overload
123138 def to_schema (
124139 self ,
125- data : "Sequence[ModelOrRowMappingT]" ,
140+ data : "Union[ModelProtocol, RowMapping]" ,
141+ total : "Optional[int]" = None ,
142+ * ,
143+ schema_type : "type[ModelDTOT]" ,
144+ ) -> "ModelDTOT" : ...
145+
146+ @overload
147+ def to_schema (
148+ self ,
149+ data : "ModelOrRowMappingT" ,
126150 total : "Optional[int]" = None ,
127151 filters : "Union[Sequence[Union[StatementFilter, ColumnElement[bool]]], Sequence[StatementFilter], None]" = None ,
128152 * ,
129153 schema_type : None = None ,
130- ) -> OffsetPagination [ ModelOrRowMappingT ] : ...
154+ ) -> " ModelOrRowMappingT" : ...
131155
132156 @overload
133157 def to_schema (
@@ -136,8 +160,34 @@ def to_schema(
136160 total : "Optional[int]" = None ,
137161 filters : "Union[Sequence[Union[StatementFilter, ColumnElement[bool]]], Sequence[StatementFilter], None]" = None ,
138162 * ,
139- schema_type : type [ModelDTOT ],
140- ) -> ModelDTOT : ...
163+ schema_type : "type[ModelDTOT]" ,
164+ ) -> "ModelDTOT" : ...
165+
166+ @overload
167+ def to_schema (
168+ self ,
169+ data : "Sequence[ModelOrRowMappingT]" ,
170+ * ,
171+ schema_type : None = None ,
172+ ) -> "OffsetPagination[ModelOrRowMappingT]" : ...
173+
174+ @overload
175+ def to_schema (
176+ self ,
177+ data : "Union[Sequence[ModelProtocol], Sequence[RowMapping]]" ,
178+ * ,
179+ schema_type : "type[ModelDTOT]" ,
180+ ) -> "OffsetPagination[ModelDTOT]" : ...
181+
182+ @overload
183+ def to_schema (
184+ self ,
185+ data : "Sequence[ModelOrRowMappingT]" ,
186+ total : "Optional[int]" = None ,
187+ filters : "Union[Sequence[Union[StatementFilter, ColumnElement[bool]]], Sequence[StatementFilter], None]" = None ,
188+ * ,
189+ schema_type : None = None ,
190+ ) -> "OffsetPagination[ModelOrRowMappingT]" : ...
141191
142192 @overload
143193 def to_schema (
@@ -146,7 +196,7 @@ def to_schema(
146196 total : "Optional[int]" = None ,
147197 filters : "Union[Sequence[Union[StatementFilter, ColumnElement[bool]]], Sequence[StatementFilter], None]" = None ,
148198 * ,
149- schema_type : type [ModelDTOT ],
199+ schema_type : " type[ModelDTOT]" ,
150200 ) -> "OffsetPagination[ModelDTOT]" : ...
151201
152202 def to_schema (
0 commit comments