UNPKG

@lobehub/chat

Version:

Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.

42 lines (32 loc) 1.06 kB
import { ModelProvider } from '@/libs/model-runtime'; import { createHeaderWithAuth } from '@/services/_auth'; import { OpenAIImagePayload } from '@/types/openai/image'; import { API_ENDPOINTS } from './_url'; interface FetchOptions { signal?: AbortSignal | undefined; } class ImageGenerationService { generateImage = async ( 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, }); const res = await fetch(API_ENDPOINTS.images(provider), { body: JSON.stringify(payload), headers: headers, method: 'POST', signal: options?.signal, }); if (!res.ok) { throw await res.json(); } const urls = await res.json(); return urls[0] as string; }; } export const imageGenerationService = new ImageGenerationService();