@chat-lab/ui 0.1.0-beta.57 → 0.1.0-beta.58

package/dist/packages/core/dist/index.js

@@ -1,8 +1,8 @@
 import dayjs from '../../../node_modules/.pnpm/dayjs@1.11.18/node_modules/dayjs/dayjs.min.js';
 import { ClientFactory, ClientFactoryOptions, DefaultAgentCardResolver } from '../../../node_modules/.pnpm/@a2a-js_sdk@0.3.7_express@5.1.0/node_modules/@a2a-js/sdk/dist/client/index.js';
 import mitt from '../../../node_modules/.pnpm/mitt@3.0.1/node_modules/mitt/dist/mitt.js';
-import { proxy } from '../../../node_modules/.pnpm/valtio@2.1.8_patch_hash_boxi3qpsg2dztwe4vz7tsjpr3m_@types_react@18.2.0_react@18.2.0/node_modules/valtio/esm/vanilla.js';
-export { snapshot, subscribe } from '../../../node_modules/.pnpm/valtio@2.1.8_patch_hash_boxi3qpsg2dztwe4vz7tsjpr3m_@types_react@18.2.0_react@18.2.0/node_modules/valtio/esm/vanilla.js';
+import { proxy } from '../../../node_modules/.pnpm/valtio@2.1.8_patch_hash_boxi3qpsg2dztwe4vz7tsjpr3m_@types_react@17.0.2_react@17.0.2/node_modules/valtio/esm/vanilla.js';
+export { snapshot, subscribe } from '../../../node_modules/.pnpm/valtio@2.1.8_patch_hash_boxi3qpsg2dztwe4vz7tsjpr3m_@types_react@17.0.2_react@17.0.2/node_modules/valtio/esm/vanilla.js';
 import v4 from '../../../node_modules/.pnpm/uuid@13.0.0/node_modules/uuid/dist/v4.js';
 import last from '../../../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/last.js';
 import isNil from '../../../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/isNil.js';
@@ -1281,7 +1281,7 @@ var Compiler = class {
         beforeDestroy: new AsyncSeriesWaterfallHook()
       },
       message: {
-        received: new AsyncSeriesWaterfallHook(),
+        received: new SyncHook(),
         beforeSend: new AsyncSeriesWaterfallHook(),
         error: new SyncWaterfallHook(),
         end: new AsyncSeriesWaterfallHook()
@@ -1720,6 +1720,7 @@ async function fetchChatCompletion({
   completions,
   protocolAdaptor,
   onResponse,
+  onReceivedMessage,
   storeManager,
   thread
 }) {
@@ -1741,12 +1742,16 @@ async function fetchChatCompletion({
         response
       });
       currentMessage = updatedMessage;
+      onReceivedMessage?.(currentMessage);
       if (updatedMessage.status === "error" /* ERROR */) {
         throttledOnResponse.cancel();
-        onResponse({ ...updatedMessage, status: "error" /* ERROR */ });
+        onResponse(() => ({ ...currentMessage, status: "error" /* ERROR */ }));
         return;
       }
-      throttledOnResponse({ ...updatedMessage, status: "pending" /* PENDING */ });
+      throttledOnResponse(() => ({
+        ...currentMessage,
+        status: "pending" /* PENDING */
+      }));
     };
     try {
       if (abortController?.signal) {
@@ -2059,13 +2064,19 @@ var ChatController = class {
         protocolAdaptor,
         storeManager,
         thread,
-        onResponse: async (updatedMessage) => {
+        onReceivedMessage: async (message) => {
+          this.pluginManager.hooks.message.received.call(message);
+        },
+        onResponse: async (updatedMessageOrFactory) => {
+          let updatedMessage;
+          if (typeof updatedMessageOrFactory === "function") {
+            updatedMessage = updatedMessageOrFactory();
+          } else {
+            updatedMessage = updatedMessageOrFactory;
+          }
           if (updatedMessage.status === "pending" /* PENDING */ && !updatedMessage.createdAt) {
             updatedMessage.createdAt = Date.now();
           }
-          this.pluginManager.hooks.message.received.callAsync(
-            toRaw(updatedMessage)
-          );
           storeManager.updateMessage(threadId, updatedMessage.id, {
             updatedMessage,
             partial: true