@difizen/magent-chat
Version:
117 lines • 6.98 kB
JavaScript
function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
var _dec, _dec2, _dec3, _dec4, _dec5, _dec6, _dec7, _class, _class2, _descriptor, _descriptor2, _descriptor3, _descriptor4, _descriptor5, _descriptor6, _dec8, _dec9, _class3, _class4, _descriptor7;
function _initializerDefineProperty(target, property, descriptor, context) { if (!descriptor) return; Object.defineProperty(target, property, { enumerable: descriptor.enumerable, configurable: descriptor.configurable, writable: descriptor.writable, value: descriptor.initializer ? descriptor.initializer.call(context) : void 0 }); }
function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, _toPropertyKey(descriptor.key), descriptor); } }
function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); Object.defineProperty(Constructor, "prototype", { writable: false }); return Constructor; }
function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : String(i); }
function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
function _applyDecoratedDescriptor(target, property, decorators, descriptor, context) { var desc = {}; Object.keys(descriptor).forEach(function (key) { desc[key] = descriptor[key]; }); desc.enumerable = !!desc.enumerable; desc.configurable = !!desc.configurable; if ('value' in desc || desc.initializer) { desc.writable = true; } desc = decorators.slice().reverse().reduce(function (desc, decorator) { return decorator(target, property, desc) || desc; }, desc); if (context && desc.initializer !== void 0) { desc.value = desc.initializer ? desc.initializer.call(context) : void 0; desc.initializer = undefined; } if (desc.initializer === void 0) { Object.defineProperty(target, property, desc); desc = null; } return desc; }
function _initializerWarningHelper(descriptor, context) { throw new Error('Decorating class property failed. Please ensure that ' + 'transform-class-properties is enabled and runs after the decorators transform.'); }
import { AutoFactoryOption } from '@difizen/magent-core';
import { autoFactory, toAutoFactory } from '@difizen/magent-core';
import { inject, prop, singleton } from '@difizen/mana-app';
import dayjs from 'dayjs';
export var DefaultChatMessageSummary = (_dec = autoFactory(), _dec2 = prop(), _dec3 = prop(), _dec4 = prop(), _dec5 = prop(), _dec6 = prop(), _dec7 = prop(), _dec(_class = (_class2 = /*#__PURE__*/function () {
function DefaultChatMessageSummary(option) {
_classCallCheck(this, DefaultChatMessageSummary);
/**
* token usage for completion
*/
_initializerDefineProperty(this, "completionTokens", _descriptor, this);
/**
* token usage for prompt input
*/
_initializerDefineProperty(this, "promptTokens", _descriptor2, this);
/**
* total token usage
*/
_initializerDefineProperty(this, "totalTokens", _descriptor3, this);
/**
* the time chat started
*/
_initializerDefineProperty(this, "startTime", _descriptor4, this);
/**
* the time chat ended
*/
_initializerDefineProperty(this, "endTime", _descriptor5, this);
/**
* the time chat ended
*/
_initializerDefineProperty(this, "responseTime", _descriptor6, this);
this.fromMeta(option);
}
DefaultChatMessageSummary = inject(AutoFactoryOption)(DefaultChatMessageSummary, undefined, 0) || DefaultChatMessageSummary;
_createClass(DefaultChatMessageSummary, [{
key: "fromMeta",
value: function fromMeta(option) {
if (option.completion_tokens !== undefined) {
this.completionTokens = option.completion_tokens;
}
if (option.prompt_tokens !== undefined) {
this.promptTokens = option.prompt_tokens;
}
if (option.total_tokens !== undefined) {
this.totalTokens = option.total_tokens;
}
if (option.start_time !== undefined) {
this.startTime = dayjs(option.start_time);
}
if (option.end_time !== undefined) {
this.endTime = dayjs(option.end_time);
}
if (option.response_time !== undefined) {
this.responseTime = option.response_time;
}
}
}]);
return DefaultChatMessageSummary;
}(), (_descriptor = _applyDecoratedDescriptor(_class2.prototype, "completionTokens", [_dec2], {
configurable: true,
enumerable: true,
writable: true,
initializer: null
}), _descriptor2 = _applyDecoratedDescriptor(_class2.prototype, "promptTokens", [_dec3], {
configurable: true,
enumerable: true,
writable: true,
initializer: null
}), _descriptor3 = _applyDecoratedDescriptor(_class2.prototype, "totalTokens", [_dec4], {
configurable: true,
enumerable: true,
writable: true,
initializer: null
}), _descriptor4 = _applyDecoratedDescriptor(_class2.prototype, "startTime", [_dec5], {
configurable: true,
enumerable: true,
writable: true,
initializer: null
}), _descriptor5 = _applyDecoratedDescriptor(_class2.prototype, "endTime", [_dec6], {
configurable: true,
enumerable: true,
writable: true,
initializer: null
}), _descriptor6 = _applyDecoratedDescriptor(_class2.prototype, "responseTime", [_dec7], {
configurable: true,
enumerable: true,
writable: true,
initializer: null
})), _class2)) || _class);
export var ChatMessageSummaryProvider = (_dec8 = singleton(), _dec9 = inject(toAutoFactory(DefaultChatMessageSummary)), _dec8(_class3 = (_class4 = /*#__PURE__*/function () {
function ChatMessageSummaryProvider() {
_classCallCheck(this, ChatMessageSummaryProvider);
_initializerDefineProperty(this, "tokenSummaryFactory", _descriptor7, this);
}
_createClass(ChatMessageSummaryProvider, [{
key: "create",
value: function create(opt) {
return this.tokenSummaryFactory(opt);
}
}]);
return ChatMessageSummaryProvider;
}(), (_descriptor7 = _applyDecoratedDescriptor(_class4.prototype, "tokenSummaryFactory", [_dec9], {
configurable: true,
enumerable: true,
writable: true,
initializer: null
})), _class4)) || _class3);