@athenaintel/react 0.6.0 → 0.6.1

package/dist/index.cjs

@@ -7650,7 +7650,7 @@ const toAppendMessage = (messages, message) => {
     startRun: message.startRun
   };
 };
-const getThreadState = (runtime, threadListItemState) => {
+const getThreadState$1 = (runtime, threadListItemState) => {
   const lastMessage = runtime.messages.at(-1);
   return Object.freeze({
     threadId: threadListItemState.id,
@@ -7673,7 +7673,7 @@ class ThreadRuntimeImpl {
     __publicField(this, "_eventSubscriptionSubjects", /* @__PURE__ */ new Map());
     const stateBinding = new ShallowMemoizeSubject({
       path: threadBinding.path,
-      getState: () => getThreadState(threadBinding.getState(), threadListItemBinding.getState()),
+      getState: () => getThreadState$1(threadBinding.getState(), threadListItemBinding.getState()),
       subscribe: (callback) => {
         const sub1 = threadBinding.subscribe(callback);
         const sub2 = threadListItemBinding.subscribe(callback);
@@ -20637,6 +20637,73 @@ const parseLangGraphState = (state) => {
   }
   return result;
 };
+function getAuthHeaders(auth) {
+  if (auth.token) {
+    return { Authorization: `Bearer ${auth.token}` };
+  }
+  if (auth.apiKey) {
+    return { "X-API-KEY": auth.apiKey };
+  }
+  return {};
+}
+function getAgoraBaseUrl(backendUrl) {
+  const stripped = backendUrl.replace(/\/api\/assistant-ui\/?$/, "");
+  if (stripped === backendUrl) {
+    return backendUrl.replace(/\/$/, "");
+  }
+  return stripped;
+}
+async function listThreads(backendUrl, auth, opts = {}) {
+  const base2 = getAgoraBaseUrl(backendUrl);
+  const res = await fetch(`${base2}/api/conversations/threads/list`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json", ...getAuthHeaders(auth) },
+    body: JSON.stringify({ limit: opts.limit ?? 50, offset: opts.offset ?? 0 })
+  });
+  if (!res.ok) {
+    throw new Error(`[AthenaSDK] Failed to list threads: ${res.status}`);
+  }
+  return res.json();
+}
+function deserializeMessage(msg) {
+  if (msg && typeof msg === "object" && "lc" in msg && "kwargs" in msg && msg.type === "constructor") {
+    const kwargs = msg.kwargs;
+    const deserializedToolCalls = Array.isArray(kwargs.tool_calls) ? kwargs.tool_calls.map((tc) => {
+      if (tc && typeof tc === "object" && "lc" in tc && "kwargs" in tc) {
+        return tc.kwargs;
+      }
+      return tc;
+    }) : kwargs.tool_calls;
+    return { ...kwargs, tool_calls: deserializedToolCalls };
+  }
+  return msg;
+}
+async function getThreadState(backendUrl, auth, threadId) {
+  const base2 = getAgoraBaseUrl(backendUrl);
+  const res = await fetch(`${base2}/api/unstable/threads/${threadId}`, {
+    method: "GET",
+    headers: { ...getAuthHeaders(auth) }
+  });
+  if (!res.ok) {
+    throw new Error(`[AthenaSDK] Failed to get thread state: ${res.status}`);
+  }
+  const data = await res.json();
+  if (Array.isArray(data.messages)) {
+    data.messages = data.messages.map(deserializeMessage);
+  }
+  return data;
+}
+async function archiveThread(backendUrl, auth, threadId) {
+  const base2 = getAgoraBaseUrl(backendUrl);
+  const res = await fetch(`${base2}/api/conversations/threads/archive`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json", ...getAuthHeaders(auth) },
+    body: JSON.stringify({ thread_id: threadId })
+  });
+  if (!res.ok) {
+    throw new Error(`[AthenaSDK] Failed to archive thread: ${res.status}`);
+  }
+}
 const DEFAULT_API_URL = "https://sync.athenaintel.com/api/chat";
 const DEFAULT_BACKEND_URL = "https://api.athenaintel.com/api/assistant-ui";
 const DEFAULT_MODEL = "claude-sonnet-4-6-low";
@@ -20937,15 +21004,33 @@ const useAthenaRuntime = (config2) => {
     }
   });
   const hasResumedRef = React.useRef(false);
+  const prevThreadIdRef = React.useRef(threadId);
+  if (prevThreadIdRef.current !== threadId) {
+    prevThreadIdRef.current = threadId;
+    hasResumedRef.current = false;
+  }
   React.useEffect(() => {
     if (isExistingThread && !hasResumedRef.current) {
       hasResumedRef.current = true;
-      try {
-        runtime.thread.unstable_resumeRun({ parentId: null });
-      } catch {
-      }
+      (async () => {
+        try {
+          const auth = { apiKey: apiKeyRef.current, token: tokenRef.current };
+          const state = await getThreadState(backendUrl, auth, threadId);
+          runtime.thread.importExternalState({
+            messages: state.messages
+          });
+        } catch (err) {
+          if (process.env.NODE_ENV !== "production") {
+            console.error("[AthenaSDK] Failed to load thread state:", err);
+          }
+          try {
+            runtime.thread.unstable_resumeRun({ parentId: null });
+          } catch {
+          }
+        }
+      })();
     }
-  }, [isExistingThread, runtime]);
+  }, [isExistingThread, runtime, threadId, backendUrl]);
   return runtime;
 };
 function r(e) {
@@ -24179,50 +24264,13 @@ function useAthenaConfig() {
   }
   return ctx;
 }
