@difizen/libro-ai-native 0.2.41-next.0

package/es/libro-ai-native-chat-view.js

@@ -0,0 +1,88 @@
+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, _class, _class2, _descriptor;
+function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
+function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
+function _defineProperty(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
+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 _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function"); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, writable: true, configurable: true } }); Object.defineProperty(subClass, "prototype", { writable: false }); if (superClass) _setPrototypeOf(subClass, superClass); }
+function _setPrototypeOf(o, p) { _setPrototypeOf = Object.setPrototypeOf ? Object.setPrototypeOf.bind() : function _setPrototypeOf(o, p) { o.__proto__ = p; return o; }; return _setPrototypeOf(o, p); }
+function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = _getPrototypeOf(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = _getPrototypeOf(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return _possibleConstructorReturn(this, result); }; }
+function _possibleConstructorReturn(self, call) { if (call && (_typeof(call) === "object" || typeof call === "function")) { return call; } else if (call !== void 0) { throw new TypeError("Derived constructors may only return object or undefined"); } return _assertThisInitialized(self); }
+function _assertThisInitialized(self) { if (self === void 0) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return self; }
+function _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); return true; } catch (e) { return false; } }
+function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf.bind() : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); }
+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 { ChatView } from '@difizen/magent-chat';
+import { prop, transient, view } from '@difizen/mana-app';
+import breaks from 'remark-breaks';
+import remarkGfm from 'remark-gfm';
+import { CodeBlockInChat } from "./ai-native-code-block.js";
+import "./index.less";
+import { ImageModal } from "./utils.js";
+// import { ImageModal } from './utils.js';
+
+var viewId = 'magent-chat';
+export var LibroAiNativeChatView = (_dec = transient(), _dec2 = view(viewId), _dec3 = prop(), _dec(_class = _dec2(_class = (_class2 = /*#__PURE__*/function (_ChatView) {
+  _inherits(LibroAiNativeChatView, _ChatView);
+  var _super = _createSuper(LibroAiNativeChatView);
+  function LibroAiNativeChatView() {
+    var _this;
+    _classCallCheck(this, LibroAiNativeChatView);
+    for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) {
+      args[_key] = arguments[_key];
+    }
+    _this = _super.call.apply(_super, [this].concat(args));
+    _initializerDefineProperty(_this, "isCellChat", _descriptor, _assertThisInitialized(_this));
+    return _this;
+  }
+  _createClass(LibroAiNativeChatView, [{
+    key: "getMarkdownProps",
+    value: function getMarkdownProps() {
+      return {
+        components: {
+          code: CodeBlockInChat,
+          img: ImageModal
+        },
+        remarkPlugins: [remarkGfm, breaks]
+      };
+    }
+  }, {
+    key: "toMessageOption",
+    value: function toMessageOption(msgContent) {
+      if (this.isCellChat) {
+        var _this$libro;
+        return _objectSpread(_objectSpread({
+          stream: true
+        }, this.option), {}, {
+          sender: {
+            type: 'HUMAN'
+          },
+          input: msgContent,
+          system_prompt: "\u6709\u5982\u4E0B\u7684\u4EE3\u7801\u4F5C\u4E3A\u5BF9\u8BDD\u7684\u4E0A\u4E0B\u6587\uFF1A".concat((_this$libro = this.libro) === null || _this$libro === void 0 || (_this$libro = _this$libro.activeCell) === null || _this$libro === void 0 ? void 0 : _this$libro.model.value)
+        });
+      }
+      return _objectSpread(_objectSpread({
+        stream: true
+      }, this.option), {}, {
+        sender: {
+          type: 'HUMAN'
+        },
+        input: msgContent
+      });
+    }
+  }]);
+  return LibroAiNativeChatView;
+}(ChatView), (_descriptor = _applyDecoratedDescriptor(_class2.prototype, "isCellChat", [_dec3], {
+  configurable: true,
+  enumerable: true,
+  writable: true,
+  initializer: function initializer() {
+    return false;
+  }
+})), _class2)) || _class) || _class);

package/es/libro-ai-native-color-registry.d.ts

