From 1a3680acd653f352e5f34de28132891b90ad23c0 Mon Sep 17 00:00:00 2001
From: Stefan Nica <stefan@zenml.io>
Date: Tue, 24 Sep 2024 10:43:35 +0200
Subject: [PATCH] Add the audience back in

---
 src/zenml/zen_server/cloud_utils.py | 2 ++
 1 file changed, 2 insertions(+)

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: