@liujitcn/kratos-admin-system 0.0.62 → 0.0.65

package/dist/package/src/views/ai/chat/index.vue

@@ -27,7 +27,11 @@
       :sending="currentSessionSending"
       :shortcuts="starterShortcuts"
       :loading-shortcuts="loadingShortcuts"
+      :model-providers="modelProviders"
+      :provider-id="selectedProviderId"
+      :model-name="selectedModelName"
       @submit="handleSubmit"
+      @model-change="handleModelChange"
       @message-action="handleMessageAction"
       @message-edit="handleEditMessage"
     >
@@ -54,8 +58,10 @@ import { getAdminAiExtension } from "@liujitcn/kratos-admin-system";
 import { defAiMessageService } from "@liujitcn/kratos-admin-system/api/base/v1/ai_message";
 import { defAiSessionService } from "@liujitcn/kratos-admin-system/api/base/v1/ai_session";
 import { defAiToolService } from "@liujitcn/kratos-admin-system/api/base/v1/ai_tool";
+import { defAiModelService } from "@liujitcn/kratos-admin-system/api/base/v1/ai_provider";
 import type { AiAction } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_message";
 import type { AiSession } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_session";
+import type { AiProviderModelOption } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_provider";
 import type { AiShortcut } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_tool";
 import { AiMessageStatus } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_session";
 import { Terminal } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_tool";
@@ -79,6 +85,7 @@ import {
   sortMessages
 } from "./message";
 import { readAiEventStream } from "./stream";
