From cc891463af36da3e804d277f5ceba3f09d6e0b40 Mon Sep 17 00:00:00 2001
From: Maxime Goyette <maxgoyette0@gmail.com>
Date: Mon, 10 Mar 2025 18:29:35 -0400
Subject: [PATCH] Feature: jsonable_encoder should pass context to pydantic
 models

---
 fastapi/_compat.py             |  2 ++
 fastapi/encoders.py            | 16 ++++++++++++++++
 tests/test_compat.py           | 11 +++++++++++
 tests/test_jsonable_encoder.py | 29 ++++++++++++++++++++++++++++-
 4 files changed, 57 insertions(+), 1 deletion(-)

diff --git a/fastapi/_compat.py b/fastapi/_compat.py
index 26b6638c8..279db8b74 100644
--- a/fastapi/_compat.py
+++ b/fastapi/_compat.py
@@ -453,6 +453,8 @@ else:
     def _model_dump(
         model: BaseModel, mode: Literal["json", "python"] = "json", **kwargs: Any
     ) -> Any:
+        if not PYDANTIC_V2:
+            kwargs.pop("context", None)
         return model.dict(**kwargs)
 
     def _get_model_config(model: BaseModel) -> Any:
diff --git a/fastapi/encoders.py b/fastapi/encoders.py
index b037f8bb5..a4b4cea02 100644
--- a/fastapi/encoders.py
+++ b/fastapi/encoders.py
@@ -126,6 +126,15 @@ def jsonable_encoder(
             """
         ),
     ] = None,
+    context: Annotated[
+        Optional[Any],
+        Doc(
+            """
+            Pydantic's `context` parameter, passed to Pydantic serializers.
+            This only works with Pydantic V2.
+            """
+        ),
+    ] = None,
     by_alias: Annotated[
         bool,
         Doc(
@@ -225,6 +234,7 @@ def jsonable_encoder(
             mode="json",
             include=include,
             exclude=exclude,
+            context=context,
             by_alias=by_alias,
             exclude_unset=exclude_unset,
             exclude_none=exclude_none,
@@ -234,6 +244,7 @@ def jsonable_encoder(
             obj_dict = obj_dict["__root__"]
         return jsonable_encoder(
             obj_dict,
+            context=context,
             exclude_none=exclude_none,
             exclude_defaults=exclude_defaults,
             # TODO: remove when deprecating Pydantic v1
@@ -247,6 +258,7 @@ def jsonable_encoder(
             obj_dict,
             include=include,
             exclude=exclude,
+            context=context,
             by_alias=by_alias,
             exclude_unset=exclude_unset,
             exclude_defaults=exclude_defaults,
@@ -281,6 +293,7 @@ def jsonable_encoder(
             ):
                 encoded_key = jsonable_encoder(
                     key,
+                    context=context,
                     by_alias=by_alias,
                     exclude_unset=exclude_unset,
                     exclude_none=exclude_none,
@@ -289,6 +302,7 @@ def jsonable_encoder(
                 )
                 encoded_value = jsonable_encoder(
                     value,
+                    context=context,
                     by_alias=by_alias,
                     exclude_unset=exclude_unset,
                     exclude_none=exclude_none,
@@ -305,6 +319,7 @@ def jsonable_encoder(
                     item,
                     include=include,
                     exclude=exclude,
+                    context=context,
                     by_alias=by_alias,
                     exclude_unset=exclude_unset,
                     exclude_defaults=exclude_defaults,
@@ -335,6 +350,7 @@ def jsonable_encoder(
         data,
         include=include,
         exclude=exclude,
+        context=context,
         by_alias=by_alias,
         exclude_unset=exclude_unset,
         exclude_defaults=exclude_defaults,
diff --git a/tests/test_compat.py b/tests/test_compat.py
index 43c686489..39f45730a 100644
--- a/tests/test_compat.py
+++ b/tests/test_compat.py
@@ -5,6 +5,7 @@ from fastapi._compat import (
     ModelField,
     Undefined,
     _get_model_config,
+    _model_dump,
     get_cached_model_fields,
     get_model_fields,
     is_bytes_sequence_annotation,
@@ -163,3 +164,13 @@ def test_get_model_fields_cached():
 
     assert non_cached_fields is not non_cached_fields2
     assert cached_fields is cached_fields2
+
+
+@needs_pydanticv1
+def test_model_dump_remove_context_kwarg_in_pydanticv1() -> None:
+    class Model(BaseModel):
+        foo: str
+
+    # The following instruction would throw an error
+    # in pv1 if the context kwarg was not removed.
+    _model_dump(Model(foo="bar"), context=1)
diff --git a/tests/test_jsonable_encoder.py b/tests/test_jsonable_encoder.py
index 447c5b4d6..843007af4 100644
--- a/tests/test_jsonable_encoder.py
+++ b/tests/test_jsonable_encoder.py
@@ -9,7 +9,11 @@ from typing import Optional
 import pytest
 from fastapi._compat import PYDANTIC_V2, Undefined
 from fastapi.encoders import jsonable_encoder
-from pydantic import BaseModel, Field, ValidationError
+from pydantic import (
+    BaseModel,
+    Field,
+    ValidationError,
+)
 
 from .utils import needs_pydanticv1, needs_pydanticv2
 
@@ -319,3 +323,26 @@ def test_encode_deque_encodes_child_models():
 def test_encode_pydantic_undefined():
     data = {"value": Undefined}
     assert jsonable_encoder(data) == {"value": None}
+
+
+@needs_pydanticv2
+def test_encode_with_context() -> None:
+    from pydantic import SerializationInfo, field_serializer
+
+    class ModelWithContextualSerializer(BaseModel):
+        value: int
+
+        @field_serializer("value")
+        def serialize_value(value: int, info: SerializationInfo) -> int:
+            if info.context is not None:
+                if isinstance(
+                    value_from_context := info.context.get("value"), int
+                ):
+                    return value_from_context
+
+            return value
+
+    model = ModelWithContextualSerializer(value=1)
+
+    assert jsonable_encoder(model) == {"value": 1}
+    assert jsonable_encoder(model, context={"value": 2}) == {"value": 2}