@truefoundry/assistant-ui-runtime 0.1.9 → 0.1.10

package/dist/index.js

@@ -11,7 +11,7 @@ import {
   isTfyMcpToolInfo,
   isTfySystemToolInfo,
   isTfyToolInfo
-} from "./chunk-CXBZ6WLZ.js";
+} from "./chunk-6ZOGBD7E.js";
 
 // src/useTrueFoundryAgentRuntime.ts
 import {
@@ -1929,7 +1929,11 @@ function projectHistoryTurns(snapshot, options) {
       content,
       record
     );
-    const custom = record.sandboxId != null ? { ...baseCustom, sandboxId: record.sandboxId } : baseCustom;
+    const custom = {
+      ...baseCustom,
+      turnId: record.id,
+      ...record.sandboxId != null ? { sandboxId: record.sandboxId } : {}
+    };
     if (record.userText) {
       messages.push(
         buildUserMessageFromTurnInput(
@@ -2374,7 +2378,7 @@ function turnStreamUpdateToAssistantMessage(turnId, update, existing, options) {
       unstable_annotations: [],
       unstable_data: [],
       steps: [],
-      custom: update.metadata?.custom ?? {}
+      custom: { ...update.metadata?.custom, turnId }
     }
   };
 }
@@ -2411,6 +2415,13 @@ function createDraftSessionBridge(server) {
   };
 }
 
+// src/sessionListStartTimestamp.ts
+function sessionListStartTimestamp() {
+  const start = /* @__PURE__ */ new Date();
+  start.setFullYear(start.getFullYear() - 1);
+  return start.toISOString();
+}
+
 // src/draft/agentSpec.ts
 function mergeAgentSpec(base, update) {
   const { model: modelUpdate, ...rest } = update;
@@ -2432,33 +2443,47 @@ function draftSessionTitle(draft) {
   return draft.title ?? draft.agentSpec.model.name;
 }
 
-// src/sessionListStartTimestamp.ts
-function sessionListStartTimestamp() {
-  const start = /* @__PURE__ */ new Date();
-  start.setFullYear(start.getFullYear() - 1);
-  return start.toISOString();
+// src/sessionThreadMetadata.ts
+function sessionDisplayTitle(session, defaultAgentSpec) {
+  if (session.isMutable) {
+    const agentSpec = session.agentSpec ?? defaultAgentSpec;
+    if (agentSpec != null) {
+      return draftSessionTitle({
+        title: session.title,
+        agentSpec
+      });
+    }
+  }
+  return session.title ?? session.agentName ?? session.id;
+}
+function sessionToThreadMetadata(session, title) {
+  return {
+    status: "regular",
+    remoteId: session.id,
+    title,
+    lastMessageAt: new Date(session.updatedAt),
+    ...session.agentName != null ? { custom: { agentName: session.agentName } } : {}
+  };
 }
 
 // src/draft/truefoundryDraftThreadListAdapter.ts
 var THREAD_LIST_PAGE_SIZE = 20;
 function createTrueFoundryDraftThreadListAdapter(options) {
-  const { server, defaultAgentSpec, getAgentSpec } = options;
+  const { server, defaultAgentSpec, getAgentSpec, listSessionsAgentId } = options;
   return {
     async list({ after } = {}) {
       const page = await server.listSessions({
+        ...listSessionsAgentId != null ? { agentId: listSessionsAgentId } : {},
         limit: THREAD_LIST_PAGE_SIZE,
         pageToken: after,
         startTimestamp: sessionListStartTimestamp()
       });
-      const threads = page.data.filter((session) => session.isMutable).map((draft) => ({
-        status: "regular",
-        remoteId: draft.id,
-        title: draftSessionTitle({
-          title: draft.title,
-          agentSpec: draft.agentSpec ?? defaultAgentSpec
-        }),
-        lastMessageAt: new Date(draft.updatedAt)
-      }));
+      const threads = page.data.map(
+        (session) => sessionToThreadMetadata(
+          session,
+          sessionDisplayTitle(session, defaultAgentSpec)
+        )
+      );
       return {
         threads,
         nextCursor: page.nextPageToken ?? void 0
@@ -2472,15 +2497,10 @@ function createTrueFoundryDraftThreadListAdapter(options) {
     },
     async fetch(remoteId) {
       const draft = await server.getSession({ sessionId: remoteId });
-      return {
-        status: "regular",
-        remoteId: draft.id,
-        title: draftSessionTitle({
-          title: draft.title,
-          agentSpec: draft.agentSpec ?? defaultAgentSpec
-        }),
-        lastMessageAt: new Date(draft.updatedAt)
-      };
+      return sessionToThreadMetadata(
+        draft,
+        sessionDisplayTitle(draft, defaultAgentSpec)
+      );
     },
     async rename() {
     },
@@ -2532,21 +2552,18 @@ function getSession(server, sessionId) {
 // src/truefoundryThreadListAdapter.ts
 var THREAD_LIST_PAGE_SIZE2 = 20;
 function createTrueFoundryThreadListAdapter(options) {
-  const { server, agentName } = options;
+  const { server, agentName, listSessionsAgentId } = options;
   return {
     async list({ after } = {}) {
       const page = await server.listSessions({
-        agentName,
+        ...listSessionsAgentId != null ? { agentId: listSessionsAgentId } : {},
         limit: THREAD_LIST_PAGE_SIZE2,
         pageToken: after,
         startTimestamp: sessionListStartTimestamp()
       });
-      const threads = page.data.map((session) => ({
-        status: "regular",
-        remoteId: session.id,
-        title: session.title ?? void 0,
-        lastMessageAt: new Date(session.updatedAt)
-      }));
+      const threads = page.data.map(
+        (session) => sessionToThreadMetadata(session, session.title ?? void 0)
+      );
       return {
         threads,
         nextCursor: page.nextPageToken ?? void 0
@@ -2558,12 +2575,7 @@ function createTrueFoundryThreadListAdapter(options) {
     },
     async fetch(remoteId) {
       const session = await getSession(server, remoteId);
-      return {
-        status: "regular",
-        remoteId: session.id,
-        title: session.title ?? void 0,
-        lastMessageAt: new Date(session.updatedAt)
-      };
+      return sessionToThreadMetadata(session, session.title ?? void 0);
     },
     async rename() {
     },
@@ -3803,18 +3815,28 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
     [sendTurn]
   );
   const downloadSandboxFile = useCallback3(
-    async (path) => {
+    async ({ turnId, path }) => {
       if (server.downloadSandboxFile == null) {
         throw new Error(
           "Downloading a sandbox file requires AgentChatServer.downloadSandboxFile."
         );
       }
+      if (sessionId == null) {
+        throw new Error(
+          "This session has not been saved yet, so its files cannot be downloaded."
+        );
+      }
       if (sandboxId == null) {
         throw new Error("No sandbox is available yet for this session.");
       }
-      return await server.downloadSandboxFile(sandboxId, { path });
+      return await server.downloadSandboxFile({
+        sessionId,
+        turnId,
+        sandboxId,
+        path
+      });
     },
-    [server, sandboxId]
+    [server, sessionId, sandboxId]
   );
   const draftExtras = useMemo3(() => {
     if (agent.mode !== "draft") {
@@ -3912,20 +3934,23 @@ function useTrueFoundryAgentRuntime(options) {
   );
   const agentMode = agent.mode;
   const namedAgentName = agent.mode === "named" ? agent.agentName : void 0;
+  const listSessionsAgentId = resolved.listSessionsAgentId;
   const threadListAdapter = useMemo3(() => {
     if (agentMode === "draft") {
       const draftAgent = agent;
       return createTrueFoundryDraftThreadListAdapter({
         server,
         defaultAgentSpec: draftAgent.defaultAgentSpec,
-        getAgentSpec: () => pendingAgentSpecRef.current ?? draftAgent.defaultAgentSpec
+        getAgentSpec: () => pendingAgentSpecRef.current ?? draftAgent.defaultAgentSpec,
+        listSessionsAgentId
       });
     }
     return createTrueFoundryThreadListAdapter({
       server,
-      agentName: namedAgentName
+      agentName: namedAgentName,
+      listSessionsAgentId
     });
-  }, [agentMode, namedAgentName, server]);
+  }, [agentMode, namedAgentName, listSessionsAgentId, server]);
   return useRemoteThreadListRuntime({
     allowNesting: true,
     adapter: threadListAdapter,
@@ -3938,30 +3963,19 @@ function useTrueFoundryAgentRuntime(options) {
 
 // src/truefoundryOwnedSessionsThreadListAdapter.ts
 var THREAD_LIST_PAGE_SIZE3 = 20;
-function ownedSessionTitle(session) {
-  if (session.isMutable && session.agentSpec != null) {
-    return draftSessionTitle({
-      title: session.title,
-      agentSpec: session.agentSpec
-    });
-  }
-  return session.title ?? session.agentName ?? session.id;
-}
 function createTrueFoundryOwnedSessionsThreadListAdapter(options) {
-  const { server } = options;
+  const { server, listSessionsAgentId } = options;
   return {
     async list({ after } = {}) {
       const page = await server.listSessions({
+        ...listSessionsAgentId != null ? { agentId: listSessionsAgentId } : {},
         limit: THREAD_LIST_PAGE_SIZE3,
         pageToken: after,
         startTimestamp: sessionListStartTimestamp()
       });
-      const threads = page.data.map((session) => ({
-        status: "regular",
-        remoteId: session.id,
-        title: ownedSessionTitle(session),
-        lastMessageAt: new Date(session.updatedAt)
-      }));
+      const threads = page.data.map(
+        (session) => sessionToThreadMetadata(session, sessionDisplayTitle(session))
+      );
       return {
         threads,
         nextCursor: page.nextPageToken ?? void 0
@@ -3974,12 +3988,7 @@ function createTrueFoundryOwnedSessionsThreadListAdapter(options) {
     },
     async fetch(remoteId) {
       const session = await server.getSession({ sessionId: remoteId });
-      return {
-        status: "regular",
-        remoteId: session.id,
-        title: ownedSessionTitle(session),
-        lastMessageAt: new Date(session.updatedAt)
-      };
+      return sessionToThreadMetadata(session, sessionDisplayTitle(session));
     },
     async rename() {
     },
@@ -3999,7 +4008,7 @@ function createTrueFoundryOwnedSessionsThreadListAdapter(options) {
 
 // src/hooks.ts
 import { useMemo as useMemo4 } from "react";
-import { useAui as useAui2 } from "@assistant-ui/store";
+import { useAui as useAui2, useAuiState as useAuiState2 } from "@assistant-ui/store";
 var useTrueFoundryApprovals = () => {
   const extras = trueFoundryExtras.use((e) => e, void 0);
   return useMemo4(
@@ -4049,9 +4058,20 @@ var useTrueFoundryResumeMcpAuth = () => {
   return () => trueFoundryExtras.get(aui).resumeMcpAuth();
 };
 var useTrueFoundrySandboxId = () => trueFoundryExtras.use((e) => e.sandboxId, void 0);
+var useTrueFoundryTurnId = () => useAuiState2(
+  (state) => state.message.metadata?.custom?.turnId
+);
 var useTrueFoundryDownloadSandboxFile = () => {
   const aui = useAui2();
-  return (path) => trueFoundryExtras.get(aui).downloadSandboxFile(path);
+  const turnId = useTrueFoundryTurnId();
+  return (path) => {
+    if (turnId == null) {
+      throw new Error(
+        "Downloading a sandbox file requires a message scope to resolve its turn."
+      );
+    }
+    return trueFoundryExtras.get(aui).downloadSandboxFile({ turnId, path });
+  };
 };
 var useTrueFoundryCancel = () => {
   const aui = useAui2();
@@ -4191,6 +4211,7 @@ export {
   useTrueFoundryResumeMcpAuth,
   useTrueFoundrySandboxId,
   useTrueFoundryToolResponses,
+  useTrueFoundryTurnId,
   useTrueFoundryUpdateAgentSpec
 };
 //# sourceMappingURL=index.js.map