feat(ai): 添加 AI 对话聊天和 API 密钥管理功能

- 新增 AI 对话聊天管理页面，包括对话列表和消息列表 - 新增 API 密钥管理页面，包括密钥列表和表单 - 添加相关 API 接口和数据模型 - 集成表单和表格组件，实现基本的 CRUD 操作
2025-06-06 17:09:14 +08:00
parent 75c5669a97
commit 3ef362508a
28 changed files with 2509 additions and 105 deletions
--- a/apps/web-antd/src/api/ai/chat/conversation/index.ts
+++ b/apps/web-antd/src/api/ai/chat/conversation/index.ts
@@ -0,0 +1,75 @@
+import type { PageResult } from '@vben/request';
+
+import { requestClient } from '#/api/request';
+
+export namespace AiChatConversationApi {
+  export interface ChatConversationVO {
+    id: number; // ID 编号
+    userId: number; // 用户编号
+    title: string; // 对话标题
+    pinned: boolean; // 是否置顶
+    roleId: number; // 角色编号
+    modelId: number; // 模型编号
+    model: string; // 模型标志
+    temperature: number; // 温度参数
+    maxTokens: number; // 单条回复的最大 Token 数量
+    maxContexts: number; // 上下文的最大 Message 数量
+    createTime?: Date; // 创建时间
+    // 额外字段
+    systemMessage?: string; // 角色设定
+    modelName?: string; // 模型名字
+    roleAvatar?: string; // 角色头像
+    modelMaxTokens?: string; // 模型的单条回复的最大 Token 数量
+    modelMaxContexts?: string; // 模型的上下文的最大 Message 数量
+  }
+}
+
+// 获得【我的】聊天对话
+export function getChatConversationMy(id: number) {
+  return requestClient.get<
+    PageResult<AiChatConversationApi.ChatConversationVO>
+  >(`/ai/chat/conversation/get-my?id=${id}`);
+}
+
+// 新增【我的】聊天对话
+export function createChatConversationMy(
+  data: AiChatConversationApi.ChatConversationVO,
+) {
+  return requestClient.post('/ai/chat/conversation/create-my', data);
+}
+
+//  更新【我的】聊天对话
+export function updateChatConversationMy(
+  data: AiChatConversationApi.ChatConversationVO,
+) {
+  return requestClient.put(`/ai/chat/conversation/update-my`, data);
+}
+
+//  删除【我的】聊天对话
+export function deleteChatConversationMy(id: string) {
+  return requestClient.delete(`/ai/chat/conversation/delete-my?id=${id}`);
+}
+
+//  删除【我的】所有对话，置顶除外
+export function deleteChatConversationMyByUnpinned() {
+  return requestClient.delete(`/ai/chat/conversation/delete-by-unpinned`);
+}
+
+//  获得【我的】聊天对话列表
+export function getChatConversationMyList() {
+  return requestClient.get<AiChatConversationApi.ChatConversationVO[]>(
+    `/ai/chat/conversation/my-list`,
+  );
+}
+
+//  获得【我的】聊天对话列表
+export function getChatConversationPage(params: any) {
+  return requestClient.get<
+    PageResult<AiChatConversationApi.ChatConversationVO[]>
+  >(`/ai/chat/conversation/page`, { params });
+}
+
+//  管理员删除消息
+export function deleteChatConversationByAdmin(id: number) {
+  return requestClient.delete(`/ai/chat/conversation/delete-by-admin?id=${id}`);
+}
--- a/apps/web-antd/src/api/ai/chat/message/index.ts
+++ b/apps/web-antd/src/api/ai/chat/message/index.ts
@@ -0,0 +1,96 @@
+import type { PageResult } from '@vben/request';
+
+import { fetchEventSource } from '@vben/request';
+import { useAccessStore } from '@vben/stores';
+
+import { requestClient } from '#/api/request';
+
+const accessStore = useAccessStore();
+export namespace AiChatMessageApi {
+  export interface ChatMessageVO {
+    id: number; // 编号
+    conversationId: number; // 对话编号
+    type: string; // 消息类型
+    userId: string; // 用户编号
+    roleId: string; // 角色编号
+    model: number; // 模型标志
+    modelId: number; // 模型编号
+    content: string; // 聊天内容
+    tokens: number; // 消耗 Token 数量
+    segmentIds?: number[]; // 段落编号
+    segments?: {
+      content: string; // 段落内容
+      documentId: number; // 文档编号
+      documentName: string; // 文档名称
+      id: number; // 段落编号
+    }[];
+    createTime: Date; // 创建时间
+    roleAvatar: string; // 角色头像
+    userAvatar: string; // 用户头像
+  }
+}
+
+// 消息列表
+export function getChatMessageListByConversationId(
+  conversationId: null | number,
+) {
+  return requestClient.get<AiChatMessageApi.ChatMessageVO[]>(
+    `/ai/chat/message/list-by-conversation-id?conversationId=${conversationId}`,
+  );
+}
+
+// 发送 Stream 消息
+export function sendChatMessageStream(
+  conversationId: number,
+  content: string,
+  ctrl: any,
+  enableContext: boolean,
+  onMessage: any,
+  onError: any,
+  onClose: any,
+) {
+  const token = accessStore.accessToken;
+  return fetchEventSource(
+    `${import.meta.env.VITE_BASE_URL}/ai/chat/message/send-stream`,
+    {
+      method: 'post',
+      headers: {
+        'Content-Type': 'application/json',
+        Authorization: `Bearer ${token}`,
+      },
+      openWhenHidden: true,
+      body: JSON.stringify({
+        conversationId,
+        content,
+        useContext: enableContext,
+      }),
+      onmessage: onMessage,
+      onerror: onError,
+      onclose: onClose,
+      signal: ctrl.signal,
+    },
+  );
+}
+
+// 删除消息
+export function deleteChatMessage(id: string) {
+  return requestClient.delete(`/ai/chat/message/delete?id=${id}`);
+}
+
+// 删除指定对话的消息
+export function deleteByConversationId(conversationId: number) {
+  return requestClient.delete(
+    `/ai/chat/message/delete-by-conversation-id?conversationId=${conversationId}`,
+  );
+}
+// 获得消息分页
+export function getChatMessagePage(params: any) {
+  return requestClient.get<PageResult<AiChatMessageApi.ChatMessageVO>>(
+    '/ai/chat/message/page',
+    { params },
+  );
+}
+// 管理员删除消息
+export function deleteChatMessageByAdmin(id: number) {
+  return requestClient.delete(`/ai/chat/message/delete-by-admin?id=${id}`);
+}
--- a/apps/web-antd/src/api/ai/knowledge/knowledge/index.ts
+++ b/apps/web-antd/src/api/ai/knowledge/knowledge/index.ts
@@ -0,0 +1,50 @@
+import type { PageParam, PageResult } from '@vben/request';
+
+import { requestClient } from '#/api/request';
+
+export namespace AiKnowledgeKnowledgeApi {
+  export interface KnowledgeVO {
+    id: number; // 编号
+    name: string; // 知识库名称
+    description: string; // 知识库描述
+    embeddingModelId: number; // 嵌入模型编号，高质量模式时维护
+    topK: number; // topK
+    similarityThreshold: number; // 相似度阈值
+  }
+}
+
+// 查询知识库分页
+export function getKnowledgePage(params: PageParam) {
+  return requestClient.get<PageResult<AiKnowledgeKnowledgeApi.KnowledgeVO>>(
+    '/ai/knowledge/page',
+    { params },
+  );
+}
+
+// 查询知识库详情
+export function getKnowledge(id: number) {
+  return requestClient.get<AiKnowledgeKnowledgeApi.KnowledgeVO>(
+    `/ai/knowledge/get?id=${id}`,
+  );
+}
+// 新增知识库
+export function createKnowledge(data: AiKnowledgeKnowledgeApi.KnowledgeVO) {
+  return requestClient.post('/ai/knowledge/create', data);
+}
+
+// 修改知识库
+export function updateKnowledge(data: AiKnowledgeKnowledgeApi.KnowledgeVO) {
+  return requestClient.put('/ai/knowledge/update', data);
+}
+
+// 删除知识库
+export function deleteKnowledge(id: number) {
+  return requestClient.delete(`/ai/knowledge/delete?id=${id}`);
+}
+
+// 获取知识库简单列表
+export function getSimpleKnowledgeList() {
+  return requestClient.get<AiKnowledgeKnowledgeApi.KnowledgeVO[]>(
+    '/ai/knowledge/simple-list',
+  );
+}
--- a/apps/web-antd/src/api/ai/model/apiKey/index.ts
+++ b/apps/web-antd/src/api/ai/model/apiKey/index.ts
@@ -0,0 +1,50 @@
+import type { PageParam, PageResult } from '@vben/request';
+
+import { requestClient } from '#/api/request';
+
+export namespace AiModelApiKeyApi {
+  export interface ApiKeyVO {
+    id: number; // 编号
+    name: string; // 名称
+    apiKey: string; // 密钥
+    platform: string; // 平台
+    url: string; // 自定义 API 地址
+    status: number; // 状态
+  }
+}
+
+// 查询 API 密钥分页
+export function getApiKeyPage(params: PageParam) {
+  return requestClient.get<PageResult<AiModelApiKeyApi.ApiKeyVO>>(
+    '/ai/api-key/page',
+    { params },
+  );
+}
+
+// 获得 API 密钥列表
+export function getApiKeySimpleList() {
+  return requestClient.get<AiModelApiKeyApi.ApiKeyVO[]>(
+    '/ai/api-key/simple-list',
+  );
+}
+
+// 查询 API 密钥详情
+export function getApiKey(id: number) {
+  return requestClient.get<AiModelApiKeyApi.ApiKeyVO>(
+    `/ai/api-key/get?id=${id}`,
+  );
+}
+// 新增 API 密钥
+export function createApiKey(data: AiModelApiKeyApi.ApiKeyVO) {
+  return requestClient.post('/ai/api-key/create', data);
+}
+
+// 修改 API 密钥
+export function updateApiKey(data: AiModelApiKeyApi.ApiKeyVO) {
+  return requestClient.put('/ai/api-key/update', data);
+}
+
+// 删除 API 密钥
+export function deleteApiKey(id: number) {
+  return requestClient.delete(`/ai/api-key/delete?id=${id}`);
+}
--- a/apps/web-antd/src/api/ai/model/chatRole/index.ts
+++ b/apps/web-antd/src/api/ai/model/chatRole/index.ts
@@ -0,0 +1,85 @@
+import type { PageParam, PageResult } from '@vben/request';
+
+import { requestClient } from '#/api/request';
+
+export namespace AiModelChatRoleApi {
+  export interface ChatRoleVO {
+    id: number; // 角色编号
+    modelId: number; // 模型编号
+    name: string; // 角色名称
+    avatar: string; // 角色头像
+    category: string; // 角色类别
+    sort: number; // 角色排序
+    description: string; // 角色描述
+    systemMessage: string; // 角色设定
+    welcomeMessage: string; // 角色设定
+    publicStatus: boolean; // 是否公开
+    status: number; // 状态
+    knowledgeIds?: number[]; // 引用的知识库 ID 列表
+    toolIds?: number[]; // 引用的工具 ID 列表
+  }
+
+  // AI 聊天角色 分页请求 vo
+  export interface ChatRolePageReqVO {
+    name?: string; // 角色名称
+    category?: string; // 角色类别
+    publicStatus: boolean; // 是否公开
+    pageNo: number; // 是否公开
+    pageSize: number; // 是否公开
+  }
+}
+
+// 查询聊天角色分页
+export function getChatRolePage(params: PageParam) {
+  return requestClient.get<PageResult<AiModelChatRoleApi.ChatRoleVO>>(
+    '/ai/chat-role/page',
+    { params },
+  );
+}
+
+// 查询聊天角色详情
+export function getChatRole(id: number) {
+  return requestClient.get<AiModelChatRoleApi.ChatRoleVO>(
+    `/ai/chat-role/get?id=${id}`,
+  );
+}
+// 新增聊天角色
+export function createChatRole(data: AiModelChatRoleApi.ChatRoleVO) {
+  return requestClient.post('/ai/chat-role/create', data);
+}
+
+// 修改聊天角色
+export function updateChatRole(data: AiModelChatRoleApi.ChatRoleVO) {
+  return requestClient.put('/ai/chat-role/update', data);
+}
+
+// 删除聊天角色
+export function deleteChatRole(id: number) {
+  return requestClient.delete(`/ai/chat-role/delete?id=${id}`);
+}
+
+// ======= chat 聊天
+// 获取 my role
+export function getMyPage(params: AiModelChatRoleApi.ChatRolePageReqVO) {
+  return requestClient.get('/ai/chat-role/my-page', { params });
+}
+
+// 获取角色分类
+export function getCategoryList() {
+  return requestClient.get('/ai/chat-role/category-list');
+}
+
+// 创建角色
+export function createMy(data: AiModelChatRoleApi.ChatRoleVO) {
+  return requestClient.post('/ai/chat-role/create-my', data);
+}
+
+// 更新角色
+export function updateMy(data: AiModelChatRoleApi.ChatRoleVO) {
+  return requestClient.put('/ai/chat-role/update', data);
+}
+
+// 删除角色 my
+export function deleteMy(id: number) {
+  return requestClient.delete(`/ai/chat-role/delete-my?id=${id}`);
+}
--- a/apps/web-antd/src/api/ai/model/model/index.ts
+++ b/apps/web-antd/src/api/ai/model/model/index.ts
@@ -0,0 +1,55 @@
+import type { PageParam, PageResult } from '@vben/request';
+
+import { requestClient } from '#/api/request';
+
+export namespace AiModelModelApi {
+  export interface ModelVO {
+    id: number; // 编号
+    keyId: number; // API 秘钥编号
+    name: string; // 模型名字
+    model: string; // 模型标识
+    platform: string; // 模型平台
+    type: number; // 模型类型
+    sort: number; // 排序
+    status: number; // 状态
+    temperature?: number; // 温度参数
+    maxTokens?: number; // 单条回复的最大 Token 数量
+    maxContexts?: number; // 上下文的最大 Message 数量
+  }
+}
+
+// 查询模型分页
+export function getModelPage(params: PageParam) {
+  return requestClient.get<PageResult<AiModelModelApi.ModelVO>>(
+    '/ai/model/page',
+    { params },
+  );
+}
+
+// 获得模型列表
+export function getModelSimpleList(type?: number) {
+  return requestClient.get<AiModelModelApi.ModelVO[]>('/ai/model/simple-list', {
+    params: {
+      type,
+    },
+  });
+}
+
+// 查询模型详情
+export function getModel(id: number) {
+  return requestClient.get<AiModelModelApi.ModelVO>(`/ai/model/get?id=${id}`);
+}
+// 新增模型
+export function createModel(data: AiModelModelApi.ModelVO) {
+  return requestClient.post('/ai/model/create', data);
+}
+
+// 修改模型
+export function updateModel(data: AiModelModelApi.ModelVO) {
+  return requestClient.put('/ai/model/update', data);
+}
+
+// 删除模型
+export function deleteModel(id: number) {
+  return requestClient.delete(`/ai/model/delete?id=${id}`);
+}
--- a/apps/web-antd/src/api/ai/model/tool/index.ts
+++ b/apps/web-antd/src/api/ai/model/tool/index.ts
@@ -0,0 +1,43 @@
+import type { PageParam, PageResult } from '@vben/request';
+
+import { requestClient } from '#/api/request';
+
+export namespace AiModelToolApi {
+  export interface ToolVO {
+    id: number; // 工具编号
+    name: string; // 工具名称
+    description: string; // 工具描述
+    status: number; // 状态
+  }
+}
+
+// 查询工具分页
+export function getToolPage(params: PageParam) {
+  return requestClient.get<PageResult<AiModelToolApi.ToolVO>>('/ai/tool/page', {
+    params,
+  });
+}
+
+// 查询工具详情
+export function getTool(id: number) {
+  return requestClient.get<AiModelToolApi.ToolVO>(`/ai/tool/get?id=${id}`);
+}
+// 新增工具
+export function createTool(data: AiModelToolApi.ToolVO) {
+  return requestClient.post('/ai/tool/create', data);
+}
+
+// 修改工具
+export function updateTool(data: AiModelToolApi.ToolVO) {
+  return requestClient.put('/ai/tool/update', data);
+}
+
+// 删除工具
+export function deleteTool(id: number) {
+  return requestClient.delete(`/ai/tool/delete?id=${id}`);
+}
+
+// 获取工具简单列表
+export function getToolSimpleList() {
+  return requestClient.get<AiModelToolApi.ToolVO[]>('/ai/tool/simple-list');
+}