diff --git a/src/zenml/zen_server/cloud_utils.py b/src/zenml/zen_server/cloud_utils.py index c8adb0e4f8..e21f84cf12 100644 --- a/src/zenml/zen_server/cloud_utils.py +++ b/src/zenml/zen_server/cloud_utils.py @@ -22,6 +22,7 @@ class ZenMLCloudConfiguration(BaseModel): api_url: str oauth2_client_id: str oauth2_client_secret: str + oauth2_audience: str @field_validator("api_url") @classmethod @@ -213,6 +214,7 @@ def _fetch_auth_token(self) -> str: payload = { "client_id": self._config.oauth2_client_id, "client_secret": self._config.oauth2_client_secret, + "audience": self._config.oauth2_audience, "grant_type": "client_credentials", } try: