@maximai/maxim-js
Version:
Maxim AI JS SDK. Visit https://getmaxim.ai for more info.
111 lines • 3.91 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.Span = void 0;
const base_1 = require("./base");
const error_1 = require("./error");
const generation_1 = require("./generation");
const retrieval_1 = require("./retrieval");
const toolCall_1 = require("./toolCall");
const types_1 = require("./types");
class Span extends base_1.EventEmittingBaseContainer {
constructor(config, writer) {
super(types_1.Entity.SPAN, config, writer);
this.commit("create");
}
generation(config) {
const generation = new generation_1.Generation(config, this.writer);
this.commit("add-generation", {
id: config.id,
messages: JSON.parse(JSON.stringify(config.messages)),
...generation.data(),
});
return generation;
}
static generation_(writer, id, config) {
const generation = new generation_1.Generation(config, writer);
base_1.EventEmittingBaseContainer.commit_(writer, types_1.Entity.SPAN, id, "add-generation", {
id: config.id,
messages: JSON.parse(JSON.stringify(config.messages)),
...generation.data(),
});
return generation;
}
span(config) {
const span = new Span(config, this.writer);
this.commit("add-span", {
id: config.id,
...span.data(),
});
return span;
}
static span_(writer, id, config) {
const span = new Span(config, writer);
base_1.EventEmittingBaseContainer.commit_(writer, types_1.Entity.SPAN, id, "add-span", {
id: config.id,
...span.data(),
});
return span;
}
error(config) {
const error = new error_1.Error(config, this.writer);
this.commit("add-error", {
id: config.id,
...error.data(),
});
return error;
}
static error_(writer, id, config) {
const error = new error_1.Error(config, writer);
base_1.EventEmittingBaseContainer.commit_(writer, types_1.Entity.SPAN, id, "add-error", {
id: config.id,
...error.data(),
});
return error;
}
retrieval(config) {
const retrieval = new retrieval_1.Retrieval(config, this.writer);
this.commit("add-retrieval", {
id: config.id,
...retrieval.data(),
});
return retrieval;
}
static retrieval_(writer, id, config) {
const retrieval = new retrieval_1.Retrieval(config, writer);
base_1.EventEmittingBaseContainer.commit_(writer, types_1.Entity.SPAN, id, "add-retrieval", {
id: config.id,
...retrieval.data(),
});
return retrieval;
}
toolCall(config) {
const toolCall = new toolCall_1.ToolCall(config, this.writer);
this.commit("add-tool-call", {
id: config.id,
...toolCall.data(),
});
return toolCall;
}
static toolCall_(writer, id, config) {
const toolCall = new toolCall_1.ToolCall(config, writer);
base_1.EventEmittingBaseContainer.commit_(writer, types_1.Entity.SPAN, id, "add-tool-call", {
id: config.id,
...toolCall.data(),
});
return toolCall;
}
addAttachment(attachment) {
this.commit("upload-attachment", attachment);
}
static addAttachment_(writer, id, attachment) {
base_1.EventEmittingBaseContainer.commit_(writer, types_1.Entity.SPAN, id, "upload-attachment", attachment);
}
static end_(writer, id, data) {
base_1.EventEmittingBaseContainer.end_(writer, types_1.Entity.SPAN, id, data);
}
static addTag_(writer, id, key, value) {
base_1.EventEmittingBaseContainer.addTag_(writer, types_1.Entity.SPAN, id, key, value);
}
}
exports.Span = Span;
//# sourceMappingURL=span.js.map