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

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

@@ -42,6 +42,7 @@ export interface AIStreamHooks {
     onMessageParse: (msgItem: StoredMessageObject, result: {
         messages: ChatMessage[];
     }) => void;
+    onChatMessageSent: (userMessage: ChatMessage, respMessage: ChatMessage) => void;
     onTextCompose: (respMsg: ChatMessage, status: ChatMessageStatus, result: {
         text: string;
     }) => void;
@@ -81,6 +82,7 @@ export declare function withAIStream(options?: AIStreamOptions): (agent: ChatAge
             data: string;
         }>;
         onSkillCompose: (fn: AIStreamHooks['onSkillCompose']) => () => void;
+        onChatMessageSent: (fn: AIStreamHooks['onChatMessageSent']) => () => void;
         onTextCompose: (fn: AIStreamHooks['onTextCompose']) => () => void;
         onSkillsEnd: (fn: AIStreamHooks['onSkillsEnd']) => () => void;
         onCardsReceived: (fn: AIStreamHooks['onCardsReceived']) => () => void;

package/dist/withAIStream.js

@@ -350,6 +350,7 @@ export function withAIStream() {
       const message = createMessage({
         role: responseBy
       });
+      await hooks.callHook('onChatMessageSent', userMsg, message);
       const skills = [];
       logger.debug('withAIStream chat agent.flushStreamToShow');
       const result = {
@@ -562,6 +563,9 @@ export function withAIStream() {
         onSkillCompose: fn => {
           return hooks.hook('onSkillCompose', fn);
         },
+        onChatMessageSent: fn => {
+          return hooks.hook('onChatMessageSent', fn);
+        },
         onTextCompose: fn => {
           return hooks.hook('onTextCompose', fn);
         },

package/package.json

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