AIClass/src/api/questions.js

import axios from "@/utils/request";

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

// 数字人对话框
export function CreateDialogue (data){
  return axios({
    url: "bjdxWeb/ai/create-dialogue",
    method: 'post',
    data 
  })
}


// 发送 Stream 消息
// 为什么不用 axios 呢？因为它不支持 SSE 调用
export async function sendChatMessageStream (
    conversationId,
    content,
    contentAnswer = undefined,
    ctrl,
    enableContext,
    onMessage,
    onError,
    onClose
) {
  // 获取 token
  const token = localStorage.getItem('token');
  return fetchEventSource(import.meta.env.VITE_BASE_URL + `/bjdxWeb/ai/dialogue-send-stream`, {
    method: 'post',
    headers: {
      'Content-Type': 'application/json',
      // 添加 token 到请求头
      ...(token ? { 'Authorization': `Bearer ${token}` } : {})
      // Authorization: `Bearer b55bd67fba3e4bb49608168f078fde63`
    },
    openWhenHidden: true,
    body: JSON.stringify({
      conversationId,
      content,
      useContext: enableContext,
      contentAnswer
    }),
    onmessage: onMessage,
    onerror: onError,
    onclose: onClose,
    signal: ctrl.signal
  })
}


//  AI 图像生成状态的枚举
export const AiImageStatusEnum = {
  IN_PROGRESS: 10, // 进行中
  SUCCESS: 20, // 已完成
  FAIL: 30 // 已失败
}

// 生成图片
export function CreatePainting (data){
  return axios({
    url: "bjdxWeb/ai/create-painting",
    method: 'post',
    data
  })
}
// 绘画
export async function PaintingGetMys(ids) {
  return await axios({
    url: "bjdxWeb/ai/painting-get-mys",
    method: 'get',
    data: {ids: ids.join(',')}
  })
}