From 5a2e90cbcd1980b54b25edc161b047b9ce0149bc Mon Sep 17 00:00:00 2001
From: xuanmiss <xuanmine@gmail.com>
Date: Sat, 13 Jul 2024 01:00:31 +0800
Subject: [PATCH] Fix: custom_llm_provider needs to be passed when using
 self-built openai model

---
 app/src/renderer/src/pages/Setting/index.tsx | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/app/src/renderer/src/pages/Setting/index.tsx b/app/src/renderer/src/pages/Setting/index.tsx
index dafd375..b778458 100644
--- a/app/src/renderer/src/pages/Setting/index.tsx
+++ b/app/src/renderer/src/pages/Setting/index.tsx
@@ -84,6 +84,10 @@ export function Setting(){
        <Button
               onClick={async () => {
                 const hide = message.loading('检测中...', 0);
+                const llmConfig = {
+                  ...form.getFieldValue("llm"),
+                  custom_llm_provider: form.getFieldValue("format")
+                };
                 const res = await fetch(`${localServerBaseUrl}/llm`,
                   {
                     method: 'POST',
@@ -92,7 +96,7 @@ export function Setting(){
                     },
                     body: JSON.stringify(
                       { "messages": [{ "role": "user", "content": "hello" }],
-                        "llm_config": JSON.stringify(form.getFieldValue("llm"))}),
+                        "llm_config": JSON.stringify(llmConfig)}),
                   }
                 )
                 hide();