From cc01095107d1683d51731869de06384bf061eeaf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=B5=B5=E5=BF=A0=E6=9E=97?= <170083662@qq.com>
Date: Sat, 28 Feb 2026 00:38:19 +0800
Subject: [PATCH] =?UTF-8?q?refactor(ai):=20=E7=A7=BB=E9=99=A4=20OpenAI=20?=
 =?UTF-8?q?=E6=94=AF=E6=8C=81=E5=B9=B6=E4=BC=98=E5=8C=96=20Ollama=20?=
 =?UTF-8?q?=E9=9B=86=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 移除了 OpenAI 相关的 API 导入和类型定义
- 删除了 provider 切换逻辑和相关的条件分支代码
- 简化了模型列表加载逻辑，仅保留 Ollama 模型获取
- 移除了聊天消息发送中的 provider 分支判断
- 删除了历史消息 JSON 计算属性
- 移除了 API Key 输入字段和相关错误处理
- 统一使用 Ollama API 类型定义替代 OpenAI 类型
- 移除了 provider 监听器和切换功能组件
---
 src/views/ai/index.vue | 197 ++++++++++-------------------------------
 1 file changed, 45 insertions(+), 152 deletions(-)

diff --git a/src/views/ai/index.vue b/src/views/ai/index.vue
index 49c282a..372bb75 100644
--- a/src/views/ai/index.vue
+++ b/src/views/ai/index.vue
@@ -1,13 +1,6 @@
 <script setup lang="ts">
-  import { computed, onBeforeUnmount, ref, watch } from 'vue';
+  import { computed, onBeforeUnmount, ref } from 'vue';
   import { message } from 'ant-design-vue';
-  import {
-    chatCompletions,
-    chatCompletionsStream,
-    listModels,
-    type OpenAIChatMessage,
-    type OpenAIModel
-  } from '@/api/ai/openai';
   import { OLLAMA_API_URL } from '@/config/setting';
   import {
     listOllamaModels,
@@ -16,20 +9,13 @@
     type OllamaChatMessage
   } from '@/api/ai/ollama';
 
-  type Msg = OpenAIChatMessage;
+  type Msg = OllamaChatMessage;
 
-  type Provider = 'openai' | 'ollama';
-  const provider = ref<Provider>('ollama');
-
-  // Force direct calls to the hosted OpenAI-compatible gateway (no local /ai-proxy).
-  const DIRECT_OPENAI_BASE_URL = 'https://ai-api.websoft.top/api/v1';
-  const baseURL = ref<string>(
-    provider.value === 'openai' ? DIRECT_OPENAI_BASE_URL : OLLAMA_API_URL
-  );
-  const apiKey = ref<string>('');
+  // Only keep Ollama native API mode.
+  const baseURL = ref<string>(OLLAMA_API_URL);
 
   const modelLoading = ref(false);
-  const models = ref<OpenAIModel[]>([]);
+  const models = ref<Array<{ id: string; name?: string }>>([]);
   const modelId = ref<string>('');
 
   const systemPrompt = ref<string>('你是一个有帮助的助手。');
@@ -43,7 +29,6 @@
   const errorText = ref<string>('');
 
   const history = ref<Msg[]>([]);
