ipms-sjy-ui/src/api/ai/writer/index.ts

import { fetchEventSource } from '@microsoft/fetch-event-source'

import { getAccessToken } from '@/utils/auth'
import { config } from '@/config/axios/config'

// TODO @hhhero：可以改成 WriteVO 哈，主要是保持一致
export interface WriteParams {
  // TODO @hhhero：注释。每个属性的后面哈。会更简洁一点
  /**
   * 1:撰写 2:回复
   */
  type: 1 | 2
  /**
   * 写作内容提示 1。撰写 2回复
   */
  prompt: string
  /**
   *  原文
   */
  originalContent: string
  /**
   * 长度
   */
  length: number
  /**
   * 格式
   */
  format: number
  /**
   * 语气
   */
  tone: number
  /**
   * 语言
   */
  language: number
}

export const writeStream = ({
  data,
  onClose,
  onMessage,
  onError,
  ctrl
}: {
  data: WriteParams
  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
  })
}