@ray-js/t-agent-plugin-aistream 0.2.6-beta-4 → 0.2.6-beta-6

package/dist/utils/promisify.js

@@ -79,19 +79,19 @@ export function promisify(fn) {
 export function listening(on, off) {
   let enableMock = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : false;
   return listener => {
+    const cb = event => {
+      logger.debug('listening on event', on.name, event.detail);
+      listener(event.detail);
+    };
+    on(listener);
     if (getEnableMock() && enableMock) {
-      const cb = event => {
-        logger.debug('listening on event', on.name, event.detail);
-        listener(event.detail);
-      };
       mock.emitter.addEventListener(on.name, cb);
-      return () => {
-        mock.emitter.removeEventListener(on.name, cb);
-      };
     }
-    on(listener);
     return () => {
       off(listener);
+      if (getEnableMock() && enableMock) {
+        mock.emitter.removeEventListener(on.name, cb);
+      }
     };
   };
 }

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-4",
+  "version": "0.2.6-beta-6",
   "author": "Tuya.inc",
   "license": "MIT",
   "private": false,
@@ -35,5 +35,5 @@
   "devDependencies": {
     "@types/url-parse": "^1.4.11"
   },
-  "gitHead": "c1afc79e00950e8eb385ec233f74e1677a9c93cf"
+  "gitHead": "2f18554f908818b62ab24ead773d4e74fd5ef7fc"
 }