From f0d0f6b3c52aece23b53e6e6320eb9c37b4c1136 Mon Sep 17 00:00:00 2001
From: arvinxx <arvinx@foxmail.com>
Date: Sun, 6 Oct 2024 02:04:54 +0900
Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20refactor:=20refactor=20the?=
 =?UTF-8?q?=20text=20to=20image?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/services/_url.ts        | 2 +-
 src/services/textToImage.ts | 6 ++++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/services/_url.ts b/src/services/_url.ts
index d8d0d56e058a..f01e23290e18 100644
--- 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
index f60eb9afc4be..3de461a41052 100644
--- 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',