+import { loadStoredModelSelection, saveModelSelection } from "./modelSelection";
 import type {
   AiStreamEvent,
   AiStreamPayload,
@@ -102,6 +109,9 @@ const loadingSessionID = ref("");
 const loadingShortcuts = ref(false);
 const sessions = ref<AiSession[]>([]);
 const starterShortcuts = ref<AiShortcut[]>([]);
+const modelProviders = ref<AiProviderModelOption[]>([]);
+const selectedProviderId = ref(0);
+const selectedModelName = ref("");
 const messages = ref<Record<string, ChatMessageItem[]>>({});
 const pendingDeltaMap = new Map<string, AiStreamPayload>();
 const runningStreamTaskMap = new Map<string, AiStreamTask>();
@@ -217,6 +227,8 @@ async function runAiStreamTask(sessionID: string, payload: SubmitPayload) {
       {
         session_id: sessionID,
         content: payload.text,
+        provider_id: selectedProviderId.value,
+        model_name: selectedModelName.value,
         attachments: payload.attachments.map(item => ({
           id: item.id,
           name: item.name,
@@ -645,15 +657,50 @@ async function loadAiShortcuts() {
   if (loadingShortcuts.value) return;
   loadingShortcuts.value = true;
   try {
-    const response = await defAiToolService.ListAiShortcut({ terminal: Terminal.TERMINAL_ADMIN });
+    const [response, providerResponse] = await Promise.all([
+      defAiToolService.ListAiShortcut({ terminal: Terminal.TERMINAL_ADMIN }),
+      defAiModelService.ListAiProviderModelOptions({})
+    ]);
     starterShortcuts.value = normalizeStarterShortcuts(response.shortcuts);
+    modelProviders.value = providerResponse.providers ?? [];
+    applyModelSelection();
   } catch {
     starterShortcuts.value = [];
+    modelProviders.value = [];
+    selectedProviderId.value = 0;
+    selectedModelName.value = "";
   } finally {
     loadingShortcuts.value = false;
   }
 }
 
+/** 优先恢复本地缓存的模型选择，缓存失效时回退到首个可用模型。 */
+function applyModelSelection() {
+  const stored = loadStoredModelSelection();
+  const provider = stored
+    ? modelProviders.value.find(
+        item =>
+          item.provider_id === stored.provider_id &&
+          item.models.some(model => model.model_name === stored.model_name)
+      )
+    : undefined;
+  if (provider && stored) {
+    selectedProviderId.value = provider.provider_id;
+    selectedModelName.value = stored.model_name;
+    return;
+  }
+  const firstProvider = modelProviders.value[0];
+  selectedProviderId.value = firstProvider?.provider_id ?? 0;
+  selectedModelName.value = firstProvider?.models[0]?.model_name ?? "";
+}
+
+/** 更新当前聊天使用的供应商与模型并持久化到本地缓存。 */
+function handleModelChange(providerId: number, modelName: string) {
+  selectedProviderId.value = providerId;
+  selectedModelName.value = modelName;
+  saveModelSelection({ provider_id: providerId, model_name: modelName });
+}
+
 /** 使指定会话仍在途的消息列表请求失效。 */
 function invalidateLoadMessages(sessionID: string) {
   loadMessagesSeq.set(sessionID, (loadMessagesSeq.get(sessionID) ?? 0) + 1);

package/dist/package/src/views/ai/chat/modelDisplay.ts

@@ -0,0 +1,67 @@
+import type { AiProviderModelOption } from "@liujitcn/kratos-admin-system/rpc/base/v1/ai_provider";
+
+/** 常见模型的展示名映射；未命中时按通用规则美化原始名。 */
+const MODEL_DISPLAY_NAMES: Record<string, string> = {
+  "deepseek-chat": "DeepSeek Chat",
+  "deepseek-reasoner": "DeepSeek Reasoner",
+  "deepseek-v3": "DeepSeek V3",
+  "deepseek-r1": "DeepSeek R1",
+  "gpt-4o": "GPT-4o",
+  "gpt-4o-mini": "GPT-4o mini",
+  "gpt-4.1": "GPT-4.1",
+  "gpt-4.1-mini": "GPT-4.1 mini",
+  "gpt-4-turbo": "GPT-4 Turbo",
+  "o3-mini": "O3-mini",
+  "claude-3-5-sonnet": "Claude 3.5 Sonnet",
+  "claude-3-7-sonnet": "Claude 3.7 Sonnet",
+  "qwen-max": "Qwen Max",
+  "qwen-plus": "Qwen Plus"
+};
+
+/** 需要整体大写的模型名片段。 */
+const UPPERCASE_TOKENS = new Set([
+  "gpt",
+  "llama",
+  "qwen",
+  "glm",
+  "kimi",
+  "gemini",
+  "claude",
+  "doubao",
+  "deepseek",
+  "moonshot",
+  "hunyuan"
+]);
+
+/** 对未配置展示名的模型名做通用美化。 */
+export function fallbackModelDisplayName(modelName: string): string {
+  if (!modelName) return "";
+  const mapped = MODEL_DISPLAY_NAMES[modelName.toLowerCase()];
+  if (mapped) return mapped;
+  return modelName
+    .split(/[-_]/)
+    .filter(Boolean)
+    .map(token => {
+      const lower = token.toLowerCase();
+      if (UPPERCASE_TOKENS.has(lower)) return lower.toUpperCase();
+      return lower.charAt(0).toUpperCase() + lower.slice(1);
+    })
+    .join(" ");
+}
+
+/** 从供应商模型选项中解析模型展示名称，兼容历史消息中「供应商 / 模型」格式的原始值。 */
+export function resolveModelDisplayName(
+  providers: AiProviderModelOption[] | undefined,
+  modelName: string
+): string {
+  if (!modelName) return "";
+  const separator = " / ";
+  const modelPart = modelName.includes(separator)
+    ? modelName.slice(modelName.lastIndexOf(separator) + separator.length)
+    : modelName;
+  for (const provider of providers ?? []) {
+    const matched = provider.models.find(item => item.model_name === modelPart);
+    if (matched?.display_name) return matched.display_name;
+  }
+  return fallbackModelDisplayName(modelPart);
+}

package/dist/package/src/views/ai/chat/modelSelection.ts

@@ -0,0 +1,31 @@
+/** 用户最近一次选择的供应商与模型，仅保存在浏览器本地。 */
+export interface AiModelSelection {
+  provider_id: number;
+  model_name: string;
+}
+
+const MODEL_SELECTION_STORAGE_KEY = "kratos-admin:ai:model-selection";
+
+/** 读取本地缓存的模型选择，缓存缺失或损坏时返回 undefined。 */
+export function loadStoredModelSelection(): AiModelSelection | undefined {
+  try {
+    const raw = localStorage.getItem(MODEL_SELECTION_STORAGE_KEY);
+    if (!raw) return undefined;
+    const parsed = JSON.parse(raw) as Partial<AiModelSelection>;
+    if (typeof parsed.provider_id === "number" && typeof parsed.model_name === "string" && parsed.model_name) {
+      return { provider_id: parsed.provider_id, model_name: parsed.model_name };
+    }
+  } catch {
+    // 缓存损坏或存储不可用时忽略，回退到默认选择。
+  }
+  return undefined;
+}
+
+/** 持久化模型选择，存储不可用时静默失败，选择仍对当前会话生效。 */
+export function saveModelSelection(selection: AiModelSelection): void {
+  try {
+    localStorage.setItem(MODEL_SELECTION_STORAGE_KEY, JSON.stringify(selection));
+  } catch {
+    // 忽略存储失败，不影响当前会话内的选择。
+  }
+}