From 3fef83e3c68750a0b005c45a84d7714e6879b968 Mon Sep 17 00:00:00 2001
From: Arvin Xu <arvinx@foxmail.com>
Date: Fri, 24 Jan 2025 00:32:21 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix:=20fix=20tts=20in=20new=20pr?=
 =?UTF-8?q?ovider=20model=20(#5569)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/services/_header.ts | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/services/_header.ts b/src/services/_header.ts
index e4c2e6d637400..d1525ec7baf16 100644
--- a/src/services/_header.ts
+++ b/src/services/_header.ts
@@ -4,6 +4,8 @@ import {
   OPENAI_API_KEY_HEADER_KEY,
   OPENAI_END_POINT,
 } from '@/const/fetch';
+import { isDeprecatedEdition } from '@/const/version';
+import { aiProviderSelectors, useAiInfraStore } from '@/store/aiInfra';
 import { useUserStore } from '@/store/user';
 import { keyVaultsConfigSelectors } from '@/store/user/selectors';
 
@@ -14,14 +16,23 @@ import { keyVaultsConfigSelectors } from '@/store/user/selectors';
 // eslint-disable-next-line no-undef
 export const createHeaderWithOpenAI = (header?: HeadersInit): HeadersInit => {
   const state = useUserStore.getState();
-  const openAIConfig = keyVaultsConfigSelectors.openAIConfig(state);
 
+  let keyVaults: Record<string, any> = {};
+
+  // TODO: remove this condition in V2.0
+  if (isDeprecatedEdition) {
+    keyVaults = keyVaultsConfigSelectors.getVaultByProvider('openai' as any)(
+      useUserStore.getState(),
+    );
+  } else {
+    keyVaults = aiProviderSelectors.providerKeyVaults('openai')(useAiInfraStore.getState()) || {};
+  }
   // eslint-disable-next-line no-undef
   return {
     ...header,
     [LOBE_CHAT_ACCESS_CODE]: keyVaultsConfigSelectors.password(state),
     [LOBE_USER_ID]: state.user?.id || '',
-    [OPENAI_API_KEY_HEADER_KEY]: openAIConfig.apiKey || '',
-    [OPENAI_END_POINT]: openAIConfig.baseURL || '',
+    [OPENAI_API_KEY_HEADER_KEY]: keyVaults.apiKey || '',
+    [OPENAI_END_POINT]: keyVaults.baseURL || '',
   };
 };