-function getAuthHeaders(auth) {
-  if (auth.token) {
-    return { Authorization: `Bearer ${auth.token}` };
-  }
-  if (auth.apiKey) {
-    return { "X-API-KEY": auth.apiKey };
-  }
-  return {};
-}
-function getAgoraBaseUrl(backendUrl) {
-  return backendUrl.replace(/\/api\/assistant-ui\/?$/, "");
-}
-async function listThreads(backendUrl, auth, opts = {}) {
-  const base2 = getAgoraBaseUrl(backendUrl);
-  const res = await fetch(`${base2}/api/conversations/threads/list`, {
-    method: "POST",
-    headers: { "Content-Type": "application/json", ...getAuthHeaders(auth) },
-    body: JSON.stringify({ limit: opts.limit ?? 50, offset: opts.offset ?? 0 })
-  });
-  if (!res.ok) {
-    throw new Error(`[AthenaSDK] Failed to list threads: ${res.status}`);
-  }
-  return res.json();
-}
-async function archiveThread(backendUrl, auth, threadId) {
-  const base2 = getAgoraBaseUrl(backendUrl);
-  const res = await fetch(`${base2}/api/conversations/threads/archive`, {
-    method: "POST",
-    headers: { "Content-Type": "application/json", ...getAuthHeaders(auth) },
-    body: JSON.stringify({ thread_id: threadId })
-  });
-  if (!res.ok) {
-    throw new Error(`[AthenaSDK] Failed to archive thread: ${res.status}`);
-  }
-}
 const AthenaThreadIdContext = React.createContext(void 0);
 function useAthenaThreadId() {
   return React.useContext(AthenaThreadIdContext);
 }
 function useAthenaThreadListAdapter(config2) {
   const configRef = React.useRef(config2);
-  React.useEffect(() => {
-    configRef.current = config2;
-  }, [config2]);
+  configRef.current = config2;
   const auth = React.useMemo(
     () => ({ apiKey: config2.apiKey, token: config2.token }),
     [config2.apiKey, config2.token]
@@ -24265,7 +24313,9 @@ function useAthenaThreadListAdapter(config2) {
       await archiveThread(configRef.current.backendUrl, auth, remoteId);
     },
     async generateTitle(_remoteId, _messages) {
-      return new Response("").body;
+      return new ReadableStream({ start(c) {
+        c.close();
+      } });
     },
     async fetch(remoteId) {
       return {
@@ -24523,7 +24573,7 @@ function AthenaWithThreadList({
     apiKey,
     token
   });
-  const runtimeConfig = React.useMemo(() => ({
+  const runtimeConfigRef = React.useRef({
     apiUrl,
     backendUrl,
     apiKey,
@@ -24535,10 +24585,23 @@ function AthenaWithThreadList({
     workbench,
     knowledgeBase,
     systemPrompt
-  }), [apiUrl, backendUrl, apiKey, token, model, agent2, tools, frontendToolIds, workbench, knowledgeBase, systemPrompt]);
+  });
+  runtimeConfigRef.current = {
+    apiUrl,
+    backendUrl,
+    apiKey,
+    token,
+    model,
+    agent: agent2,
+    tools,
+    frontendToolIds,
+    workbench,
+    knowledgeBase,
+    systemPrompt
+  };
   const runtimeHook = React.useCallback(
-    () => useAthenaRuntimeHook(runtimeConfig),
-    [runtimeConfig]
+    () => useAthenaRuntimeHook(runtimeConfigRef.current),
+    []
   );
   const runtime = useRemoteThreadListRuntime({
     runtimeHook,
@@ -61660,6 +61723,7 @@ function AssetToolCard({
       assetId && isComplete && !isCancelled && CREATE_ASSET_TOOLS.includes(toolName.toLowerCase()) && /* @__PURE__ */ jsxRuntime.jsxs(
         "button",
         {
+          type: "button",
           onClick: () => openAsset(assetId, { name: title ?? void 0, type: assetType }),
           className: "flex shrink-0 items-center gap-1 rounded-md border border-border/60 px-2 py-0.5 text-[11px] font-medium text-muted-foreground transition-colors hover:bg-muted/50 hover:text-foreground",
           children: [
@@ -61672,6 +61736,7 @@ function AssetToolCard({
       !isCancelled && /* @__PURE__ */ jsxRuntime.jsx(
         "button",
         {
+          type: "button",
           onClick: () => setDetailsOpen((o) => !o),
           className: cn(
             "flex size-5 shrink-0 items-center justify-center rounded transition-all",
@@ -61703,6 +61768,7 @@ function AssetOpenLink({ assetId, toolName }) {
   return /* @__PURE__ */ jsxRuntime.jsx(
     "button",
     {
+      type: "button",
       onClick: (e) => {
         e.stopPropagation();
         openAsset(assetId, { type: assetType });
@@ -62127,7 +62193,7 @@ function ToolCard({
             "flex size-8 shrink-0 items-center justify-center rounded-lg",
             isRunning && "bg-blue-50 text-blue-600",
             isComplete && "bg-emerald-50 text-emerald-600",
-            isError && "bg-red-50 text-red-500"
+            isError && "bg-destructive/10 text-destructive"
           ),
           children: isRunning ? /* @__PURE__ */ jsxRuntime.jsx(Loader, { className: "size-4 animate-spin" }) : isError ? /* @__PURE__ */ jsxRuntime.jsx(CircleAlert, { className: "size-4" }) : /* @__PURE__ */ jsxRuntime.jsx(Icon2, { className: "size-4" })
         }
@@ -62146,7 +62212,7 @@ function ToolCard({
         /* @__PURE__ */ jsxRuntime.jsx("div", { className: "h-1.5 w-6 animate-pulse rounded-full bg-blue-100" })
       ] })
     ] }),
-    error2 && /* @__PURE__ */ jsxRuntime.jsx("div", { className: "border-t border-border/40 bg-red-50/50 px-4 py-2.5", children: /* @__PURE__ */ jsxRuntime.jsx("p", { className: "text-[12px] leading-relaxed text-red-600", children: error2 }) }),
+    error2 && /* @__PURE__ */ jsxRuntime.jsx("div", { className: "border-t border-border/40 bg-destructive/10 px-4 py-2.5", children: /* @__PURE__ */ jsxRuntime.jsx("p", { className: "text-[12px] leading-relaxed text-destructive", children: error2 }) }),
     children
   ] });
 }
@@ -62392,6 +62458,7 @@ function CreateAssetToolUIImpl({
       children: assetId && isComplete && !isCancelled && /* @__PURE__ */ jsxRuntime.jsx("div", { className: "border-t border-border/40 px-4 py-2", children: /* @__PURE__ */ jsxRuntime.jsxs(
         "button",
         {
+          type: "button",
           onClick: handleOpen,
           className: "flex items-center gap-1.5 rounded-md border border-border/60 px-3 py-1.5 text-xs font-medium text-muted-foreground transition-colors hover:bg-muted/50 hover:text-foreground",
           children: [
@@ -62547,12 +62614,12 @@ const RunPythonCodeToolUIImpl = ({
           ] }),
           /* @__PURE__ */ jsxRuntime.jsx("pre", { className: "max-h-48 overflow-auto whitespace-pre-wrap break-words rounded-md bg-muted/30 p-2 text-[11px] leading-relaxed font-mono text-foreground/80", children: [parsed.value, parsed.stdout, parsed.stderr].filter(Boolean).join("\n") })
         ] }),
-        isComplete && hasError && /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "border-t border-border/40 bg-red-50/50 px-4 py-2.5", children: [
+        isComplete && hasError && /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "border-t border-border/40 bg-destructive/10 px-4 py-2.5", children: [
           /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-1.5 mb-1", children: [
-            /* @__PURE__ */ jsxRuntime.jsx(TriangleAlert, { className: "size-3 text-red-500" }),
-            /* @__PURE__ */ jsxRuntime.jsx("span", { className: "text-[11px] font-medium text-red-600", children: parsed.exception ? `${parsed.exception.name}: ${parsed.exception.value}` : "Error" })
+            /* @__PURE__ */ jsxRuntime.jsx(TriangleAlert, { className: "size-3 text-destructive" }),
+            /* @__PURE__ */ jsxRuntime.jsx("span", { className: "text-[11px] font-medium text-destructive", children: parsed.exception ? `${parsed.exception.name}: ${parsed.exception.value}` : "Error" })
           ] }),
-          /* @__PURE__ */ jsxRuntime.jsx("pre", { className: "max-h-48 overflow-auto whitespace-pre-wrap break-words rounded-md bg-red-50 p-2 text-[11px] leading-relaxed font-mono text-red-600/80", children: ((_a2 = parsed.exception) == null ? void 0 : _a2.traceback) ?? parsed.error })
+          /* @__PURE__ */ jsxRuntime.jsx("pre", { className: "max-h-48 overflow-auto whitespace-pre-wrap break-words rounded-md bg-destructive/5 p-2 text-[11px] leading-relaxed font-mono text-destructive/80", children: ((_a2 = parsed.exception) == null ? void 0 : _a2.traceback) ?? parsed.error })
         ] }),
         isComplete && hasImage && /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "border-t border-border/40 px-4 py-2.5", children: [
           /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-1.5 mb-1.5", children: [
@@ -62573,6 +62640,7 @@ const RunPythonCodeToolUIImpl = ({
           /* @__PURE__ */ jsxRuntime.jsx("div", { className: "mt-1 flex flex-wrap gap-1.5", children: parsed.createdAssets.map((a) => /* @__PURE__ */ jsxRuntime.jsxs(
             "button",
             {
+              type: "button",
               onClick: () => openAsset(a.asset_id),
               className: "flex items-center gap-1 rounded-md border border-border/60 px-2 py-1 text-[11px] font-medium text-muted-foreground transition-colors hover:bg-muted/50 hover:text-foreground",
               children: [
@@ -62627,6 +62695,7 @@ const OpenAssetToolUIImpl = ({
       children: assetId && isComplete && !isCancelled && /* @__PURE__ */ jsxRuntime.jsx("div", { className: "border-t border-border/40 px-4 py-2", children: /* @__PURE__ */ jsxRuntime.jsxs(
         "button",
         {
+          type: "button",
           onClick: () => openAsset(assetId),
           className: "flex items-center gap-1.5 rounded-md border border-border/60 px-3 py-1.5 text-xs font-medium text-muted-foreground transition-colors hover:bg-muted/50 hover:text-foreground",
           children: [