@@ -275,10 +275,6 @@ def get_responses(
275
275
}
276
276
)
277
277
_schemas [UnprocessableEntity .__name__ ] = Schema (** UnprocessableEntity .schema ())
278
- # if not responses.get("500"):
279
- # _responses["500"] = Response(description=HTTP_STATUS["500"])
280
- # handle extra_responses
281
-
282
278
for key , response in responses .items ():
283
279
# Verify that the response is a class and that class is a subclass of `pydantic.BaseModel`
284
280
if inspect .isclass (response ) and issubclass (response , BaseModel ):
@@ -302,17 +298,13 @@ def get_responses(
302
298
if definitions :
303
299
for name , value in definitions .items ():
304
300
_schemas [name ] = Schema (** value )
305
-
306
- # Verify that if the key is "204", the response is None,
307
- # because http status code "204" means return "No Content"
301
+ # Verify that if the response is None, because http status code "204" means return "No Content"
308
302
elif response is None :
309
303
_responses [key ] = Response (
310
304
description = HTTP_STATUS .get (key , "" ),
311
305
)
312
-
313
306
else :
314
- raise AttributeError (f'{ response } is invalid `pydantic.BaseModel` '
315
- f'or if the key is "204" the type should be None' )
307
+ raise TypeError (f'{ response } is invalid `pydantic.BaseModel`.' )
316
308
# handle extra_responses
317
309
for key , value in extra_responses .items ():
318
310
# key "200" value {"content":{"text/csv":{"schema":{"type": "string"}}}}
0 commit comments