@maximai/maxim-js/src/lib/logger/components/retrieval.js

Maxim AI JS SDK. Visit https://getmaxim.ai for more info.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.Retrieval = void 0;
const utils_1 = require("../utils");
const base_1 = require("./base");
const types_1 = require("./types");
class Retrieval extends base_1.EvaluatableBaseContainer {
    constructor(config, writer) {
        super(types_1.Entity.RETRIEVAL, config, writer);
    }
    input(query) {
        this.commit("update", { input: query });
    }
    static input_(writer, id, query) {
        base_1.EvaluatableBaseContainer.commit_(writer, types_1.Entity.RETRIEVAL, id, "update", { input: query });
    }
    output(docs) {
        let finalDocs = docs;
        if (typeof docs === "string") {
            finalDocs = [docs];
        }
        this.commit("end", { docs: finalDocs, endTimestamp: (0, utils_1.utcNow)() });
    }
    static output_(writer, id, docs) {
        let finalDocs = docs;
        if (typeof docs === "string") {
            finalDocs = [docs];
        }
        base_1.EvaluatableBaseContainer.commit_(writer, types_1.Entity.RETRIEVAL, id, "end", { docs: finalDocs, endTimestamp: (0, utils_1.utcNow)() });
    }
    static end_(writer, id, data) {
        base_1.EvaluatableBaseContainer.end_(writer, types_1.Entity.RETRIEVAL, id, data);
    }
    static addTag_(writer, id, key, value) {
        base_1.EvaluatableBaseContainer.addTag_(writer, types_1.Entity.RETRIEVAL, id, key, value);
    }
}
exports.Retrieval = Retrieval;
//# sourceMappingURL=retrieval.js.map