@@ -0,0 +1,6 @@
+import type { ColorRegistry } from '@difizen/mana-app';
+import { ColorContribution } from '@difizen/mana-app';
+export declare class LibroAINativeColorRegistry implements ColorContribution {
+    registerColors(colors: ColorRegistry): void;
+}
+//# sourceMappingURL=libro-ai-native-color-registry.d.ts.map

package/es/libro-ai-native-color-registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"libro-ai-native-color-registry.d.ts","sourceRoot":"","sources":["../src/libro-ai-native-color-registry.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAEvD,OAAO,EAAa,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAEjE,qBACa,0BAA2B,YAAW,iBAAiB;IAClE,cAAc,CAAC,MAAM,EAAE,aAAa,GAAG,IAAI;CA0B5C"}

package/es/libro-ai-native-color-registry.js

@@ -0,0 +1,46 @@
+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 _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); }
+import { Color } from '@difizen/mana-app';
+import { singleton, ColorContribution } from '@difizen/mana-app';
+export var LibroAINativeColorRegistry = (_dec = singleton({
+  contrib: ColorContribution
+}), _dec(_class = /*#__PURE__*/function () {
+  function LibroAINativeColorRegistry() {
+    _classCallCheck(this, LibroAINativeColorRegistry);
+  }
+  _createClass(LibroAINativeColorRegistry, [{
+    key: "registerColors",
+    value: function registerColors(colors) {
+      colors.register(
+      // #region antd variable
+      {
+        id: 'libro.ai.native.color',
+        defaults: {
+          dark: '#bd74e8',
+          light: '#bd74e8'
+        },
+        description: ''
+      }, {
+        id: 'libro.ai.native.box.shadow',
+        defaults: {
+          dark: Color.rgba(203, 146, 197, 0.37),
+          light: Color.rgba(203, 146, 197, 0.37)
+        },
+        description: ''
+      }, {
+        id: 'libro.ai.native.btn.hover.color',
+        defaults: {
+          dark: Color.rgba(255, 255, 255, 0.2),
+          light: Color.rgba(0, 10, 26, 0.16)
+        },
+        description: ''
+      });
+    }
+  }]);
+  return LibroAINativeColorRegistry;
+}()) || _class);

package/es/module.d.ts

@@ -0,0 +1,3 @@
+import { ManaModule } from '@difizen/mana-app';
+export declare const LibroAINativeModule: ManaModule;
+//# sourceMappingURL=module.d.ts.map

package/es/module.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"module.d.ts","sourceRoot":"","sources":["../src/module.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAc/C,eAAO,MAAM,mBAAmB,YA0BJ,CAAC"}

package/es/module.js

@@ -0,0 +1,28 @@
+import { CellOutputTopProvider, ErrorOutputModel } from '@difizen/libro-jupyter';
+import { ChatView } from '@difizen/magent-chat';
+import { LibroChatModule, LibroChatService } from '@difizen/magent-libro';
+import { ManaModule } from '@difizen/mana-app';
+import { LibroAINativeCommandContribution } from "./ai-native-command-contribution.js";
+import { LibroAINativeForCellView } from "./ai-native-for-cell-view.js";
+import { LibroAINativeCellTopBlank } from "./ai-native-output-top.js";
+import { LibroAINativeService } from "./ai-native-service.js";
+import { LibroAIChatSlotContribution } from "./chat-slot-contribution.js";
+import { LibroChatView } from "./chat-view.js";
+import { AIErrorOutputModel } from "./error-output-model.js";
+import { LibroAIChatMessageItemModel } from "./libro-ai-msg-item-model.js";
+import { LibroAINativeChatService } from "./libro-ai-native-chat-service.js";
+import { LibroAiNativeChatView } from "./libro-ai-native-chat-view.js";
+import { LibroAINativeColorRegistry } from "./libro-ai-native-color-registry.js";
+export var LibroAINativeModule = ManaModule.create().register(LibroAINativeColorRegistry, LibroChatView, LibroAIChatSlotContribution, LibroAINativeCommandContribution, LibroAINativeService, LibroAIChatMessageItemModel, LibroAINativeForCellView, {
+  token: ErrorOutputModel,
+  useClass: AIErrorOutputModel
+}, {
+  token: ChatView,
+  useClass: LibroAiNativeChatView
+}, {
+  token: LibroChatService,
+  useClass: LibroAINativeChatService
+}, {
+  token: CellOutputTopProvider,
+  useValue: LibroAINativeCellTopBlank
+}).dependOn(LibroChatModule);

