@webeleon/nestjs-openai
Version:
An openAi API client for nestjs
71 lines • 3.06 kB
JavaScript
;
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
return c > 3 && r && Object.defineProperty(target, key, r), r;
};
var __metadata = (this && this.__metadata) || function (k, v) {
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
};
var __param = (this && this.__param) || function (paramIndex, decorator) {
return function (target, key) { decorator(target, key, paramIndex); }
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.OpenAIService = void 0;
const common_1 = require("@nestjs/common");
const open_ai_client_provider_1 = require("./open-ai-client.provider");
const options_interface_1 = require("./options.interface");
const roles_enum_1 = require("./roles.enum");
let OpenAIService = class OpenAIService {
constructor(config, openaiProvider) {
this.config = config;
this.openaiProvider = openaiProvider;
this.openai = openaiProvider.openai;
}
async chat({ prompt, userId, history = [], role = roles_enum_1.Role.USER, temperature = 1, numberOfCompletions = 1, }) {
const messages = [
...history,
{
role,
content: prompt,
name: userId,
},
];
const completion = await this.openai.chat.completions.create({
model: this.config.model,
user: userId,
messages,
temperature,
n: numberOfCompletions,
});
return completion.choices.map((choice) => choice.message.content);
}
async generateImagePngBuffers({ prompt, n = 1, size = '512x512', user, }) {
const { data } = await this.openai.images.generate({
prompt,
n,
size,
user,
response_format: 'b64_json',
});
return data.map((img) => Buffer.from(img.b64_json, 'base64'));
}
async generateImageUrls({ prompt, n = 1, size = '512x512', user, }) {
const { data } = await this.openai.images.generate({
prompt,
n,
size,
user,
response_format: 'url',
});
return data.map((img) => img.url);
}
};
exports.OpenAIService = OpenAIService;
exports.OpenAIService = OpenAIService = __decorate([
(0, common_1.Injectable)(),
__param(0, (0, common_1.Inject)(options_interface_1.CONFIG_OPTIONS)),
__metadata("design:paramtypes", [Object, open_ai_client_provider_1.OpenAIClientProvider])
], OpenAIService);
//# sourceMappingURL=open-ai.service.js.map