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

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-5",
   "author": "Tuya.inc",
   "license": "MIT",
   "private": false,
@@ -35,5 +35,5 @@
   "devDependencies": {
     "@types/url-parse": "^1.4.11"
   },
-  "gitHead": "c1afc79e00950e8eb385ec233f74e1677a9c93cf"
+  "gitHead": "60f0d3f33a78b76aae90b22942d9bf64c430e565"
 }