@ray-js/t-agent-plugin-aistream 0.2.6-beta-3 → 0.2.6-beta-5

package/dist/utils/defaultMock.js

@@ -272,7 +272,8 @@ mock.hooks.hook('sendEventEnd', async context => {
     }
     const ctx = {
       data: event.data,
-      responseText: ''
+      responseText: '',
+      wordDelayMs: 100
     };
     const [error] = await tryCatch(() => mock.hooks.callHook('sendToAIStream', ctx));
     if (error) {
@@ -297,6 +298,7 @@ mock.hooks.hook('sendEventEnd', async context => {
       return;
     }
     const text = ctx.responseText || '⚠️ No mock text response matched!';
+    const wordDelayMs = ctx.wordDelayMs || 100;
     const words = splitString(text);
     if (event.controller.signal.aborted || session.closed) {
       return;
@@ -309,7 +311,7 @@ mock.hooks.hook('sendEventEnd', async context => {
     const bizId = generateId();
     event.replyText(StreamFlag.START);
     for (const word of words) {
-      await mock.sleep(100);
+      await mock.sleep(wordDelayMs);
       if (event.controller.signal.aborted || session.closed) {
         return;
       }

package/dist/utils/mock.d.ts

@@ -15,6 +15,7 @@ export interface SendToAIStreamContext {
         userData?: Attribute[];
     }>;
     responseText: string | undefined;
+    wordDelayMs: number;
     responseSkills?: ReceivedTextSkillPacketBody[];
 }
 interface AsrDetectionContext {

package/dist/withAIStream.d.ts

@@ -1,4 +1,4 @@
-import { ChatAgent, ChatCardObject, ChatMessage, ChatTile, GetChatPluginHandler, InputBlock } from '@ray-js/t-agent';
+import { ChatAgent, ChatCardObject, ChatMessage, ChatMessageStatus, ChatTile, GetChatPluginHandler, InputBlock } from '@ray-js/t-agent';
 import { TTTAction } from './utils';
 import { ConnectClientType, ReceivedTextSkillPacketBody } from './AIStreamTypes';
 import { ChatHistoryStore, StoredMessageObject } from './ChatHistoryStore';
@@ -42,6 +42,9 @@ export interface AIStreamHooks {
     onMessageParse: (msgItem: StoredMessageObject, result: {
         messages: ChatMessage[];
     }) => void;
+    onTextCompose: (respMsg: ChatMessage, status: ChatMessageStatus, result: {
+        text: string;
+    }) => void;
     onSkillCompose: (skill: ReceivedTextSkillPacketBody[], respMsg: ChatMessage, result: {
         messages: ChatMessage[];
     }) => void;
@@ -78,6 +81,7 @@ export declare function withAIStream(options?: AIStreamOptions): (agent: ChatAge
             data: string;
         }>;
         onSkillCompose: (fn: AIStreamHooks['onSkillCompose']) => () => void;
+        onTextCompose: (fn: AIStreamHooks['onTextCompose']) => () => void;
         onSkillsEnd: (fn: AIStreamHooks['onSkillsEnd']) => () => void;
         onCardsReceived: (fn: AIStreamHooks['onCardsReceived']) => () => void;
         onTTTAction: (fn: AIStreamHooks['onTTTAction']) => () => void;

package/dist/withAIStream.js

@@ -354,6 +354,13 @@ export function withAIStream() {
       logger.debug('withAIStream chat agent.flushStreamToShow');
       const result = {
         messages: await agent.flushStreamToShow(message, response, {
+          textCompose: async (message, text, status) => {
+            const result = {
+              text
+            };
+            await hooks.callHook('onTextCompose', message, status, result);
+            return result.text;
+          },
           attachmentCompose: async (respMsg, part) => {
             const result = {
               messages: []
@@ -555,6 +562,9 @@ export function withAIStream() {
         onSkillCompose: fn => {
           return hooks.hook('onSkillCompose', fn);
         },
+        onTextCompose: fn => {
+          return hooks.hook('onTextCompose', fn);
+        },
         onSkillsEnd,
         onCardsReceived: fn => {
           return hooks.hook('onCardsReceived', fn);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ray-js/t-agent-plugin-aistream",
-  "version": "0.2.6-beta-3",
+  "version": "0.2.6-beta-5",
   "author": "Tuya.inc",
   "license": "MIT",
   "private": false,
@@ -35,5 +35,5 @@
   "devDependencies": {
     "@types/url-parse": "^1.4.11"
   },
-  "gitHead": "e478ba7ce360563e061efc24d6a8dfb495c5f9e9"
+  "gitHead": "60f0d3f33a78b76aae90b22942d9bf64c430e565"
 }