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 (
|
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());
|
||||||
|
|
Loading…
Add table
Reference in a new issue