From ac4eb4a965174b9e72b05d373a893d2adbcbefe0 Mon Sep 17 00:00:00 2001 From: kshramt Date: Wed, 27 Mar 2024 21:20:42 +0900 Subject: [PATCH] Support the `schema_generator` argument --- fastapi/openapi/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/fastapi/openapi/utils.py b/fastapi/openapi/utils.py index 5bfb5acef..b341a0445 100644 --- a/fastapi/openapi/utils.py +++ b/fastapi/openapi/utils.py @@ -450,6 +450,7 @@ def get_openapi( contact: Optional[Dict[str, Union[str, Any]]] = None, license_info: Optional[Dict[str, Union[str, Any]]] = None, separate_input_output_schemas: bool = True, + schema_generator: Optional[GenerateJsonSchema] = None, ) -> Dict[str, Any]: info: Dict[str, Any] = {"title": title, "version": version} if summary: @@ -471,7 +472,8 @@ def get_openapi( operation_ids: Set[str] = set() all_fields = get_fields_from_routes(list(routes or []) + list(webhooks or [])) model_name_map = get_compat_model_name_map(all_fields) - schema_generator = GenerateJsonSchema(ref_template=REF_TEMPLATE) + if schema_generator is None: + schema_generator = GenerateJsonSchema(ref_template=REF_TEMPLATE) field_mapping, definitions = get_definitions( fields=all_fields, schema_generator=schema_generator,