diff --git a/apps/viewer/src/features/chat/api/continueChat.ts b/apps/viewer/src/features/chat/api/continueChat.ts index 03b666dd5..438073514 100644 --- a/apps/viewer/src/features/chat/api/continueChat.ts +++ b/apps/viewer/src/features/chat/api/continueChat.ts @@ -55,7 +55,11 @@ export const continueChat = publicProcedure ) { if (origin && session.state.allowedOrigins.includes(origin)) res.setHeader('Access-Control-Allow-Origin', origin) - else res.removeHeader('Access-Control-Allow-Origin') + else + res.setHeader( + 'Access-Control-Allow-Origin', + session.state.allowedOrigins[0] + ) } const { diff --git a/apps/viewer/src/features/chat/api/startChat.ts b/apps/viewer/src/features/chat/api/startChat.ts index b2445d13c..084dafbe6 100644 --- a/apps/viewer/src/features/chat/api/startChat.ts +++ b/apps/viewer/src/features/chat/api/startChat.ts @@ -59,7 +59,11 @@ export const startChat = publicProcedure ) { if (origin && newSessionState.allowedOrigins.includes(origin)) res.setHeader('Access-Control-Allow-Origin', origin) - else res.removeHeader('Access-Control-Allow-Origin') + else + res.setHeader( + 'Access-Control-Allow-Origin', + newSessionState.allowedOrigins[0] + ) } const session = isOnlyRegistering