From b3a023e4ac19c936e009f6f6e1048c9cec35bf3a Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Mon, 15 Jul 2024 17:58:58 +0800
Subject: [PATCH] fix: model checking error

---
 .../src/pages/settings/components/ProviderSetting.tsx    | 3 ++-
 src/renderer/src/services/api.ts                         | 9 +++++----
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/renderer/src/pages/settings/components/ProviderSetting.tsx b/src/renderer/src/pages/settings/components/ProviderSetting.tsx
index 7af7aff8..fd5057b5 100644
--- a/src/renderer/src/pages/settings/components/ProviderSetting.tsx
+++ b/src/renderer/src/pages/settings/components/ProviderSetting.tsx
@@ -18,7 +18,8 @@ interface Props {
   provider: Provider
 }
 
-const ProviderSetting: FC<Props> = ({ provider }) => {
+const ProviderSetting: FC<Props> = ({ provider: _provider }) => {
+  const { provider } = useProvider(_provider.id)
   const [apiKey, setApiKey] = useState(provider.apiKey)
   const [apiHost, setApiHost] = useState(provider.apiHost)
   const [apiValid, setApiValid] = useState(false)
diff --git a/src/renderer/src/services/api.ts b/src/renderer/src/services/api.ts
index 47aa57b6..6c0b8538 100644
--- a/src/renderer/src/services/api.ts
+++ b/src/renderer/src/services/api.ts
@@ -127,17 +127,17 @@ export async function checkApi(provider: Provider) {
   const style = { marginTop: '3vh' }
 
   if (!provider.apiKey) {
-    window.message.error({ content: t('error.enter.api.key'), key, style })
+    window.message.error({ content: t('message.error.enter.api.key'), key, style })
     return false
   }
 
   if (!provider.apiHost) {
-    window.message.error({ content: t('error.enter.api.host'), key, style })
+    window.message.error({ content: t('message.error.enter.api.host'), key, style })
     return false
   }
 
   if (!model) {
-    window.message.error({ content: t('error.enter.model'), key, style })
+    window.message.error({ content: t('message.error.enter.model'), key, style })
     return false
   }
 
@@ -147,7 +147,8 @@ export async function checkApi(provider: Provider) {
   try {
     const response = await openaiProvider.chat.completions.create({
       model: model.id,
-      messages: [{ role: 'user', content: 'hello' }],
+      messages: [{ role: 'user', content: 'hi' }],
+      max_tokens: 100,
       stream: false
     })