From a762b87d2b85e09c672c3f8a8bac267e22afb1f5 Mon Sep 17 00:00:00 2001
From: fofolee <fofolee@qq.com>
Date: Wed, 23 Apr 2025 19:21:31 +0800
Subject: [PATCH] =?UTF-8?q?ai=E8=AE=BE=E7=BD=AE=E6=94=AF=E6=8C=81=E6=B7=BB?=
 =?UTF-8?q?=E5=8A=A0utools=E5=86=85=E7=BD=AE=E7=9A=84ai?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

diff --git a/src/components/ai/AIConfig.vue b/src/components/ai/AIConfig.vue
index ee2cee3..c97ad99 100644
--- a/src/components/ai/AIConfig.vue
+++ b/src/components/ai/AIConfig.vue
@@ -6,10 +6,7 @@
         <ButtonGroup
           v-model="apiToAdd"
           class="col"
-          :options="[
-            { label: 'OPENAI', value: 'openai' },
-            { label: 'OLLAMA', value: 'ollama' },
-          ]"
+          :options="aiOptions"
           height="26px"
         />
         <q-icon
@@ -78,8 +75,11 @@
                     :placeholder="
                       aiConfig.apiType === 'openai'
                         ? '例：https://api.openai.com'
-                        : '例：http://localhost:11434'
+                        : aiConfig.apiType === 'ollama'
+                        ? '例：http://localhost:11434'
+                        : '无需配置'
                     "
+                    :disable="aiConfig.apiType === 'utools'"
                   >
                     <template v-slot:prepend>
                       <q-badge
@@ -179,7 +179,7 @@
 </template>
 
 <script>
-import { defineComponent } from "vue";
+import { defineComponent, ref } from "vue";
 import { dbManager } from "js/utools.js";
 import ButtonGroup from "components/composer/common/ButtonGroup.vue";
 import draggable from "vuedraggable";
@@ -191,6 +191,18 @@ export default defineComponent({
     ButtonGroup,
     draggable,
   },
+  setup() {
+    const aiOptions = ref([
+      { label: "OPENAI接口", value: "openai" },
+      { label: "OLLAMA接口", value: "ollama" },
+    ]);
+    if (utools.allAiModels) {
+      aiOptions.value.push({ label: "uTools内置", value: "utools" });
+    }
+    return {
+      aiOptions,
+    };
+  },
   data() {
     return {
       apiToAdd: "openai",
@@ -202,6 +214,19 @@ export default defineComponent({
   emits: ["save"],
   methods: {
     async getModels(aiConfig) {
+      if (aiConfig.apiType === "utools") {
+        try {
+          const models = await utools.allAiModels();
+          this.models = models.map((model) => model.id);
+        } catch (error) {
+          quickcommand.showMessageBox(
+            "获取 uTools AI 模型失败: " + error.message,
+            "error"
+          );
+          this.models = [];
+        }
+        return;
+      }
       const { success, result, error } = await window.getModelsFromAiApi(
         aiConfig
       );
@@ -223,13 +248,21 @@ export default defineComponent({
       this.aiConfigs.splice(index, 1);
     },
     addModel() {
-      this.aiConfigs.push({
+      const defaultConfig = {
         id: getUniqueId(),
         apiType: this.apiToAdd,
         apiUrl: "",
         apiToken: "",
         model: "",
         name: "",
+      };
+
+      if (this.apiToAdd === "utools") {
+        defaultConfig.apiUrl = "";
+      }
+
+      this.aiConfigs.push(defaultConfig);
+
       });
     },
     getConfigListHeight() {