fix base models

pentest-gg · pentest-gg · commit 902611f1aa86 · 2025-09-01T21:59:29.000+02:00
diff --git a/fastapi_users_db_dynamodb/__init__.py b/fastapi_users_db_dynamodb/__init__.py
@@ -39,7 +39,7 @@
 class DynamoDBBaseUserTable(BaseModel, Generic[ID]):
     """Base user table schema for DynamoDB."""
 
-    model_config = ConfigDict(arbitrary_types_allowed=True)
+    model_config = ConfigDict(arbitrary_types_allowed=True, from_attributes=True)
 
     __tablename__ = "user"
 
diff --git a/fastapi_users_db_dynamodb/access_token.py b/fastapi_users_db_dynamodb/access_token.py
@@ -26,7 +26,7 @@
 class DynamoDBBaseAccessTokenTable(BaseModel, Generic[ID]):
     """Base access token table schema for DynamoDB."""
 
-    model_config = ConfigDict(arbitrary_types_allowed=True)
+    model_config = ConfigDict(arbitrary_types_allowed=True, from_attributes=True)
 
     __tablename__ = "accesstoken"
 
diff --git a/tests/test_users.py b/tests/test_users.py
@@ -18,15 +18,15 @@
 
 
 class Base(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True)
+    model_config = ConfigDict(arbitrary_types_allowed=True, from_attributes=True)
 
 
 class User(DynamoDBBaseUserTableUUID, Base):
     first_name: str | None = Field(default=None, description="First name of the user")
 
 
-class OAuthBase:
-    pass
+class OAuthBase(BaseModel):
+    model_config = ConfigDict(arbitrary_types_allowed=True, from_attributes=True)
 
 
 class OAuthAccount(DynamoDBBaseOAuthAccountTableUUID, OAuthBase):
