519 lines
14 KiB
Vue
519 lines
14 KiB
Vue
<template>
|
||
<div ref="aiChatRef" class="ai-chat" :class="log ? 'chart-log' : ''">
|
||
<el-scrollbar ref="scrollDiv" @scroll="handleScrollTop">
|
||
<div ref="dialogScrollbar" class="ai-chat__content p-24">
|
||
<div class="item-content mb-16">
|
||
<div class="avatar">
|
||
<AppAvatar class="avatar-gradient">
|
||
<img src="@/assets/icon_robot.svg" style="width: 54%" alt="" />
|
||
</AppAvatar>
|
||
</div>
|
||
|
||
<div class="content">
|
||
<el-card shadow="always" class="dialog-card">
|
||
<template v-for="(item, index) in prologueList" :key="index">
|
||
<div
|
||
v-if="isMdArray(item)"
|
||
@click="quickProblemHandel(item)"
|
||
class="problem-button ellipsis-2 mb-8"
|
||
:class="log ? 'disabled' : 'cursor'"
|
||
>
|
||
<el-icon><EditPen /></el-icon>
|
||
{{ item }}
|
||
</div>
|
||
<MdPreview
|
||
v-else
|
||
class="mb-8"
|
||
ref="editorRef"
|
||
editorId="preview-only"
|
||
:modelValue="item"
|
||
/>
|
||
</template>
|
||
</el-card>
|
||
</div>
|
||
</div>
|
||
<template v-for="(item, index) in chatList" :key="index">
|
||
<!-- 问题 -->
|
||
<div class="item-content mb-16 lighter">
|
||
<div class="avatar">
|
||
<AppAvatar>
|
||
<img src="@/assets/user-icon.svg" style="width: 54%" alt="" />
|
||
</AppAvatar>
|
||
</div>
|
||
<div class="content">
|
||
<div class="text break-all">
|
||
{{ item.problem_text }}
|
||
</div>
|
||
</div>
|
||
</div>
|
||
<!-- 回答 -->
|
||
<div class="item-content mb-16 lighter">
|
||
<div class="avatar">
|
||
<AppAvatar class="avatar-gradient">
|
||
<img src="@/assets/icon_robot.svg" style="width: 54%" alt="" />
|
||
</AppAvatar>
|
||
</div>
|
||
<div class="content">
|
||
<div class="flex" v-if="!item.answer_text">
|
||
<el-card shadow="always" class="dialog-card">
|
||
回答中 <span class="dotting"></span>
|
||
</el-card>
|
||
</div>
|
||
|
||
<el-card v-else shadow="always" class="dialog-card">
|
||
<MdRenderer :source="item.answer_text"></MdRenderer>
|
||
<div v-if="item.write_ed || log">
|
||
<el-divider> <el-text type="info">知识来源</el-text> </el-divider>
|
||
<div class="mb-8">
|
||
<el-space wrap>
|
||
<el-button
|
||
v-for="(dataset, index) in item.dataset_list"
|
||
:key="index"
|
||
type="primary"
|
||
plain
|
||
size="small"
|
||
>{{ dataset.name }}</el-button
|
||
>
|
||
</el-space>
|
||
</div>
|
||
|
||
<div>
|
||
<el-button class="mr-8" type="primary" plain size="small"
|
||
>引用分段：{{ item.paragraph_list.length }}</el-button
|
||
>
|
||
<el-tag type="info" effect="plain">
|
||
消耗 tokens: {{ item?.message_tokens + item?.answer_tokens }}
|
||
</el-tag>
|
||
<el-tag class="ml-8" type="info" effect="plain">
|
||
耗时: {{ item.run_time.toFixed(2) }} s
|
||
</el-tag>
|
||
</div>
|
||
</div>
|
||
</el-card>
|
||
<div class="flex-between mt-8" v-if="log">
|
||
<LogOperationButton v-model:data="chatList[index]" :applicationId="appId" />
|
||
</div>
|
||
|
||
<div class="flex-between mt-8" v-else>
|
||
<div>
|
||
<el-button
|
||
type="primary"
|
||
v-if="item.is_stop && !item.write_ed"
|
||
@click="startChat(item)"
|
||
link
|
||
>继续</el-button
|
||
>
|
||
<el-button type="primary" v-else-if="!item.write_ed" @click="stopChat(item)" link
|
||
>停止回答</el-button
|
||
>
|
||
</div>
|
||
|
||
<div v-if="item.write_ed && props.appId">
|
||
<OperationButton
|
||
:data="item"
|
||
:applicationId="appId"
|
||
:chartId="chartOpenId"
|
||
@regeneration="regenerationChart(item)"
|
||
/>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</template>
|
||
</div>
|
||
</el-scrollbar>
|
||
<div class="ai-chat__operate p-24" v-if="!log">
|
||
<div class="operate-textarea flex">
|
||
<el-input
|
||
ref="quickInputRef"
|
||
v-model="inputValue"
|
||
type="textarea"
|
||
placeholder="请输入"
|
||
:autosize="{ minRows: 1, maxRows: 8 }"
|
||
@keydown.enter="sendChatHandle($event)"
|
||
/>
|
||
<div class="operate">
|
||
<el-button
|
||
text
|
||
class="sent-button"
|
||
:disabled="isDisabledChart || loading"
|
||
@click="sendChatHandle"
|
||
>
|
||
<img v-show="isDisabledChart || loading" src="@/assets/icon_send.svg" alt="" />
|
||
<img
|
||
v-show="!isDisabledChart && !loading"
|
||
src="@/assets/icon_send_colorful.svg"
|
||
alt=""
|
||
/>
|
||
</el-button>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</template>
|
||
<script setup lang="ts">
|
||
import { ref, nextTick, computed, watch } from 'vue'
|
||
import { useRoute } from 'vue-router'
|
||
import LogOperationButton from './LogOperationButton.vue'
|
||
import OperationButton from './OperationButton.vue'
|
||
import applicationApi from '@/api/application'
|
||
import { ChatManagement, type chatType } from '@/api/type/application'
|
||
import { randomId } from '@/utils/utils'
|
||
import useStore from '@/stores'
|
||
import MdRenderer from '@/components/markdown-renderer/MdRenderer.vue'
|
||
import { MdPreview } from 'md-editor-v3'
|
||
defineOptions({ name: 'AiChat' })
|
||
const route = useRoute()
|
||
const {
|
||
params: { accessToken }
|
||
} = route as any
|
||
const props = defineProps({
|
||
data: {
|
||
type: Object,
|
||
default: () => {}
|
||
},
|
||
appId: String,
|
||
log: Boolean,
|
||
record: {
|
||
type: Array<chatType[]>,
|
||
default: () => []
|
||
}
|
||
})
|
||
const { application } = useStore()
|
||
|
||
const aiChatRef = ref()
|
||
const quickInputRef = ref()
|
||
const scrollDiv = ref()
|
||
const dialogScrollbar = ref()
|
||
const loading = ref(false)
|
||
const inputValue = ref('')
|
||
const chartOpenId = ref('')
|
||
const chatList = ref<any[]>([])
|
||
|
||
const isDisabledChart = computed(
|
||
() => !(inputValue.value && (props.appId || (props.data?.name && props.data?.model_id)))
|
||
)
|
||
|
||
const prologueList = computed(() => {
|
||
const temp = props.data?.prologue
|
||
const lines = temp?.split('\n')
|
||
return lines
|
||
})
|
||
const isMdArray = (val: string) => val.match(/^-\s.*/m)
|
||
|
||
watch(
|
||
() => props.record,
|
||
(value) => {
|
||
if (props.log) {
|
||
chatList.value = value
|
||
}
|
||
},
|
||
{
|
||
immediate: true
|
||
}
|
||
)
|
||
|
||
function quickProblemHandel(val: string) {
|
||
if (!props.log) {
|
||
inputValue.value = val
|
||
nextTick(() => {
|
||
quickInputRef.value?.focus()
|
||
})
|
||
}
|
||
}
|
||
|
||
function sendChatHandle(event: any) {
|
||
if (!event.ctrlKey) {
|
||
// 如果没有按下组合键ctrl，则会阻止默认事件
|
||
event.preventDefault()
|
||
if (!isDisabledChart.value && !loading.value) {
|
||
chatMessage()
|
||
}
|
||
} else {
|
||
// 如果同时按下ctrl+回车键，则会换行
|
||
inputValue.value += '\n'
|
||
}
|
||
}
|
||
const stopChat = (chat: chatType) => {
|
||
ChatManagement.stop(chat.id)
|
||
}
|
||
const startChat = (chat: chatType) => {
|
||
ChatManagement.write(chat.id)
|
||
}
|
||
/**
|
||
* 对话
|
||
*/
|
||
function getChartOpenId() {
|
||
loading.value = true
|
||
const obj = props.data
|
||
if (props.appId) {
|
||
applicationApi
|
||
.getChatOpen(props.appId)
|
||
.then((res) => {
|
||
chartOpenId.value = res.data
|
||
chatMessage()
|
||
})
|
||
.catch((res) => {
|
||
if (res.response.status === 403) {
|
||
application.asyncAppAuthentication(accessToken).then(() => {
|
||
getChartOpenId()
|
||
})
|
||
}
|
||
loading.value = false
|
||
})
|
||
} else {
|
||
applicationApi
|
||
.postChatOpen(obj)
|
||
.then((res) => {
|
||
chartOpenId.value = res.data
|
||
chatMessage()
|
||
})
|
||
.catch(() => {
|
||
loading.value = false
|
||
})
|
||
}
|
||
}
|
||
|
||
function chatMessage() {
|
||
loading.value = true
|
||
if (!chartOpenId.value) {
|
||
getChartOpenId()
|
||
} else {
|
||
const problem_text = inputValue.value
|
||
const id = randomId()
|
||
chatList.value.push({
|
||
id: id,
|
||
problem_text: problem_text,
|
||
answer_text: '',
|
||
buffer: [],
|
||
write_ed: false,
|
||
is_stop: false,
|
||
record_id: '',
|
||
vote_status: '-1'
|
||
})
|
||
inputValue.value = ''
|
||
nextTick(() => {
|
||
scrollDiv.value.setScrollTop(Number.MAX_SAFE_INTEGER)
|
||
})
|
||
|
||
applicationApi.postChatMessage(chartOpenId.value, problem_text).then((response) => {
|
||
const row = chatList.value.find((item) => item.id === id)
|
||
if (row) {
|
||
ChatManagement.addChatRecord(row, 50, loading)
|
||
ChatManagement.write(id)
|
||
const reader = response.body.getReader()
|
||
let tempResult = ''
|
||
/*eslint no-constant-condition: ["error", { "checkLoops": false }]*/
|
||
const write = ({ done, value }: { done: boolean; value: any }) => {
|
||
try {
|
||
if (done) {
|
||
ChatManagement.close(id)
|
||
return
|
||
}
|
||
|
||
const decoder = new TextDecoder('utf-8')
|
||
let str = decoder.decode(value, { stream: true })
|
||
// 这里解释一下 start 因为数据流返回流并不是按照后端chunk返回 我们希望得到的chunk是data:{xxx}\n\n 但是它获取到的可能是 data:{ -> xxx}\n\n 总而言之就是 fetch不能保证每个chunk都说以data:开始 \n\n结束
|
||
tempResult += str
|
||
if (tempResult.endsWith('\n\n')) {
|
||
str = tempResult
|
||
tempResult = ''
|
||
} else {
|
||
return reader.read().then(write)
|
||
}
|
||
// 这里解释一下 end
|
||
if (str && str.startsWith('data:')) {
|
||
const split = str.match(/data:.*}\n\n/g)
|
||
if (split) {
|
||
for (const index in split) {
|
||
const chunk = JSON?.parse(split[index].replace('data:', ''))
|
||
row.record_id = chunk.id
|
||
const content = chunk?.content
|
||
if (content) {
|
||
ChatManagement.append(id, content)
|
||
}
|
||
if (chunk.is_end) {
|
||
// 流处理成功 返回成功回调
|
||
return Promise.resolve()
|
||
}
|
||
}
|
||
}
|
||
}
|
||
} catch (e) {
|
||
console.log(e)
|
||
// console
|
||
}
|
||
return reader.read().then(write)
|
||
}
|
||
reader
|
||
.read()
|
||
.then(write)
|
||
.finally((ok: any) => {
|
||
ChatManagement.close(id)
|
||
})
|
||
.catch((e: any) => {
|
||
ChatManagement.close(id)
|
||
})
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
function regenerationChart(item: chatType) {
|
||
inputValue.value = item.problem_text
|
||
chatMessage()
|
||
}
|
||
|
||
/**
|
||
* 滚动条距离最上面的高度
|
||
*/
|
||
const scrollTop = ref(0)
|
||
|
||
const scorll = ref(true)
|
||
|
||
const handleScrollTop = ($event: any) => {
|
||
scrollTop.value = $event.scrollTop
|
||
if (
|
||
dialogScrollbar.value.scrollHeight - (scrollTop.value + scrollDiv.value.wrapRef.offsetHeight) <=
|
||
10
|
||
) {
|
||
scorll.value = true
|
||
} else {
|
||
scorll.value = false
|
||
}
|
||
}
|
||
|
||
const handleScroll = () => {
|
||
if (!props.log && scrollDiv.value) {
|
||
// 内部高度小于外部高度 就需要出滚动条
|
||
if (scrollDiv.value.wrapRef.offsetHeight < dialogScrollbar.value.scrollHeight) {
|
||
// 如果当前滚动条距离最下面的距离在 规定距离 滚动条就跟随
|
||
if (scorll.value) {
|
||
scrollDiv.value.setScrollTop(Number.MAX_SAFE_INTEGER)
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
watch(
|
||
chatList,
|
||
() => {
|
||
handleScroll()
|
||
},
|
||
{ deep: true, immediate: true }
|
||
)
|
||
</script>
|
||
<style lang="scss" scoped>
|
||
.ai-chat {
|
||
--padding-left: 40px;
|
||
height: 100%;
|
||
display: flex;
|
||
flex-direction: column;
|
||
box-sizing: border-box;
|
||
position: relative;
|
||
color: var(--app-text-color);
|
||
&.chart-log {
|
||
.ai-chat__content {
|
||
padding-bottom: 0;
|
||
}
|
||
}
|
||
&__content {
|
||
padding-top: 0;
|
||
padding-bottom: 96px;
|
||
box-sizing: border-box;
|
||
|
||
.avatar {
|
||
float: left;
|
||
}
|
||
.content {
|
||
padding-left: var(--padding-left);
|
||
:deep(ol) {
|
||
margin-left: 16px !important;
|
||
}
|
||
}
|
||
.text {
|
||
padding: 6px 0;
|
||
}
|
||
.problem-button {
|
||
width: 100%;
|
||
border: none;
|
||
border-radius: 8px;
|
||
background: var(--app-layout-bg-color);
|
||
height: 46px;
|
||
padding: 0 12px;
|
||
line-height: 46px;
|
||
box-sizing: border-box;
|
||
color: var(--el-text-color-regular);
|
||
-webkit-line-clamp: 1;
|
||
word-break: break-all;
|
||
&:hover {
|
||
background: var(--el-color-primary-light-9);
|
||
}
|
||
&.disabled {
|
||
&:hover {
|
||
background: var(--app-layout-bg-color);
|
||
}
|
||
}
|
||
:deep(.el-icon) {
|
||
color: var(--el-color-primary);
|
||
}
|
||
}
|
||
}
|
||
&__operate {
|
||
background: #f3f7f9;
|
||
position: absolute;
|
||
bottom: 0;
|
||
left: 0;
|
||
width: 100%;
|
||
box-sizing: border-box;
|
||
z-index: 10;
|
||
&:before {
|
||
background: linear-gradient(0deg, #f3f7f9 0%, rgba(243, 247, 249, 0) 100%);
|
||
content: '';
|
||
position: absolute;
|
||
width: 100%;
|
||
top: -16px;
|
||
left: 0;
|
||
height: 16px;
|
||
}
|
||
.operate-textarea {
|
||
box-shadow: 0px 6px 24px 0px rgba(31, 35, 41, 0.08);
|
||
background-color: #ffffff;
|
||
border-radius: 8px;
|
||
border: 1px solid #ffffff;
|
||
box-sizing: border-box;
|
||
|
||
&:has(.el-textarea__inner:focus) {
|
||
border: 1px solid var(--el-color-primary);
|
||
}
|
||
|
||
:deep(.el-textarea__inner) {
|
||
border-radius: 8px !important;
|
||
box-shadow: none;
|
||
resize: none;
|
||
padding: 12px 16px;
|
||
}
|
||
.operate {
|
||
padding: 6px 10px;
|
||
.sent-button {
|
||
max-height: none;
|
||
.el-icon {
|
||
font-size: 24px;
|
||
}
|
||
}
|
||
:deep(.el-loading-spinner) {
|
||
margin-top: -15px;
|
||
.circular {
|
||
width: 31px;
|
||
height: 31px;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
.dialog-card {
|
||
border: none;
|
||
}
|
||
}
|
||
</style>
|