-  const historyJson = computed(() => JSON.stringify(history.value, null, 2));
 
   const canSend = computed(() => {
     return !!modelId.value && !!userPrompt.value.trim() && !sending.value;
@@ -61,46 +46,22 @@
     modelLoading.value = true;
     errorText.value = '';
     try {
-      if (provider.value === 'openai') {
-        if (!baseURL.value.trim()) {
-          baseURL.value = DIRECT_OPENAI_BASE_URL;
-        }
-        const res = await listModels({
-          baseURL: baseURL.value.trim() || DIRECT_OPENAI_BASE_URL,
-          apiKey: apiKey.value.trim() || undefined
-        });
-        models.value = res.data ?? [];
-        if (!modelId.value && models.value.length) {
-          modelId.value = models.value[0].id;
-        }
-      } else {
-        if (!baseURL.value.trim()) {
-          baseURL.value = OLLAMA_API_URL;
-        }
-        const res = await listOllamaModels({
-          baseURL: baseURL.value.trim() || OLLAMA_API_URL
-        });
-        models.value = (res.models ?? []).map((m) => ({
-          id: m.name,
-          name: m.name,
-          object: 'model'
-        }));
-        if (!modelId.value && models.value.length) {
-          modelId.value = models.value[0].id;
-        }
+      if (!baseURL.value.trim()) {
+        baseURL.value = OLLAMA_API_URL;
+      }
+      const res = await listOllamaModels({
+        baseURL: baseURL.value.trim() || OLLAMA_API_URL
+      });
+      models.value = (res.models ?? []).map((m) => ({
+        id: m.name,
+        name: m.name
+      }));
+      if (!modelId.value && models.value.length) {
+        modelId.value = models.value[0].id;
       }
     } catch (e: any) {
       errorText.value = e?.message ?? String(e);
-      if (
-        provider.value === 'openai' &&
-        String(errorText.value).includes('401')
-      ) {
-        message.error(
-          '未认证(401)：请填写 API Key（如需走本地代理注入，可将 BaseURL 改为 /ai-proxy）'
-        );
-      } else {
-        message.error('加载模型列表失败');
-      }
+      message.error('加载模型列表失败');
     } finally {
       modelLoading.value = false;
     }
@@ -131,72 +92,34 @@
     abortController.value = controller;
 
     try {
-      if (provider.value === 'openai') {
-        if (stream.value) {
-          await chatCompletionsStream(
-            {
-              model: modelId.value,
-              messages,
-              temperature: temperature.value
-            },
-            {
-              baseURL: baseURL.value.trim() || DIRECT_OPENAI_BASE_URL,
-              apiKey: apiKey.value.trim() || undefined,
-              signal: controller.signal,
-              onDelta: (t) => {
-                assistantText.value += t;
-              }
+      if (stream.value) {
+        await ollamaChatStream(
+          {
+            model: modelId.value,
+            messages,
+            options: { temperature: temperature.value }
+          },
+          {
+            baseURL: baseURL.value.trim() || OLLAMA_API_URL,
+            signal: controller.signal,
+            onDelta: (t) => {
+              assistantText.value += t;
             }
-          );
-        } else {
-          const res = await chatCompletions(
-            {
-              model: modelId.value,
-              messages,
-              temperature: temperature.value
-            },
-            {
-              baseURL: baseURL.value.trim() || DIRECT_OPENAI_BASE_URL,
-              apiKey: apiKey.value.trim() || undefined,
-              signal: controller.signal
-            }
-          );
-          assistantText.value = res.choices?.[0]?.message?.content ?? '';
-        }
+          }
+        );
       } else {
-        const ollamaMessages: OllamaChatMessage[] = messages.map((m) => ({
-          role: m.role as any,
-          content: m.content
-        }));
-        if (stream.value) {
-          await ollamaChatStream(
-            {
-              model: modelId.value,
-              messages: ollamaMessages,
-              options: { temperature: temperature.value }
-            },
-            {
-              baseURL: baseURL.value.trim() || OLLAMA_API_URL,
-              signal: controller.signal,
-              onDelta: (t) => {
-                assistantText.value += t;
-              }
-            }
-          );
-        } else {
-          const res = await ollamaChat(
-            {
-              model: modelId.value,
-              messages: ollamaMessages,
-              options: { temperature: temperature.value }
-            },
-            {
-              baseURL: baseURL.value.trim() || OLLAMA_API_URL,
-              signal: controller.signal
-            }
-          );
-          assistantText.value = res.message?.content ?? '';
-        }
+        const res = await ollamaChat(
+          {
+            model: modelId.value,
+            messages,
+            options: { temperature: temperature.value }
+          },
+          {
+            baseURL: baseURL.value.trim() || OLLAMA_API_URL,
+            signal: controller.signal
+          }
+        );
+        assistantText.value = res.message?.content ?? '';
       }
 
       history.value = [
@@ -226,19 +149,6 @@
   // Load once for convenience; if the gateway blocks CORS you can still paste output from curl.
   loadModels();
 
-  watch(
-    () => provider.value,
-    (p) => {
-      stop();
-      clearChat();
-      models.value = [];
-      modelId.value = '';
-      errorText.value = '';
-      baseURL.value = p === 'openai' ? DIRECT_OPENAI_BASE_URL : OLLAMA_API_URL;
-      loadModels();
-    }
-  );
-
   onBeforeUnmount(() => {
     stop();
   });
@@ -256,28 +166,11 @@
         />
 
         <a-row :gutter="12">
-          <a-col :xs="24" :md="6">
-            <a-select
-              v-model:value="provider"
-              :options="[
-                { label: 'OpenAI兼容(/v1)', value: 'openai' },
-                { label: 'Ollama原生(/api)', value: 'ollama' }
-              ]"
-              style="width: 100%"
-            />
-          </a-col>
           <a-col :xs="24" :md="12">
             <a-input
               v-model:value="baseURL"
               addon-before="BaseURL"
-              placeholder="https://ai-api.websoft.top/api/v1"
-            />
-          </a-col>
-          <a-col :xs="24" :md="12" v-if="provider === 'openai'">
-            <a-input-password
-              v-model:value="apiKey"
-              addon-before="API Key"
-              placeholder="可选（不建议在前端保存）"
+              placeholder="http://localhost:11434"
             />
           </a-col>
         </a-row>