@chat-lab/ui 0.1.0-beta.84 → 0.1.0-beta.85

package/dist/core/dist/index.js

@@ -1,8 +1,8 @@
 import { last, isNil, throttle, keys, compact, findLastIndex } from 'lodash-es';
 import { v4 } from 'uuid';
+import dayjs from 'dayjs';
 import { proxy } from 'valtio';
 export { snapshot, subscribe, useSnapshot } from 'valtio';
-import dayjs from 'dayjs';
 import { DefaultAgentCardResolver, ClientFactory, ClientFactoryOptions } from '@a2a-js/sdk/client';
 import { proxy as proxy$1 } from 'valtio/vanilla';
 import mitt from 'mitt';
@@ -46,15 +46,14 @@ var MessageStatus = /* @__PURE__ */ ((MessageStatus2) => {
   MessageStatus2["ABORTED"] = "aborted";
   return MessageStatus2;
 })(MessageStatus || {});
-
-// src/utils/Message.ts
 function generateModelMessage(message) {
   return {
     id: v4(),
     role: "assistant",
     status: "sending" /* SENDING */,
     content: [],
-    createdAt: Date.now(),
+    createdAt: dayjs().unix(),
+    updatedAt: dayjs().unix(),
     metadata: {},
     ...message
   };
@@ -884,15 +883,33 @@ function getHostUrl(fullUrl) {
     return "";
   }
 }
+function getQueryString(fullUrl) {
+  if (!fullUrl) return "";
+  try {
+    const urlObj = new URL(fullUrl);
+    return urlObj.search;
+  } catch (error) {
+    return "";
+  }
+}
+function buildSessionUrl(api, appName, userId, sessionId) {
+  const hostUrl = getHostUrl(api);
+  const queryString = getQueryString(api);
+  const baseUrl = `${hostUrl}/apps/${appName}/users/${userId}/sessions/${sessionId}`;
+  return queryString ? `${baseUrl}${queryString}` : baseUrl;
+}
 var AdkCompletion = async (params) => {
   try {
     await completion(params);
   } catch (error) {
     if (error instanceof SSEError && error.message === '{"detail":"Session not found"}') {
-      await fetch(
-        `${getHostUrl(params.requestOptions.api)}/apps/${params.requestOptions.data?.appName}/users/${params.requestOptions.data?.userId}/sessions/${params.requestOptions.data?.sessionId}`,
-        { method: "post", headers: params.requestOptions.headers }
+      const url = buildSessionUrl(
+        params.requestOptions.api,
+        params.requestOptions.data?.appName,
+        params.requestOptions.data?.userId,
+        params.requestOptions.data?.sessionId
       );
+      await fetch(url, { method: "post", headers: params.requestOptions.headers });
       await completion(params);
       return;
     }
@@ -1414,7 +1431,7 @@ var LocalStorage = class {
   }
 };
 var pluginName = "ADKPlugin";
-function buildSessionUrl(api, appName, userId, sessionId) {
+function buildSessionUrl2(api, appName, userId, sessionId) {
   const apiUrlObj = new URL(api);
   const baseUrl = `${apiUrlObj.origin}/apps/${appName}/users/${userId}/sessions/${sessionId}`;
   const queryString = apiUrlObj.search;
@@ -1483,7 +1500,7 @@ var AdkPlugin = class extends ChatkitPlugin {
         };
         let res;
         try {
-          const url = buildSessionUrl(
+          const url = buildSessionUrl2(
             requestOptions.api,
             this.options.appName,
             this.options.userId,
@@ -1569,7 +1586,7 @@ var AdkPlugin = class extends ChatkitPlugin {
         );
         return thread;
       }
-      const url = buildSessionUrl(
+      const url = buildSessionUrl2(
         requestOptions.api,
         this.options.appName,
         this.options.userId,
@@ -1671,7 +1688,7 @@ var AdkPlugin = class extends ChatkitPlugin {
         }
         const threadId = thread.id;
         const requestOptions = this.chatController?.getRequestOptions();
-        const url = buildSessionUrl(
+        const url = buildSessionUrl2(
           requestOptions.api,
           this.options.appName,
           this.options.userId,
@@ -1787,7 +1804,7 @@ async function fetchChatCompletion({
         ...currentMessage,
         status: "error" /* ERROR */,
         sending: false,
-        endedAt: Date.now(),
+        endedAt: dayjs().unix(),
         error: {
           message: error.message || "\u672A\u77E5\u9519\u8BEF",
           extra: error.stack
@@ -1802,7 +1819,9 @@ async function fetchChatCompletion({
       id: currentMessage.id,
       threadId: currentMessage.threadId,
       status: currentMessage.status !== "error" /* ERROR */ ? "success" /* SUCCESS */ : "error" /* ERROR */,
-      endedAt: Date.now()
+      updatedAt: dayjs().unix(),
+      endedAt: dayjs().unix(),
+      sending: false
     });
   } catch (error) {
     console.log(error);
@@ -2114,7 +2133,8 @@ var ChatController = class {
       });
       if (userMessage) storeManager.addMessage(threadId, userMessage);
       assistantMessage = assistantMessageFromProps || generateModelMessage({
-        threadId
+        threadId,
+        sending: true
       });
       if (!assistantMessageFromProps) {
         if (override) {
@@ -2150,8 +2170,9 @@ var ChatController = class {
             updatedMessage = updatedMessageOrFactory;
           }
           if (updatedMessage.status === "pending" /* PENDING */ && !updatedMessage.createdAt) {
-            updatedMessage.createdAt = Date.now();
+            updatedMessage.createdAt = dayjs().unix();
           }
+          updatedMessage.updatedAt = dayjs().unix();
           storeManager.updateMessage(threadId, updatedMessage.id, {
             updatedMessage,
             partial: true