46 lines
1.3 KiB
TypeScript
46 lines
1.3 KiB
TypeScript
import { fetchEventSource } from '@microsoft/fetch-event-source'
|
||
|
||
import { getAccessToken } from '@/utils/auth'
|
||
import { config } from '@/config/axios/config'
|
||
import { AiWriteTypeEnum } from '@/views/ai/utils/constants'
|
||
|
||
export interface WriteVO {
|
||
type: AiWriteTypeEnum.WRITING | AiWriteTypeEnum.REPLY // 1:撰写 2:回复
|
||
prompt: string // 写作内容提示 1。撰写 2回复
|
||
originalContent: string // 原文
|
||
length: number // 长度
|
||
format: number // 格式
|
||
tone: number // 语气
|
||
language: number // 语言
|
||
}
|
||
|
||
// TODO @hhero：搞成 WriteApi，类似 ConversationApi 一样。这样更有类的概念，后续引入某个 Api，然后调用它的方法就可以了。
|
||
export const writeStream = ({
|
||
data,
|
||
onClose,
|
||
onMessage,
|
||
onError,
|
||
ctrl
|
||
}: {
|
||
data: WriteVO
|
||
onMessage?: (res: any) => void
|
||
onError?: (...args: any[]) => void
|
||
onClose?: (...args: any[]) => void
|
||
ctrl: AbortController
|
||
}) => {
|
||
const token = getAccessToken()
|
||
return fetchEventSource(`${config.base_url}/ai/write/generate-stream`, {
|
||
method: 'post',
|
||
headers: {
|
||
'Content-Type': 'application/json',
|
||
Authorization: `Bearer ${token}`
|
||
},
|
||
openWhenHidden: true,
|
||
body: JSON.stringify(data),
|
||
onmessage: onMessage,
|
||
onerror: onError,
|
||
onclose: onClose,
|
||
signal: ctrl.signal
|
||
})
|
||
}
|