migrate pydantic config classes to ConfigDict

reflex/base.py

@@ -50,12 +50,12 @@ class Base(pydantic.BaseModel):
     frontend and backend should subclass this class.
     """
 
-    class Config:
+    # Pydantic config
-        """Pydantic config."""
+    model_config = pydantic.ConfigDict(
-
+        arbitrary_types_allowed=True,
-        arbitrary_types_allowed = True
+        use_enum_values=True,
-        use_enum_values = True
+        extra="allow",
-        extra = "allow"
+    )
 
     def json(self) -> str:
         """Convert the object to a json string.

reflex/config.py

@@ -127,10 +127,10 @@ class DBConfig(Base):
 class Config(Base):
     """A Reflex config."""
 
-    class Config:
+    # Pydantic config
-        """Pydantic config for the config."""
+    model_config = pydantic.ConfigDict(
-
+        validate_assignment=True,
-        validate_assignment = True
+    )
 
     # The name of the app.
     app_name: str

reflex/event.py

@@ -16,6 +16,8 @@ from typing import (
     Union,
 )
 
+from pydantic import ConfigDict
+
 from reflex import constants
 from reflex.base import Base
 from reflex.utils import console, format
@@ -147,11 +149,10 @@ class EventHandler(EventActionsMixin):
     # The function to call in response to the event.
     fn: Any
 
-    class Config:
+    # Pydantic config
-        """The Pydantic config."""
+    model_config = ConfigDict(
-
+        frozen=True,  # Needed to allow serialization of Callable.
-        # Needed to allow serialization of Callable.
+    )
-        frozen = True
 
     @property
     def is_background(self) -> bool:
@@ -219,11 +220,10 @@ class EventSpec(EventActionsMixin):
     # TODO: pydantic v2 add rx.Var type annotation?
     args: Tuple[Tuple[Any, Any], ...] = ()
 
-    class Config:
+    # Pydantic config
-        """The Pydantic config."""
+    model_config = ConfigDict(
-
+        frozen=True,  # Required to allow tuple fields.
-        # Required to allow tuple fields.
+    )
-        frozen = True
 
     def with_args(self, args: Tuple[Tuple[Var, Var], ...]) -> EventSpec:
         """Copy the event spec, with updated args.

reflex/model.py

@@ -62,8 +62,7 @@ class Model(Base, sqlmodel.SQLModel):
         non_default_primary_key_fields = [
             field_name
             for field_name, field in cls.model_fields.items()
-            if field_name != "id"
+            if field_name != "id" and getattr(field, "primary_key", None) is True
-            and getattr(field, "primary_key", None) is True
         ]
         if non_default_primary_key_fields:
             cls.model_fields.pop("id", None)

reflex/state.py

@@ -2104,17 +2104,19 @@ class StateManagerMemory(StateManager):
     # The dict of mutexes for each client
     _states_locks: Dict[str, asyncio.Lock] = pydantic.PrivateAttr({})
 
-    class Config:
+    # Pydantic config
-        """The Pydantic config."""
+    model_config = pydantic.ConfigDict(
-
+        arbitrary_types_allowed=True,
-        # TODO: pydantic v2
+        use_enum_values=True,
-        fields = {
+        extra="allow",
-            "_states_locks": {"exclude": True},
-        }
-
         #  json_encoders = {
-        #      MutableProxy: lambda v: v.__wrapped__,
+        #      MutableProxy: lambda v: v.__wrapped__, # this is currently done in _get_value
         #  }
+        # TODO: pydantic v2
+        #  fields = {
+        #      "_states_locks": {"exclude": True},
+        #  }
+    )
 
     async def get_state(self, token: str) -> BaseState:
         """Get the state for a token.