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: {