feat: generate images

2024-11-22 10:00:34 +08:00
parent 84e0c46711
commit f24aef300b
18 changed files with 725 additions and 114 deletions
@@ -2,6 +2,8 @@ import { request } from '@umijs/max';

 export const CHAT_API = '/v1-openai/chat/completions';

+export const CREAT_IMAGE_API = '/v1-openai/images/generations';
+
 export const EMBEDDING_API = '/v1-openai/embeddings';

 export const OPENAI_MODELS = '/v1-openai/models';
@@ -51,3 +53,19 @@ export const handleEmbedding = async (
    cancelToken: options?.cancelToken
  });
 };
+
+export const createImages = async (
+  params: {
+    model: string;
+    prompt: string;
+    n: number;
+    size: string;
+  },
+  options?: any
+) => {
+  return request(`${CREAT_IMAGE_API}`, {
+    method: 'POST',
+    data: params,
+    cancelToken: options?.cancelToken
+  });
+};