2424 import orjson # noqa
2525 from fastapi .responses import ORJSONResponse as JSONResponse
2626except ImportError : # pragma: nocover
27- from starlette .responses import JSONResponse
27+ from starlette .responses import JSONResponse # type: ignore
2828
2929
3030def create_request_model (
3131 model_name = "SearchGetRequest" ,
3232 base_model : Union [Type [BaseModel ], Type [APIRequest ]] = BaseSearchGetRequest ,
3333 extensions : Optional [List [ApiExtension ]] = None ,
3434 mixins : Optional [Union [List [Type [BaseModel ]], List [Type [APIRequest ]]]] = None ,
35- request_type : Optional [ str ] = "GET" ,
35+ request_type : str = "GET" ,
3636) -> Union [Type [BaseModel ], Type [APIRequest ]]:
3737 """Create a pydantic model for validating request bodies."""
3838 fields = {}
39- extension_models = []
39+ extension_models : List [ Union [ Type [ BaseModel ], Type [ APIRequest ]]] = []
4040
4141 # Check extensions for additional parameters to search
4242 for extension in extensions or []:
@@ -54,7 +54,7 @@ def create_request_model(
5454 # Handle POST requests
5555 elif all ([issubclass (m , BaseModel ) for m in models ]):
5656 for model in models :
57- for k , field_info in model .model_fields .items ():
57+ for k , field_info in model .model_fields .items (): # type: ignore
5858 fields [k ] = (field_info .annotation , field_info )
5959
6060 return create_model (model_name , ** fields , __base__ = base_model ) # type: ignore
@@ -64,11 +64,10 @@ def create_request_model(
6464
6565def create_get_request_model (
6666 extensions : Optional [List [ApiExtension ]],
67- base_model : BaseSearchGetRequest = BaseSearchGetRequest ,
67+ base_model : Type [ BaseSearchGetRequest ] = BaseSearchGetRequest ,
6868) -> Type [APIRequest ]:
6969 """Wrap create_request_model to create the GET request model."""
70-
71- return create_request_model (
70+ return create_request_model ( # type: ignore
7271 "SearchGetRequest" ,
7372 base_model = base_model ,
7473 extensions = extensions ,
@@ -78,10 +77,10 @@ def create_get_request_model(
7877
7978def create_post_request_model (
8079 extensions : Optional [List [ApiExtension ]],
81- base_model : BaseSearchPostRequest = BaseSearchPostRequest ,
80+ base_model : Type [ BaseSearchPostRequest ] = BaseSearchPostRequest ,
8281) -> Type [BaseModel ]:
8382 """Wrap create_request_model to create the POST request model."""
84- return create_request_model (
83+ return create_request_model ( # type: ignore
8584 "SearchPostRequest" ,
8685 base_model = base_model ,
8786 extensions = extensions ,
0 commit comments