package/es/protocol.d.ts

@@ -0,0 +1,10 @@
+import type { CellView } from '@difizen/libro-jupyter';
+import type { IChatMessageItem } from '@difizen/magent-chat';
+export interface IAINativeForCellViewOption {
+    cell: CellView;
+    [key: string]: any;
+}
+export interface LibroAINativeChatMessageItemOption extends IChatMessageItem {
+    chat_key: string;
+}
+//# sourceMappingURL=protocol.d.ts.map

package/es/protocol.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"protocol.d.ts","sourceRoot":"","sources":["../src/protocol.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAE7D,MAAM,WAAW,0BAA0B;IACzC,IAAI,EAAE,QAAQ,CAAC;IACf,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB;AAED,MAAM,WAAW,kCAAmC,SAAQ,gBAAgB;IAC1E,QAAQ,EAAE,MAAM,CAAC;CAClB"}

package/es/protocol.js

@@ -0,0 +1 @@
+export {};

package/es/utils.js

@@ -0,0 +1,114 @@
+function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
+function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
+function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
+function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
+function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t.return && (u = t.return(), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
+function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
+import { Modal } from 'antd';
+import { useEffect, useState } from 'react';
+import { TransformComponent, TransformWrapper } from 'react-zoom-pan-pinch';
+import { jsx as _jsx } from "react/jsx-runtime";
+import { jsxs as _jsxs } from "react/jsx-runtime";
+export function stringToReadableStream(inputString) {
+  // Convert the string into a Uint8Array
+  var encoder = new TextEncoder();
+  var uint8Array = encoder.encode(inputString);
+
+  // Create a new ReadableStream
+  var readableStream = new ReadableStream({
+    start: function start(controller) {
+      // Enqueue the Uint8Array into the stream
+      controller.enqueue(uint8Array);
+      // Close the stream
+      controller.close();
+    }
+  });
+  return readableStream;
+}
+export function ImageModal(_ref) {
+  var src = _ref.src,
+    alt = _ref.alt;
+  var _useState = useState(false),
+    _useState2 = _slicedToArray(_useState, 2),
+    visible = _useState2[0],
+    setVisible = _useState2[1];
+  var _useState3 = useState({
+      width: 0,
+      height: 0
+    }),
+    _useState4 = _slicedToArray(_useState3, 2),
+    imageDimensions = _useState4[0],
+    setImageDimensions = _useState4[1];
+  useEffect(function () {
+    var img = new Image();
+    img.src = src;
+    img.onload = function () {
+      setImageDimensions({
+        width: img.width,
+        height: img.height
+      });
+    };
+  }, [src]);
+  var maxModalWidth = window.innerWidth * 0.8; // 80% of the viewport width
+  var maxModalHeight = window.innerHeight * 0.8; // 80% of the viewport height
+
+  var adjustedWidth, adjustedHeight;
+  var aspectRatio = imageDimensions.width / imageDimensions.height;
+  if (imageDimensions.width > maxModalWidth) {
+    adjustedWidth = maxModalWidth;
+    adjustedHeight = adjustedWidth / aspectRatio;
+  } else if (imageDimensions.height > maxModalHeight) {
+    adjustedHeight = maxModalHeight;
+    adjustedWidth = adjustedHeight * aspectRatio;
+  } else {
+    adjustedWidth = imageDimensions.width;
+    adjustedHeight = imageDimensions.height;
+  }
+  return /*#__PURE__*/_jsxs("div", {
+    children: [/*#__PURE__*/_jsx("img", {
+      src: src,
+      alt: alt,
+      style: {
+        cursor: 'pointer'
+      },
+      onClick: function onClick() {
+        return setVisible(true);
+      },
+      onLoad: function onLoad() {
+        var _document$getElementB;
+        // 解决生成图片没有滚动到最下方的问题。
+        (_document$getElementB = document.getElementById('chat-main-scroll')) === null || _document$getElementB === void 0 || _document$getElementB.scrollIntoView(false);
+      }
+    }), /*#__PURE__*/_jsx(Modal, {
+      open: visible,
+      closeIcon: false,
+      footer: null,
+      width: adjustedWidth + 30,
+      onCancel: function onCancel() {
+        return setVisible(false);
+      },
+      bodyStyle: {
+        padding: 0,
+        display: 'flex',
+        justifyContent: 'center',
+        alignItems: 'center',
+        height: adjustedHeight
+      },
+      children: /*#__PURE__*/_jsx(TransformWrapper, {
+        initialScale: 1,
+        initialPositionX: 0,
+        initialPositionY: 0,
+        children: /*#__PURE__*/_jsx(TransformComponent, {
+          children: /*#__PURE__*/_jsx("img", {
+            src: src,
+            alt: alt,
+            style: {
+              width: '100%',
+              height: '100%'
+            }
+          })
+        })
+      })
+    })]
+  });
+}

package/package.json

@@ -0,0 +1,74 @@
+{
+  "name": "@difizen/libro-ai-native",
+  "version": "0.2.41-next.0",
+  "description": "",
+  "keywords": [
+    "libro",
+    "ai-native"
+  ],
+  "repository": "git@github.com:difizen/libro.git",
+  "license": "MIT",
+  "type": "module",
+  "exports": {
+    ".": {
+      "typings": "./es/index.d.ts",
+      "default": "./es/index.js"
+    },
+    "./mock": {
+      "typings": "./es/mock/index.d.ts",
+      "default": "./es/mock/index.js"
+    },
+    "./es/mock": {
+      "typings": "./es/mock/index.d.ts",
+      "default": "./es/mock/index.js"
+    },
+    "./package.json": "./package.json"
+  },
+  "main": "es/index.js",
+  "module": "es/index.js",
+  "typings": "es/index.d.ts",
+  "files": [
+    "es",
+    "src"
+  ],
+  "dependencies": {
+    "@difizen/libro-jupyter": "^0.2.41-next.0",
+    "@ant-design/icons": "^5.4.0",
+    "react-zoom-pan-pinch": "^3.6.1",
+    "@difizen/mana-app": "latest",
+    "@difizen/mana-l10n": "latest",
+    "@difizen/magent-libro": "^0.1.27",
+    "@difizen/magent-chat": "^0.1.27",
+    "@difizen/magent-core": "^0.1.27",
+    "copy-to-clipboard": "^3.3.3",
+    "eventsource-parser": "^1.1.2",
+    "react-syntax-highlighter": "^15.5.0",
+    "markdown-it": "^13.0.1",
+    "remark-breaks": "^4.0.0",
+    "remark-gfm": "^4.0.0",
+    "uuid": "^9.0.0"
+  },
+  "peerDependencies": {
+    "antd": "^5.8.6",
+    "react": ">=16"
+  },
+  "devDependencies": {
+    "@types/react": "^18.2.25",
+    "@types/react-syntax-highlighter": "^15.5.13",
+    "@types/sanitize-html": "^2.6.2",
+    "@types/uuid": "^9.0.2"
+  },
+  "scripts": {
+    "setup": "father build",
+    "build": "father build",
+    "test": ": Note: lint task is delegated to test:* scripts",
+    "test:vitest": "vitest run",
+    "test:jest": "jest",
+    "coverage": ": Note: lint task is delegated to coverage:* scripts",
+    "coverage:vitest": "vitest run --coverage",
+    "coverage:jest": "jest --coverage",
+    "lint": ": Note: lint task is delegated to lint:* scripts",
+    "lint:eslint": "eslint src",
+    "typecheck:tsc": "tsc --noEmit"
+  }
+}

package/src/ai-native-code-block.tsx

@@ -0,0 +1,138 @@
+import { CopyOutlined } from '@ant-design/icons';
+import { LibroCodeCellView } from '@difizen/libro-jupyter';
+import { useInject, ViewInstance } from '@difizen/mana-app';
+import { l10n } from '@difizen/mana-l10n';
+import { Button, message } from 'antd';
+import copy from 'copy-to-clipboard';
+import { Prism as SyntaxHighlighter } from 'react-syntax-highlighter';
+import { v4 } from 'uuid';
+
+import './index.less';
+import type { LibroAINativeForCellView } from './ai-native-for-cell-view.js';
+import type { LibroAiNativeChatView } from './libro-ai-native-chat-view.js';
+
+export const CodeBlockInCell = (props: any) => {
+  const { className, children } = props;
+  const instance = useInject<LibroAINativeForCellView>(ViewInstance);
+  const cell = instance.cell;
+
+  const replace = (data: any) => {
+    if (cell instanceof LibroCodeCellView && cell.editor) {
+      const length = cell.editor.model.value.length;
+      const start = cell.editor.getPositionAt(0);
+      const end = cell.editor.getPositionAt(length);
+      if (start && end) {
+        cell.editor.replaceSelection(data, {
+          start,
+          end,
+        });
+      }
+    }
+  };
+
+  if (!props.inline && className) {
+    const [, lang] = /language-(\w+)/.exec(className || '') || [];
+
+    return (
+      <pre className={`chat-msg-md-code-wrap`}>
+        {lang && <div className={`chat-msg-md-code-lang`}>{lang}</div>}
+        <CopyOutlined
+          onClick={() => {
+            copy(children);
+            message.success(l10n.t('代码已复制'));
+          }}
+          className={`chat-msg-md-code-copy`}
+        />
+        <SyntaxHighlighter
+          className={`libro-llm-syntax-highlighter`}
+          language={lang}
+          style={{}}
+        >
+          {typeof children === 'string' ? children.trim() : children}
+        </SyntaxHighlighter>
+        <div className="libro-ai-native-debug-code-btn-container">
+          <Button color="primary" variant="outlined" onClick={() => replace(children)}>
+            {l10n.t('替换代码')}
+          </Button>
+        </div>
+      </pre>
+    );
+  }
+
+  return <code className={`chat-msg-md-code-code`}>{children}</code>;
+};
+
+export const CodeBlockInChat = (props: any) => {
+  const { className, children } = props;
+  const instance = useInject<LibroAiNativeChatView>(ViewInstance);
+  const cell = instance.libro?.activeCell;
+
+  const insertCell = async (data: any) => {
+    const libro = instance.libro;
+    if (!libro) {
+      return;
+    }
+    const insertIndex = libro.model.cells.findIndex(
+      (c) => c.id === libro.activeCell?.id,
+    );
+    await libro.addCell(
+      {
+        id: v4(),
+        cell: { cell_type: 'code', source: data, metadata: {} },
+      },
+      insertIndex + 1,
+    );
+  };
+
+  const replace = (data: any) => {
+    if (cell instanceof LibroCodeCellView && cell.editor) {
+      const length = cell.editor.model.value.length;
+      const start = cell.editor.getPositionAt(0);
+      const end = cell.editor.getPositionAt(length);
+      if (start && end) {
+        cell.editor.replaceSelection(data, {
+          start,
+          end,
+        });
+      }
+    }
+  };
+
+  if (!props.inline && className) {
+    const [, lang] = /language-(\w+)/.exec(className || '') || [];
+
+    return (
+      <pre className={`chat-msg-md-code-wrap`}>
+        {lang && <div className={`chat-msg-md-code-lang`}>{lang}</div>}
+        <CopyOutlined
+          onClick={() => {
+            copy(children);
+            message.success(l10n.t('代码已复制'));
+          }}
+          className={`chat-msg-md-code-copy`}
+        />
+        <SyntaxHighlighter
+          className={`libro-llm-syntax-highlighter`}
+          language={lang}
+          style={{}}
+        >
+          {typeof children === 'string' ? children.trim() : children}
+        </SyntaxHighlighter>
+        <div className="libro-ai-native-debug-code-btn-container">
+          <Button
+            color="primary"
+            variant="outlined"
+            onClick={() => insertCell(children)}
+          >
+            {l10n.t('插入 Cell')}
+          </Button>
+          <Button color="primary" variant="outlined" onClick={() => replace(children)}>
+            {l10n.t('替换 Cell')}
+          </Button>
+        </div>
+      </pre>
+    );
+  }
+
+  return <code className={`chat-msg-md-code-code`}>{children}</code>;
+};