✨ Enhanced OpenAPI docstrings (#121)

vortico · Sep 3, 2024 · d944180 · d944180
1 parent 96229a1
commit d944180
Show file tree

Hide file tree

Showing 4 changed files with 729 additions and 518 deletions.
diff --git a/flama/schemas/generator.py b/flama/schemas/generator.py
@@ -364,22 +364,22 @@ def _build_endpoint_parameters(
     def _build_endpoint_body(
         self, endpoint: EndpointInfo, metadata: t.Dict[str, t.Any]
     ) -> t.Optional[openapi.RequestBody]:
-        if not endpoint.body_parameter:
-            return None
+        content = {k: v for k, v in metadata.get("requestBody", {}).get("content", {}).items()}
+
+        if endpoint.body_parameter:
+            if endpoint.body_parameter.schema.schema not in self.schemas:
+                self.schemas.register(schema=endpoint.body_parameter.schema.schema)
+
+            content["application/json"] = openapi.MediaType(
+                schema=self.schemas.get_openapi_ref(endpoint.body_parameter.schema.schema, multiple=False),
+            )
 
-        if endpoint.body_parameter.schema.schema not in self.schemas:
-            self.schemas.register(schema=endpoint.body_parameter.schema.schema)
+        if not content:
+            return None
 
         return openapi.RequestBody(
-            content={
-                "application/json": openapi.MediaType(
-                    schema=self.schemas.get_openapi_ref(endpoint.body_parameter.schema.schema, multiple=False),
-                )
-            },
-            **{
-                x: metadata.get("requestBody", {}).get("content", {}).get("application/json", {}).get(x)
-                for x in ("description", "required")
-            },
+            content=content,
+            **{x: metadata.get("requestBody", {}).get(x) for x in ("description", "required")},
         )
 
     def _build_endpoint_default_response(self, metadata: t.Dict[str, t.Any]) -> openapi.Response: