4
4
OPENAI_API_KEY_HEADER_KEY ,
5
5
OPENAI_END_POINT ,
6
6
} from '@/const/fetch' ;
7
+ import { isDeprecatedEdition } from '@/const/version' ;
8
+ import { aiProviderSelectors , useAiInfraStore } from '@/store/aiInfra' ;
7
9
import { useUserStore } from '@/store/user' ;
8
10
import { keyVaultsConfigSelectors } from '@/store/user/selectors' ;
9
11
@@ -14,14 +16,23 @@ import { keyVaultsConfigSelectors } from '@/store/user/selectors';
14
16
// eslint-disable-next-line no-undef
15
17
export const createHeaderWithOpenAI = ( header ?: HeadersInit ) : HeadersInit => {
16
18
const state = useUserStore . getState ( ) ;
17
- const openAIConfig = keyVaultsConfigSelectors . openAIConfig ( state ) ;
18
19
20
+ let keyVaults : Record < string , any > = { } ;
21
+
22
+ // TODO: remove this condition in V2.0
23
+ if ( isDeprecatedEdition ) {
24
+ keyVaults = keyVaultsConfigSelectors . getVaultByProvider ( 'openai' as any ) (
25
+ useUserStore . getState ( ) ,
26
+ ) ;
27
+ } else {
28
+ keyVaults = aiProviderSelectors . providerKeyVaults ( 'openai' ) ( useAiInfraStore . getState ( ) ) || { } ;
29
+ }
19
30
// eslint-disable-next-line no-undef
20
31
return {
21
32
...header ,
22
33
[ LOBE_CHAT_ACCESS_CODE ] : keyVaultsConfigSelectors . password ( state ) ,
23
34
[ LOBE_USER_ID ] : state . user ?. id || '' ,
24
- [ OPENAI_API_KEY_HEADER_KEY ] : openAIConfig . apiKey || '' ,
25
- [ OPENAI_END_POINT ] : openAIConfig . baseURL || '' ,
35
+ [ OPENAI_API_KEY_HEADER_KEY ] : keyVaults . apiKey || '' ,
36
+ [ OPENAI_END_POINT ] : keyVaults . baseURL || '' ,
26
37
} ;
27
38
} ;
0 commit comments