Pass prometheus_metrics_include_in_schema to include_in_schema

vrslev · vrslev · commit 6bb2ccdf943e · 2024-10-11T15:20:17.000+03:00
diff --git a/microbootstrap/bootstrappers/fastapi.py b/microbootstrap/bootstrappers/fastapi.py
@@ -106,7 +106,7 @@ def bootstrap_after(self, application: fastapi.FastAPI) -> fastapi.FastAPI:
         ).expose(
             application,
             endpoint=self.instrument_config.prometheus_metrics_path,
-            include_in_schema=False,
+            include_in_schema=self.instrument_config.prometheus_metrics_include_in_schema,
             **self.instrument_config.prometheus_expose_params,
         )
         return application
diff --git a/microbootstrap/bootstrappers/litestar.py b/microbootstrap/bootstrappers/litestar.py
@@ -126,7 +126,7 @@ class LitestarPrometheusInstrument(PrometheusInstrument[LitestarPrometheusConfig
     def bootstrap_before(self) -> dict[str, typing.Any]:
         class LitestarPrometheusController(PrometheusController):
             path = self.instrument_config.prometheus_metrics_path
-            include_in_schema = False
+            include_in_schema = self.instrument_config.prometheus_metrics_include_in_schema
             openmetrics_format = True
 
         litestar_prometheus_config: typing.Final = PrometheusConfig(

Original file line number	Diff line number	Diff line change
`@@ -106,7 +106,7 @@ def bootstrap_after(self, application: fastapi.FastAPI) -> fastapi.FastAPI:`
`106`	`106`	`).expose(`
`107`	`107`	`application,`
`108`	`108`	`endpoint=self.instrument_config.prometheus_metrics_path,`
`109`		`- include_in_schema=False,`
	`109`	`+ include_in_schema=self.instrument_config.prometheus_metrics_include_in_schema,`
`110`	`110`	`**self.instrument_config.prometheus_expose_params,`
`111`	`111`	`)`
`112`	`112`	`return application`