From 9d967cbaf407a243a00540f6226ad083f571fc85 Mon Sep 17 00:00:00 2001
From: Matt Aitken <matt@mattaitken.com>
Date: Fri, 30 May 2025 09:19:58 +0100
Subject: [PATCH] Set the branch header for the SDK realtime functions

---
 packages/core/src/v3/apiClient/index.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/packages/core/src/v3/apiClient/index.ts b/packages/core/src/v3/apiClient/index.ts
index 117792466c..12147df109 100644
--- a/packages/core/src/v3/apiClient/index.ts
+++ b/packages/core/src/v3/apiClient/index.ts
@@ -1016,11 +1016,15 @@ export class ApiClient {
   }
 
   #getRealtimeHeaders() {
-    const headers: Record<string, string> = {
+    let headers: Record<string, string> = {
       Authorization: `Bearer ${this.accessToken}`,
       "trigger-version": VERSION,
     };
 
+    if (this.previewBranch) {
+      headers["x-trigger-branch"] = this.previewBranch;
+    }
+
     return headers;
   }
 }