@fe-free/ai 4.1.8 → 4.1.9

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @fe-free/ai
 
+## 4.1.9
+
+### Patch Changes
+
+- feat: ai
+  - @fe-free/core@4.1.9
+  - @fe-free/icons@4.1.9
+  - @fe-free/tool@4.1.9
+
 ## 4.1.8
 
 ### Patch Changes

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fe-free/ai",
-  "version": "4.1.8",
+  "version": "4.1.9",
   "description": "",
   "main": "./src/index.ts",
   "author": "",
@@ -15,7 +15,7 @@
     "lodash-es": "^4.17.21",
     "uuid": "^13.0.0",
     "zustand": "^4.5.7",
-    "@fe-free/core": "4.1.8"
+    "@fe-free/core": "4.1.9"
   },
   "peerDependencies": {
     "@ant-design/x-sdk": "^2.1.3",
@@ -26,8 +26,8 @@
     "i18next-icu": "^2.4.1",
     "react": "^19.2.0",
     "react-i18next": "^16.4.0",
-    "@fe-free/tool": "4.1.8",
-    "@fe-free/icons": "4.1.8"
+    "@fe-free/tool": "4.1.9",
+    "@fe-free/icons": "4.1.9"
   },
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1",

package/src/ai.stories.tsx

@@ -9,6 +9,7 @@ import {
 } from '@fe-free/ai';
 import { sleep } from '@fe-free/tool';
 import type { Meta } from '@storybook/react-vite';
+import { set } from 'lodash-es';
 import { useCallback, useEffect, useMemo } from 'react';
 
 const meta: Meta = {
@@ -85,11 +86,6 @@ function Component() {
       user: {
         ...v,
       },
-      ai: {
-        data: {
-          text: '',
-        },
-      },
     };
 
     addMessage(message);
@@ -99,7 +95,9 @@ function Component() {
       onUpdate: ({ event, data }) => {
         if (event === 'message') {
           message.status = EnumChatMessageStatus.STREAMING;
-          message.ai!.data!.text += data;
+
+          const preText = message.ai?.data?.text || '';
+          set(message, 'ai.data.text', preText + data);
 
           updateMessage({
             ...message,
@@ -131,12 +129,12 @@ function Component() {
       >
         <Messages
           messages={messages}
-          renderMessageOfUser={(message) => (
+          renderMessageOfUser={({ message }) => (
             <div className="p-2">
               <div className="rounded-xl bg-primary p-2 text-white">{message.user?.text}</div>
             </div>
           )}
-          renderMessageOfAI={(message) => (
+          renderMessageOfAI={({ message }) => (
             <div className="p-2">
               <div>{message.status}</div>
               <pre className="whitespace-pre-wrap">{message.ai?.data?.text}</pre>

package/src/m_sender/record.tsx

@@ -159,7 +159,7 @@ function RecordAction(props: MSenderProps & { setType }) {
 
   return (
     <div
-      className={classNames('absolute inset-0 flex items-center justify-center rounded-full', {
+      className={classNames('absolute inset-0 flex items-center justify-center rounded-xl', {
         'bg-white': !isRecording,
         'bg-red-500': isRecording && isCancel,
         'bg-primary': isRecording && !isCancel,

package/src/messages/index.tsx

@@ -4,8 +4,8 @@ import type { ChatMessage } from '../store/types';
 
 interface MessagesProps<AIData> {
   messages?: ChatMessage<AIData>[];
-  renderMessageOfUser?: (message: ChatMessage<AIData>) => React.ReactNode;
-  renderMessageOfAI?: (message: ChatMessage<AIData>) => React.ReactNode;
+  renderMessageOfUser?: (props: { message: ChatMessage<AIData> }) => React.ReactNode;
+  renderMessageOfAI?: (props: { message: ChatMessage<AIData> }) => React.ReactNode;
 }
 
 function Messages<AIData>(props: MessagesProps<AIData>) {
@@ -55,8 +55,8 @@ function Messages<AIData>(props: MessagesProps<AIData>) {
       <div ref={ref} className="flex h-full flex-col overflow-y-auto">
         {messages?.map((message) => (
           <div key={message.uuid} data-uuid={message.uuid} className="flex flex-col">
-            <div className="flex justify-end">{renderMessageOfUser?.(message)}</div>
-            <div className="flex justify-start">{renderMessageOfAI?.(message)}</div>
+            <div className="flex justify-end">{renderMessageOfUser?.({ message })}</div>
+            <div className="flex justify-start">{renderMessageOfAI?.({ message })}</div>
           </div>
         ))}
       </div>

package/src/sender/index.tsx

@@ -100,7 +100,7 @@ function Sender(originProps: SenderProps) {
       <div
         ref={refContainer}
         className={classNames(
-          'fea-sender relative flex flex-col rounded-lg border border-01 bg-white p-2',
+          'fea-sender relative flex flex-col rounded-xl border border-01 bg-white p-2',
           {
             'fea-sender-drag-hover': dragHover,
           },