import request from '@/utils/request';
import type { AxiosRequestConfig } from 'axios';
import type { ApiResult } from '@/api';
import type {
  ChatMessageParam,
  ChatMessageStopParam
} from '@/api/oa/chatgpt/model';

/**
 * 发送聊天消息
 */
export async function sendMessage(
  data: ChatMessageParam,
  config?: AxiosRequestConfig
) {
  const res = await request.post<ApiResult<unknown>>(
    '/ai/chat/message',
    data,
    config
  );
  if (res.data.code === 0) {
    return res.data.data ?? res.data.message;
  }
  return Promise.reject(new Error(res.data.message));
}

/**
 * 停止AI生成
 */
export async function stopMessage(data: ChatMessageStopParam) {
  const res = await request.post<ApiResult<unknown>>('/ai/messageStop', data);
  if (res.data.code === 0) {
    return res.data.data ?? res.data.message;
  }
  return Promise.reject(new Error(res.data.message || '停止失败'));
}