MINOR: fix model dump dict key names (#19484) (#19485)

Co-authored-by: Abdallah Serghine <[email protected]>
open-metadata · Jan 23, 2025 · a631d22 · a631d22
1 parent 06759c5
commit a631d22
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/ingestion/src/metadata/clients/aws_client.py b/ingestion/src/metadata/clients/aws_client.py
@@ -58,12 +58,13 @@ class AWSAssumeRoleCredentialResponse(BaseModel):
 
 
 class AWSAssumeRoleCredentialWrapper(BaseModel):
-    accessKeyId: str = Field()
-    secretAccessKey: CustomSecretStr = Field()
-    sessionToken: Optional[str] = Field(
-        default=None,
-    )
-    expiryTime: Optional[str] = Field()
+    accessKeyId: str = Field(alias="access_key")
+    secretAccessKey: CustomSecretStr = Field(alias="secret_key")
+    sessionToken: Optional[str] = Field(default=None, alias="token")
+    expiryTime: Optional[str] = Field(alias="expiry_time")
+
+    class Config:
+        populate_by_name = True
 
 
 AWSAssumeRoleCredentialFormat = TypeVar(