@difizen/libro-prompt-cell 0.1.13 → 0.1.14

package/es/libro-formatter-prompt-magic-contribution.js

@@ -24,7 +24,7 @@ export var FormatterPromptMagicContribution = (_dec = singleton({
   };
   this.encode = function (source) {
     var promptObj = {
-      model_name: source.modelType || 'CodeGPT',
+      model_name: source.modelType || 'chatgpt',
       prompt: source.value
     };
     var encodeValue = "%%prompt \n".concat(JSON.stringify(promptObj));

package/es/module.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"module.d.ts","sourceRoot":"","sources":["../src/module.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAW/C,eAAO,MAAM,qBAAqB,YAwB2B,CAAC"}
+{"version":3,"file":"module.d.ts","sourceRoot":"","sources":["../src/module.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAY/C,eAAO,MAAM,qBAAqB,YAyB2B,CAAC"}

package/es/module.js

@@ -7,9 +7,10 @@ import { PromptCellContribution } from "./prompt-cell-contribution.js";
 import { LibroPromptCellModel } from "./prompt-cell-model.js";
 import { LibroPromptOutputArea } from "./prompt-cell-output-area.js";
 import { LibroPromptCellModelFactory } from "./prompt-cell-protocol.js";
+import { PromptScript } from "./prompt-cell-script.js";
 import { LibroPromptCellView } from "./prompt-cell-view.js";
 import { LibroPromptOutputMimeTypeContribution } from "./prompt-output-rendermime-contribution.js";
-export var LibroPromptCellModule = ManaModule.create().register(PromptCellContribution, LibroPromptCellView, LibroPromptCellModel, LibroPromptOutputArea, LibroPromptOutputMimeTypeContribution, LibroPromptCellCommandContribution, FormatterPromptMagicContribution, {
+export var LibroPromptCellModule = ManaModule.create().register(PromptCellContribution, PromptScript, LibroPromptCellView, LibroPromptCellModel, LibroPromptOutputArea, LibroPromptOutputMimeTypeContribution, LibroPromptCellCommandContribution, FormatterPromptMagicContribution, {
   token: LibroPromptCellModelFactory,
   useFactory: function useFactory(ctx) {
     return function (options) {

package/es/prompt-cell-script.d.ts

@@ -1,4 +1,4 @@
-export declare namespace PromptScript {
-    const get_models = "import json\nfrom aistudio_notebook.prompt_flow import prompt_model_registry\nmodel_list = list(prompt_model_registry.promptModelRegistry.get_models().keys())\nmodel_data = json.dumps(model_list)\nprint(model_data)\n";
+export declare class PromptScript {
+    readonly toList = "from libro_server import chat_provider\nchat_provider.dump_list_json()";
 }
 //# sourceMappingURL=prompt-cell-script.d.ts.map

package/es/prompt-cell-script.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"prompt-cell-script.d.ts","sourceRoot":"","sources":["../src/prompt-cell-script.ts"],"names":[],"mappings":"AAAA,yBAAiB,YAAY,CAAC;IACrB,MAAM,UAAU,6NACqM,CAAC;CAC9N"}
+{"version":3,"file":"prompt-cell-script.d.ts","sourceRoot":"","sources":["../src/prompt-cell-script.ts"],"names":[],"mappings":"AAEA,qBACa,YAAY;IACvB,SAAgB,MAAM,4EACQ;CAC/B"}

package/es/prompt-cell-script.js

@@ -1,4 +1,12 @@
-export var PromptScript;
-(function (_PromptScript) {
-  var get_models = _PromptScript.get_models = 'import json\nfrom aistudio_notebook.prompt_flow import prompt_model_registry\nmodel_list = list(prompt_model_registry.promptModelRegistry.get_models().keys())\nmodel_data = json.dumps(model_list)\nprint(model_data)\n';
-})(PromptScript || (PromptScript = {}));
+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, _class;
+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 _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
+import { singleton } from '@difizen/mana-app';
+export var PromptScript = (_dec = singleton(), _dec(_class = /*#__PURE__*/_createClass(function PromptScript() {
+  _classCallCheck(this, PromptScript);
+  this.toList = "from libro_server import chat_provider\nchat_provider.dump_list_json()";
+})) || _class);

package/es/prompt-cell-view.d.ts

@@ -8,20 +8,19 @@ import { ViewManager } from '@difizen/mana-app';
 import { Deferred } from '@difizen/mana-app';
 import React from 'react';
 import type { LibroPromptCellModel } from './prompt-cell-model.js';
-export interface IModelSelectionItem {
-    value: string;
-    label: string;
-}
-export interface IModelItem {
-    value: string;
+import { PromptScript } from './prompt-cell-script.js';
+export interface IChatSelectionItem {
+    name: string;
+    type: string;
 }
 export declare const CellEditorMemo: React.NamedExoticComponent<{}>;
 export declare class LibroPromptCellView extends LibroExecutableCellView {
     view: React.ForwardRefExoticComponent<React.RefAttributes<HTMLDivElement>>;
     model: LibroPromptCellModel;
-    modelSelection: IModelSelectionItem[];
+    selection: IChatSelectionItem[];
     viewManager: ViewManager;
     codeEditorManager: CodeEditorManager;
+    promptScript: PromptScript;
     outputs: IOutput[];
     libroViewTracker: LibroViewTracker;
     editorView?: CodeEditorView;
@@ -29,7 +28,7 @@ export declare class LibroPromptCellView extends LibroExecutableCellView {
     get outputAreaReady(): Promise<LibroOutputArea>;
     protected editorViewReadyDeferred: Deferred<void>;
     get editorReady(): Promise<void>;
-    constructor(options: CellViewOptions, cellService: CellService, viewManager: ViewManager, libroViewTracker: LibroViewTracker, codeEditorManager: CodeEditorManager);
+    constructor(options: CellViewOptions, cellService: CellService, viewManager: ViewManager, libroViewTracker: LibroViewTracker);
     outputWatch(): void;
     toJSON(): LibroCell;
     onViewMount(): void;
@@ -47,6 +46,11 @@ export declare class LibroPromptCellView extends LibroExecutableCellView {
     };
     run(): Promise<boolean>;
     fetch: (content: KernelMessage.IExecuteRequestMsg['content'], ioCallback: (msg: KernelMessage.IIOPubMessage) => any) => Promise<KernelMessage.IExecuteReplyMsg>;
+    updateChatList: () => Promise<KernelMessage.IExecuteReplyMsg>;
+    toSelectionOption: (item: IChatSelectionItem) => {
+        value: string;
+        label: import("react/jsx-runtime").JSX.Element;
+    };
     handleQueryResponse: (response: KernelMessage.IIOPubMessage) => void;
 }
 //# sourceMappingURL=prompt-cell-view.d.ts.map

package/es/prompt-cell-view.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"prompt-cell-view.d.ts","sourceRoot":"","sources":["../src/prompt-cell-view.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,KAAK,EACV,qBAAqB,EACrB,MAAM,EACN,cAAc,EACf,MAAM,4BAA4B,CAAC;AACpC,OAAO,KAAK,EAAa,OAAO,EAAE,MAAM,uBAAuB,CAAC;AAEhE,OAAO,KAAK,EAEV,SAAS,EACT,eAAe,EAChB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,WAAW,EACX,uBAAuB,EACvB,eAAe,EACf,gBAAgB,EAEjB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAiB,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAE3E,OAAO,EAQL,WAAW,EAIZ,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,OAAO,KAA8B,MAAM,OAAO,CAAC;AAEnD,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAGnE,MAAM,WAAW,mBAAmB;IAClC,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;CACf;AACD,MAAM,WAAW,UAAU;IACzB,KAAK,EAAE,MAAM,CAAC;CACf;AAYD,eAAO,MAAM,cAAc,gCAAyB,CAAC;AAsErD,qBAEa,mBAAoB,SAAQ,uBAAuB;IACrD,IAAI,uEAA6B;IAElC,KAAK,EAAE,oBAAoB,CAAC;IAGpC,cAAc,EAAE,mBAAmB,EAAE,CAAM;IAE3C,WAAW,EAAE,WAAW,CAAC;IAEzB,iBAAiB,EAAE,iBAAiB,CAAC;IAErC,OAAO,EAAE,OAAO,EAAE,CAAC;IAEnB,gBAAgB,EAAE,gBAAgB,CAAC;IAGnC,UAAU,CAAC,EAAE,cAAc,CAAC;IAE5B,SAAS,CAAC,kBAAkB,4BAAmC;IAC/D,IAAI,eAAe,6BAElB;IAED,SAAS,CAAC,uBAAuB,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAwB;IAEzE,IAAI,WAAW,kBAEd;gBAGqB,OAAO,EAAE,eAAe,EACvB,WAAW,EAAE,WAAW,EACxB,WAAW,EAAE,WAAW,EACnB,gBAAgB,EAAE,gBAAgB,EACjC,iBAAiB,EAAE,iBAAiB;IAgCxD,WAAW;IAQX,MAAM,IAAI,SAAS;IASnB,WAAW;IAQpB,SAAS,CAAC,eAAe,IAAI,qBAAqB;IAY5C,YAAY;cAmBF,gBAAgB;IAUvB,qBAAqB,CAAC,CAAC,EAAE,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC;IAQtD,IAAI,aAGX;IAEO,KAAK,WAAY,OAAO,UAqC/B;IAEO,cAAc,aAGrB;IAEF,aAAa,QAAO,EAAE,CAEpB;IAEF,qBAAqB,cAAe,MAAM;;;MAKxC;IAEa,GAAG;IA4ElB,KAAK,YACM,cAAc,kBAAkB,CAAC,SAAS,CAAC,oBAClC,cAAc,aAAa,KAAK,GAAG,6CAUrD;IAEF,mBAAmB,aAAc,cAAc,aAAa,UAmC1D;CACH"}
+{"version":3,"file":"prompt-cell-view.d.ts","sourceRoot":"","sources":["../src/prompt-cell-view.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,KAAK,EACV,qBAAqB,EACrB,MAAM,EACN,cAAc,EACf,MAAM,4BAA4B,CAAC;AACpC,OAAO,KAAK,EAAa,OAAO,EAAE,MAAM,uBAAuB,CAAC;AAEhE,OAAO,KAAK,EAEV,SAAS,EACT,eAAe,EAChB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,WAAW,EACX,uBAAuB,EACvB,eAAe,EACf,gBAAgB,EAEjB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAiB,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAE3E,OAAO,EAQL,WAAW,EAIZ,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,OAAO,KAA8B,MAAM,OAAO,CAAC;AAEnD,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AACnE,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAEvD,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACd;AAmCD,eAAO,MAAM,cAAc,gCAAyB,CAAC;AAgErD,qBAEa,mBAAoB,SAAQ,uBAAuB;IACrD,IAAI,uEAA6B;IAElC,KAAK,EAAE,oBAAoB,CAAC;IAGpC,SAAS,EAAE,kBAAkB,EAAE,CAAM;IAErC,WAAW,EAAE,WAAW,CAAC;IAEE,iBAAiB,EAAE,iBAAiB,CAAC;IAC1C,YAAY,EAAE,YAAY,CAAC;IAEjD,OAAO,EAAE,OAAO,EAAE,CAAC;IAEnB,gBAAgB,EAAE,gBAAgB,CAAC;IAGnC,UAAU,CAAC,EAAE,cAAc,CAAC;IAE5B,SAAS,CAAC,kBAAkB,4BAAmC;IAC/D,IAAI,eAAe,6BAElB;IAED,SAAS,CAAC,uBAAuB,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAwB;IAEzE,IAAI,WAAW,kBAEd;gBAGqB,OAAO,EAAE,eAAe,EACvB,WAAW,EAAE,WAAW,EACxB,WAAW,EAAE,WAAW,EACnB,gBAAgB,EAAE,gBAAgB;IA+BrD,WAAW;IAQX,MAAM,IAAI,SAAS;IASnB,WAAW;IAQpB,SAAS,CAAC,eAAe,IAAI,qBAAqB;IAY5C,YAAY;cAmBF,gBAAgB;IAUvB,qBAAqB,CAAC,CAAC,EAAE,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC;IAQtD,IAAI,aAGX;IAEO,KAAK,WAAY,OAAO,UAqC/B;IAEO,cAAc,aAGrB;IAEF,aAAa,QAAO,EAAE,CAEpB;IAEF,qBAAqB,cAAe,MAAM;;;MAKxC;IAEa,GAAG;IA4ElB,KAAK,YACM,cAAc,kBAAkB,CAAC,SAAS,CAAC,oBAClC,cAAc,aAAa,KAAK,GAAG,6CAUrD;IAEF,cAAc,gDAKZ;IAEF,iBAAiB,SAAU,kBAAkB;;;MAK3C;IAEF,mBAAmB,aAAc,cAAc,aAAa,UA4B1D;CACH"}

package/es/prompt-cell-view.js

@@ -1,5 +1,5 @@
 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, _class, _class2, _descriptor, _descriptor2;
+var _dec, _dec2, _dec3, _dec4, _dec5, _dec6, _class, _class2, _descriptor, _descriptor2, _descriptor3, _descriptor4;
 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); } }
@@ -35,12 +35,32 @@ import { CellService, LibroExecutableCellView, LibroOutputArea, LibroViewTracker
 import { KernelError, LibroJupyterModel } from '@difizen/libro-jupyter';
 import { getOrigin, inject, prop, transient, useInject, view, ViewInstance, ViewManager, ViewOption, ViewRender, watch } from '@difizen/mana-app';
 import { Deferred } from '@difizen/mana-app';
-import { Select } from 'antd';
+import { Select, Tag } from 'antd';
 import React, { useEffect, useState } from 'react';
 import { PromptScript } from "./prompt-cell-script.js";
 import { jsx as _jsx } from "react/jsx-runtime";
-import { Fragment as _Fragment } from "react/jsx-runtime";
 import { jsxs as _jsxs } from "react/jsx-runtime";
+import { Fragment as _Fragment } from "react/jsx-runtime";
+var SelectionItemLabel = function SelectionItemLabel(props) {
+  var item = props.item;
+  var colorMap = {
+    VARIABLE: 'red',
+    LLM: 'blue',
+    API: 'green',
+    CUSTOM: undefined
+  };
+  return /*#__PURE__*/_jsxs("span", {
+    className: "libro-prompt-cell-selection-label",
+    children: [/*#__PURE__*/_jsx(Tag, {
+      color: colorMap[item.type] || undefined,
+      className: "libro-prompt-cell-header-selection-type",
+      children: item.type
+    }), /*#__PURE__*/_jsx("span", {
+      className: "libro-prompt-cell-header-selection-name",
+      children: item.name
+    })]
+  });
+};
 var CellEditor = function CellEditor() {
   var _instance$editorView3;
   var instance = useInject(ViewInstance);
@@ -65,14 +85,11 @@ var PropmtEditorViewComponent = /*#__PURE__*/React.forwardRef(function MaxPropmt
     selectedModel = _useState2[0],
     setSelectedModel = _useState2[1];
   useEffect(function () {
-    instance.fetch({
-      code: PromptScript.get_models,
-      store_history: false
-    }, instance.handleQueryResponse).then(function () {
-      var len = instance.modelSelection.length;
+    instance.updateChatList().then(function () {
+      var len = instance.selection.length;
       if (len > 0) {
         instance.model.decodeObject = _objectSpread({
-          modelType: instance.modelSelection[len - 1].label
+          modelType: instance.selection[len - 1].name
         }, instance.model.decodeObject);
         setSelectedModel(instance.model.decodeObject['modelType']);
         instance.model.modelType = instance.model.decodeObject['modelType'];
@@ -106,17 +123,14 @@ var PropmtEditorViewComponent = /*#__PURE__*/React.forwardRef(function MaxPropmt
             width: 160
           },
           onChange: handleChange,
-          options: instance.modelSelection,
+          options: instance.selection.map(instance.toSelectionOption),
           bordered: false,
           onFocus: /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee() {
             return _regeneratorRuntime().wrap(function _callee$(_context) {
               while (1) switch (_context.prev = _context.next) {
                 case 0:
                   _context.next = 2;
-                  return instance.fetch({
-                    code: PromptScript.get_models,
-                    store_history: false
-                  }, instance.handleQueryResponse);
+                  return instance.updateChatList();
                 case 2:
                 case "end":
                   return _context.stop();
@@ -128,21 +142,22 @@ var PropmtEditorViewComponent = /*#__PURE__*/React.forwardRef(function MaxPropmt
     }), /*#__PURE__*/_jsx(CellEditorMemo, {})]
   });
 });
-export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-editor-cell-view'), _dec3 = prop(), _dec4 = prop(), _dec(_class = _dec2(_class = (_class2 = /*#__PURE__*/function (_LibroExecutableCellV) {
+export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-editor-cell-view'), _dec3 = prop(), _dec4 = inject(CodeEditorManager), _dec5 = inject(PromptScript), _dec6 = prop(), _dec(_class = _dec2(_class = (_class2 = /*#__PURE__*/function (_LibroExecutableCellV) {
   _inherits(LibroPromptCellView, _LibroExecutableCellV);
   var _super = _createSuper(LibroPromptCellView);
-  function LibroPromptCellView(options, cellService, viewManager, libroViewTracker, codeEditorManager) {
+  function LibroPromptCellView(options, cellService, viewManager, libroViewTracker) {
     var _options$cell;
     var _this;
     _classCallCheck(this, LibroPromptCellView);
     _this = _super.call(this, options, cellService);
     _this.view = PropmtEditorViewComponent;
-    _initializerDefineProperty(_this, "modelSelection", _descriptor, _assertThisInitialized(_this));
+    _initializerDefineProperty(_this, "selection", _descriptor, _assertThisInitialized(_this));
     _this.viewManager = void 0;
-    _this.codeEditorManager = void 0;
+    _initializerDefineProperty(_this, "codeEditorManager", _descriptor2, _assertThisInitialized(_this));
+    _initializerDefineProperty(_this, "promptScript", _descriptor3, _assertThisInitialized(_this));
     _this.outputs = void 0;
     _this.libroViewTracker = void 0;
-    _initializerDefineProperty(_this, "editorView", _descriptor2, _assertThisInitialized(_this));
+    _initializerDefineProperty(_this, "editorView", _descriptor4, _assertThisInitialized(_this));
     _this.outputAreaDeferred = new Deferred();
     _this.editorViewReadyDeferred = new Deferred();
     _this.blur = function () {
@@ -247,6 +262,28 @@ export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-edito
         return _ref3.apply(this, arguments);
       };
     }();
+    _this.updateChatList = /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee4() {
+      return _regeneratorRuntime().wrap(function _callee4$(_context4) {
+        while (1) switch (_context4.prev = _context4.next) {
+          case 0:
+            return _context4.abrupt("return", _this.fetch({
+              code: _this.promptScript.toList,
+              store_history: false
+            }, _this.handleQueryResponse));
+          case 1:
+          case "end":
+            return _context4.stop();
+        }
+      }, _callee4);
+    }));
+    _this.toSelectionOption = function (item) {
+      return {
+        value: item.name,
+        label: /*#__PURE__*/_jsx(SelectionItemLabel, {
+          item: item
+        })
+      };
+    };
     _this.handleQueryResponse = function (response) {
       var msgType = response.header.msg_type;
       switch (msgType) {
@@ -258,13 +295,7 @@ export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-edito
               content = content.slice(1, -1);
               content = content.replace(/\\"/g, '"').replace(/\\'/g, "'");
             }
-            var update = JSON.parse(content);
-            _this.modelSelection = update.map(function (item) {
-              return {
-                value: item,
-                label: item
-              };
-            });
+            _this.selection = JSON.parse(content);
             break;
           }
         case 'stream':
@@ -275,13 +306,7 @@ export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-edito
               contentStream = contentStream.slice(1, -1);
               contentStream = contentStream.replace(/\\"/g, '"').replace(/\\'/g, "'");
             }
-            var updateStream = JSON.parse(contentStream);
-            _this.modelSelection = updateStream.map(function (item) {
-              return {
-                value: item,
-                label: item
-              };
-            });
+            _this.selection = JSON.parse(contentStream);
             break;
           }
         default:
@@ -291,7 +316,6 @@ export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-edito
     _this.options = options;
     _this.viewManager = viewManager;
     _this.className = _this.className + ' prompt';
-    _this.codeEditorManager = codeEditorManager;
     _this.outputs = (_options$cell = options.cell) === null || _options$cell === void 0 ? void 0 : _options$cell.outputs;
     _this.libroViewTracker = libroViewTracker;
 
@@ -300,38 +324,37 @@ export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-edito
       cellId: _this.id,
       cell: _assertThisInitialized(_this)
     }).then( /*#__PURE__*/function () {
-      var _ref4 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee4(outputArea) {
+      var _ref5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee5(outputArea) {
         var output;
-        return _regeneratorRuntime().wrap(function _callee4$(_context4) {
-          while (1) switch (_context4.prev = _context4.next) {
+        return _regeneratorRuntime().wrap(function _callee5$(_context5) {
+          while (1) switch (_context5.prev = _context5.next) {
             case 0:
               _this.outputArea = outputArea;
               output = _this.outputs;
               if (!isOutput(output)) {
-                _context4.next = 5;
+                _context5.next = 5;
                 break;
               }
-              _context4.next = 5;
+              _context5.next = 5;
               return _this.outputArea.fromJSON(output);
             case 5:
               _this.outputAreaDeferred.resolve(outputArea);
               _this.outputWatch();
-              return _context4.abrupt("return");
+              return _context5.abrupt("return");
             case 8:
             case "end":
-              return _context4.stop();
+              return _context5.stop();
           }
-        }, _callee4);
+        }, _callee5);
       }));
       return function (_x3) {
-        return _ref4.apply(this, arguments);
+        return _ref5.apply(this, arguments);
       };
     }()).catch(function () {
       //
     });
     return _this;
   }
-  LibroPromptCellView = inject(CodeEditorManager)(LibroPromptCellView, undefined, 4) || LibroPromptCellView;
   LibroPromptCellView = inject(LibroViewTracker)(LibroPromptCellView, undefined, 3) || LibroPromptCellView;
   LibroPromptCellView = inject(ViewManager)(LibroPromptCellView, undefined, 2) || LibroPromptCellView;
   LibroPromptCellView = inject(CellService)(LibroPromptCellView, undefined, 1) || LibroPromptCellView;
@@ -391,36 +414,36 @@ export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-edito
   }, {
     key: "createEditor",
     value: function () {
-      var _createEditor = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee5() {
+      var _createEditor = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee6() {
         var option, editorView;
-        return _regeneratorRuntime().wrap(function _callee5$(_context5) {
-          while (1) switch (_context5.prev = _context5.next) {
+        return _regeneratorRuntime().wrap(function _callee6$(_context6) {
+          while (1) switch (_context6.prev = _context6.next) {
             case 0:
               option = this.getEditorOption();
               this.editorStatus = EditorStatus.LOADING;
 
               // 防止虚拟滚动中编辑器被频繁创建
               if (!this.editorView) {
-                _context5.next = 5;
+                _context6.next = 5;
                 break;
               }
               this.editorStatus = EditorStatus.LOADED;
-              return _context5.abrupt("return");
+              return _context6.abrupt("return");
             case 5:
-              _context5.next = 7;
+              _context6.next = 7;
               return this.codeEditorManager.getOrCreateEditorView(option);
             case 7:
-              editorView = _context5.sent;
+              editorView = _context6.sent;
               this.editorView = editorView;
               this.editorViewReadyDeferred.resolve();
               this.editorStatus = EditorStatus.LOADED;
-              _context5.next = 13;
+              _context6.next = 13;
               return this.afterEditorReady();
             case 13:
             case "end":
-              return _context5.stop();
+              return _context6.stop();
           }
-        }, _callee5, this);
+        }, _callee6, this);
       }));
       function createEditor() {
         return _createEditor.apply(this, arguments);
@@ -430,11 +453,11 @@ export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-edito
   }, {
     key: "afterEditorReady",
     value: function () {
-      var _afterEditorReady = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee6() {
+      var _afterEditorReady = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee7() {
         var _this3 = this,
           _this$editorView9;
-        return _regeneratorRuntime().wrap(function _callee6$(_context6) {
-          while (1) switch (_context6.prev = _context6.next) {
+        return _regeneratorRuntime().wrap(function _callee7$(_context7) {
+          while (1) switch (_context7.prev = _context7.next) {
             case 0:
               watch(this.parent.model, 'readOnly', function () {
                 var _this3$editorView;
@@ -445,9 +468,9 @@ export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-edito
               });
             case 2:
             case "end":
-              return _context6.stop();
+              return _context7.stop();
           }
-        }, _callee6, this);
+        }, _callee7, this);
       }));
       function afterEditorReady() {
         return _afterEditorReady.apply(this, arguments);
@@ -463,22 +486,22 @@ export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-edito
   }, {
     key: "run",
     value: function () {
-      var _run = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee7() {
+      var _run = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee8() {
         var _this4 = this;
         var libroModel, kernelConnection, cellContent, future, startTimeStr, msgPromise, endTimeStr;
-        return _regeneratorRuntime().wrap(function _callee7$(_context7) {
-          while (1) switch (_context7.prev = _context7.next) {
+        return _regeneratorRuntime().wrap(function _callee8$(_context8) {
+          while (1) switch (_context8.prev = _context8.next) {
             case 0:
               libroModel = this.parent.model;
               if (!(!libroModel || !(libroModel instanceof LibroJupyterModel) || !libroModel.kernelConnection || libroModel.kernelConnection.isDisposed)) {
-                _context7.next = 3;
+                _context8.next = 3;
                 break;
               }
-              return _context7.abrupt("return", false);
+              return _context8.abrupt("return", false);
             case 3:
               kernelConnection = getOrigin(libroModel.kernelConnection);
               cellContent = this.model.source;
-              _context7.prev = 5;
+              _context8.prev = 5;
               // Promise.resolve().then(() => {
               this.clearExecution();
               // });
@@ -508,10 +531,10 @@ export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-edito
                   _this4.model.hasExecutedError = true;
                 }
               };
-              _context7.next = 14;
+              _context8.next = 14;
               return future.done;
             case 14:
-              msgPromise = _context7.sent;
+              msgPromise = _context8.sent;
               this.model.executing = false;
               this.model.kernelExecuting = false;
               this.model.hasExecutedSuccess = !this.model.hasExecutedError;
@@ -520,36 +543,36 @@ export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-edito
               this.model.metadata['execution']['shell.execute_reply.started'] = startTimeStr;
               this.model.metadata['execution']['shell.execute_reply.end'] = endTimeStr;
               if (msgPromise) {
-                _context7.next = 24;
+                _context8.next = 24;
                 break;
               }
-              return _context7.abrupt("return", true);
+              return _context8.abrupt("return", true);
             case 24:
               if (!(msgPromise.content.status === 'ok')) {
-                _context7.next = 28;
+                _context8.next = 28;
                 break;
               }
-              return _context7.abrupt("return", true);
+              return _context8.abrupt("return", true);
             case 28:
               throw new KernelError(msgPromise.content);
             case 29:
-              _context7.next = 36;
+              _context8.next = 36;
               break;
             case 31:
-              _context7.prev = 31;
-              _context7.t0 = _context7["catch"](5);
-              if (!_context7.t0.message.startsWith('Canceled')) {
-                _context7.next = 35;
+              _context8.prev = 31;
+              _context8.t0 = _context8["catch"](5);
+              if (!_context8.t0.message.startsWith('Canceled')) {
+                _context8.next = 35;
                 break;
               }
-              return _context7.abrupt("return", false);
+              return _context8.abrupt("return", false);
             case 35:
-              throw _context7.t0;
+              throw _context8.t0;
             case 36:
             case "end":
-              return _context7.stop();
+              return _context8.stop();
           }
-        }, _callee7, this, [[5, 31]]);
+        }, _callee8, this, [[5, 31]]);
       }));
       function run() {
         return _run.apply(this, arguments);
@@ -558,14 +581,24 @@ export var LibroPromptCellView = (_dec = transient(), _dec2 = view('prompt-edito
     }()
   }]);
   return LibroPromptCellView;
-}(LibroExecutableCellView), (_descriptor = _applyDecoratedDescriptor(_class2.prototype, "modelSelection", [_dec3], {
+}(LibroExecutableCellView), (_descriptor = _applyDecoratedDescriptor(_class2.prototype, "selection", [_dec3], {
   configurable: true,
   enumerable: true,
   writable: true,
   initializer: function initializer() {
     return [];
   }
-}), _descriptor2 = _applyDecoratedDescriptor(_class2.prototype, "editorView", [_dec4], {
+}), _descriptor2 = _applyDecoratedDescriptor(_class2.prototype, "codeEditorManager", [_dec4], {
+  configurable: true,
+  enumerable: true,
+  writable: true,
+  initializer: null
+}), _descriptor3 = _applyDecoratedDescriptor(_class2.prototype, "promptScript", [_dec5], {
+  configurable: true,
+  enumerable: true,
+  writable: true,
+  initializer: null
+}), _descriptor4 = _applyDecoratedDescriptor(_class2.prototype, "editorView", [_dec6], {
   configurable: true,
   enumerable: true,
   writable: true,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@difizen/libro-prompt-cell",
-  "version": "0.1.13",
+  "version": "0.1.14",
   "description": "",
   "keywords": [
     "libro",
@@ -32,12 +32,12 @@
     "src"
   ],
   "dependencies": {
-    "@difizen/libro-code-editor": "^0.1.13",
-    "@difizen/libro-jupyter": "^0.1.13",
-    "@difizen/libro-rendermime": "^0.1.13",
-    "@difizen/libro-codemirror": "^0.1.13",
-    "@difizen/libro-common": "^0.1.13",
-    "@difizen/libro-core": "^0.1.13",
+    "@difizen/libro-code-editor": "^0.1.14",
+    "@difizen/libro-jupyter": "^0.1.14",
+    "@difizen/libro-rendermime": "^0.1.14",
+    "@difizen/libro-codemirror": "^0.1.14",
+    "@difizen/libro-common": "^0.1.14",
+    "@difizen/libro-core": "^0.1.14",
     "@difizen/mana-app": "latest",
     "highlight.js": "^11.8.0",
     "marked": "^5.1.1",

package/src/libro-formatter-prompt-magic-contribution.ts

@@ -27,7 +27,7 @@ export class FormatterPromptMagicContribution
 
   encode = (source: PromptDecodedFormatter) => {
     const promptObj = {
-      model_name: source.modelType || 'CodeGPT',
+      model_name: source.modelType || 'chatgpt',
       prompt: source.value,
     };
     const encodeValue = `%%prompt \n${JSON.stringify(promptObj)}`;

package/src/module.ts

@@ -8,12 +8,14 @@ import { PromptCellContribution } from './prompt-cell-contribution.js';
 import { LibroPromptCellModel } from './prompt-cell-model.js';
 import { LibroPromptOutputArea } from './prompt-cell-output-area.js';
 import { LibroPromptCellModelFactory } from './prompt-cell-protocol.js';
+import { PromptScript } from './prompt-cell-script.js';
 import { LibroPromptCellView } from './prompt-cell-view.js';
 import { LibroPromptOutputMimeTypeContribution } from './prompt-output-rendermime-contribution.js';
 
 export const LibroPromptCellModule = ManaModule.create()
   .register(
     PromptCellContribution,
+    PromptScript,
     LibroPromptCellView,
     LibroPromptCellModel,
     LibroPromptOutputArea,

package/src/prompt-cell-script.ts

@@ -1,4 +1,7 @@
-export namespace PromptScript {
-  export const get_models =
-    'import json\nfrom aistudio_notebook.prompt_flow import prompt_model_registry\nmodel_list = list(prompt_model_registry.promptModelRegistry.get_models().keys())\nmodel_data = json.dumps(model_list)\nprint(model_data)\n';
+import { singleton } from '@difizen/mana-app';
+
+@singleton()
+export class PromptScript {
+  public readonly toList = `from libro_server import chat_provider
+chat_provider.dump_list_json()`;
 }

package/src/prompt-cell-view.tsx

@@ -34,20 +34,40 @@ import {
   watch,
 } from '@difizen/mana-app';
 import { Deferred } from '@difizen/mana-app';
-import { Select } from 'antd';
+import { Select, Tag } from 'antd';
 import React, { useEffect, useState } from 'react';
 
 import type { LibroPromptCellModel } from './prompt-cell-model.js';
 import { PromptScript } from './prompt-cell-script.js';
 
-export interface IModelSelectionItem {
-  value: string;
-  label: string;
-}
-export interface IModelItem {
-  value: string;
+export interface IChatSelectionItem {
+  name: string;
+  type: string;
 }
 
+const SelectionItemLabel: React.FC<{ item: IChatSelectionItem }> = (props: {
+  item: IChatSelectionItem;
+}) => {
+  const item = props.item;
+  const colorMap: Record<string, any> = {
+    VARIABLE: 'red',
+    LLM: 'blue',
+    API: 'green',
+    CUSTOM: undefined,
+  };
+
+  return (
+    <span className="libro-prompt-cell-selection-label">
+      <Tag
+        color={colorMap[item.type] || undefined}
+        className="libro-prompt-cell-header-selection-type"
+      >
+        {item.type}
+      </Tag>
+      <span className="libro-prompt-cell-header-selection-name">{item.name}</span>
+    </span>
+  );
+};
 const CellEditor: React.FC = () => {
   const instance = useInject<LibroPromptCellView>(ViewInstance);
   useEffect(() => {
@@ -66,15 +86,12 @@ const PropmtEditorViewComponent = React.forwardRef<HTMLDivElement>(
     const [selectedModel, setSelectedModel] = useState<string>('暂无内置模型');
     useEffect(() => {
       instance
-        .fetch(
-          { code: PromptScript.get_models, store_history: false },
-          instance.handleQueryResponse,
-        )
+        .updateChatList()
         .then(() => {
-          const len = instance.modelSelection.length;
+          const len = instance.selection.length;
           if (len > 0) {
             instance.model.decodeObject = {
-              modelType: instance.modelSelection[len - 1].label,
+              modelType: instance.selection[len - 1].name,
               ...instance.model.decodeObject,
             };
             setSelectedModel(instance.model.decodeObject['modelType']);
@@ -111,13 +128,10 @@ const PropmtEditorViewComponent = React.forwardRef<HTMLDivElement>(
               value={selectedModel}
               style={{ width: 160 }}
               onChange={handleChange}
-              options={instance.modelSelection}
+              options={instance.selection.map(instance.toSelectionOption)}
               bordered={false}
               onFocus={async () => {
-                await instance.fetch(
-                  { code: PromptScript.get_models, store_history: false },
-                  instance.handleQueryResponse,
-                );
+                await instance.updateChatList();
               }}
             />
           </div>
@@ -136,11 +150,12 @@ export class LibroPromptCellView extends LibroExecutableCellView {
   declare model: LibroPromptCellModel;
 
   @prop()
-  modelSelection: IModelSelectionItem[] = [];
+  selection: IChatSelectionItem[] = [];
 
   viewManager: ViewManager;
 
-  codeEditorManager: CodeEditorManager;
+  @inject(CodeEditorManager) codeEditorManager: CodeEditorManager;
+  @inject(PromptScript) promptScript: PromptScript;
 
   outputs: IOutput[];
 
@@ -165,13 +180,11 @@ export class LibroPromptCellView extends LibroExecutableCellView {
     @inject(CellService) cellService: CellService,
     @inject(ViewManager) viewManager: ViewManager,
     @inject(LibroViewTracker) libroViewTracker: LibroViewTracker,
-    @inject(CodeEditorManager) codeEditorManager: CodeEditorManager,
   ) {
     super(options, cellService);
     this.options = options;
     this.viewManager = viewManager;
     this.className = this.className + ' prompt';
-    this.codeEditorManager = codeEditorManager;
 
     this.outputs = options.cell?.outputs as IOutput[];
     this.libroViewTracker = libroViewTracker;
@@ -421,6 +434,20 @@ export class LibroPromptCellView extends LibroExecutableCellView {
     return future.done as Promise<KernelMessage.IExecuteReplyMsg>;
   };
 
+  updateChatList = async () => {
+    return this.fetch(
+      { code: this.promptScript.toList, store_history: false },
+      this.handleQueryResponse,
+    );
+  };
+
+  toSelectionOption = (item: IChatSelectionItem) => {
+    return {
+      value: item.name,
+      label: <SelectionItemLabel item={item} />,
+    };
+  };
+
   handleQueryResponse = (response: KernelMessage.IIOPubMessage) => {
     const msgType = response.header.msg_type;
     switch (msgType) {
@@ -432,10 +459,7 @@ export class LibroPromptCellView extends LibroExecutableCellView {
           content = content.replace(/\\"/g, '"').replace(/\\'/g, "'");
         }
 
-        const update = JSON.parse(content) as string[];
-        this.modelSelection = update.map((item) => {
-          return { value: item, label: item };
-        });
+        this.selection = JSON.parse(content) as IChatSelectionItem[];
         break;
       }
       case 'stream': {
@@ -446,11 +470,7 @@ export class LibroPromptCellView extends LibroExecutableCellView {
           contentStream = contentStream.replace(/\\"/g, '"').replace(/\\'/g, "'");
         }
 
-        const updateStream = JSON.parse(contentStream) as string[];
-
-        this.modelSelection = updateStream.map((item) => {
-          return { value: item, label: item };
-        });
+        this.selection = JSON.parse(contentStream) as IChatSelectionItem[];
         break;
       }
       default: