From e5e8807312cc9380631dc180b5f4c47c219c8ecf Mon Sep 17 00:00:00 2001
From: Junyan Qin <rockchinq@gmail.com>
Date: Tue, 20 May 2025 16:01:20 +0800
Subject: [PATCH] perf: no longer ask for apikeys for ollama and lm studio

---
 .../models/component/llm-form/LLMForm.tsx     | 43 +++++++++++--------
 1 file changed, 26 insertions(+), 17 deletions(-)

diff --git a/web/src/app/home/models/component/llm-form/LLMForm.tsx b/web/src/app/home/models/component/llm-form/LLMForm.tsx
index 0e86dd51..9bdf67f3 100644
--- a/web/src/app/home/models/component/llm-form/LLMForm.tsx
+++ b/web/src/app/home/models/component/llm-form/LLMForm.tsx
@@ -103,7 +103,7 @@ export default function LLMForm({
       name: '',
       model_provider: '',
       url: '',
-      api_key: '',
+      api_key: 'sk-xxxxx',
       abilities: [],
       extra_args: [],
     },
@@ -131,6 +131,7 @@ export default function LLMForm({
     string[]
   >([]);
   const [modelTesting, setModelTesting] = useState(false);
+  const [currentModelProvider, setCurrentModelProvider] = useState('');
 
   useEffect(() => {
     initLLMModelFormComponent().then(() => {
@@ -138,6 +139,7 @@ export default function LLMForm({
         getLLMConfig(initLLMId).then((val) => {
           form.setValue('name', val.name);
           form.setValue('model_provider', val.model_provider);
+          setCurrentModelProvider(val.model_provider);
           form.setValue('url', val.url);
           form.setValue('api_key', val.api_key);
           form.setValue(
@@ -409,6 +411,7 @@ export default function LLMForm({
                     <Select
                       onValueChange={(value) => {
                         field.onChange(value);
+                        setCurrentModelProvider(value);
                         const index = requesterNameList.findIndex(
                           (item) => item.value === value,
                         );
@@ -455,22 +458,28 @@ export default function LLMForm({
                 </FormItem>
               )}
             />
-            <FormField
-              control={form.control}
-              name="api_key"
-              render={({ field }) => (
-                <FormItem>
-                  <FormLabel>
-                    {t('models.apiKey')}
-                    <span className="text-red-500">*</span>
-                  </FormLabel>
-                  <FormControl>
-                    <Input {...field} />
-                  </FormControl>
-                  <FormMessage />
-                </FormItem>
-              )}
-            />
+
+            {!['lmstudio-chat-completions', 'ollama-chat'].includes(
+              currentModelProvider,
+            ) && (
+              <FormField
+                control={form.control}
+                name="api_key"
+                render={({ field }) => (
+                  <FormItem>
+                    <FormLabel>
+                      {t('models.apiKey')}
+                      <span className="text-red-500">*</span>
+                    </FormLabel>
+                    <FormControl>
+                      <Input {...field} />
+                    </FormControl>
+                    <FormMessage />
+                  </FormItem>
+                )}
+              />
+            )}
+
             <FormField
               control={form.control}
               name="abilities"