From 3fb17dcd99e579e0ecd1cfac2278344cd06fc848 Mon Sep 17 00:00:00 2001
From: fofolee <fofolee@qq.com>
Date: Wed, 23 Apr 2025 21:28:20 +0800
Subject: [PATCH] =?UTF-8?q?=E7=AE=80=E5=8C=96ai=E8=AE=BE=E7=BD=AE=E4=B8=AD?=
 =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E9=85=8D=E7=BD=AE=E7=9A=84=E6=93=8D=E4=BD=9C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/ai/AIConfig.vue | 47 +++++++++++++++-------------------
 1 file changed, 21 insertions(+), 26 deletions(-)

diff --git a/src/components/ai/AIConfig.vue b/src/components/ai/AIConfig.vue
index 2ee78c5..bdc0f04 100644
--- a/src/components/ai/AIConfig.vue
+++ b/src/components/ai/AIConfig.vue
@@ -2,19 +2,16 @@
   <q-card style="width: 800px" class="q-pa-sm">
     <div class="text-h5 q-my-md q-px-sm">API配置</div>
     <div>
-      <div class="flex q-mb-md q-px-sm" style="height: 26px">
-        <ButtonGroup
-          v-model="apiToAdd"
-          class="col"
-          :options="aiOptions"
-          height="26px"
-        />
-        <q-icon
-          name="add_box"
-          @click="addModel"
+      <div class="q-pa-sm row q-gutter-sm">
+        <q-btn
+          v-for="option in aiOptions"
+          :key="option.value"
+          icon="add_link"
+          dense
           color="primary"
-          size="26px"
-          class="cursor-pointer q-ml-sm"
+          :label="option.label"
+          @click="addModel(option.value)"
+          class="col"
         />
       </div>
       <q-scroll-area
@@ -76,11 +73,9 @@
                     :placeholder="
                       aiConfig.apiType === 'openai'
                         ? '例：https://api.openai.com'
-                        : aiConfig.apiType === 'ollama'
-                        ? '例：http://localhost:11434'
-                        : '无需配置'
+                        : '例：http://localhost:11434'
                     "
-                    :disable="aiConfig.apiType === 'utools'"
+                    v-show="aiConfig.apiType !== 'utools'"
                   >
                     <template v-slot:prepend>
                       <q-badge
@@ -182,31 +177,31 @@
 <script>
 import { defineComponent, ref } from "vue";
 import { dbManager } from "js/utools.js";
-import ButtonGroup from "components/composer/common/ButtonGroup.vue";
 import draggable from "vuedraggable";
 import { getUniqueId } from "js/common/uuid.js";
 
 export default defineComponent({
   name: "AIConfig",
   components: {
-    ButtonGroup,
     draggable,
   },
   setup() {
+    const initAiOptions = utools.allAiModels
+      ? [{ label: "uTools内置AI", value: "utools" }]
+      : [];
+
     const aiOptions = ref([
-      { label: "OPENAI接口", value: "openai" },
+      ...initAiOptions,
+      { label: "OPENAI接口(需Key)", value: "openai" },
       { label: "OLLAMA接口", value: "ollama" },
     ]);
-    if (utools.allAiModels) {
-      aiOptions.value.push({ label: "uTools内置", value: "utools" });
-    }
+
     return {
       aiOptions,
     };
   },
   data() {
     return {
-      apiToAdd: "openai",
       aiConfigs: [],
       models: [],
       tokenInputTypes: [],
@@ -248,17 +243,17 @@ export default defineComponent({
     deleteModel(index) {
       this.aiConfigs.splice(index, 1);
     },
-    addModel() {
+    addModel(apiType) {
       const defaultConfig = {
         id: getUniqueId(),
-        apiType: this.apiToAdd,
+        apiType: apiType,
         apiUrl: "",
         apiToken: "",
         model: "",
         name: "",
       };
 
-      if (this.apiToAdd === "utools") {
+      if (apiType === "utools") {
         defaultConfig.apiUrl = "";
       }