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