@ray-js/t-agent-plugin-aistream 0.2.0-beta-3 → 0.2.0-beta-4

package/dist/ChatHistoryStore.d.ts

@@ -36,8 +36,11 @@ export interface ChatHistoryStore<Key = number> {
     }>;
     /** 更新消息 */
     update(id: Key, body: ChatMessageObject): Promise<void>;
+    /** 删除消息 */
     remove(id: Key): Promise<void>;
-    removeAll(): Promise<void>;
+    /** 批量删除消息 */
+    removeAll(ids?: Key[]): Promise<void>;
+    /** 插入消息 */
     insert(body: ChatMessageObject): Promise<{
         id: Key;
     }>;
@@ -57,7 +60,7 @@ export declare class ChatHistoryLocalStore implements ChatHistoryStore {
     }>;
     update(id: number, body: ChatMessageObject): Promise<void>;
     remove(id: number): Promise<void>;
-    removeAll(): Promise<void>;
+    removeAll(ids?: number[]): Promise<void>;
     insert(message: ChatMessageObject): Promise<{
         id: number;
     }>;

package/dist/ChatHistoryStore.js

@@ -150,8 +150,20 @@ export class ChatHistoryLocalStore {
       id: [id]
     });
   }
-  async removeAll() {
-    await deleteRecordList(this.getQueryCondition());
+  async removeAll(ids) {
+    if (ids) {
+      if (!Array.isArray(ids)) {
+        throw new Error('ids is not array');
+      }
+      if (ids.length === 0) {
+        return;
+      }
+      await deleteRecordList(_objectSpread({
+        id: ids
+      }, this.getQueryCondition()));
+    } else {
+      await deleteRecordList(this.getQueryCondition());
+    }
   }
   async insert(message) {
     const nowTime = Date.now();

package/dist/utils/defaultMock.js

@@ -35,19 +35,20 @@ mock.hooks.hook('getMiniAppConfig', context => {
     config: {}
   };
 });
+const getCurrentConnection = () => {
+  return mock.data.get('currentConnection');
+};
 mock.hooks.hook('connect', context => {
-  const {
-    options
-  } = context;
-  const key = "connection-".concat(options.clientType, "-").concat(options.deviceId || '');
-  if (!mock.data.has(key)) {
-    const connectionId = generateId();
-    mock.data.set(key, {
-      connectionId
-    });
-    mock.data.set("connection-".concat(connectionId), key);
+  let connection = getCurrentConnection();
+  if (connection) {
+    throw new Error('already connected');
+  } else {
+    connection = {
+      connectionId: generateId()
+    };
+    mock.data.set('currentConnection', connection);
   }
-  context.result = mock.data.get(key);
+  context.result = connection;
 });
 mock.hooks.hook('disconnect', context => {
   const {
@@ -55,9 +56,15 @@ mock.hooks.hook('disconnect', context => {
       connectionId
     }
   } = context;
-  const key = mock.data.get("connection-".concat(connectionId));
-  mock.data.delete(key);
-  mock.data.delete("connection-".concat(connectionId));
+  const connection = getCurrentConnection();
+  if (!connection) {
+    throw new Error('not connected');
+  }
+  if (connection.connectionId !== connectionId) {
+    throw new Error('connectionId mismatch');
+  }
+  mock.data.set('currentConnection', null);
+  mock.data.set('sessionMap', new Map());
 });
 mock.hooks.hook('queryAgentToken', context => {
   context.result = {
@@ -107,6 +114,14 @@ mock.hooks.hook('createSession', context => {
     sendDataChannels: session.sendDataChannels,
     revDataChannels: session.revDataChannels
   };
+
+  // setTimeout(() => {
+  //   dispatch('onConnectStateChanged', {
+  //     connectionId: getCurrentConnection().connectionId,
+  //     connectState: ConnectState.DISCONNECTED,
+  //     code: 200,
+  //   });
+  // }, 1000);
 });
 mock.hooks.hook('closeSession', context => {
   const map = mock.data.get('sessionMap');
@@ -259,6 +274,7 @@ mock.hooks.hook('startRecordAndSendAudioData', context => {
     let text = '';
     controller.signal.addEventListener('abort', async () => {
       var _finishResolve;
+      // 终止识别到出完整结果的延迟
       await mock.sleep(300);
       session.replyText(StreamFlag.IN_PROGRESS, {
         bizType: ReceivedTextPacketType.ASR,
@@ -278,6 +294,8 @@ mock.hooks.hook('startRecordAndSendAudioData', context => {
       }
       (_finishResolve = finishResolve) === null || _finishResolve === void 0 || _finishResolve();
     });
+
+    // 识别 ASR 的延迟
     await mock.sleep(100);
     if (controller.signal.aborted) {
       return;

package/dist/utils/ttt.js

@@ -2,7 +2,7 @@ import { listening, promisify } from './promisify';
 import { ConnectClientType, ConnectState } from '../AIStreamTypes';
 import { generateId } from '@ray-js/t-agent';
 export const getMiniAppConfig = promisify(ty.getMiniAppConfig, true);
-export const getAccountInfo = promisify(ty.getAccountInfo, true);
+export const getAccountInfo = promisify(ty.getAccountInfo);
 export const isDevTools = () => {
   return ty.getSystemInfoSync().brand === 'devtools';
 };

package/dist/withAIStream.js

@@ -351,10 +351,19 @@ export function withAIStream() {
         payload
       } = context;
       const message = session.messages.get(payload.messageId);
+      if (!message) {
+        throw new Error('message not found');
+      }
       const {
         eventId,
         sessionId
-      } = (message === null || message === void 0 ? void 0 : message.meta) || {};
+      } = message.meta;
+      if (!eventId || !sessionId) {
+        context.result = {
+          success: true
+        };
+        return;
+      }
       await feedback({
         requestId: [eventId, sessionId].join('#'),
         type: payload.rate
@@ -443,9 +452,6 @@ export function withAIStream() {
           Array.from(messages).forEach(message => {
             message.remove();
           });
-          if (options !== null && options !== void 0 && options.indexId) {
-            return;
-          }
           // 清空缓存的数据
           const historyStore = getHistoryStore();
           if (historyStore) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ray-js/t-agent-plugin-aistream",
-  "version": "0.2.0-beta-3",
+  "version": "0.2.0-beta-4",
   "author": "Tuya.inc",
   "license": "MIT",
   "private": false,
@@ -35,5 +35,5 @@
   "devDependencies": {
     "@types/url-parse": "^1.4.11"
   },
-  "gitHead": "03c4af7fe4cfe27069db83d99a61042c8053399e"
+  "gitHead": "1728f8a1cac6b2ca53e506e06171808fbf38efdc"
 }