@chat-lab/ui 0.1.0-beta.44 → 0.1.0-beta.45

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

@@ -18,6 +18,12 @@ var defaultThreadManagerState = {
   currentThreadId: null,
   threadMap: {}
 };
+var createThreadManagerState = () => {
+  return proxy({
+    currentThreadId: null,
+    threadMap: {}
+  });
+};
 var createThreadManagerStateWithDefault = () => {
   const id = v4();
   return proxy({
@@ -214,7 +220,9 @@ function rewriteAgentCardUrl(original, serviceUrl) {
 var A2ACompletion = async ({
   onChunk,
   requestOptions,
-  abortController
+  abortController,
+  thread,
+  onUpdateThread
 }) => {
   const cardResolver = new DefaultAgentCardResolver({
     fetchImpl: (input, init) => {
@@ -228,6 +236,14 @@ var A2ACompletion = async ({
     }
   });
   const rawCard = await cardResolver.resolve(requestOptions.api);
+  if (onUpdateThread && thread) {
+    onUpdateThread({
+      metadata: {
+        ...thread.metadata,
+        agentCard: rawCard
+      }
+    });
+  }
   const serviceUrl = requestOptions.api;
   const rewrittenCard = rewriteAgentCardUrl(rawCard, serviceUrl);
   const factory = new ClientFactory(
@@ -1328,8 +1344,6 @@ var LocalStorage = class {
     keysToRemove.forEach((key) => localStorage.removeItem(key));
   }
 };
-
-// src/plugins/adk-plugin.ts
 var pluginName = "ADKPlugin";
 var AdkPlugin = class extends ChatkitPlugin {
   constructor(options) {
@@ -1345,7 +1359,8 @@ var AdkPlugin = class extends ChatkitPlugin {
           name: id,
           metadata: {},
           sending: false,
-          messages: []
+          messages: [],
+          createdAt: dayjs().unix()
         }).then((res) => {
           this.runtimeMap.setItem(runtime, res.id);
           this.chatController?.setCurrentThread(res.id);
@@ -1602,7 +1617,8 @@ async function fetchChatCompletion({
   completions,
   protocolAdaptor,
   onResponse,
-  storeManager
+  storeManager,
+  thread
 }) {
   try {
     let currentMessage = storeManager.getMessage(
@@ -1645,7 +1661,11 @@ async function fetchChatCompletion({
       await completions({
         abortController,
         requestOptions,
-        onChunk
+        onChunk,
+        thread,
+        onUpdateThread: (updates) => {
+          storeManager.updateThread(message.threadId, updates);
+        }
       });
     } catch (error) {
       throttledOnResponse.cancel();
@@ -1922,6 +1942,7 @@ var ChatController = class {
         completions: this.requestOptions.completions || AdkCompletion,
         protocolAdaptor,
         storeManager,
+        thread,
         onResponse: async (updatedMessage) => {
           if (updatedMessage.status === "pending" /* PENDING */ && !updatedMessage.createdAt) {
             updatedMessage.createdAt = Date.now();
@@ -1943,5 +1964,5 @@ var ChatController = class {
   }
 };
 
-export { A2ACompletion, AdkCompletion, AdkPlugin, AsyncSeriesWaterfallHook, BaseProtocols_exports as BaseProtocol, ChatController, ChatkitPlugin, ChatkitPluginManager, Compiler, MessageStatus, ProtocolEnum, SyncBailHook, SyncHook, SyncWaterfallHook, ThreadMessageManager, completion, createThreadManagerStateWithDefault, getProtocolAdaptor, storeActions, threadManagerState };
+export { A2ACompletion, AdkCompletion, AdkPlugin, AsyncSeriesWaterfallHook, BaseProtocols_exports as BaseProtocol, ChatController, ChatkitPlugin, ChatkitPluginManager, Compiler, MessageStatus, ProtocolEnum, SyncBailHook, SyncHook, SyncWaterfallHook, ThreadMessageManager, completion, createThreadManagerState, createThreadManagerStateWithDefault, getProtocolAdaptor, storeActions, threadManagerState };
 //# sourceMappingURL=index.js.map