UNPKG

@webeleon/nestjs-openai

Version:
23 lines (22 loc) 934 B
/// <reference types="node" /> import { Message } from './message.dto'; import { OpenAIClientProvider } from './open-ai-client.provider'; import OpenAIApi from 'openai'; import { Role } from './roles.enum'; import { GenerateImageInput } from './generate-image-input.interface'; export declare class OpenAIService { private readonly config; private readonly openaiProvider; openai: OpenAIApi; constructor(config: any, openaiProvider: OpenAIClientProvider); chat({ prompt, userId, history, role, temperature, numberOfCompletions, }: { prompt: string; userId: string; history?: Message[]; role?: Role; temperature?: number; numberOfCompletions?: number; }): Promise<string[]>; generateImagePngBuffers({ prompt, n, size, user, }: GenerateImageInput): Promise<Buffer[]>; generateImageUrls({ prompt, n, size, user, }: GenerateImageInput): Promise<string[]>; }