[代码优化]AI: 写作添加注释,增加可读性,调整代码,方便后续调整

This commit is contained in:
hhhero
2024-07-09 00:44:27 +08:00
parent 6224602152
commit 5730707be9
8 changed files with 154 additions and 174 deletions

View File

@@ -0,0 +1,67 @@
<template>
<div class="h-[calc(100vh-var(--top-tool-height)-var(--app-footer-height)-40px)] -m-5 flex">
<Left :is-writing="isWriting" class="h-full" @submit="submit" @example="handleExampleClick" />
<!-- TODO @hhhero顶部应该有个预览的 header -->
<!-- TODO @hhhero整个 Right 组件的框没铺满的感觉 -->
<Right
:is-writing="isWriting"
@stop-stream="stopStream"
ref="rightRef"
class="flex-grow"
v-model:content="writeResult"
/>
</div>
</template>
<script setup lang="ts">
import Left from '../components/Left.vue'
import Right from '../components/Right.vue'
import * as WriteApi from '@/api/ai/writer'
import { WriteExampleDataJson } from '@/views/ai/utils/utils'
const message = useMessage()
const writeResult = ref('') // 写作结果
const isWriting = ref(false) // 是否正在写作中
const abortController = ref<AbortController>() // // 写作进行中 abort 控制器(控制 stream 写作)
/** 停止 stream 生成 */
const stopStream = () => {
abortController.value?.abort()
isWriting.value = false
}
/** 执行写作 */
const rightRef = ref<InstanceType<typeof Right>>()
const submit = (data) => {
abortController.value = new AbortController()
writeResult.value = ''
isWriting.value = true
WriteApi.writeStream({
data,
onMessage: async (res) => {
const { code, data, msg } = JSON.parse(res.data)
if (code !== 0) {
message.alert(`写作异常! ${msg}`)
stopStream()
return
}
writeResult.value = writeResult.value + data
nextTick(() => {
rightRef.value?.scrollToBottom()
})
},
ctrl: abortController.value,
onClose: stopStream,
onError: (...err) => {
console.error('写作异常', ...err)
stopStream()
}
})
}
/** 点击示例触发 */
const handleExampleClick = (type: keyof typeof WriteExampleDataJson) => {
writeResult.value = WriteExampleDataJson[type].data
}
</script>