@truefoundry/agent-ui-sdk 0.0.5 → 0.0.6

package/dist/index.js

@@ -949,19 +949,13 @@ function MessageTimestamp({ className }) {
 }
 
 // src/atoms/UserMessageActionBar.tsx
-import { ActionBarPrimitive, useAuiState as useAuiState2 } from "@assistant-ui/react";
+import { ActionBarPrimitive, useAui } from "@assistant-ui/react";
 import { useActionBarCopy, useActionBarEdit } from "@assistant-ui/core/react";
-import { useTrueFoundryResetFromTurn } from "@truefoundry/assistant-ui-runtime";
 import { jsx as jsx36, jsxs as jsxs13 } from "react/jsx-runtime";
-function parseTurnIdFromMessageId(messageId) {
-  return messageId.replace(/-user$/, "");
-}
 function UserMessageActionBar({ className }) {
   const IconButton5 = useSlot("IconButton");
   const MessageTimestamp2 = useSlot("MessageTimestamp");
-  const messageId = useAuiState2((s) => s.message.id);
-  const turnId = parseTurnIdFromMessageId(messageId);
-  const resetFromTurn = useTrueFoundryResetFromTurn();
+  const aui = useAui();
   const { edit, disabled: editDisabled } = useActionBarEdit();
   const { copy, isCopied } = useActionBarCopy({
     copyToClipboard: (text) => navigator.clipboard.writeText(text)
@@ -980,7 +974,10 @@ function UserMessageActionBar({ className }) {
           {
             icon: "rotate-right",
             tooltip: "Try again",
-            onClick: () => void resetFromTurn(turnId)
+            onClick: () => {
+              aui.message().composer().beginEdit();
+              aui.message().composer().send({ startRun: true });
+            }
           }
         ),
         /* @__PURE__ */ jsx36(
@@ -1293,7 +1290,7 @@ import { default as default6 } from "tfy-web-components/components/atoms/Skeleto
 // src/containers/ThreadContainer.tsx
 import { useEffect as useEffect6 } from "react";
 import { preloadMarkdownOpenUI } from "tfy-web-components/components/molecules/Markdown";
-import { ThreadPrimitive, useAuiState as useAuiState11 } from "@assistant-ui/react";
+import { ThreadPrimitive, useAuiState as useAuiState10 } from "@assistant-ui/react";
 
 // src/hooks/useComposerBusyState.ts
 import { useThreadIsRunning } from "@assistant-ui/core/react";
@@ -1339,7 +1336,7 @@ function useComposerBusyState() {
 }
 
 // src/containers/AssistantMessageContainer.tsx
-import { MessagePrimitive as MessagePrimitive2, useAuiState as useAuiState6 } from "@assistant-ui/react";
+import { MessagePrimitive as MessagePrimitive2, useAuiState as useAuiState5 } from "@assistant-ui/react";
 import { useActionBarCopy as useActionBarCopy2, useMessageError, useThreadIsRunning as useThreadIsRunning2 } from "@assistant-ui/core/react";
 
 // src/utils/computeAgentStepsSplit.ts
@@ -1386,7 +1383,7 @@ function computeAgentStepsSplit(parts, isRunning) {
 
 // src/containers/AssistantTextContainer.tsx
 import { useCallback as useCallback3 } from "react";
-import { useAuiState as useAuiState3 } from "@assistant-ui/react";
+import { useAuiState as useAuiState2 } from "@assistant-ui/react";
 import { useTrueFoundryDownloadSandboxFile } from "@truefoundry/assistant-ui-runtime";
 
 // src/containers/ErrorToasterContainer.tsx
@@ -1459,10 +1456,10 @@ function AssistantTextContainer() {
   const Markdown2 = useSlot("Markdown");
   const downloadSandboxFile = useTrueFoundryDownloadSandboxFile();
   const errorToaster = useErrorToasterOptional();
-  const text = useAuiState3(
+  const text = useAuiState2(
     (s) => s.part.type === "text" || s.part.type === "reasoning" ? s.part.text : ""
   );
-  const isStreaming = useAuiState3((s) => {
+  const isStreaming = useAuiState2((s) => {
     if (s.message.status?.type !== "running") return false;
     const lastIndex = s.message.parts.length - 1;
     if (lastIndex < 0) return false;
@@ -1483,13 +1480,13 @@ function AssistantTextContainer() {
 }
 
 // src/containers/MessageImageContainer.tsx
-import { useAuiState as useAuiState4 } from "@assistant-ui/react";
+import { useAuiState as useAuiState3 } from "@assistant-ui/react";
 import { jsx as jsx43 } from "react/jsx-runtime";
 function MessageImageContainer() {
   const AttachmentCard2 = useSlot("AttachmentCard");
   const AttachmentPreviewDialog2 = useSlot("AttachmentPreviewDialog");
-  const image = useAuiState4((s) => s.part.type === "image" ? s.part.image : "");
-  const filename = useAuiState4(
+  const image = useAuiState3((s) => s.part.type === "image" ? s.part.image : "");
+  const filename = useAuiState3(
     (s) => s.part.type === "image" ? s.part.filename : void 0
   );
   if (!image) {
@@ -1510,14 +1507,14 @@ function MessageImageContainer() {
 
 // src/containers/ReasoningContainer.tsx
 import { useCallback as useCallback4, useRef, useState as useState4 } from "react";
-import { useAuiState as useAuiState5 } from "@assistant-ui/react";
+import { useAuiState as useAuiState4 } from "@assistant-ui/react";
 import { jsx as jsx44 } from "react/jsx-runtime";
 function ReasoningContainer({ group }) {
   const ReasoningCard2 = useSlot("ReasoningCard");
-  const content = useAuiState5(
+  const content = useAuiState4(
     (s) => group.indices.map((i) => s.message.parts[i]).filter((p) => p != null && p.type === "reasoning").map((p) => p.text ?? "").join("")
   );
-  const streaming = useAuiState5((s) => {
+  const streaming = useAuiState4((s) => {
     if (s.message.status?.type !== "running") return false;
     const lastIndex = s.message.parts.length - 1;
     if (lastIndex < 0) return false;
@@ -2218,7 +2215,7 @@ function AssistantMessageContainer() {
   const { copy, isCopied } = useActionBarCopy2({
     copyToClipboard: (text) => navigator.clipboard.writeText(text)
   });
-  const parts = useAuiState6((s) => s.message.parts);
+  const parts = useAuiState5((s) => s.message.parts);
   const { cutIndex, hasFinal, toolCount, thinkingCount } = computeAgentStepsSplit(parts, isRunning);
   const groupBy = (part, _context) => {
     const index = parts.findIndex((p) => p === part);
@@ -2324,16 +2321,16 @@ function HistoryLoaderContainer() {
 import {
   ComposerPrimitive as ComposerPrimitive2,
   MessagePrimitive as MessagePrimitive4,
-  useAuiState as useAuiState9
+  useAuiState as useAuiState8
 } from "@assistant-ui/react";
 import { useComposerCancel, useComposerSend, useThreadIsRunning as useThreadIsRunning3 } from "@assistant-ui/core/react";
 
 // src/containers/AttachmentsContainer.tsx
-import { ComposerPrimitive, MessagePrimitive as MessagePrimitive3, useAui, useAuiState as useAuiState8 } from "@assistant-ui/react";
+import { ComposerPrimitive, MessagePrimitive as MessagePrimitive3, useAui as useAui2, useAuiState as useAuiState7 } from "@assistant-ui/react";
 
 // src/containers/useAttachmentPreviewSrc.ts
 import { useEffect as useEffect5, useState as useState11 } from "react";
-import { useAuiState as useAuiState7 } from "@assistant-ui/react";
+import { useAuiState as useAuiState6 } from "@assistant-ui/react";
 function isImageAttachment(type, contentType) {
   return type === "image" || (contentType?.startsWith("image/") ?? false);
 }
@@ -2351,10 +2348,10 @@ function useFileObjectUrl(file) {
   return src;
 }
 function useAttachmentPreviewSrc() {
-  const type = useAuiState7((s) => s.attachment.type);
-  const contentType = useAuiState7((s) => s.attachment.contentType);
-  const file = useAuiState7((s) => "file" in s.attachment ? s.attachment.file : void 0);
-  const contentSrc = useAuiState7((s) => {
+  const type = useAuiState6((s) => s.attachment.type);
+  const contentType = useAuiState6((s) => s.attachment.contentType);
+  const file = useAuiState6((s) => "file" in s.attachment ? s.attachment.file : void 0);
+  const contentSrc = useAuiState6((s) => {
     if (!isImageAttachment(s.attachment.type, s.attachment.contentType)) return void 0;
     const imagePart = s.attachment.content?.find((part) => part.type === "image");
     if (imagePart?.type === "image" && imagePart.image) return imagePart.image;
@@ -2371,10 +2368,10 @@ import { jsx as jsx52 } from "react/jsx-runtime";
 function ComposerAttachmentItem() {
   const AttachmentCard2 = useSlot("AttachmentCard");
   const AttachmentPreviewDialog2 = useSlot("AttachmentPreviewDialog");
-  const aui = useAui();
-  const name = useAuiState8((s) => s.attachment.name);
-  const contentType = useAuiState8((s) => s.attachment.contentType);
-  const type = useAuiState8((s) => s.attachment.type);
+  const aui = useAui2();
+  const name = useAuiState7((s) => s.attachment.name);
+  const contentType = useAuiState7((s) => s.attachment.contentType);
+  const type = useAuiState7((s) => s.attachment.type);
   const isImage = isImageAttachment(type, contentType);
   const previewSrc = useAttachmentPreviewSrc();
   return /* @__PURE__ */ jsx52(AttachmentPreviewDialog2, { previewSrc, children: /* @__PURE__ */ jsx52(
@@ -2392,9 +2389,9 @@ function ComposerAttachmentItem() {
 function MessageAttachmentItem() {
   const AttachmentCard2 = useSlot("AttachmentCard");
   const AttachmentPreviewDialog2 = useSlot("AttachmentPreviewDialog");
-  const name = useAuiState8((s) => s.attachment.name);
-  const contentType = useAuiState8((s) => s.attachment.contentType);
-  const type = useAuiState8((s) => s.attachment.type);
+  const name = useAuiState7((s) => s.attachment.name);
+  const contentType = useAuiState7((s) => s.attachment.contentType);
+  const type = useAuiState7((s) => s.attachment.type);
   const isImage = isImageAttachment(type, contentType);
   const previewSrc = useAttachmentPreviewSrc();
   const card = /* @__PURE__ */ jsx52(
@@ -2427,7 +2424,7 @@ function ComposerAttachmentPickerContainer() {
 // src/containers/UserEditComposerContainer.tsx
 import { jsx as jsx53, jsxs as jsxs16 } from "react/jsx-runtime";
 function ReadOnlyMessageAttachments() {
-  const hasAttachments = useAuiState9(
+  const hasAttachments = useAuiState8(
     (s) => (s.message.attachments?.length ?? 0) > 0
   );
   if (!hasAttachments) {
@@ -2486,14 +2483,14 @@ function UserEditComposerContainer() {
 }
 
 // src/containers/UserMessageContainer.tsx
-import { MessagePrimitive as MessagePrimitive5, useAuiState as useAuiState10 } from "@assistant-ui/react";
+import { MessagePrimitive as MessagePrimitive5, useAuiState as useAuiState9 } from "@assistant-ui/react";
 import { useThreadIsRunning as useThreadIsRunning4 } from "@assistant-ui/core/react";
 import { jsx as jsx54 } from "react/jsx-runtime";
 function UserMessageContainer() {
   const UserMessageBubble2 = useSlot("UserMessageBubble");
   const UserMessageActionBar2 = useSlot("UserMessageActionBar");
   const isRunning = useThreadIsRunning4();
-  const text = useAuiState10(
+  const text = useAuiState9(
     (s) => s.message.content.filter((part) => part.type === "text").map((part) => part.text).join("\n")
   );
   return /* @__PURE__ */ jsx54(MessagePrimitive5.Root, { "data-role": "user", children: /* @__PURE__ */ jsx54(
@@ -2510,7 +2507,7 @@ function UserMessageContainer() {
 import { Fragment as Fragment3, jsx as jsx55, jsxs as jsxs17 } from "react/jsx-runtime";
 var isNewChatView = (s) => s.thread.messages.length === 0 && (!s.thread.isLoading || s.threads.isLoading);
 function ThreadMessage({ isEditing }) {
-  const role = useAuiState11((s) => s.message.role);
+  const role = useAuiState10((s) => s.message.role);
   if (role === "user") {
     if (isEditing) {
       return /* @__PURE__ */ jsx55(UserEditComposerContainer, {});
@@ -2530,8 +2527,8 @@ function ThreadContainer({ composer }) {
   const WelcomeScreen2 = useSlot("WelcomeScreen");
   const MessageListSkeleton2 = useSlot("MessageListSkeleton");
   const ScrollToBottomButton2 = useSlot("ScrollToBottomButton");
-  const isEmpty = useAuiState11(isNewChatView);
-  const isLoading = useAuiState11((s) => s.thread.isLoading);
+  const isEmpty = useAuiState10(isNewChatView);
+  const isLoading = useAuiState10((s) => s.thread.isLoading);
   return /* @__PURE__ */ jsx55(ComposerBusyProvider, { children: /* @__PURE__ */ jsx55(ThreadPrimitive.Root, { asChild: true, children: /* @__PURE__ */ jsxs17(ThreadRootShell2, { children: [
     /* @__PURE__ */ jsx55(ThreadPrimitive.Viewport, { asChild: true, turnAnchor: "top", autoScroll: true, children: /* @__PURE__ */ jsxs17(ThreadViewportShell2, { isEmpty, children: [
       isEmpty && /* @__PURE__ */ jsx55(WelcomeScreen2, {}),
@@ -2554,7 +2551,7 @@ function ThreadContainer({ composer }) {
 
 // src/containers/ToolGroupContainer.tsx
 import { useState as useState12 } from "react";
-import { useAuiState as useAuiState12 } from "@assistant-ui/react";
+import { useAuiState as useAuiState11 } from "@assistant-ui/react";
 import { jsx as jsx56 } from "react/jsx-runtime";
 function ToolGroupContainer({
   children,
@@ -2562,7 +2559,7 @@ function ToolGroupContainer({
 }) {
   const ToolGroupCard2 = useSlot("ToolGroupCard");
   const toolCallCount = group.indices.length;
-  const active = useAuiState12((s) => {
+  const active = useAuiState11((s) => {
     if (s.message.status?.type !== "running") return false;
     const lastIndex = s.message.parts.length - 1;
     if (lastIndex < 0) return false;
@@ -2590,11 +2587,11 @@ function ToolGroupContainer({
 
 // src/containers/ComposerContainer.tsx
 import { useRef as useRef5 } from "react";
-import { useAui as useAui2, useAuiState as useAuiState14 } from "@assistant-ui/react";
+import { useAui as useAui3, useAuiState as useAuiState13 } from "@assistant-ui/react";
 import { useTrueFoundryCancel } from "@truefoundry/assistant-ui-runtime";
 
 // src/hooks/useComposerPauseView.ts
-import { useAuiState as useAuiState13 } from "@assistant-ui/react";
+import { useAuiState as useAuiState12 } from "@assistant-ui/react";
 import { useTrueFoundryToolResponses } from "@truefoundry/assistant-ui-runtime";
 function threadHasPendingMcpAuth(s) {
   const messages = s.thread.messages;
@@ -2604,7 +2601,7 @@ function threadHasPendingMcpAuth(s) {
   return last.metadata?.custom?.pendingMcpAuth === true;
 }
 function useComposerPauseView() {
-  const mcpPending = useAuiState13(threadHasPendingMcpAuth);
+  const mcpPending = useAuiState12(threadHasPendingMcpAuth);
   const { pending: toolResponsesPending } = useTrueFoundryToolResponses();
   if (mcpPending) {
     return { kind: "mcp" };
@@ -2739,8 +2736,8 @@ function ComposerContainer({
   placeholder = "Ask anything... (Shift+Enter for new line)"
 }) {
   const ComposerShell2 = useSlot("ComposerShell");
-  const aui = useAui2();
-  const text = useAuiState14((s) => s.composer.text);
+  const aui = useAui3();
+  const text = useAuiState13((s) => s.composer.text);
   const { isBusy, send, resetBusy } = useComposerBusyState();
   const pauseView = useComposerPauseView();
   const cancel = useTrueFoundryCancel();
@@ -2791,7 +2788,7 @@ function ComposerContainer({
 }
 
 // src/containers/ThreadListContainer.tsx
-import { useAui as useAui3, useAuiState as useAuiState15 } from "@assistant-ui/react";
+import { useAui as useAui4, useAuiState as useAuiState14 } from "@assistant-ui/react";
 import { useEffect as useEffect8, useRef as useRef6 } from "react";
 import { Fragment as Fragment5, jsx as jsx60, jsxs as jsxs19 } from "react/jsx-runtime";
 function ThreadListContainer() {
@@ -2800,13 +2797,13 @@ function ThreadListContainer() {
   const ThreadListRow2 = useSlot("ThreadListRow");
   const ThreadListRowSkeleton2 = useSlot("ThreadListRowSkeleton");
   const ThreadListEmptyState2 = useSlot("ThreadListEmptyState");
-  const aui = useAui3();
-  const isLoading = useAuiState15((s) => s.threads.isLoading);
-  const isLoadingMore = useAuiState15((s) => s.threads.isLoadingMore);
-  const hasMore = useAuiState15((s) => s.threads.hasMore);
-  const threadIds = useAuiState15((s) => s.threads.threadIds);
-  const threadItems = useAuiState15((s) => s.threads.threadItems);
-  const mainThreadId = useAuiState15((s) => s.threads.mainThreadId);
+  const aui = useAui4();
+  const isLoading = useAuiState14((s) => s.threads.isLoading);
+  const isLoadingMore = useAuiState14((s) => s.threads.isLoadingMore);
+  const hasMore = useAuiState14((s) => s.threads.hasMore);
+  const threadIds = useAuiState14((s) => s.threads.threadIds);
+  const threadItems = useAuiState14((s) => s.threads.threadItems);
+  const mainThreadId = useAuiState14((s) => s.threads.mainThreadId);
   const sentinelRef = useRef6(null);
   useEffect8(() => {
     const node = sentinelRef.current;