This commit is contained in:
sagitchu 2024-08-30 15:15:49 +08:00
parent f51541195e
commit 441b218245

View file

@ -130,6 +130,7 @@ const useSocket = (
if ( if (
embeddingModelProvider && embeddingModelProvider &&
embeddingModelProvider != 'custom_openai' &&
!embeddingModelProviders[embeddingModelProvider][embeddingModel] !embeddingModelProviders[embeddingModelProvider][embeddingModel]
) { ) {
embeddingModel = Object.keys( embeddingModel = Object.keys(
@ -159,6 +160,17 @@ const useSocket = (
searchParams.append('embeddingModel', embeddingModel!); searchParams.append('embeddingModel', embeddingModel!);
searchParams.append('embeddingModelProvider', embeddingModelProvider); searchParams.append('embeddingModelProvider', embeddingModelProvider);
if (embeddingModelProvider === 'custom_openai') {
searchParams.append(
'openAIApiKey',
localStorage.getItem('openAIApiKey')!,
);
searchParams.append(
'openAIBaseURL',
localStorage.getItem('openAIBaseURL')!,
);
}
wsURL.search = searchParams.toString(); wsURL.search = searchParams.toString();
const ws = new WebSocket(wsURL.toString()); const ws = new WebSocket(wsURL.toString());