tprompter/dist/llm/OpenAIService.js

```bash $ ask anything ```

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); }
};
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
import { Service } from 'typedi';
import { AppConfig } from '../config/AppConfig.js';
import OpenAI from 'openai';
import { MissconfigurationError } from '../utils/errors/MisconfigurationError.js';
import { InjectLogger } from '../logger/logger.decorator.js';
let OpenAIService = class OpenAIService {
    constructor(config, logger) {
        this.config = config;
        this.logger = logger;
    }
    get client() {
        if (!this._client) {
            const { openAIApiKey } = this.config.getConfig();
            if (!openAIApiKey) {
                throw new MissconfigurationError('openAIApiKey', 'OpenAI API key is missing');
            }
            this._client = new OpenAI({
                apiKey: openAIApiKey,
            });
        }
        return this._client;
    }
    getCompletion(userMessages, options) {
        return __awaiter(this, void 0, void 0, function* () {
            var _a, _b, _c;
            const actualModel = options.model;
            this.logger.debug('Getting completion from OpenAI', { actualModel });
            const messages = [];
            if ((_a = options.developerMessages) === null || _a === void 0 ? void 0 : _a.length) {
                const roleName = actualModel.startsWith('gpt-') ? 'system' : 'developer';
                messages.push(...options.developerMessages.map((content) => ({ role: roleName, content })));
            }
            if (Array.isArray(userMessages)) {
                messages.push(...userMessages.map((content) => ({ role: 'user', content })));
            }
            else {
                messages.push({ role: 'user', content: userMessages });
            }
            const completion = yield this.client.chat.completions.create({
                model: actualModel,
                messages,
            });
            this.logger.debug('Got completion from OpenAI', {
                message: completion.choices[0].message,
                promptUsage: (_b = completion.usage) === null || _b === void 0 ? void 0 : _b.prompt_tokens,
                completionUsage: (_c = completion.usage) === null || _c === void 0 ? void 0 : _c.completion_tokens,
            });
            return completion.choices[0].message.content;
        });
    }
    getSupportedModels() {
        return __awaiter(this, void 0, void 0, function* () {
            return ['gpt-4o-mini', 'gpt-4o', 'gpt-4o-large', 'o1', 'o1-mini', 'o3-mini'];
        });
    }
};
OpenAIService = __decorate([
    Service(),
    __param(1, InjectLogger()),
    __metadata("design:paramtypes", [AppConfig, Object])
], OpenAIService);
export { OpenAIService };