Merge remote-tracking branch 'origin/dev' into dev

pull/449/head^2
cherishsince 2024-05-15 23:11:49 +08:00
commit 5c02e79105
4 changed files with 126 additions and 122 deletions

View File

@ -13,6 +13,7 @@ export interface ChatConversationVO {
maxTokens: number // 单条回复的最大 Token 数量 maxTokens: number // 单条回复的最大 Token 数量
maxContexts: number // 上下文的最大 Message 数量 maxContexts: number // 上下文的最大 Message 数量
// 额外字段 // 额外字段
modelName?: string // 模型名字
roleAvatar?: string // 角色头像 roleAvatar?: string // 角色头像
modelMaxTokens?: string // 模型的单条回复的最大 Token 数量 modelMaxTokens?: string // 模型的单条回复的最大 Token 数量
modelMaxContexts?: string // 模型的上下文的最大 Message 数量 modelMaxContexts?: string // 模型的上下文的最大 Message 数量
@ -21,19 +22,25 @@ export interface ChatConversationVO {
// AI 聊天会话 API // AI 聊天会话 API
export const ChatConversationApi = { export const ChatConversationApi = {
// 获得【我的】聊天会话 // 获得【我的】聊天会话
getChatConversationMy: async (id: string) => { getChatConversationMy: async (id: number) => {
return await request.get({ url: `/ai/chat/conversation/get-my?id=${id}` }) return await request.get({ url: `/ai/chat/conversation/get-my?id=${id}` })
}, },
// 更新【我的】聊天会话
updateChatConversationMy: async (data: ChatConversationVO) => {
return await request.put({ url: `/ai/chat/conversation/update-my`, data })
},
// 新增【我的】聊天会话 // 新增【我的】聊天会话
createChatConversationMy: async (data?: ChatConversationVO) => { createChatConversationMy: async (data?: ChatConversationVO) => {
return await request.post({ url: `/ai/chat/conversation/create-my`, data }) return await request.post({ url: `/ai/chat/conversation/create-my`, data })
}, },
// 更新【我的】聊天会话
updateChatConversationMy: async (data: ChatConversationVO) => {
return await request.put({ url: `/ai/chat/conversation/update-my`, data })
},
// 删除【我的】聊天会话
deleteChatConversationMy: async (id: number) => {
return await request.delete({ url: `/ai/chat/conversation/delete-my?id=${id}` })
},
// 获得【我的】聊天会话列表 // 获得【我的】聊天会话列表
getChatConversationMyList: async () => { getChatConversationMyList: async () => {
return await request.get({ url: `/ai/chat/conversation/my-list` }) return await request.get({ url: `/ai/chat/conversation/my-list` })

View File

@ -31,19 +31,16 @@ export const ChatMessageApi = {
}) })
}, },
// 发送 add 消息
add: async (data: ChatMessageSendVO) => {
return await request.post({ url: `/ai/chat/message/add`, data })
},
// 发送 send 消息
send: async (data: ChatMessageSendVO) => {
return await request.post({ url: `/ai/chat/message/send`, data })
},
// 发送 send stream 消息 // 发送 send stream 消息
// TODO axios 可以么? https://apifox.com/apiskills/how-to-create-axios-stream/ // TODO axios 可以么? https://apifox.com/apiskills/how-to-create-axios-stream/
sendStream: async (id: string, ctrl, onMessage, onError, onClose) => { sendStream: async (
conversationId: number,
content: string,
ctrl,
onMessage,
onError,
onClose
) => {
const token = getAccessToken() const token = getAccessToken()
return fetchEventSource(`${config.base_url}/ai/chat/message/send-stream`, { return fetchEventSource(`${config.base_url}/ai/chat/message/send-stream`, {
method: 'post', method: 'post',
@ -53,7 +50,8 @@ export const ChatMessageApi = {
}, },
openWhenHidden: true, openWhenHidden: true,
body: JSON.stringify({ body: JSON.stringify({
id: id conversationId,
content
}), }),
onmessage: onMessage, onmessage: onMessage,
onerror: onError, onerror: onError,

View File

@ -4,23 +4,46 @@
ref="formRef" ref="formRef"
:model="formData" :model="formData"
:rules="formRules" :rules="formRules"
label-width="100px" label-width="130px"
v-loading="formLoading" v-loading="formLoading"
> >
<el-form-item label="角色设定" prop="systemContext"> <el-form-item label="角色设定" prop="systemMessage">
<el-input type="textarea" v-model="formData.systemContext" placeholder="请输入角色设定" /> <el-input type="textarea" v-model="formData.systemMessage" placeholder="请输入角色设定" />
</el-form-item> </el-form-item>
<el-form-item label="模型" prop="modelId"> <el-form-item label="模型" prop="modelId">
<UploadImg v-model="formData.modelId" /> <el-select v-model="formData.modelId" placeholder="请选择模型">
<el-option
v-for="chatModel in chatModelList"
:key="chatModel.id"
:label="chatModel.name"
:value="chatModel.id"
/>
</el-select>
</el-form-item> </el-form-item>
<el-form-item label="上下文数量" prop="category"> <el-form-item label="温度参数" prop="temperature">
<el-input v-model="formData.category" placeholder="请输入角色类别" /> <el-input-number
v-model="formData.temperature"
placeholder="请输入温度参数"
:min="0"
:max="2"
:precision="2"
/>
</el-form-item> </el-form-item>
<el-form-item label="话题随机性" prop="description"> <el-form-item label="回复数 Token 数" prop="maxTokens">
<el-input type="textarea" v-model="formData.description" placeholder="请输入角色描述" /> <el-input-number
v-model="formData.maxTokens"
placeholder="请输入回复数 Token 数"
:min="0"
:max="4096"
/>
</el-form-item> </el-form-item>
<el-form-item label="回复数" prop="systemMessage"> <el-form-item label="上下文数量" prop="maxContexts">
<el-input type="textarea" v-model="formData.systemMessage" placeholder="请输入角色设定" /> <el-input-number
v-model="formData.maxContexts"
placeholder="请输入上下文数量"
:min="0"
:max="20"
/>
</el-form-item> </el-form-item>
</el-form> </el-form>
<template #footer> <template #footer>
@ -30,7 +53,6 @@
</Dialog> </Dialog>
</template> </template>
<script setup lang="ts"> <script setup lang="ts">
import { ChatRoleApi, ChatRoleVO } from '@/api/ai/model/chatRole'
import { CommonStatusEnum } from '@/utils/constants' import { CommonStatusEnum } from '@/utils/constants'
import { ChatModelApi, ChatModelVO } from '@/api/ai/model/chatModel' import { ChatModelApi, ChatModelVO } from '@/api/ai/model/chatModel'
import { ChatConversationApi, ChatConversationVO } from '@/api/ai/chat/conversation' import { ChatConversationApi, ChatConversationVO } from '@/api/ai/chat/conversation'
@ -45,25 +67,18 @@ const dialogVisible = ref(false) // 弹窗的是否展示
const formLoading = ref(false) // 12 const formLoading = ref(false) // 12
const formData = ref({ const formData = ref({
id: undefined, id: undefined,
systemContext: undefined,
modelId: undefined,
name: undefined,
avatar: undefined,
category: undefined,
sort: undefined,
description: undefined,
systemMessage: undefined, systemMessage: undefined,
publicStatus: true, modelId: undefined,
status: CommonStatusEnum.ENABLE temperature: undefined,
maxTokens: undefined,
maxContexts: undefined
}) })
const formRules = reactive({ const formRules = reactive({
name: [{ required: true, message: '角色名称不能为空', trigger: 'blur' }], modelId: [{ required: true, message: '模型不能为空', trigger: 'blur' }],
avatar: [{ required: true, message: '角色头像不能为空', trigger: 'blur' }], status: [{ required: true, message: '状态不能为空', trigger: 'blur' }],
category: [{ required: true, message: '角色类别不能为空', trigger: 'blur' }], temperature: [{ required: true, message: '温度参数不能为空', trigger: 'blur' }],
sort: [{ required: true, message: '角色排序不能为空', trigger: 'blur' }], maxTokens: [{ required: true, message: '回复数 Token 数不能为空', trigger: 'blur' }],
description: [{ required: true, message: '角色描述不能为空', trigger: 'blur' }], maxContexts: [{ required: true, message: '上下文数量不能为空', trigger: 'blur' }]
systemMessage: [{ required: true, message: '角色设定不能为空', trigger: 'blur' }],
publicStatus: [{ required: true, message: '是否公开不能为空', trigger: 'blur' }]
}) })
const formRef = ref() // Ref const formRef = ref() // Ref
const chatModelList = ref([] as ChatModelVO[]) // const chatModelList = ref([] as ChatModelVO[]) //
@ -76,7 +91,13 @@ const open = async (id: number) => {
if (id) { if (id) {
formLoading.value = true formLoading.value = true
try { try {
formData.value = await ChatConversationApi.getChatConversationMy(id) const data = await ChatConversationApi.getChatConversationMy(id)
formData.value = Object.keys(formData.value).reduce((obj, key) => {
if (data.hasOwnProperty(key)) {
obj[key] = data[key]
}
return obj
}, {})
} finally { } finally {
formLoading.value = false formLoading.value = false
} }
@ -94,9 +115,9 @@ const submitForm = async () => {
// //
formLoading.value = true formLoading.value = true
try { try {
const data = formData.value as unknown as ChatRoleVO const data = formData.value as unknown as ChatConversationVO
await ChatRoleApi.updateChatRole(data) await ChatConversationApi.updateChatConversationMy(data)
message.success(t('common.updateSuccess')) message.success('对话配置已更新')
dialogVisible.value = false dialogVisible.value = false
// //
emit('success') emit('success')
@ -109,15 +130,11 @@ const submitForm = async () => {
const resetForm = () => { const resetForm = () => {
formData.value = { formData.value = {
id: undefined, id: undefined,
modelId: undefined,
name: undefined,
avatar: undefined,
category: undefined,
sort: undefined,
description: undefined,
systemMessage: undefined, systemMessage: undefined,
publicStatus: true, modelId: undefined,
status: CommonStatusEnum.ENABLE temperature: undefined,
maxTokens: undefined,
maxContexts: undefined
} }
formRef.value?.resetFields() formRef.value?.resetFields()
} }

View File

@ -22,7 +22,7 @@
</el-input> </el-input>
<!-- 左中间对话列表 --> <!-- 左中间对话列表 -->
<div class="conversation-list"> <div class="conversation-list">
<!-- TODO @芋艿置顶聊天记录一星期钱30天前前端对数据重新做一下分组或者后端接口改一下 --> <!-- TODO @fain置顶聊天记录一星期钱30天前前端对数据重新做一下分组或者后端接口改一下 -->
<div> <div>
<el-text class="mx-1" size="small" tag="b">置顶</el-text> <el-text class="mx-1" size="small" tag="b">置顶</el-text>
</div> </div>
@ -35,11 +35,12 @@
<img class="avatar" :src="conversation.roleAvatar" /> <img class="avatar" :src="conversation.roleAvatar" />
<span class="title">{{ conversation.title }}</span> <span class="title">{{ conversation.title }}</span>
</div> </div>
<!-- TODO @fan缺一个置顶按钮效果改成 hover 上去展示 -->
<div class="button-wrapper"> <div class="button-wrapper">
<el-icon title="编辑" @click="updateConversationTitle(conversation)"> <el-icon title="编辑" @click="updateConversationTitle(conversation)">
<Icon icon="ep:edit" /> <Icon icon="ep:edit" />
</el-icon> </el-icon>
<el-icon title="删除会话" @click="deleteConversationTitle(conversation)"> <el-icon title="删除会话" @click="deleteChatConversation(conversation)">
<Icon icon="ep:delete" /> <Icon icon="ep:delete" />
</el-icon> </el-icon>
</div> </div>
@ -68,17 +69,10 @@
</div> </div>
<div> <div>
<!-- TODO @fan样式改下这里我已经改成点击后弹出了 --> <!-- TODO @fan样式改下这里我已经改成点击后弹出了 -->
<el-dropdown style="margin-right: 12px" @command="openChatConversationUpdateForm"> <el-button type="primary" @click="openChatConversationUpdateForm">
<el-button type="primary"> <span v-html="useConversation?.modelName"></span>
<span v-html="useModal?.name"></span> <Icon icon="ep:setting" style="margin-left: 10px" />
<Icon icon="ep:setting" style="margin-left: 10px" /> </el-button>
</el-button>
<template #dropdown>
<el-dropdown-menu v-for="(item, index) in modalList" :key="index">
<el-dropdown-item :command="item">{{ item.name }}</el-dropdown-item>
</el-dropdown-menu>
</template>
</el-dropdown>
<el-button> <el-button>
<Icon icon="ep:user" /> <Icon icon="ep:user" />
</el-button> </el-button>
@ -96,6 +90,7 @@
<div class="message-container" ref="messageContainer"> <div class="message-container" ref="messageContainer">
<div class="chat-list" v-for="(item, index) in list" :key="index"> <div class="chat-list" v-for="(item, index) in list" :key="index">
<!-- 靠左 message --> <!-- 靠左 message -->
<!-- TODO 芋艿类型判断 -->
<div class="left-message message-item" v-if="item.type === 'system'"> <div class="left-message message-item" v-if="item.type === 'system'">
<div class="avatar"> <div class="avatar">
<el-avatar <el-avatar
@ -187,18 +182,16 @@
</el-container> </el-container>
</el-container> </el-container>
<ChatConversationUpdateForm ref="chatConversationUpdateFormRef" /> <ChatConversationUpdateForm
ref="chatConversationUpdateFormRef"
@success="getChatConversationList"
/>
</template> </template>
<script setup lang="ts"> <script setup lang="ts">
import { ChatMessageApi, ChatMessageSendVO, ChatMessageVO } from '@/api/ai/chat/message' import { ChatMessageApi, ChatMessageSendVO, ChatMessageVO } from '@/api/ai/chat/message'
import { import { ChatConversationApi, ChatConversationVO } from '@/api/ai/chat/conversation'
ChatConversationApi,
ChatConversationUpdateVO,
ChatConversationVO
} from '@/api/ai/chat/conversation'
import ChatConversationUpdateForm from './components/ChatConversationUpdateForm.vue' import ChatConversationUpdateForm from './components/ChatConversationUpdateForm.vue'
import { ChatModelApi, ChatModelVO } from '@/api/ai/model/chatModel'
import { formatDate } from '@/utils/formatTime' import { formatDate } from '@/utils/formatTime'
import { useClipboard } from '@vueuse/core' import { useClipboard } from '@vueuse/core'
// markdown // markdown
@ -207,7 +200,6 @@ import { marked } from 'marked'
import 'highlight.js/styles/vs2015.min.css' import 'highlight.js/styles/vs2015.min.css'
import hljs from 'highlight.js' import hljs from 'highlight.js'
const { t } = useI18n() //
const message = useMessage() // const message = useMessage() //
// //
@ -229,8 +221,7 @@ const { copy } = useClipboard()
const searchName = ref('') // const searchName = ref('') //
const inputTimeout = ref<any>() // const inputTimeout = ref<any>() //
const conversationId = ref(0) // const conversationId = ref(0) //
const conversationInProgress = ref<Boolean>() // const conversationInProgress = ref(false) //
conversationInProgress.value = false
const conversationInAbortController = ref<any>() // abort ( stream ) const conversationInAbortController = ref<any>() // abort ( stream )
const prompt = ref<string>() // prompt const prompt = ref<string>() // prompt
@ -243,9 +234,7 @@ const isComposing = ref(false) // 判断用户是否在输入
/** chat message 列表 */ /** chat message 列表 */
// defineOptions({ name: 'chatMessageList' }) // defineOptions({ name: 'chatMessageList' })
const list = ref<ChatMessageVO[]>([]) // const list = ref<ChatMessageVO[]>([]) //
const useModal = ref<ChatModelVO>() // 使modal
const useConversation = ref<ChatConversationVO>() // 使 Conversation const useConversation = ref<ChatConversationVO>() // 使 Conversation
const modalList = ref<ChatModelVO[]>([]) //
/** 新建对话 */ /** 新建对话 */
const createConversation = async () => { const createConversation = async () => {
@ -282,13 +271,20 @@ const updateConversationTitle = async (conversation: ChatConversationVO) => {
} }
/** 删除聊天会话 */ /** 删除聊天会话 */
const deleteConversationTitle = async (conversation: ChatConversationVO) => { const deleteChatConversation = async (conversation: ChatConversationVO) => {
console.log(conversation) try {
// TODO //
await message.delConfirm(`是否确认删除会话 - ${conversation.title}?`)
//
await ChatConversationApi.deleteChatConversationMy(conversation.id)
message.success('会话已删除')
//
await getChatConversationList()
} catch {}
} }
const searchConversation = () => { const searchConversation = () => {
// TODO // TODO fan
} }
/** send */ /** send */
@ -302,24 +298,28 @@ const onSend = async () => {
return return
} }
const content = prompt.value?.trim() const content = prompt.value?.trim()
if (content?.length < 2) { if (content.length < 2) {
ElMessage({ ElMessage({
message: '请输入内容!', message: '请输入内容!',
type: 'error' type: 'error'
}) })
return return
} }
// TODO UI
// //
prompt.value = '' prompt.value = ''
const requestParams = { // const requestParams = {
// conversationId: conversationId.value,
// content: content
// } as unknown as ChatMessageSendVO
// // message
const userMessage = {
conversationId: conversationId.value, conversationId: conversationId.value,
content: content content: content
} as unknown as ChatMessageSendVO }
// message // list.value.push(userMessage)
const userMessage = (await ChatMessageApi.add(requestParams)) as unknown as ChatMessageVO // //
list.value.push(userMessage) // scrollToBottom()
//
scrollToBottom()
// stream // stream
await doSendStream(userMessage) await doSendStream(userMessage)
// //
@ -335,13 +335,15 @@ const doSendStream = async (userMessage: ChatMessageVO) => {
let isFirstMessage = true let isFirstMessage = true
let content = '' let content = ''
ChatMessageApi.sendStream( ChatMessageApi.sendStream(
userMessage.id, userMessage.conversationId, // TODO
userMessage.content,
conversationInAbortController.value, conversationInAbortController.value,
(message) => { (message) => {
console.log('message', message) console.log('message', message)
const data = JSON.parse(message.data) as unknown as ChatMessageVO const data = JSON.parse(message.data) // TODO
// debugger
// //
if (data.content === '') { if (data.receive.content === '') {
// //
conversationInProgress.value = false conversationInProgress.value = false
// stream // stream
@ -350,9 +352,12 @@ const doSendStream = async (userMessage: ChatMessageVO) => {
// message // message
if (isFirstMessage) { if (isFirstMessage) {
isFirstMessage = false isFirstMessage = false
list.value.push(data) // debugger
list.value.push(data.send)
list.value.push(data.receive)
} else { } else {
content = content + data.content // debugger
content = content + data.receive.content
const lastMessage = list.value[list.value.length - 1] const lastMessage = list.value[list.value.length - 1]
lastMessage.content = marked(content) as unknown as string lastMessage.content = marked(content) as unknown as string
list.value[list.value - 1] = lastMessage list.value[list.value - 1] = lastMessage
@ -460,23 +465,10 @@ const stopStream = async () => {
/** 修改聊天会话 */ /** 修改聊天会话 */
const chatConversationUpdateFormRef = ref() const chatConversationUpdateFormRef = ref()
const openChatConversationUpdateForm = async (command) => { const openChatConversationUpdateForm = async () => {
// const update = {
// id: conversationId.value,
// modelId: command.id
// } as unknown as ChatConversationUpdateVO
// // modal
// useModal.value = command
// //
// await ChatConversationApi.updateChatConversationMy(update)
chatConversationUpdateFormRef.value.open(conversationId.value) chatConversationUpdateFormRef.value.open(conversationId.value)
} }
const getModalList = async () => {
// as unknown as ChatModelVO
modalList.value = (await ChatModelApi.getChatModelSimpleList(0)) as unknown as ChatModelVO[]
}
// //
const onCompositionstart = () => { const onCompositionstart = () => {
console.log('onCompositionstart。。。.') console.log('onCompositionstart。。。.')
@ -516,14 +508,6 @@ const getConversation = async (conversationId: string) => {
// //
useConversation.value = await ChatConversationApi.getChatConversationMy(conversationId) useConversation.value = await ChatConversationApi.getChatConversationMy(conversationId)
console.log('useConversation.value', useConversation.value) console.log('useConversation.value', useConversation.value)
// modal
if (useConversation.value) {
modalList.value.forEach((item) => {
if (useConversation.value?.modelId === item.id) {
useModal.value = item
}
})
}
} }
/** 获得聊天会话列表 */ /** 获得聊天会话列表 */
@ -545,8 +529,6 @@ const getChatConversationList = async () => {
onMounted(async () => { onMounted(async () => {
// //
await getChatConversationList() await getChatConversationList()
//
getModalList()
// //
getConversation(conversationId.value) getConversation(conversationId.value)
// //