UNPKG

@difizen/libro-ai-native

Version:
344 lines 27 kB
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, _class, _class2, _descriptor, _descriptor2, _descriptor3, _descriptor4, _descriptor5; function _regeneratorRuntime() { "use strict"; /*! regenerator-runtime -- Copyright (c) 2014-present, Facebook, Inc. -- license (MIT): https://github.com/facebook/regenerator/blob/main/LICENSE */ _regeneratorRuntime = function _regeneratorRuntime() { return e; }; var t, e = {}, r = Object.prototype, n = r.hasOwnProperty, o = Object.defineProperty || function (t, e, r) { t[e] = r.value; }, i = "function" == typeof Symbol ? Symbol : {}, a = i.iterator || "@@iterator", c = i.asyncIterator || "@@asyncIterator", u = i.toStringTag || "@@toStringTag"; function define(t, e, r) { return Object.defineProperty(t, e, { value: r, enumerable: !0, configurable: !0, writable: !0 }), t[e]; } try { define({}, ""); } catch (t) { define = function define(t, e, r) { return t[e] = r; }; } function wrap(t, e, r, n) { var i = e && e.prototype instanceof Generator ? e : Generator, a = Object.create(i.prototype), c = new Context(n || []); return o(a, "_invoke", { value: makeInvokeMethod(t, r, c) }), a; } function tryCatch(t, e, r) { try { return { type: "normal", arg: t.call(e, r) }; } catch (t) { return { type: "throw", arg: t }; } } e.wrap = wrap; var h = "suspendedStart", l = "suspendedYield", f = "executing", s = "completed", y = {}; function Generator() {} function GeneratorFunction() {} function GeneratorFunctionPrototype() {} var p = {}; define(p, a, function () { return this; }); var d = Object.getPrototypeOf, v = d && d(d(values([]))); v && v !== r && n.call(v, a) && (p = v); var g = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(p); function defineIteratorMethods(t) { ["next", "throw", "return"].forEach(function (e) { define(t, e, function (t) { return this._invoke(e, t); }); }); } function AsyncIterator(t, e) { function invoke(r, o, i, a) { var c = tryCatch(t[r], t, o); if ("throw" !== c.type) { var u = c.arg, h = u.value; return h && "object" == _typeof(h) && n.call(h, "__await") ? e.resolve(h.__await).then(function (t) { invoke("next", t, i, a); }, function (t) { invoke("throw", t, i, a); }) : e.resolve(h).then(function (t) { u.value = t, i(u); }, function (t) { return invoke("throw", t, i, a); }); } a(c.arg); } var r; o(this, "_invoke", { value: function value(t, n) { function callInvokeWithMethodAndArg() { return new e(function (e, r) { invoke(t, n, e, r); }); } return r = r ? r.then(callInvokeWithMethodAndArg, callInvokeWithMethodAndArg) : callInvokeWithMethodAndArg(); } }); } function makeInvokeMethod(e, r, n) { var o = h; return function (i, a) { if (o === f) throw new Error("Generator is already running"); if (o === s) { if ("throw" === i) throw a; return { value: t, done: !0 }; } for (n.method = i, n.arg = a;;) { var c = n.delegate; if (c) { var u = maybeInvokeDelegate(c, n); if (u) { if (u === y) continue; return u; } } if ("next" === n.method) n.sent = n._sent = n.arg;else if ("throw" === n.method) { if (o === h) throw o = s, n.arg; n.dispatchException(n.arg); } else "return" === n.method && n.abrupt("return", n.arg); o = f; var p = tryCatch(e, r, n); if ("normal" === p.type) { if (o = n.done ? s : l, p.arg === y) continue; return { value: p.arg, done: n.done }; } "throw" === p.type && (o = s, n.method = "throw", n.arg = p.arg); } }; } function maybeInvokeDelegate(e, r) { var n = r.method, o = e.iterator[n]; if (o === t) return r.delegate = null, "throw" === n && e.iterator.return && (r.method = "return", r.arg = t, maybeInvokeDelegate(e, r), "throw" === r.method) || "return" !== n && (r.method = "throw", r.arg = new TypeError("The iterator does not provide a '" + n + "' method")), y; var i = tryCatch(o, e.iterator, r.arg); if ("throw" === i.type) return r.method = "throw", r.arg = i.arg, r.delegate = null, y; var a = i.arg; return a ? a.done ? (r[e.resultName] = a.value, r.next = e.nextLoc, "return" !== r.method && (r.method = "next", r.arg = t), r.delegate = null, y) : a : (r.method = "throw", r.arg = new TypeError("iterator result is not an object"), r.delegate = null, y); } function pushTryEntry(t) { var e = { tryLoc: t[0] }; 1 in t && (e.catchLoc = t[1]), 2 in t && (e.finallyLoc = t[2], e.afterLoc = t[3]), this.tryEntries.push(e); } function resetTryEntry(t) { var e = t.completion || {}; e.type = "normal", delete e.arg, t.completion = e; } function Context(t) { this.tryEntries = [{ tryLoc: "root" }], t.forEach(pushTryEntry, this), this.reset(!0); } function values(e) { if (e || "" === e) { var r = e[a]; if (r) return r.call(e); if ("function" == typeof e.next) return e; if (!isNaN(e.length)) { var o = -1, i = function next() { for (; ++o < e.length;) if (n.call(e, o)) return next.value = e[o], next.done = !1, next; return next.value = t, next.done = !0, next; }; return i.next = i; } } throw new TypeError(_typeof(e) + " is not iterable"); } return GeneratorFunction.prototype = GeneratorFunctionPrototype, o(g, "constructor", { value: GeneratorFunctionPrototype, configurable: !0 }), o(GeneratorFunctionPrototype, "constructor", { value: GeneratorFunction, configurable: !0 }), GeneratorFunction.displayName = define(GeneratorFunctionPrototype, u, "GeneratorFunction"), e.isGeneratorFunction = function (t) { var e = "function" == typeof t && t.constructor; return !!e && (e === GeneratorFunction || "GeneratorFunction" === (e.displayName || e.name)); }, e.mark = function (t) { return Object.setPrototypeOf ? Object.setPrototypeOf(t, GeneratorFunctionPrototype) : (t.__proto__ = GeneratorFunctionPrototype, define(t, u, "GeneratorFunction")), t.prototype = Object.create(g), t; }, e.awrap = function (t) { return { __await: t }; }, defineIteratorMethods(AsyncIterator.prototype), define(AsyncIterator.prototype, c, function () { return this; }), e.AsyncIterator = AsyncIterator, e.async = function (t, r, n, o, i) { void 0 === i && (i = Promise); var a = new AsyncIterator(wrap(t, r, n, o), i); return e.isGeneratorFunction(r) ? a : a.next().then(function (t) { return t.done ? t.value : a.next(); }); }, defineIteratorMethods(g), define(g, u, "Generator"), define(g, a, function () { return this; }), define(g, "toString", function () { return "[object Generator]"; }), e.keys = function (t) { var e = Object(t), r = []; for (var n in e) r.push(n); return r.reverse(), function next() { for (; r.length;) { var t = r.pop(); if (t in e) return next.value = t, next.done = !1, next; } return next.done = !0, next; }; }, e.values = values, Context.prototype = { constructor: Context, reset: function reset(e) { if (this.prev = 0, this.next = 0, this.sent = this._sent = t, this.done = !1, this.delegate = null, this.method = "next", this.arg = t, this.tryEntries.forEach(resetTryEntry), !e) for (var r in this) "t" === r.charAt(0) && n.call(this, r) && !isNaN(+r.slice(1)) && (this[r] = t); }, stop: function stop() { this.done = !0; var t = this.tryEntries[0].completion; if ("throw" === t.type) throw t.arg; return this.rval; }, dispatchException: function dispatchException(e) { if (this.done) throw e; var r = this; function handle(n, o) { return a.type = "throw", a.arg = e, r.next = n, o && (r.method = "next", r.arg = t), !!o; } for (var o = this.tryEntries.length - 1; o >= 0; --o) { var i = this.tryEntries[o], a = i.completion; if ("root" === i.tryLoc) return handle("end"); if (i.tryLoc <= this.prev) { var c = n.call(i, "catchLoc"), u = n.call(i, "finallyLoc"); if (c && u) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } else if (c) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); } else { if (!u) throw new Error("try statement without catch or finally"); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } } } }, abrupt: function abrupt(t, e) { for (var r = this.tryEntries.length - 1; r >= 0; --r) { var o = this.tryEntries[r]; if (o.tryLoc <= this.prev && n.call(o, "finallyLoc") && this.prev < o.finallyLoc) { var i = o; break; } } i && ("break" === t || "continue" === t) && i.tryLoc <= e && e <= i.finallyLoc && (i = null); var a = i ? i.completion : {}; return a.type = t, a.arg = e, i ? (this.method = "next", this.next = i.finallyLoc, y) : this.complete(a); }, complete: function complete(t, e) { if ("throw" === t.type) throw t.arg; return "break" === t.type || "continue" === t.type ? this.next = t.arg : "return" === t.type ? (this.rval = this.arg = t.arg, this.method = "return", this.next = "end") : "normal" === t.type && e && (this.next = e), y; }, finish: function finish(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.finallyLoc === t) return this.complete(r.completion, r.afterLoc), resetTryEntry(r), y; } }, catch: function _catch(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.tryLoc === t) { var n = r.completion; if ("throw" === n.type) { var o = n.arg; resetTryEntry(r); } return o; } } throw new Error("illegal catch attempt"); }, delegateYield: function delegateYield(e, r, n) { return this.delegate = { iterator: values(e), resultName: r, nextLoc: n }, "next" === this.method && (this.arg = t), y; } }, e; } function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } } function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; } 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 { CommentOutlined } from '@ant-design/icons'; import { LibroCommandRegister, LibroSlotManager, LibroSlotView, LibroToolbarArea, LibroCellView, LibroView } from '@difizen/libro-jupyter'; import { inject } from '@difizen/mana-app'; import { CommandContribution } from '@difizen/mana-app'; import { singleton, ToolbarContribution, ViewManager } from '@difizen/mana-app'; import { l10n } from '@difizen/mana-l10n'; import { AINativeCommands } from "./ai-native-command.js"; import { LibroAINativeService } from "./ai-native-service.js"; import { AIToolbarSelector } from "./ai-side-toolbar-selector.js"; import { LibroAIChatSlotContribution } from "./chat-slot-contribution.js"; import { AIIcon } from "./icon.js"; import { addCellAIClassname } from "./utils.js"; import { jsx as _jsx } from "react/jsx-runtime"; export var LibroAINativeCommandContribution = (_dec = singleton({ contrib: [CommandContribution, ToolbarContribution] }), _dec2 = inject(LibroCommandRegister), _dec3 = inject(LibroAIChatSlotContribution), _dec4 = inject(LibroSlotManager), _dec5 = inject(LibroAINativeService), _dec6 = inject(ViewManager), _dec(_class = (_class2 = /*#__PURE__*/function () { function LibroAINativeCommandContribution() { _classCallCheck(this, LibroAINativeCommandContribution); _initializerDefineProperty(this, "libroCommand", _descriptor, this); _initializerDefineProperty(this, "libroAIChatSlotContribution", _descriptor2, this); _initializerDefineProperty(this, "libroSlotManager", _descriptor3, this); _initializerDefineProperty(this, "libroAINativeService", _descriptor4, this); _initializerDefineProperty(this, "viewManager", _descriptor5, this); } _createClass(LibroAINativeCommandContribution, [{ key: "registerToolbarItems", value: function registerToolbarItems(registry) { registry.registerItem({ id: AINativeCommands['AISideToolbarSelect'].id, command: AINativeCommands['AISideToolbarSelect'].id, icon: AIToolbarSelector, showLabelInline: true, order: 'b' }); registry.registerItem({ id: AINativeCommands['Chat'].id, command: AINativeCommands['Chat'].id, icon: AIIcon, order: 'a' }); registry.registerItem({ id: AINativeCommands['CellChat'].id, command: AINativeCommands['CellChat'].id, icon: /*#__PURE__*/_jsx(CommentOutlined, { className: "libro-ai-native-cell-chat-icon" }), order: 'a' }); } }, { key: "registerCommands", value: function registerCommands(command) { var _this = this; this.libroCommand.registerLibroCommand(command, AINativeCommands['AISideToolbarSelect'], { execute: function () { var _execute = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee() { return _regeneratorRuntime().wrap(function _callee$(_context) { while (1) switch (_context.prev = _context.next) { case 0: case "end": return _context.stop(); } }, _callee); })); function execute() { return _execute.apply(this, arguments); } return execute; }(), isVisible: function isVisible(cell, libro, path) { if (!cell || !libro || !(libro instanceof LibroView)) { return false; } return path === LibroToolbarArea.CellRight; }, isEnabled: function isEnabled(cell, libro) { if (!libro || !(libro instanceof LibroView)) { return false; } return true; } }); this.libroCommand.registerLibroCommand(command, AINativeCommands['CellChat'], { execute: function () { var _execute2 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2(cell, libro) { var chatView, showChat; return _regeneratorRuntime().wrap(function _callee2$(_context2) { while (1) switch (_context2.prev = _context2.next) { case 0: if (!(!libro || !cell)) { _context2.next = 2; break; } return _context2.abrupt("return"); case 2: addCellAIClassname(cell); chatView = _this.libroAINativeService.chatViewMap.get(libro.id); showChat = _this.libroAINativeService.showChatMap.get(libro.id); if (!chatView) { _context2.next = 12; break; } chatView.setAINativeChatView({ id: cell.id, isCellChat: true, cell: cell }); if (!showChat) { _context2.next = 9; break; } return _context2.abrupt("return"); case 9: _this.libroAINativeService.showChatMap.set(libro.id, !showChat); _this.libroAINativeService.cellAIChatMap.set(cell.id, !showChat); _this.libroSlotManager.slotViewManager.addView(chatView, _this.libroSlotManager.getSlotName(libro, _this.libroAIChatSlotContribution.slot), { reveal: true, order: 'a' }); case 12: //优化交互,用于控制点击按钮后菜单消失 _this.libroAINativeService.showSideToolbar = false; case 13: case "end": return _context2.stop(); } }, _callee2); })); function execute(_x, _x2) { return _execute2.apply(this, arguments); } return execute; }(), isEnabled: function isEnabled(cell, libro) { if (!libro || !(libro instanceof LibroView)) { return false; } return true; }, isVisible: function isVisible(cell, libro, path) { if (!cell || !libro || !(libro instanceof LibroView)) { return false; } return path === LibroToolbarArea.CellRight; } }); this.libroCommand.registerLibroCommand(command, AINativeCommands['Chat'], { execute: function () { var _execute3 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee3(cell, libro) { var chatView, showChat, slotview; return _regeneratorRuntime().wrap(function _callee3$(_context3) { while (1) switch (_context3.prev = _context3.next) { case 0: if (!(!libro || !(libro instanceof LibroView))) { _context3.next = 2; break; } return _context3.abrupt("return"); case 2: chatView = _this.libroAINativeService.chatViewMap.get(libro.id); showChat = _this.libroAINativeService.showChatMap.get(libro.id); _this.libroAINativeService.showChatMap; if (!chatView) { _context3.next = 12; break; } if (!(showChat && chatView.chatView.isCellChat)) { _context3.next = 9; break; } chatView.setAINativeChatView({ id: libro.id, isCellChat: false }); return _context3.abrupt("return"); case 9: chatView.setAINativeChatView({ id: libro.id, isCellChat: false }); _this.libroAINativeService.showChatMap.set(libro.id, !showChat); if (showChat) { slotview = _this.libroSlotManager.slotViewManager.getSlotView(_this.libroSlotManager.getSlotName(libro, 'right')); if (slotview instanceof LibroSlotView) { slotview.revertActive(); } } else { _this.libroSlotManager.slotViewManager.addView(chatView, _this.libroSlotManager.getSlotName(libro, _this.libroAIChatSlotContribution.slot), { reveal: true, order: 'a' }); } case 12: _this.libroAINativeService.showSideToolbar = false; case 13: case "end": return _context3.stop(); } }, _callee3); })); function execute(_x3, _x4) { return _execute3.apply(this, arguments); } return execute; }(), isVisible: function isVisible(cell, libro, path) { if (!cell || !libro || !(libro instanceof LibroView)) { return false; } return path === LibroToolbarArea.HeaderRight; }, isEnabled: function isEnabled(cell, libro) { if (!libro || !(libro instanceof LibroView)) { return false; } return true; } }); this.libroCommand.registerLibroCommand(command, AINativeCommands['Explain'], { execute: function () { var _execute4 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee4(cell, libro) { var libroAINativeForCellView, code; return _regeneratorRuntime().wrap(function _callee4$(_context4) { while (1) switch (_context4.prev = _context4.next) { case 0: if (!(!cell || !(cell instanceof LibroCellView) || !libro || !(libro instanceof LibroView))) { _context4.next = 2; break; } return _context4.abrupt("return"); case 2: _context4.next = 4; return _this.libroAINativeService.getOrCreateLibroAINativeForCellView(cell.id, cell); case 4: libroAINativeForCellView = _context4.sent; libroAINativeForCellView.showAI = true; addCellAIClassname(cell); code = l10n.getLang() === 'en-US' ? "Could you please explain this piece of code?\uFF1A".concat(cell.model.value) : "\u5E2E\u5FD9\u89E3\u91CA\u4E00\u4E0B\u8FD9\u6BB5\u4EE3\u7801\uFF1A".concat(cell.model.value); libroAINativeForCellView.chatStream({ content: code }); // this.libroService.active?.enterEditMode(); case 9: case "end": return _context4.stop(); } }, _callee4); })); function execute(_x5, _x6) { return _execute4.apply(this, arguments); } return execute; }(), isEnabled: function isEnabled(cell, libro) { if (!libro || !(libro instanceof LibroView)) { return false; } return true; } }); this.libroCommand.registerLibroCommand(command, AINativeCommands['Optimize'], { execute: function () { var _execute5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee5(cell, libro) { var libroAINativeForCellView, code; return _regeneratorRuntime().wrap(function _callee5$(_context5) { while (1) switch (_context5.prev = _context5.next) { case 0: if (!(!cell || !(cell instanceof LibroCellView) || !libro || !(libro instanceof LibroView))) { _context5.next = 2; break; } return _context5.abrupt("return"); case 2: _context5.next = 4; return _this.libroAINativeService.getOrCreateLibroAINativeForCellView(cell.id, cell); case 4: libroAINativeForCellView = _context5.sent; libroAINativeForCellView.showAI = true; code = l10n.getLang() === 'en-US' ? "Please help optimize this piece of code: ".concat(cell.model.value, ",provide reasons and the optimized result.") : "\u5E2E\u5FD9\u4F18\u5316\u4E00\u4E0B\u8FD9\u6BB5\u4EE3\u7801\uFF1A".concat(cell.model.value, "\uFF0C\u7ED9\u51FA\u539F\u56E0\u4EE5\u53CA\u4F18\u5316\u7ED3\u679C"); addCellAIClassname(cell); libroAINativeForCellView.chatStream({ content: code }); // this.libroService.active?.enterEditMode(); case 9: case "end": return _context5.stop(); } }, _callee5); })); function execute(_x7, _x8) { return _execute5.apply(this, arguments); } return execute; }(), isEnabled: function isEnabled(cell, libro) { if (!libro || !(libro instanceof LibroView)) { return false; } return true; } }); } }]); return LibroAINativeCommandContribution; }(), (_descriptor = _applyDecoratedDescriptor(_class2.prototype, "libroCommand", [_dec2], { configurable: true, enumerable: true, writable: true, initializer: null }), _descriptor2 = _applyDecoratedDescriptor(_class2.prototype, "libroAIChatSlotContribution", [_dec3], { configurable: true, enumerable: true, writable: true, initializer: null }), _descriptor3 = _applyDecoratedDescriptor(_class2.prototype, "libroSlotManager", [_dec4], { configurable: true, enumerable: true, writable: true, initializer: null }), _descriptor4 = _applyDecoratedDescriptor(_class2.prototype, "libroAINativeService", [_dec5], { configurable: true, enumerable: true, writable: true, initializer: null }), _descriptor5 = _applyDecoratedDescriptor(_class2.prototype, "viewManager", [_dec6], { configurable: true, enumerable: true, writable: true, initializer: null })), _class2)) || _class);