From 218379aa6a22995bd0fd51a30f0b6b9253746669 Mon Sep 17 00:00:00 2001
From: houzhongjian <houzhongyi@126.com>
Date: 星期一, 12 五月 2025 10:32:04 +0800
Subject: [PATCH] ai工业大模型代码提交
---
src/api/ai/chat/message/index.ts | 57 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 56 insertions(+), 1 deletions(-)
diff --git a/src/api/ai/chat/message/index.ts b/src/api/ai/chat/message/index.ts
index ef1196a..62e82b6 100644
--- a/src/api/ai/chat/message/index.ts
+++ b/src/api/ai/chat/message/index.ts
@@ -14,9 +14,16 @@
modelId: number // 模型编号
content: string // 聊天内容
tokens: number // 消耗 Token 数量
+ segmentIds?: number[] // 段落编号
+ segments?: {
+ id: number // 段落编号
+ content: string // 段落内容
+ documentId: number // 文档编号
+ documentName: string // 文档名称
+ }[]
createTime: Date // 创建时间
roleAvatar: string // 角色头像
- userAvatar: string // 创建时间
+ userAvatar: string // 用户头像
}
// AI chat 聊天
@@ -25,6 +32,13 @@
getChatMessageListByConversationId: async (conversationId: number | null) => {
return await request.get({
url: `/ai/chat/message/list-by-conversation-id?conversationId=${conversationId}`
+ })
+ },
+
+ // 消息列表【工业大模型专用】
+ getEnergyChatMessageListByConversationId: async (conversationId: number | null) => {
+ return await request.get({
+ url: `/ai/chat/message/energy-list-by-conversation-id?conversationId=${conversationId}`
})
},
@@ -58,6 +72,35 @@
signal: ctrl.signal
})
},
+ // 发送 Stream 消息 【工业大模型专用】
+ sendEnergyChatMessageStream: async (
+ conversationId: number,
+ content: string,
+ ctrl,
+ enableContext: boolean,
+ onMessage,
+ onError,
+ onClose
+ ) => {
+ const token = getAccessToken()
+ return fetchEventSource(`${config.base_url}/ai/chat/message/send-energy-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
+ })
+ },
// 删除消息
deleteChatMessage: async (id: string) => {
@@ -71,6 +114,18 @@
})
},
+ // 删除消息【工业大模型专用】
+ deleteEnergyChatMessage: async (id: string) => {
+ return await request.delete({ url: `/ai/chat/message/delete-energy?id=${id}` })
+ },
+
+ // 删除指定对话的消息【工业大模型专用】
+ deleteEnergyByConversationId: async (conversationId: number) => {
+ return await request.delete({
+ url: `/ai/chat/message/delete-energy-by-conversation-id?conversationId=${conversationId}`
+ })
+ },
+
// 获得消息分页
getChatMessagePage: async (params: any) => {
return await request.get({ url: '/ai/chat/message/page', params })
--
Gitblit v1.9.3