plain-design 1.0.0-beta.156 → 1.0.0-beta.158

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "plain-design",
-  "version": "1.0.0-beta.156",
+  "version": "1.0.0-beta.158",
   "description": "",
   "main": "dist/plain-design.min.js",
   "module": "dist/plain-design.commonjs.min.js",

package/src/packages/components/AiChatBox/index.tsx

@@ -7,11 +7,12 @@ import Button from "../Button";
 import './ai-chat-box.scss';
 import $message from "../$message";
 import {uuid} from "@peryl/utils/uuid";
-import {$ai, iAiChatResponse, iAiConfiguration} from "../$ai";
+import {$ai, iAiChatResponse, iAiConfiguration, iAiHistory} from "../$ai";
 import $configuration from "../$configuration";
 import Scroll from "../Scroll";
 import {delay} from "@peryl/utils/delay";
 import {toArray} from "@peryl/utils/toArray";
+import {lastItem} from "@peryl/utils/lastItem";
 
 export const AiChatBox = designComponent({
   props: {
@@ -26,7 +27,8 @@ export const AiChatBox = designComponent({
     /*系统提示词*/
     systemContent: { type: String },
     /*手动处理大模型返回的message*/
-    handleMessage: { type: Function as PropType<(param: { message: string, userContent: string, chatData: iAiChatResponse, addHistory: (history: iChatBoxHistory) => void }) => void | iChatBoxHistory | iChatBoxHistory[] | Promise<void | iChatBoxHistory | iChatBoxHistory[]>> },
+    handleMessage: { type: Function as PropType<(param: { message: string, userContent: string, chatData?: iAiChatResponse, addHistory: (history: iChatBoxHistory) => void }) => void | iChatBoxHistory | iChatBoxHistory[] | Promise<void | iChatBoxHistory | iChatBoxHistory[]>> },
+    handleReceive: { type: Function as PropType<(data: { fullText?: string, chunkText?: string }) => void> },
     inputMinHeight: { type: String, default: '80px' },
     inputMaxHeight: { type: String, default: '225px' },
   },
@@ -61,25 +63,56 @@ export const AiChatBox = designComponent({
           refs.scroll.methods.scrollEnd();
         }
       },
-      addHistory: async (history: iChatBoxHistory) => {
-        state.histories.push(history);
+      addHistory: async (...history: iChatBoxHistory[]) => {
+        state.histories.push(...history);
         await methods.scrollEnd();
       },
       send: async (userContent: string) => {
         await delay(23);
         state.userContent = '';
-        state.histories.push({ role: 'user', content: userContent, id: uuid() });
-        await methods.scrollEnd();
+        await methods.addHistory({ role: 'user', content: userContent, id: uuid() });
 
-        const { message, chatData } = await $ai.chatSystem(userContent, props.systemContent, aiConfig);
+        /*const { message, chatData } = await $ai.chatSystem(userContent, props.systemContent, aiConfig);*/
+        const messages: iAiHistory[] = [{ role: 'user', content: userContent }];
+        if (!!props.systemContent?.trim().length) {messages.unshift({ role: 'system', content: props.systemContent });}
+        let chatBoxHistory: iChatBoxHistory | undefined = undefined;
+        let message: string | undefined = undefined;
+        // console.log('start chat stream');
+        await $ai.chatStream({
+          aiConfig,
+          messages,
+          onReceiving: ({ fullText, chunkText }) => {
+            // console.log('receiving', fullText);
+            props.handleReceive?.({ fullText, chunkText });
+            if (!chatBoxHistory) {
+              state.histories.push({ role: 'assistant', content: '', id: uuid() });
+              chatBoxHistory = lastItem(state.histories);
+            }
+            chatBoxHistory!.content = fullText;
+            methods.scrollEnd();
+          },
+          onFinish: (fullText) => {
+            chatBoxHistory!.content = fullText;
+            message = fullText;
+          },
+        });
+        if (!message) {
+          throw new Error('message is empty!');
+        }
 
         /*可以手动处理返回的消息，自定义往对话框插入内容*/
-        const newHistory: iChatBoxHistory | iChatBoxHistory[] = await props.handleMessage?.({ message, userContent, chatData, addHistory: methods.addHistory }) || ({ role: 'assistant', content: message, id: uuid() });
+        const newHistory: void | iChatBoxHistory | iChatBoxHistory[] = await props.handleMessage?.({
+          message,
+          userContent,
+          // chatData,
+          addHistory: methods.addHistory
+        });
 
-        state.histories.push(...toArray(newHistory));
-        await methods.scrollEnd();
+        if (newHistory != null) {
+          await methods.addHistory(...toArray(newHistory));
+        }
 
-        return { message, chatData };
+        return { message, /*chatData*/ };
       },
     };
 

package/src/packages/components/AutoTable/use/useTableOption.ai.tsx

@@ -180,7 +180,7 @@ export const useTableOptionAi = AutoModule.createRegistration((option) => {
           await methods.apply(operateItems);
           const workTitleMap = { search: '查询', sort: '排序', config: '配置', error: '' };
           return [
-            { id: uuid(), role: 'assistant', content: message },
+            // { id: uuid(), role: 'assistant', content: message },
             { id: uuid(), role: 'assistant', content: `已经为您执行 ${operateItems.map(i => workTitleMap[i.type]).join(',')} 工作` }
           ];
         } catch (e) {