init

apps/web-antd/src/api/ai/image/index.ts

@@ -0,0 +1,115 @@
+import type { PageParam, PageResult } from '@vben/request';
+
+import { requestClient } from '#/api/request';
+
+export namespace AiImageApi {
+  /** 绘图 */
+  export interface Image {
+    id: number; // 编号
+    userId: number;
+    platform: string; // 平台
+    model: string; // 模型
+    prompt: string; // 提示词
+    width: number; // 图片宽度
+    height: number; // 图片高度
+    status: number; // 状态
+    publicStatus: boolean; // 公开状态
+    picUrl: string; // 任务地址
+    errorMessage: string; // 错误信息
+    options: any; // 配置 Map<string, string>
+    taskId: number; // 任务编号
+    buttons: ImageMidjourneyButtons[]; // mj 操作按钮
+    createTime: Date; // 创建时间
+    finishTime: Date; // 完成时间
+  }
+
+  export interface ImageMidjourneyButtons {
+    customId: string; // MJ::JOB::upsample::1::85a4b4c1-8835-46c5-a15c-aea34fad1862 动作标识
+    emoji: string; // 图标 emoji
+    label: string; // Make Variations 文本
+    style: number; // 样式: 2（Primary）、3（Green）
+  }
+
+  export interface ImageDrawReqVO {
+    prompt: string; // 提示词
+    modelId: number; // 模型
+    style: string; // 图像生成的风格
+    width: string; // 图片宽度
+    height: string; // 图片高度
+    options: object; // 绘制参数，Map<String, String>
+  }
+
+  export interface ImageMidjourneyImagineReqVO {
+    prompt: string; // 提示词
+    modelId: number; // 模型
+    base64Array?: string[]; // size不能为空
+    width: string; // 图片宽度
+    height: string; // 图片高度
+    version: string; // 版本
+  }
+
+  export interface ImageMidjourneyAction {
+    id: number; // 图片编号
+    customId: string; // MJ::JOB::upsample::1::85a4b4c1-8835-46c5-a15c-aea34fad1862 动作标识
+  }
+}
+
+/** 获取【我的】绘图分页 */
+export function getImagePageMy(params: PageParam) {
+  return requestClient.get<PageResult<AiImageApi.Image>>('/ai/image/my-page', {
+    params,
+  });
+}
+
+/** 获取【我的】绘图记录 */
+export function getImageMy(id: number) {
+  return requestClient.get<AiImageApi.Image>(`/ai/image/get-my?id=${id}`);
+}
+
+/** 获取【我的】绘图记录列表 */
+export function getImageListMyByIds(ids: number[]) {
+  return requestClient.get<AiImageApi.Image[]>(`/ai/image/my-list-by-ids`, {
+    params: { ids: ids.join(',') },
+  });
+}
+
+/** 生成图片 */
+export function drawImage(data: AiImageApi.ImageDrawReqVO) {
+  return requestClient.post(`/ai/image/draw`, data);
+}
+
+/** 删除【我的】绘画记录 */
+export function deleteImageMy(id: number) {
+  return requestClient.delete(`/ai/image/delete-my?id=${id}`);
+}
+
+/** ================ midjourney 专属 ================ */
+
+/** 【Midjourney】生成图片 */
+export function midjourneyImagine(
+  data: AiImageApi.ImageMidjourneyImagineReqVO,
+) {
+  return requestClient.post(`/ai/image/midjourney/imagine`, data);
+}
+
+/** 【Midjourney】Action 操作（二次生成图片） */
+export function midjourneyAction(data: AiImageApi.ImageMidjourneyAction) {
+  return requestClient.post(`/ai/image/midjourney/action`, data);
+}
+
+/** ================ 绘图管理 ================ */
+
+/** 查询绘画分页 */
+export function getImagePage(params: any) {
+  return requestClient.get<AiImageApi.Image[]>(`/ai/image/page`, { params });
+}
+
+/** 更新绘画发布状态 */
+export function updateImage(data: any) {
+  return requestClient.put(`/ai/image/update`, data);
+}
+
+/** 删除绘画 */
+export function deleteImage(id: number) {
+  return requestClient.delete(`/ai/image/delete?id=${id}`);
+}