♻️ refactor: refactor the text to image

lobehub · Oct 5, 2024 · f0d0f6b · f0d0f6b
1 parent 0c32283
commit f0d0f6b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/services/_url.ts b/src/services/_url.ts
@@ -36,7 +36,7 @@ export const API_ENDPOINTS = mapWithBasePath({
   trace: '/webapi/trace',
 
   // image
-  images: '/webapi/text-to-image/openai',
+  images: (provider: string) => `/webapi/text-to-image/${provider}`,
 
   // STT
   stt: '/webapi/stt/openai',

diff --git a/src/services/textToImage.ts b/src/services/textToImage.ts
@@ -12,12 +12,14 @@ class ImageGenerationService {
   async generateImage(params: Omit<OpenAIImagePayload, 'model' | 'n'>, options?: FetchOptions) {
     const payload: OpenAIImagePayload = { ...params, model: 'dall-e-3', n: 1 };
 
+    const provider = ModelProvider.OpenAI;
+
     const headers = await createHeaderWithAuth({
       headers: { 'Content-Type': 'application/json' },
-      provider: ModelProvider.OpenAI,
+      provider,
     });
 
-    const res = await fetch(API_ENDPOINTS.images, {
+    const res = await fetch(API_ENDPOINTS.images(provider), {
       body: JSON.stringify(payload),
       headers: headers,
       method: 'POST',