@ray-js/t-agent 0.0.8-beta-3 → 0.0.9-beta-1

package/README-zh_CN.md

@@ -1388,3 +1388,4 @@ const renderOptions = {
 | t-agent.input.voice.require-permission             | MessageInput 切换语音输入 | 需要授权录音权限     |
 | t-agent.input.upload.failed                        | MessageInput 上传文件     | 文件上传失败         |
 | t-agent.message.feedback.success                   | BubbleTile 消息评价       | 反馈成功             |
+| t-agent-bubble-tile-aborted                        | BubbleTile 消息           | 用户中断             |

package/README.md

@@ -1383,3 +1383,4 @@ Below are the built-in multilingual keys:
 | t-agent.input.voice.require-permission             | MessageInput switch to voice input | Need to grant recording permission |
 | t-agent.input.upload.failed                        | MessageInput file upload           | File upload failed                 |
 | t-agent.message.feedback.success                   | BubbleTile message feedback        | Feedback successful                |
+| t-agent-bubble-tile-aborted                        | BubbleTile message                 | User aborted                       |

package/dist/chat/ChatAgent.js

@@ -213,8 +213,7 @@ export default class ChatAgent {
                   await message.set({
                     status: ChatMessageStatus.UPDATING
                   }).bubble.setText(text).update();
-                }
-                if (part.type === 'attachment' && attachmentCompose) {
+                } else if (part.type === 'attachment' && attachmentCompose) {
                   const list = (await attachmentCompose(message, part)) || [];
                   for (const m of list) {
                     m.set({
@@ -223,12 +222,23 @@ export default class ChatAgent {
                     await m.show();
                     messages.push(m);
                   }
-                }
-                if (part.type === 'error') {
+                } else if (part.type === 'error') {
                   if (!message.bubble.text) {
                     message.bubble.setText(part.error.message);
                   }
                   await message.bubble.setStatus(part.level === 'error' ? BubbleTileStatus.ERROR : BubbleTileStatus.WARNING).setInfo(part.error.message).update();
+                } else if (part.type === 'abort') {
+                  // 如果只有一条响应消息，则直接设置为中断状态
+                  if (messages.length === 1) {
+                    if (message.status === ChatMessageStatus.UPDATING) {
+                      await message.bubble.setStatus(BubbleTileStatus.ABORTED).update();
+                    } else {
+                      await message.remove();
+                      messages.pop();
+                    }
+                  }
+                  // stop update
+                  break;
                 }
               }
             }

package/dist/chat/StreamResponse.d.ts

@@ -1,6 +1,8 @@
 import { MessagePart, StreamResponseInstance } from './types';
 export default class StreamResponse implements StreamResponseInstance {
     private partStream;
+    private reader;
+    private aborted;
     /**
      * Create a new StreamResponse instance.
      *

package/dist/chat/StreamResponse.js

@@ -1,3 +1,4 @@
+import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
 import _awaitAsyncGenerator from "@babel/runtime/helpers/esm/awaitAsyncGenerator";
 import _wrapAsyncGenerator from "@babel/runtime/helpers/esm/wrapAsyncGenerator";
 import "core-js/modules/es.symbol.async-iterator.js";
@@ -8,6 +9,8 @@ export default class StreamResponse {
    * @param partStream
    */
   constructor(partStream) {
+    _defineProperty(this, "reader", null);
+    _defineProperty(this, "aborted", false);
     this.partStream = partStream;
   }
 
@@ -15,16 +18,41 @@ export default class StreamResponse {
    * Get all parts of the response as an async iterator.
    */
   parts() {
+    if (this.reader) {
+      throw new Error('parts() can only be called once');
+    }
     const reader = this.partStream.getReader();
+    this.reader = reader;
+
+    // eslint-disable-next-line @typescript-eslint/no-this-alias
+    const resp = this;
     return {
       [Symbol.asyncIterator]() {
         return _wrapAsyncGenerator(function* () {
           try {
             while (true) {
+              if (resp.aborted) {
+                yield {
+                  type: 'abort',
+                  level: 'warn',
+                  error: new Error('User aborted'),
+                  meta: {}
+                };
+                break;
+              }
               const {
                 done,
                 value
               } = yield _awaitAsyncGenerator(reader.read());
+              if (resp.aborted) {
+                yield {
+                  type: 'abort',
+                  level: 'warn',
+                  error: new Error('User aborted'),
+                  meta: {}
+                };
+                break;
+              }
               if (done) break;
               yield value;
             }
@@ -42,6 +70,10 @@ export default class StreamResponse {
    * @param reason
    */
   cancel(reason) {
-    return this.partStream.cancel(reason);
+    this.aborted = true;
+    if (this.reader) {
+      return this.reader.cancel(reason);
+    }
+    throw new Error('cancel() can only be called after parts()');
   }
 }

package/dist/chat/types.d.ts

@@ -97,7 +97,8 @@ export interface InputBlock {
 export declare enum BubbleTileStatus {
     NORMAL = "normal",
     WARNING = "warning",
-    ERROR = "error"
+    ERROR = "error",
+    ABORTED = "aborted"
 }
 export interface AbortSignalObject {
     aborted: boolean;
@@ -143,7 +144,7 @@ export type ChatAgentPluginHandler = {
 export type ChatAgentPlugin<T extends ChatAgentPluginHandler = Record<string, any>> = (agent: ChatAgent) => T;
 export type AddHookCallback<H, T extends keyof H> = (fn: H[T], type?: 'before' | 'after' | undefined) => () => void;
 export interface ErrorPart {
-    type: 'error';
+    type: 'error' | 'abort';
     level: 'error' | 'warn';
     error: Error;
     meta: Record<string, any>;

package/dist/chat/types.js

@@ -19,6 +19,7 @@ export let BubbleTileStatus = /*#__PURE__*/function (BubbleTileStatus) {
   BubbleTileStatus["NORMAL"] = "normal";
   BubbleTileStatus["WARNING"] = "warning";
   BubbleTileStatus["ERROR"] = "error";
+  BubbleTileStatus["ABORTED"] = "aborted";
   return BubbleTileStatus;
 }({});
 export let FinishReason = /*#__PURE__*/function (FinishReason) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ray-js/t-agent",
-  "version": "0.0.8-beta-3",
+  "version": "0.0.9-beta-1",
   "author": "Tuya.inc",
   "license": "MIT",
   "private": false,
@@ -26,5 +26,5 @@
     "build": "ray build --type=component --output dist",
     "clean": "rimraf ./dist"
   },
-  "gitHead": "686e9d17bd1a96074d3ac16ac2a3d73e1aa17f8f"
+  "gitHead": "df6f4f35caa6e62408450f14d9dcdd26f068283c"
 }