import request from '@/utils/request'

// 获取可用的AI模型列表
export function getAiModels(type) {
  return request({
    url: '/ai/models',
    method: 'GET',
    data: { type }
  })
}

// 获取首页展示的模型列表（最多返回指定数量）
export function getHomeAiModels(limit = 4) {
  return request({
    url: '/ai/models/home',
    method: 'GET',
    data: { limit }
  })
}

// 获取模型详情
export function getAiModel(id) {
  return request({
    url: `/ai/models/${id}`,
    method: 'GET'
  })
}

// 根据编码获取模型详情
export function getAiModelByCode(code, options = {}) {
  return request({
    url: `/ai/models/code/${code}`,
    method: 'GET',
    ...options
  })
}

// 创建AI任务
export function createAiTask(data) {
  return request({
    url: '/ai/tasks',
    method: 'POST',
    data
  })
}

// 获取我的任务列表
export function getMyAiTasks(params, options = {}) {
  return request({
    url: '/ai/tasks',
    method: 'GET',
    data: params,
    ...options
  })
}

// 获取任务详情
export function getAiTask(id) {
  return request({
    url: `/ai/tasks/${id}`,
    method: 'GET'
  })
}

// 根据任务编号获取任务详情
export function getAiTaskByNo(taskNo) {
  return request({
    url: `/ai/tasks/no/${taskNo}`,
    method: 'GET'
  })
}

// 取消任务
export function cancelAiTask(id) {
  return request({
    url: `/ai/tasks/${id}/cancel`,
    method: 'POST'
  })
}

// 删除任务
export function deleteAiTask(id) {
  return request({
    url: `/ai/tasks/${id}`,
    method: 'DELETE'
  })
}

// 获取任务详情（公开接口，不需要认证）
export function getAiTaskByIdPublic(id) {
  return request({
    url: `/ai/tasks/public/${id}`,
    method: 'GET'
  })
}