@lobehub/ui 1.50.1 → 1.51.0

package/es/Chat/types.d.ts

@@ -1,5 +1,5 @@
 export declare type InternalChatContext = Omit<ChatContext, 'id'>;
-export declare type MessageRoleType = 'user' | 'system' | 'assistant';
+export declare type MessageRoleType = 'user' | 'system' | 'assistant' | 'function';
 /**
  * 聊天消息错误对象
  */

package/es/ChatItem/index.d.ts

@@ -33,7 +33,7 @@ export interface ChatItemProps {
     /**
      * @description The message content of the chat item
      */
-    message?: string;
+    message?: ReactNode;
     messageExtra?: ReactNode;
     /**
      * @description Callback when the message content changes
@@ -54,6 +54,7 @@ export interface ChatItemProps {
      * @description Whether the chat item is primary
      */
     primary?: boolean;
+    renderMessage?: (content: ReactNode) => ReactNode;
     /**
      * @description Whether to show the title of the chat item
      */

package/es/ChatItem/index.js

@@ -1,6 +1,6 @@
 import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
 import _objectWithoutProperties from "@babel/runtime/helpers/esm/objectWithoutProperties";
-var _excluded = ["actions", "className", "primary", "borderSpacing", "loading", "message", "placement", "type", "avatar", "alert", "showTitle", "time", "editing", "onChange", "onEditingChange", "messageExtra"];
+var _excluded = ["actions", "className", "primary", "borderSpacing", "loading", "message", "placement", "type", "avatar", "alert", "showTitle", "time", "editing", "onChange", "onEditingChange", "messageExtra", "renderMessage"];
 function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }
 function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { _defineProperty(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }
 import { Alert } from 'antd';
@@ -34,6 +34,7 @@ var ChatItem = /*#__PURE__*/memo(function (_ref) {
     onChange = _ref.onChange,
     onEditingChange = _ref.onEditingChange,
     messageExtra = _ref.messageExtra,
+    renderMessage = _ref.renderMessage,
     properties = _objectWithoutProperties(_ref, _excluded);
   var _useStyles = useStyles({
       avatarSize: AVATAR_SIZE,
@@ -45,6 +46,17 @@ var ChatItem = /*#__PURE__*/memo(function (_ref) {
     }),
     cx = _useStyles.cx,
     styles = _useStyles.styles;
+  var content = /*#__PURE__*/_jsx(EditableMessage, {
+    classNames: {
+      input: styles.editingInput
+    },
+    editButtonSize: 'small',
+    editing: editing,
+    onChange: onChange,
+    onEditingChange: onEditingChange,
+    value: String(message || '...')
+  });
+  var messageContent = renderMessage ? renderMessage(content) : content;
   return /*#__PURE__*/_jsxs("div", _objectSpread(_objectSpread({
     className: cx(styles.container, className)
   }, properties), {}, {
@@ -80,16 +92,7 @@ var ChatItem = /*#__PURE__*/memo(function (_ref) {
           showIcon: true
         }, alert)) : /*#__PURE__*/_jsxs(Flexbox, {
           className: cx(styles.message, editing && styles.editingContainer),
-          children: [/*#__PURE__*/_jsx(EditableMessage, {
-            classNames: {
-              input: styles.editingInput
-            },
-            editButtonSize: 'small',
-            editing: editing,
-            onChange: onChange,
-            onEditingChange: onEditingChange,
-            value: String(message || '...')
-          }), messageExtra && !editing ? /*#__PURE__*/_jsx("div", {
+          children: [messageContent, messageExtra && !editing ? /*#__PURE__*/_jsx("div", {
             className: styles.messageExtra,
             children: messageExtra
           }) : null]

package/es/ChatList/index.d.ts

@@ -2,7 +2,8 @@ import { ReactNode } from 'react';
 import { ChatItemProps } from "../ChatItem";
 import type { DivProps } from "../types";
 import { ChatMessage } from "../types/chatMessage";
-declare type MessageExtra = (props: ChatMessage) => ReactNode;
+export declare type MessageExtra = (props: ChatMessage) => ReactNode;
+export declare type RenderMessage = (content: ReactNode, message: ChatMessage) => ReactNode;
 export interface ChatListProps extends DivProps {
     /**
      * @description Data of chat messages to be displayed
@@ -15,6 +16,7 @@ export interface ChatListProps extends DivProps {
      */
     onActionClick?: (actionKey: string, messageId: string) => void;
     onMessageChange?: (id: string, content: string) => void;
+    renderMessage?: RenderMessage;
     renderMessageExtra?: MessageExtra;
     /**
      * @description Whether to show name of the chat item
@@ -31,6 +33,7 @@ export interface ItemProps extends ChatMessage {
     MessageExtra?: MessageExtra;
     onActionClick?: (actionKey: string, messageId: string) => void;
     onMessageChange?: (id: string, content: string) => void;
+    renderMessage?: RenderMessage;
     showTitle?: boolean;
     type: 'docs' | 'chat';
 }

package/es/ChatList/index.js

@@ -1,13 +1,13 @@
 import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
 import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
 import _objectWithoutProperties from "@babel/runtime/helpers/esm/objectWithoutProperties";
-var _excluded = ["MessageExtra", "showTitle", "onActionClick", "onMessageChange", "type"],
-  _excluded2 = ["onActionClick", "renderMessageExtra", "className", "data", "type", "showTitle", "onMessageChange"];
+var _excluded = ["MessageExtra", "showTitle", "onActionClick", "onMessageChange", "type", "renderMessage"],
+  _excluded2 = ["onActionClick", "renderMessageExtra", "className", "data", "type", "showTitle", "onMessageChange", "renderMessage"];
 function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }
 function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { _defineProperty(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }
 import { App } from 'antd';
 import copy from 'copy-to-clipboard';
-import { memo, useState } from 'react';
+import { memo, useCallback, useState } from 'react';
 import ChatItem from "../ChatItem";
 import ActionsBar from "./ActionsBar";
 import { useStyles } from "./style";
@@ -18,6 +18,7 @@ var Item = function Item(_ref) {
     _onActionClick = _ref.onActionClick,
     onMessageChange = _ref.onMessageChange,
     type = _ref.type,
+    renderMessage = _ref.renderMessage,
     item = _objectWithoutProperties(_ref, _excluded);
   var renderMessageExtra = MessageExtra ? /*#__PURE__*/_jsx(MessageExtra, _objectSpread({}, item)) : undefined;
   var _useState = useState(false),
@@ -26,6 +27,9 @@ var Item = function Item(_ref) {
     setEditing = _useState2[1];
   var _App$useApp = App.useApp(),
     message = _App$useApp.message;
+  var innerRenderMessage = useCallback(function (content) {
+    return renderMessage === null || renderMessage === void 0 ? void 0 : renderMessage(content, item);
+  }, [renderMessage]);
   return /*#__PURE__*/_jsx(ChatItem, {
     actions: /*#__PURE__*/_jsx(ActionsBar, {
       onActionClick: function onActionClick(actionKey) {
@@ -56,6 +60,7 @@ var Item = function Item(_ref) {
     onEditingChange: setEditing,
     placement: type === 'chat' ? item.role === 'user' ? 'right' : 'left' : 'left',
     primary: item.role === 'user',
+    renderMessage: renderMessage ? innerRenderMessage : undefined,
     showTitle: showTitle,
     time: item.updateAt || item.createAt,
     type: type === 'chat' ? 'block' : 'pure'
@@ -70,6 +75,7 @@ var ChatList = /*#__PURE__*/memo(function (_ref2) {
     type = _ref2$type === void 0 ? 'chat' : _ref2$type,
     showTitle = _ref2.showTitle,
     onMessageChange = _ref2.onMessageChange,
+    renderMessage = _ref2.renderMessage,
     props = _objectWithoutProperties(_ref2, _excluded2);
   var _useStyles = useStyles(),
     cx = _useStyles.cx,
@@ -82,6 +88,7 @@ var ChatList = /*#__PURE__*/memo(function (_ref2) {
         MessageExtra: MessageExtra,
         onActionClick: onActionClick,
         onMessageChange: onMessageChange,
+        renderMessage: renderMessage,
         showTitle: showTitle,
         type: type
       }, item), item.id);

package/es/types/llm.d.ts

@@ -34,7 +34,7 @@ export interface LMParameters {
      */
     top_p?: number;
 }
-export declare type LLMRoleType = 'user' | 'system' | 'assistant';
+export declare type LLMRoleType = 'user' | 'system' | 'assistant' | 'function';
 export interface LLMMessage {
     content: string;
     role: LLMRoleType;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/ui",
-  "version": "1.50.1",
+  "version": "1.51.0",
   "description": "Lobe UI is an open-source UI component library for building chatbot web apps",
   "keywords": [
     "lobehub",