diff --git a/ingestion/src/metadata/clients/aws_client.py b/ingestion/src/metadata/clients/aws_client.py index e550c41e22a..751f64da61d 100644 --- 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(