Add polymorphic_serialization param to model_dump

Author: YuriiMotov

sqlmodel/main.py

@@ -903,6 +903,7 @@ def model_dump(
         warnings: bool | Literal["none", "warn", "error"] = True,
         fallback: Callable[[Any], Any] | None = None,  # v2.11
         serialize_as_any: bool = False,  # v2.7
+        polymorphic_serialization: bool | None = None,  # v2.13
     ) -> builtins.dict[str, Any]:
         if PYDANTIC_MINOR_VERSION < (2, 11):
             by_alias = by_alias or False
@@ -913,6 +914,8 @@ def model_dump(
             extra_kwargs["fallback"] = fallback
         if PYDANTIC_MINOR_VERSION >= (2, 12):
             extra_kwargs["exclude_computed_fields"] = exclude_computed_fields
+        if PYDANTIC_MINOR_VERSION >= (2, 13):
+            extra_kwargs["polymorphic_serialization"] = polymorphic_serialization
         return super().model_dump(
             mode=mode,
             include=include,

tests/test_pydantic/test_model_dump.py

@@ -0,0 +1,59 @@
+from sqlmodel import SQLModel
+
+
+def test_polymorphic_serialization_default():
+
+    class User(SQLModel):
+        name: str
+
+    class UserLogin(User):
+        password: str
+
+    class OuterModel(SQLModel):
+        user: User
+
+    outer_model = OuterModel(
+        user=UserLogin(name="pydantic", password="password"),
+    )
+
+    assert outer_model.model_dump() == {"user": {"name": "pydantic"}}
+
+
+def test_polymorphic_serialization_false():
+
+    class User(SQLModel):
+        name: str
+
+    class UserLogin(User):
+        password: str
+
+    class OuterModel(SQLModel):
+        user: User
+
+    outer_model = OuterModel(
+        user=UserLogin(name="pydantic", password="password"),
+    )
+
+    assert outer_model.model_dump(polymorphic_serialization=False) == {
+        "user": {"name": "pydantic"}
+    }
+
+
+def test_polymorphic_serialization_true():
+
+    class User(SQLModel):
+        name: str
+
+    class UserLogin(User):
+        password: str
+
+    class OuterModel(SQLModel):
+        user: User
+
+    outer_model = OuterModel(
+        user=UserLogin(name="pydantic", password="password"),
+    )
+
+    assert outer_model.model_dump(polymorphic_serialization=True) == {
+        "user": {"name": "pydantic", "password": "password"}
+    }