add test
This commit is contained in:
parent
f51541195e
commit
441b218245
1 changed files with 12 additions and 0 deletions
|
@ -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());
|
||||
|
|
Loading…
Add table
Reference in a new issue