/**
 * Chat API | 对话 API
 */

import { request } from '@/utils'

// 对话补全
export const chatCompletions = (data) =>
  request({
    url: `/chat/completions`,
    method: 'post',
    data
  })

// 流式对话补全
export const streamChatCompletions = async function* (data, signal, options = {}) {
  const text = data?.messages?.at?.(-1)?.content || data?.goal || ''
  const systemPrompt = data?.messages?.find?.((message) => message?.role === 'system')?.content || ''
  const response = await fetch('/api/prompt/polish', {
    method: 'POST',
    headers: {
      'Content-Type': 'application/json'
    },
    body: JSON.stringify({
      text: typeof text === 'string' ? text : JSON.stringify(text),
      system_prompt: systemPrompt,
      mode: 'chat',
      target_language: 'keep'
    }),
    signal
  })

  if (!response.ok) {
    const error = await response.json().catch(() => ({}))
    throw new Error(error?.detail || error?.message || '提示词助手请求失败')
  }

  const json = await response.json()
  yield json.text || ''
}