"use strict";
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());
    });
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.CHComponentsTLManager = void 0;
const tiktoken_instance_1 = require("../../utils/tiktoken-instance");
class CHComponentsTLManager {
    static calculateTLForCH(chistory) {
        return __awaiter(this, void 0, void 0, function* () {
            const CHString = chistory.map((message) => message.content).join(" ");
            const CHStringTokenLength = yield (0, tiktoken_instance_1.getTokenLengthByInput)(CHString);
            // Addn token length due to tokens (<|user|>, etc) prepended on actual conversation entries
            const CHArryTokenLength = chistory.length * 3;
            return CHStringTokenLength + CHArryTokenLength;
        });
    }
    constructor(conversationHistoryController) {
        this.conversationHistoryController = conversationHistoryController;
    }
    getCHComponentsTL() {
        return __awaiter(this, void 0, void 0, function* () {
            const conversationHistory = yield this.conversationHistoryController.getConversationHistory();
            const conversationHistoryTokenLength = yield CHComponentsTLManager.calculateTLForCH(conversationHistory);
            return {
                conversationHistory,
                conversationHistoryTokenLength,
            };
        });
    }
}
exports.CHComponentsTLManager = CHComponentsTLManager;
