@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
text/typescript
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();