Merge d6c58fe43c into 460f8d2cc8

fastapi/openapi/utils.py

@@ -337,6 +337,19 @@ def get_openapi_path(
                         )
                         callbacks[callback.name] = {callback.path: cb_path}
                 operation["callbacks"] = callbacks
+
+            if route.return_response_classes:
+                if route.return_response_models:
+                    response_classes = (
+                        current_response_class,
+                        *route.return_response_classes,
+                    )
+                else:
+                    response_classes = tuple(route.return_response_classes)
+            else:
+                response_classes = (current_response_class,)
+
+            for response_class in response_classes:
                 if route.status_code is not None:
                     status_code = str(route.status_code)
                 else:
@@ -345,7 +358,7 @@ def get_openapi_path(
                     # doing this inspection tricks, that would probably be in the future
                     # TODO: probably make status_code a default class attribute for all
                     # responses in Starlette
-                response_signature = inspect.signature(current_response_class.__init__)
+                    response_signature = inspect.signature(response_class.__init__)
                     status_code_param = response_signature.parameters.get("status_code")
                     if status_code_param is not None:
                         if isinstance(status_code_param.default, int):
@@ -353,7 +366,10 @@ def get_openapi_path(
                 operation.setdefault("responses", {}).setdefault(status_code, {})[
                     "description"
                 ] = route.response_description
+
                 if is_body_allowed_for_status_code(route.status_code):
+                    return_response_media_type: str | None = response_class.media_type
+
                     # Check for JSONL streaming (generator endpoints)
                     if route.is_json_stream:
                         jsonl_content: dict[str, Any] = {}
@@ -390,9 +406,9 @@ def get_openapi_path(
                         operation.setdefault("responses", {}).setdefault(
                             status_code, {}
                         ).setdefault("content", {})["text/event-stream"] = sse_content
-                elif route_response_media_type:
+                    elif return_response_media_type:
                         response_schema = {"type": "string"}
-                    if lenient_issubclass(current_response_class, JSONResponse):
+                        if lenient_issubclass(response_class, JSONResponse):
                             if route.response_field:
                                 response_schema = get_schema_from_model_field(
                                     field=route.response_field,
@@ -405,8 +421,9 @@ def get_openapi_path(
                         operation.setdefault("responses", {}).setdefault(
                             status_code, {}
                         ).setdefault("content", {}).setdefault(
-                        route_response_media_type, {}
+                            return_response_media_type, {}
                         )["schema"] = response_schema
+
             if route.responses:
                 operation_responses = operation.setdefault("responses", {})
                 for (

fastapi/routing.py

@@ -3,6 +3,7 @@ import email.message
 import functools
 import inspect
 import json
+import operator
 import types
 from collections.abc import (
     AsyncIterator,
@@ -26,7 +27,10 @@ from typing import (
     Annotated,
     Any,
     TypeVar,
+    Union,
     cast,
+    get_args,
+    get_origin,
 )
 
 import anyio
@@ -65,7 +69,7 @@ from fastapi.sse import (
     ServerSentEvent,
     format_sse_event,
 )
-from fastapi.types import DecoratedCallable, IncEx
+from fastapi.types import DecoratedCallable, IncEx, UnionType
 from fastapi.utils import (
     create_model_field,
     generate_unique_id,
@@ -844,12 +848,13 @@ class APIRoute(routing.Route):
         self.path = path
         self.endpoint = endpoint
         self.stream_item_type: Any | None = None
+        self.return_response_models = []
+        self.return_response_classes = []
+
         if isinstance(response_model, DefaultPlaceholder):
             return_annotation = get_typed_return_annotation(endpoint)
-            if lenient_issubclass(return_annotation, Response):
-                response_model = None
-            else:
             stream_item = get_stream_item_type(return_annotation)
+
             if stream_item is not None:
                 # Extract item type for JSONL or SSE streaming when
                 # response_class is DefaultPlaceholder (JSONL) or
@@ -862,9 +867,35 @@ class APIRoute(routing.Route):
                     or lenient_issubclass(response_class, EventSourceResponse)
                 ) and not lenient_issubclass(stream_item, ServerSentEvent):
                     self.stream_item_type = stream_item
+
                 response_model = None
             else:
-                    response_model = return_annotation
+                origin = get_origin(return_annotation)
+
+                if origin is Union or origin is UnionType:
+                    for arg in get_args(return_annotation):
+                        if arg is type(None):
+                            continue
+
+                        if lenient_issubclass(arg, Response):
+                            self.return_response_classes.append(arg)
+                        else:
+                            self.return_response_models.append(arg)
+                elif lenient_issubclass(return_annotation, Response):
+                    self.return_response_classes.append(return_annotation)
+                else:
+                    self.return_response_models.append(return_annotation)
+
+                if self.return_response_models:
+                    if len(self.return_response_models) == 1:
+                        response_model = self.return_response_models[0]
+                    else:
+                        response_model = functools.reduce(
+                            operator.or_, self.return_response_models
+                        )
+                else:
+                    response_model = None
+
         self.response_model = response_model
         self.summary = summary
         self.response_description = response_description

tests/test_response_model_as_return_annotation.py

@@ -1,6 +1,6 @@
 import pytest
 from fastapi import FastAPI
-from fastapi.exceptions import FastAPIError, ResponseValidationError
+from fastapi.exceptions import ResponseValidationError
 from fastapi.responses import JSONResponse, Response
 from fastapi.testclient import TestClient
 from inline_snapshot import snapshot
@@ -248,6 +248,11 @@ def no_response_model_annotation_json_response_class() -> JSONResponse:
     return JSONResponse(content={"foo": "bar"})
 
 
+@app.get("/no_response_model-annotation_response_or_none")
+def no_response_model_annotation_response_or_none() -> Response | None:
+    return Response(content="Foo")
+
+
 client = TestClient(app)
 
 
@@ -496,16 +501,10 @@ def test_no_response_model_annotation_json_response_class():
     assert response.json() == {"foo": "bar"}
 
 
-def test_invalid_response_model_field():
+def test_no_response_model_annotation_response_or_none():
-    app = FastAPI()
+    response = client.get("/no_response_model-annotation_response_or_none")
-    with pytest.raises(FastAPIError) as e:
+    assert response.status_code == 200
-
+    assert response.text == "Foo"
-        @app.get("/")
-        def read_root() -> Response | None:
-            return Response(content="Foo")  # pragma: no cover
-
-    assert "valid Pydantic field type" in e.value.args[0]
-    assert "parameter response_model=None" in e.value.args[0]
 
 
 def test_openapi_schema():
@@ -1077,7 +1076,6 @@ def test_openapi_schema():
                         "responses": {
                             "200": {
                                 "description": "Successful Response",
-                                "content": {"application/json": {"schema": {}}},
                             }
                         },
                     }
@@ -1094,6 +1092,13 @@ def test_openapi_schema():
                         },
                     }
                 },
+                "/no_response_model-annotation_response_or_none": {
+                    "get": {
+                        "summary": "No Response Model Annotation Response Or None",
+                        "operationId": "no_response_model_annotation_response_or_none_no_response_model_annotation_response_or_none_get",
+                        "responses": {"200": {"description": "Successful Response"}},
+                    }
+                },
             },
             "components": {
                 "schemas": {

tests/test_tutorial/test_response_model/test_tutorial003_02.py

@@ -45,7 +45,6 @@ def test_openapi_schema():
                         "responses": {
                             "200": {
                                 "description": "Successful Response",
-                                "content": {"application/json": {"schema": {}}},
                             },
                             "422": {
                                 "description": "Validation Error",

tests/test_tutorial/test_response_model/test_tutorial003_03.py

@@ -24,12 +24,7 @@ def test_openapi_schema():
                     "get": {
                         "summary": "Get Teleport",
                         "operationId": "get_teleport_teleport_get",
-                        "responses": {
+                        "responses": {"307": {"description": "Successful Response"}},
-                            "200": {
-                                "description": "Successful Response",
-                                "content": {"application/json": {"schema": {}}},
-                            }
-                        },
                     }
                 }
             },

tests/test_tutorial/test_response_model/test_tutorial003_04.py

@@ -1,17 +1,20 @@
 import importlib
 
 import pytest
-from fastapi.exceptions import FastAPIError
 
 from ...utils import needs_py310
 
 
+# Previously, unions including `Response` in the return annotation were
+# considered invalid and raised FastAPIError at import time.
+# They are now supported as part of the enhanced return annotation handling.
+# Importing the module should not raise FastAPIError anymore.
 @pytest.mark.parametrize(
     "module_name",
     [
         pytest.param("tutorial003_04_py310", marks=needs_py310),
     ],
 )
-def test_invalid_response_model(module_name: str) -> None:
+def test_response_union_with_response_is_valid(module_name: str) -> None:
-    with pytest.raises(FastAPIError):
+    module = importlib.import_module(f"docs_src.response_model.{module_name}")
-        importlib.import_module(f"docs_src.response_model.{module_name}")
+    assert module is not None