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

package/dist/utils/AIStream.js

@@ -253,9 +253,10 @@ export class AIStreamSession {
         bizTag: BizTag.DEFAULT,
         ownerId
       }, this.options), {}, {
-        extParams: _objectSpread({
+        extParams: _objectSpread(_objectSpread({}, this.options.extParams), {}, {
+          // chatId 之前可能是 undefined，这里要覆盖掉
           chatId
-        }, this.options.extParams)
+        })
       });
       const [error, result] = await tryCatchTTT(() => queryAgentToken(options));
       if (error) {

package/dist/utils/defaultMock.js

@@ -95,20 +95,34 @@ mock.hooks.hook('disconnect', context => {
   mock.data.set('sessionMap', new Map());
 });
 mock.hooks.hook('queryAgentToken', context => {
-  var _context$options$extP, _context$options$extP2;
+  var _context$options$extP, _context$options$extP2, _context$options$extP3;
   const agentToken = generateId();
   const map = mock.data.get('tokenMap');
   map.set(agentToken, {
     onlyAsr: ((_context$options$extP = context.options.extParams) === null || _context$options$extP === void 0 ? void 0 : _context$options$extP.onlyAsr) || false,
     needTts: ((_context$options$extP2 = context.options.extParams) === null || _context$options$extP2 === void 0 ? void 0 : _context$options$extP2.needTts) || false
   });
+  let chatId;
+  if (((_context$options$extP3 = context.options.extParams) === null || _context$options$extP3 === void 0 ? void 0 : _context$options$extP3.dialogueMode) === 1) {
+    var _context$options$extP4;
+    if ((_context$options$extP4 = context.options.extParams) !== null && _context$options$extP4 !== void 0 && _context$options$extP4.chatId) {
+      // 如果参数有传吗，则使用传入的 chatId
+      chatId = context.options.extParams.chatId;
+    } else {
+      // 使用生成的 id
+      chatId = "chatId-".concat(generateId());
+    }
+  }
   context.result = {
     agentToken,
     bizConfig: {
       bizCode: BizCode.CHAT,
       sendData: ['audio', 'video', 'text', 'image'],
       revData: ['text', 'audio']
-    }
+    },
+    extParams: JSON.stringify({
+      chatId
+    })
   };
 });
 const dispatch = (type, detail) => {
@@ -258,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) {
@@ -283,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;
@@ -295,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/utils/promisify.js

@@ -1,4 +1,5 @@
 import _objectSpread from "@babel/runtime/helpers/esm/objectSpread2";
+import "core-js/modules/es.global-this.js";
 import { isDevTools } from './ttt';
 import logger from './logger';
 import { mock } from './mock';
@@ -6,6 +7,9 @@ import { TTTError } from './errors';
 let callId = 100000;
 export const getEnableMock = () => {
   try {
+    if ((globalThis === null || globalThis === void 0 ? void 0 : globalThis.__AIStreamEnableMock) != null) {
+      return globalThis.__AIStreamEnableMock;
+    }
     const result = ty.getStorageSync({
       key: 'AIStreamEnableMock'
     });

package/package.json

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