@ray-js/t-agent 0.2.0-beta-2 → 0.2.0-beta-4

package/dist/chat/ChatMessage.js

@@ -107,7 +107,7 @@ export default class ChatMessage {
       await this.agent.hooks.callHook('onMessagePersist:before', payload, this);
       await this.agent.hooks.callHook('onMessagePersist', payload, this);
       await this.agent.hooks.callHook('onMessagePersist:after', payload, this);
-      logger.debug('%onMessagePersist', 'background: gray; color: white', payload, this);
+      logger.debug('%conMessagePersist', 'background: gray; color: white', payload, this);
     });
   }
   setTilesLocked(locked) {

package/dist/chat/ChatSession.d.ts

@@ -12,6 +12,7 @@ export default class ChatSession {
     sessionId: string | null;
     readonly messages: Map<string, ChatMessage>;
     constructor(agent: ChatAgent);
+    initValue: (key: string, value: any) => void;
     set: (key: string, value: any) => Promise<void>;
     get: <T = any>(key: string, defaultValue?: T | undefined) => any;
     getData: () => Record<string, any>;

package/dist/chat/ChatSession.js

@@ -11,6 +11,9 @@ export default class ChatSession {
     _defineProperty(this, "isNewChat", true);
     _defineProperty(this, "sessionId", null);
     _defineProperty(this, "messages", new Map());
+    _defineProperty(this, "initValue", (key, value) => {
+      this.data.set(key, value);
+    });
     _defineProperty(this, "set", async (key, value) => {
       const oldValue = this.data.get(key);
       if (oldValue === value) {

package/dist/chat/StreamRequest.d.ts

@@ -0,0 +1,14 @@
+import { InputBlock, StreamRequestInstance } from './types';
+export default class StreamRequest implements StreamRequestInstance {
+    readonly blocks: InputBlock[];
+    private aborted;
+    constructor(blocks: InputBlock[]);
+    sent(): void;
+    parts(): void;
+    /**
+     * Cancel the request stream.
+     *
+     * @param reason
+     */
+    cancel(reason: any): Promise<void>;
+}

package/dist/chat/StreamRequest.js

@@ -0,0 +1,19 @@
+import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
+export default class StreamRequest {
+  constructor(blocks) {
+    _defineProperty(this, "aborted", false);
+    this.blocks = blocks;
+  }
+  sent() {}
+  parts() {}
+
+  /**
+   * Cancel the request stream.
+   *
+   * @param reason
+   */
+  cancel() {
+    this.aborted = true;
+    throw new Error('cancel() can only be called after parts()');
+  }
+}

package/dist/chat/types.d.ts

@@ -177,6 +177,9 @@ export declare enum FinishReason {
     ABORT = "abort",
     ERROR = "error"
 }
+export type StreamRequestInstance = {
+    blocks: InputBlock[];
+};
 export type StreamResponseInstance = {
     parts: () => AsyncIterable<MessagePart>;
     cancel: (reason: any) => Promise<void>;

package/dist/plugins/debug.js

@@ -5,13 +5,6 @@ export function withDebug(options) {
       throw new Error('Debug plugin already exists');
     }
     const start = () => {
-      // debugs.push(
-      //   createDebugger(agent.hooks, { tag: 'ChatAgent', filter: options?.filter }),
-      //   createDebugger(agent.session.hooks, { tag: 'ChatSession', filter: options?.filter }),
-      // )
-      // for (const hook of agent.pluginHooks) {
-      //   debugs.push(createDebugger(hook, { tag: 'ChatPlugin', filter: options?.filter }))
-      // }
       Logger.setLogLevel('debug');
     };
     if ((options === null || options === void 0 ? void 0 : options.autoStart) !== false) {

package/dist/plugins/ui.d.ts

@@ -23,8 +23,26 @@ export interface UIEventMap {
     };
     [key: string]: any;
 }
+export interface UIHooksContext<P = any, R = any> {
+    payload: P;
+    result: R | null;
+}
+export interface UIHooks {
+    onMessageFeedback: (context: UIHooksContext<{
+        messageId: string;
+        content?: string;
+        rate?: string;
+        [key: string]: any;
+    }, {
+        success: boolean;
+    }>) => void | Promise<void>;
+    [key: string]: (context: UIHooksContext) => void;
+}
 export declare function withUI(): (agent: ChatAgent) => {
+    hooks: import("hookable").Hookable<UIHooks, string>;
     ui: {
+        callHook: <K extends keyof UIHooks>(hookName: K, payload: UIHooks[K] extends (context: UIHooksContext<infer P, any>) => void ? P : never) => Promise<UIHooks[K] extends (context: UIHooksContext<any, infer R>) => void ? R : null>;
+        hook: <K_1 extends keyof UIHooks>(hookName: K_1, fn: UIHooks[K_1]) => () => void;
         emitter: Emitter;
         emitEvent: <T extends keyof UIEventMap>(eventName: T, detail: UIEventMap[T]) => void;
         onEvent: <T_1 extends keyof UIEventMap>(eventName: T_1, handler: (detail: UIEventMap[T_1]) => void) => () => void;

package/dist/plugins/ui.js

@@ -1,7 +1,10 @@
 import "core-js/modules/esnext.iterator.constructor.js";
 import "core-js/modules/esnext.iterator.map.js";
+import "core-js/modules/web.dom-collections.iterator.js";
 import { Emitter, EmitterEvent } from '../chat';
+import { createHooks } from 'hookable';
 export function withUI() {
+  const hooks = createHooks();
   return agent => {
     if (agent.plugins.ui) {
       throw new Error('UI plugin already exists');
@@ -28,6 +31,12 @@ export function withUI() {
       };
       emitEvent('messageChange', detail);
     }, 'after');
+    agent.onAgentDispose(() => {
+      const types = Object.keys(emitter.listeners);
+      for (const type of types) {
+        delete emitter.listeners[type];
+      }
+    });
     agent.session.onChange((key, value, oldValue) => {
       emitEvent('sessionChange', {
         key,
@@ -36,7 +45,19 @@ export function withUI() {
       });
     });
     return {
+      hooks,
       ui: {
+        callHook: async (hookName, payload) => {
+          const context = {
+            payload,
+            result: null
+          };
+          await hooks.callHook(hookName, context);
+          return context.result;
+        },
+        hook: (hookName, fn) => {
+          return hooks.hook(hookName, fn);
+        },
         emitter,
         emitEvent,
         onEvent: (eventName, handler) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ray-js/t-agent",
-  "version": "0.2.0-beta-2",
+  "version": "0.2.0-beta-4",
   "author": "Tuya.inc",
   "license": "MIT",
   "private": false,
@@ -26,5 +26,5 @@
     "build": "ray build --type=component --output dist",
     "clean": "rimraf ./dist"
   },
-  "gitHead": "0784a0f440b67cc1f766a440dbad5350f633dfb1"
+  "gitHead": "1728f8a1cac6b2ca53e506e06171808fbf38efdc"
 }