@ray-js/t-agent-plugin-aistream 0.2.6 → 0.2.7-beta.10

package/dist/withAIStream.js

@@ -14,6 +14,8 @@ import { BizCode, ConnectClientType } from './AIStreamTypes';
 import { DEFAULT_TOKEN_API, DEFAULT_TOKEN_API_VERSION, globalAIStreamClient } from './global';
 import logger from './utils/logger';
 import { ChatHistoryLocalStore } from './ChatHistoryLocalStore';
+import { deepMerge } from './utils/object';
+import { trackEvent } from './utils/track';
 export function withAIStream() {
   let options = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
   const hooks = createHooks();
@@ -60,6 +62,10 @@ export function withAIStream() {
       if (clientType === ConnectClientType.APP && deviceId) {
         throw new Error('deviceId is not allowed when clientType is app');
       }
+      trackEvent(agentId, 'on_agent_start', {
+        clientType,
+        tokenApi: "".concat(tokenOptions.api, "_").concat(tokenOptions.version)
+      });
       let homeId = options.homeId;
       if (clientType === ConnectClientType.DEVELOPER) {
         homeId = 1;
@@ -88,7 +94,14 @@ export function withAIStream() {
         extParams: _objectSpread({
           needTts: !!options.enableTts,
           deviceId
-        }, tokenOptions.extParams)
+        }, tokenOptions.extParams),
+        getSessionUserData: async () => {
+          const result = {
+            userData: {}
+          };
+          await hooks.callHook('onUserDataRead', 'create-session', {}, result);
+          return result.userData;
+        }
       });
       await session.set('AIStream.streamSession', streamSession);
       if (options.earlyStart) {
@@ -221,14 +234,25 @@ export function withAIStream() {
         };
       }
     });
-    const send = (blocks, signal, extraOptions) => {
+    const send = (blocks, signal, eventUserData) => {
       const streamSession = session.get('AIStream.streamSession');
       const result = sendBlocksToAIStream({
         blocks,
         session: streamSession,
         signal,
         eventIdPrefix: options.eventIdPrefix,
-        attribute: _objectSpread({}, extraOptions)
+        getUserData: async () => {
+          const userDataResult = {
+            userData: {}
+          };
+          await hooks.callHook('onUserDataRead', 'start-event', {
+            blocks
+          }, userDataResult);
+          const userData = {};
+          deepMerge(userData, userDataResult.userData);
+          deepMerge(userData, eventUserData);
+          return userData;
+        }
       });
       signal === null || signal === void 0 || signal.addEventListener('abort', event => {
         logger.debug('withAIStream signal aborted, response.started:', result.response.started);
@@ -245,10 +269,11 @@ export function withAIStream() {
       const {
         sendBy = 'user',
         responseBy = 'assistant',
-        extraOptions
+        userData
       } = options || {};
       let audioEmitter = null;
-      for (const block of blocks) {
+      for (let i = 0; i < blocks.length; i++) {
+        const block = blocks[i];
         if (block.type === 'audio') {
           if (audioEmitter) {
             throw new Error('only one audio emitter is allowed');
@@ -345,7 +370,7 @@ export function withAIStream() {
       const {
         response,
         metaPromise
-      } = send(blocks, signal, extraOptions);
+      } = send(blocks, signal, userData);
       if (audioPromise) {
         try {
           await audioPromise;
@@ -399,7 +424,8 @@ export function withAIStream() {
       if (valid) {
         await message.persist();
         const [_, ...rest] = result.messages;
-        for (const m of rest) {
+        for (let i = 0; i < rest.length; i++) {
+          const m = rest[i];
           await m.show();
           await m.persist();
         }
@@ -531,8 +557,9 @@ export function withAIStream() {
     };
     onSkillsEnd(async (skills, respMsg, result) => {
       const cards = [];
-      for (const skill of skills) {
+      for (let i = 0; i < skills.length; i++) {
         var _skill$custom, _skill$general;
+        const skill = skills[i];
         if ((_skill$custom = skill.custom) !== null && _skill$custom !== void 0 && (_skill$custom = _skill$custom.data) !== null && _skill$custom !== void 0 && _skill$custom.aiCards) {
           for (const card of skill.custom.data.aiCards) {
             cards.push(card);
@@ -548,7 +575,8 @@ export function withAIStream() {
         cards: cards
       };
       await hooks.callHook('onCardsReceived', skills, r);
-      for (const card of cards) {
+      for (let i = 0; i < cards.length; i++) {
+        const card = cards[i];
         const m = createMessage({
           role: respMsg.role,
           status: ChatMessageStatus.FINISH,
@@ -585,6 +613,9 @@ export function withAIStream() {
         onCardsReceived: fn => {
           return hooks.hook('onCardsReceived', fn);
         },
+        onUserDataRead: fn => {
+          return hooks.hook('onUserDataRead', fn);
+        },
         onTTTAction: fn => {
           return hooks.hook('onTTTAction', fn);
         },

package/package.json

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