From 441b218245255c0b8653eef0f8e88dfeda5d4dbf Mon Sep 17 00:00:00 2001
From: sagitchu <601096721@qq.com>
Date: Fri, 30 Aug 2024 15:15:49 +0800
Subject: [PATCH] add test

---
 ui/components/ChatWindow.tsx | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/ui/components/ChatWindow.tsx b/ui/components/ChatWindow.tsx
index b3d0089..3d00cb2 100644
--- a/ui/components/ChatWindow.tsx
+++ b/ui/components/ChatWindow.tsx
@@ -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());