Add the audience back in

zenml-io · Sep 24, 2024 · 1a3680a · 1a3680a
1 parent 4dd0242
commit 1a3680a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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: