@webeleon/nestjs-openai
Version:
An openAi API client for nestjs
23 lines (22 loc) • 934 B
TypeScript
/// <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[]>;
}