From c9ee6bc9c29b0d4e39bedce33d7c1192a1845f9c Mon Sep 17 00:00:00 2001
From: Alaister Young <alaister@users.noreply.github.com>
Date: Wed, 21 May 2025 22:33:47 +0800
Subject: [PATCH] fix(studio): self-hosted assistant authorization header
 (#35804)

---
 .../components/interfaces/SQLEditor/SQLEditor.tsx   |  4 +++-
 .../components/ui/AIAssistantPanel/AIAssistant.tsx  | 13 ++++++++++---
 apps/studio/components/ui/AIEditor/index.tsx        |  4 +++-
 3 files changed, 16 insertions(+), 5 deletions(-)

diff --git a/apps/studio/components/interfaces/SQLEditor/SQLEditor.tsx b/apps/studio/components/interfaces/SQLEditor/SQLEditor.tsx
index df52ac1f564..bbb13eef394 100644
--- a/apps/studio/components/interfaces/SQLEditor/SQLEditor.tsx
+++ b/apps/studio/components/interfaces/SQLEditor/SQLEditor.tsx
@@ -500,8 +500,10 @@ export const SQLEditor = () => {
       }))
       const headerData = await constructHeaders()
 
+      const authorizationHeader = headerData.get('Authorization')
+
       await complete(prompt, {
-        headers: { Authorization: headerData.get('Authorization') ?? '' },
+        ...(authorizationHeader ? { headers: { Authorization: authorizationHeader } } : undefined),
         body: {
           completionMetadata: {
             textBeforeCursor: context.beforeSelection,
diff --git a/apps/studio/components/ui/AIAssistantPanel/AIAssistant.tsx b/apps/studio/components/ui/AIAssistantPanel/AIAssistant.tsx
index dedace208a2..12baaabfcd5 100644
--- a/apps/studio/components/ui/AIAssistantPanel/AIAssistant.tsx
+++ b/apps/studio/components/ui/AIAssistantPanel/AIAssistant.tsx
@@ -221,9 +221,16 @@ export const AIAssistant = ({ className }: AIAssistantProps) => {
     // Store the user message in the ref before appending
     lastUserMessageRef.current = payload
 
-    append(payload, {
-      headers: { Authorization: headerData.get('Authorization') ?? '' },
-    })
+    const authorizationHeader = headerData.get('Authorization')
+
+    append(
+      payload,
+      authorizationHeader
+        ? {
+            headers: { Authorization: authorizationHeader },
+          }
+        : undefined
+    )
 
     setValue('')
 
diff --git a/apps/studio/components/ui/AIEditor/index.tsx b/apps/studio/components/ui/AIEditor/index.tsx
index 4e09b1aff4a..6dc2a84ab3a 100644
--- a/apps/studio/components/ui/AIEditor/index.tsx
+++ b/apps/studio/components/ui/AIEditor/index.tsx
@@ -195,8 +195,10 @@ const AIEditor = ({
       }))
 
       const headerData = await constructHeaders()
+      const authorizationHeader = headerData.get('Authorization')
+
       await complete(prompt, {
-        headers: { Authorization: headerData.get('Authorization') ?? '' },
+        ...(authorizationHeader ? { headers: { Authorization: authorizationHeader } } : undefined),
         body: {
           ...aiMetadata,
           completionMetadata: {