@truefoundry/assistant-ui-runtime 0.1.10-rc.1 → 0.1.12

package/dist/index.js

@@ -11,7 +11,7 @@ import {
   isTfyMcpToolInfo,
   isTfySystemToolInfo,
   isTfyToolInfo
-} from "./chunk-BRPTG6VA.js";
+} from "./chunk-2NLIAJJX.js";
 
 // src/useTrueFoundryAgentRuntime.ts
 import {
@@ -1892,9 +1892,11 @@ function projectHistoryTurns(snapshot, options) {
   const messages = [];
   let lastAssistantIndex;
   let groupRootIds = [];
+  let sandboxId;
   for (let turnIndex = 0; turnIndex < snapshot.turns.length; turnIndex++) {
     const record = snapshot.turns[turnIndex];
     const turnRootIds = record.rootModelMessageIds ?? [];
+    sandboxId = record.sandboxId ?? sandboxId;
     if (record.userText) {
       groupRootIds = [...turnRootIds];
     } else {
@@ -1932,7 +1934,7 @@ function projectHistoryTurns(snapshot, options) {
     const custom = {
       ...baseCustom,
       turnId: record.id,
-      ...record.sandboxId != null ? { sandboxId: record.sandboxId } : {}
+      ...sandboxId != null ? { sandboxId } : {}
     };
     if (record.userText) {
       messages.push(
@@ -2378,7 +2380,11 @@ function turnStreamUpdateToAssistantMessage(turnId, update, existing, options) {
       unstable_annotations: [],
       unstable_data: [],
       steps: [],
-      custom: { ...update.metadata?.custom, turnId }
+      custom: {
+        ...existing?.role === "assistant" ? existing.metadata.custom : {},
+        ...update.metadata?.custom,
+        turnId
+      }
     }
   };
 }
@@ -2415,6 +2421,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;
@@ -2436,33 +2449,50 @@ 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),
+    custom: {
+      isMutable: session.isMutable,
+      ...session.agentName != null ? { 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
@@ -2476,15 +2506,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() {
     },
@@ -2536,21 +2561,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
@@ -2562,12 +2584,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() {
     },
@@ -3926,20 +3943,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,
@@ -3952,30 +3972,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
@@ -3988,12 +3997,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() {
     },