@truefoundry/assistant-ui-runtime 0.1.3-rc.2 → 0.1.4

package/dist/index.js

@@ -2809,7 +2809,6 @@ function findPausedAssistantMessage(messages) {
 }
 
 // src/streamTurn.ts
-import "truefoundry-gateway-sdk/agents";
 function buildTurnInput(options) {
   if (options.inputs != null) {
     return options.inputs;
@@ -2831,17 +2830,23 @@ function bindAbort(session, abortSignal) {
   return onAbort;
 }
 async function* streamTurnContent(session, foldState, options, abortSignal, groupRootBaseline, onTurnIdAvailable) {
-  const previousTurnId = options.previousTurnId === null ? void 0 : options.previousTurnId ?? "auto";
+  const previousTurnId = options.previousTurnId === null ? null : options.previousTurnId ?? "auto";
   const turn = session.prepareTurn({
     input: buildTurnInput(options),
-    ...previousTurnId != null ? { previousTurnId } : {}
+    ...previousTurnId !== void 0 ? { previousTurnId } : {}
   });
   const onAbort = bindAbort(session, abortSignal);
   if (abortSignal.aborted) {
     return;
   }
+  let turnIdNotified = false;
+  const notifyTurnIdIfAvailable = () => {
+    if (!turnIdNotified && turn.id != null) {
+      onTurnIdAvailable?.(turn.id);
+      turnIdNotified = true;
+    }
+  };
   try {
-    let turnIdNotified = false;
     for await (const update of streamTurnEvents(
       turn.execute(
         { stream: true },
@@ -2853,16 +2858,12 @@ async function* streamTurnContent(session, foldState, options, abortSignal, grou
       foldState,
       groupRootBaseline
     )) {
-      if (!turnIdNotified && turn.id != null) {
-        onTurnIdAvailable?.(turn.id);
-        turnIdNotified = true;
-      }
+      notifyTurnIdIfAvailable();
       yield update;
     }
-    if (!turnIdNotified && turn.id != null) {
-      onTurnIdAvailable?.(turn.id);
-    }
+    notifyTurnIdIfAvailable();
   } catch (error) {
+    notifyTurnIdIfAvailable();
     if (error instanceof Error && error.name === "AbortError") {
       return;
     }
@@ -3260,6 +3261,7 @@ function useTrueFoundryAgentMessages({
       const isContinuation = "inputs" in options || "resumeMcpAuth" in options && options.resumeMcpAuth === true;
       const continuationTurnId = snapshotRef.current.activeStream?.turnId;
       const turnId = isContinuation && continuationTurnId != null ? continuationTurnId : generateId();
+      const isFirstTurnInSession = "userMessage" in options && options.previousTurnId === void 0 && snapshotRef.current.turns.length === 0 && snapshotRef.current.pendingUser == null && snapshotRef.current.activeStream == null;
       const turnIdRef = { current: turnId };
       if ("inputs" in options) {
         applyUserToolResponsesToFold(
@@ -3335,7 +3337,7 @@ function useTrueFoundryAgentMessages({
             snapshotRef.current.fold,
             {
               userMessage: options.userMessage,
-              ...options.previousTurnId !== void 0 ? { previousTurnId: options.previousTurnId } : {},
+              ...options.previousTurnId !== void 0 ? { previousTurnId: options.previousTurnId } : isFirstTurnInSession ? { previousTurnId: null } : {},
               ...streamHeaders
             },
             signal,
@@ -3691,8 +3693,7 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
         onError?.(error);
         throw error;
       }
-      const gateway = getGatewayFromPrivateClient(privateClient);
-      const response = await gateway.agents.downloadSandboxFile(sandboxId, { path });
+      const response = await privateClient.downloadSandboxFile(sandboxId, { path });
       return await response.blob();
     },
     [privateClient, sandboxId, onError]