@truefoundry/assistant-ui-runtime 0.1.16 → 0.1.18

package/dist/index.js

@@ -11,7 +11,7 @@ import {
   isTfyMcpToolInfo,
   isTfySystemToolInfo,
   isTfyToolInfo
-} from "./chunk-S4SH2KT2.js";
+} from "./chunk-N6LJ3LR2.js";
 
 // src/useTrueFoundryAgentRuntime.ts
 import {
@@ -22,8 +22,8 @@ import {
   useRemoteThreadListRuntime,
   useRuntimeAdapters
 } from "@assistant-ui/core/react";
-import { useAui, useAuiState } from "@assistant-ui/store";
-import { useCallback as useCallback3, useMemo as useMemo3, useRef as useRef3, useState as useState3 } from "react";
+import { useAui as useAui2, useAuiState } from "@assistant-ui/store";
+import { useCallback as useCallback4, useMemo as useMemo3, useRef as useRef3, useState as useState3 } from "react";
 
 // src/constants.ts
 var ROOT_THREAD_ID = "main";
@@ -2551,7 +2551,9 @@ function createTrueFoundryDraftThreadListAdapter(options) {
 
 // src/truefoundryExtras.ts
 import { createRuntimeExtras } from "@assistant-ui/core/internal";
-var trueFoundryExtras = createRuntimeExtras(
+import { useAui } from "@assistant-ui/store";
+import { useCallback, useSyncExternalStore } from "react";
+var extrasBrand = createRuntimeExtras(
   "useTrueFoundryAgentRuntime"
 );
 var EMPTY_DRAFT_EXTRAS = {
@@ -2570,6 +2572,91 @@ var EMPTY_DRAFT_EXTRAS = {
     throw new Error("Draft agent extras are only available in draft mode.");
   }
 };
+function walkAssistantClientAncestors(client, visit) {
+  let current = client;
+  const seen = /* @__PURE__ */ new Set();
+  while (current != null && !seen.has(current)) {
+    seen.add(current);
+    if (visit(current) === "stop") return;
+    const parent = Object.getPrototypeOf(current);
+    if (parent == null || parent === Object.prototype) {
+      return;
+    }
+    current = parent;
+  }
+}
+function tryGetTrueFoundryExtras(client) {
+  let found;
+  walkAssistantClientAncestors(client, (current) => {
+    try {
+      const thread = current.thread;
+      if (typeof thread === "function") {
+        const extras = extrasBrand.tryGet(thread().getState().extras);
+        if (extras != null) {
+          found = extras;
+          return "stop";
+        }
+      }
+    } catch {
+    }
+    return "continue";
+  });
+  return found;
+}
+function getTrueFoundryExtras(client) {
+  const extras = tryGetTrueFoundryExtras(client);
+  if (extras == null) {
+    throw new Error(
+      "The current thread is not backed by the useTrueFoundryAgentRuntime runtime."
+    );
+  }
+  return extras;
+}
+function subscribeClientChain(client) {
+  return (onStoreChange) => {
+    const unsubs = [];
+    walkAssistantClientAncestors(client, (current) => {
+      try {
+        const subscribe = current.subscribe;
+        if (typeof subscribe === "function") {
+          unsubs.push(subscribe(onStoreChange));
+        }
+        return "continue";
+      } catch {
+        return "stop";
+      }
+    });
+    return () => {
+      for (const unsub of unsubs) {
+        unsub();
+      }
+    };
+  };
+}
+function useTrueFoundryRuntimeExtras() {
+  const aui = useAui();
+  const subscribe = useCallback(subscribeClientChain(aui), [aui]);
+  const getSnapshot = useCallback(() => tryGetTrueFoundryExtras(aui), [aui]);
+  return useSyncExternalStore(subscribe, getSnapshot, getSnapshot);
+}
+function useTrueFoundryExtrasApi(select, fallback) {
+  const extras = useTrueFoundryRuntimeExtras();
+  const hasFallback = arguments.length >= 2;
+  if (extras == null) {
+    if (hasFallback) return fallback;
+    throw new Error(
+      "The current thread is not backed by the useTrueFoundryAgentRuntime runtime."
+    );
+  }
+  return select != null ? select(extras) : extras;
+}
+var trueFoundryExtras = {
+  provide: extrasBrand.provide,
+  is: extrasBrand.is,
+  tryGet: extrasBrand.tryGet,
+  get: getTrueFoundryExtras,
+  use: useTrueFoundryExtrasApi
+};
 
 // src/sessions.ts
 var inflightBySessionId = /* @__PURE__ */ new Map();
@@ -2654,7 +2741,7 @@ function resolveTrueFoundryAgentRuntimeOptions(options) {
 }
 
 // src/draft/useDraftAgentSpec.ts
-import { useCallback, useEffect, useMemo, useRef, useState } from "react";
+import { useCallback as useCallback2, useEffect, useMemo, useRef, useState } from "react";
 var SPEC_SYNC_DEBOUNCE_MS = 400;
 function useDraftAgentSpec({
   draftSessionId,
@@ -2744,7 +2831,7 @@ function useDraftAgentSpec({
       setIsSpecLoading(false);
     };
   }, [defaultAgentSpec, draftBridge, draftSessionId, enabled, onError]);
-  const flushSpecSync = useCallback(
+  const flushSpecSync = useCallback2(
     async (draftId, spec, generation) => {
       if (draftBridge == null) {
         return;
@@ -2771,7 +2858,7 @@ function useDraftAgentSpec({
     },
     [draftBridge, onAgentSpecChange, onError]
   );
-  const scheduleSpecSync = useCallback(
+  const scheduleSpecSync = useCallback2(
     (draftId, spec) => {
       if (syncTimeoutRef.current != null) {
         clearTimeout(syncTimeoutRef.current);
@@ -2797,7 +2884,7 @@ function useDraftAgentSpec({
   );
   const scheduleSpecSyncRef = useRef(scheduleSpecSync);
   scheduleSpecSyncRef.current = scheduleSpecSync;
-  const flushPendingSpecSyncNow = useCallback(async () => {
+  const flushPendingSpecSyncNow = useCallback2(async () => {
     if (syncTimeoutRef.current != null) {
       clearTimeout(syncTimeoutRef.current);
       syncTimeoutRef.current = void 0;
@@ -2812,7 +2899,7 @@ function useDraftAgentSpec({
       await inFlightFlushRef.current;
     }
   }, []);
-  const adoptAgentSpec = useCallback(
+  const adoptAgentSpec = useCallback2(
     ({
       agentSpec: persistedSpec,
       updatedAt
@@ -2835,7 +2922,7 @@ function useDraftAgentSpec({
     },
     []
   );
-  const takeTurnHeaderTimestamp = useCallback(async () => {
+  const takeTurnHeaderTimestamp = useCallback2(async () => {
     await flushPendingSpecSyncNow();
     const updatedAt = lastUpdatedAtRef.current;
     lastUpdatedAtRef.current = void 0;
@@ -2849,7 +2936,7 @@ function useDraftAgentSpec({
     },
     []
   );
-  const updateAgentSpec = useCallback(
+  const updateAgentSpec = useCallback2(
     (update) => {
       if (!enabled || draftBridge == null) {
         return;
@@ -2892,7 +2979,7 @@ function useDraftAgentSpec({
 
 // src/useTrueFoundryAgentMessages.ts
 import { generateId } from "@assistant-ui/core";
-import { useCallback as useCallback2, useEffect as useEffect2, useMemo as useMemo2, useRef as useRef2, useState as useState2 } from "react";
+import { useCallback as useCallback3, useEffect as useEffect2, useMemo as useMemo2, useRef as useRef2, useState as useState2 } from "react";
 
 // src/loadSessionSnapshot.ts
 var inflightBySessionId2 = /* @__PURE__ */ new Map();
@@ -3183,7 +3270,7 @@ function useTrueFoundryAgentMessages({
   const lazilyCreatedSessionIdRef = useRef2(void 0);
   const initialLoadStartedForRef = useRef2(void 0);
   const skipInitialPromotionLoadForRef = useRef2(void 0);
-  const markResumeUnavailable = useCallback2((value) => {
+  const markResumeUnavailable = useCallback3((value) => {
     resumeUnavailableRef.current = value;
     setResumeUnavailable(value);
   }, []);
@@ -3205,7 +3292,7 @@ function useTrueFoundryAgentMessages({
     () => projectSessionMessages(snapshot, projectOptions),
     [snapshot, projectOptions]
   );
-  const runStream = useCallback2(
+  const runStream = useCallback3(
     (createStream, turnIdRef, isContinuation) => {
       const streamGeneration = ++streamGenerationRef.current;
       abortControllerRef.current?.abort();
@@ -3297,7 +3384,7 @@ function useTrueFoundryAgentMessages({
     },
     [onError]
   );
-  const load = useCallback2(async () => {
+  const load = useCallback3(async () => {
     void loadRetryTrigger;
     if (sessionId == null) {
       createdAtByMessageIdRef.current = /* @__PURE__ */ new Map();
@@ -3400,7 +3487,7 @@ function useTrueFoundryAgentMessages({
   useEffect2(() => {
     void load().catch(() => void 0);
   }, [load]);
-  const sendTurn = useCallback2(
+  const sendTurn = useCallback3(
     async (options) => {
       let gatewayTurnAccepted = false;
       let pendingUserWasSet = false;
@@ -3564,7 +3651,7 @@ function useTrueFoundryAgentMessages({
     },
     [server, runStream, sessionId]
   );
-  const cancel = useCallback2(async () => {
+  const cancel = useCallback3(async () => {
     if (sessionId == null) {
       abortControllerRef.current?.abort();
       return;
@@ -3582,7 +3669,7 @@ function useTrueFoundryAgentMessages({
   }, [server, sessionId, markResumeUnavailable]);
   const isRunningRef = useRef2(isRunning);
   isRunningRef.current = isRunning;
-  const trySendCollectedRequiredActions = useCallback2(
+  const trySendCollectedRequiredActions = useCallback3(
     (nextSnapshot) => {
       if (isRunningRef.current) {
         return;
@@ -3599,7 +3686,7 @@ function useTrueFoundryAgentMessages({
     },
     [projectOptions, sendTurn]
   );
-  const respondToToolApproval = useCallback2(
+  const respondToToolApproval = useCallback3(
     (response) => {
       const prev = snapshotRef.current;
       const approvals = new Map(prev.requiredActions.approvals);
@@ -3618,7 +3705,7 @@ function useTrueFoundryAgentMessages({
     },
     [trySendCollectedRequiredActions]
   );
-  const respondToToolResponse = useCallback2(
+  const respondToToolResponse = useCallback3(
     (response) => {
       const prev = snapshotRef.current;
       const toolResponses = new Map(prev.requiredActions.toolResponses);
@@ -3634,7 +3721,7 @@ function useTrueFoundryAgentMessages({
     },
     [trySendCollectedRequiredActions]
   );
-  const resumeRun = useCallback2(async () => {
+  const resumeRun = useCallback3(async () => {
     const turn = runningTurnRef.current;
     if (turn == null) {
       return;
@@ -3657,7 +3744,7 @@ function useTrueFoundryAgentMessages({
       true
     );
   }, [runStream, server]);
-  const branchFromTurn = useCallback2(
+  const branchFromTurn = useCallback3(
     async (turnId, userMessage) => {
       let committed;
       let previousTurnId;
@@ -3700,7 +3787,7 @@ function useTrueFoundryAgentMessages({
     },
     [cancel, server, sendTurn, sessionId]
   );
-  const resetFromTurn = useCallback2(
+  const resetFromTurn = useCallback3(
     async (turnId) => {
       const committed = commitActiveStream(snapshotRef.current);
       const originalInput = resolveTurnInput(committed, turnId);
@@ -3714,7 +3801,7 @@ function useTrueFoundryAgentMessages({
     },
     [branchFromTurn]
   );
-  const editFromTurn = useCallback2(
+  const editFromTurn = useCallback3(
     async (turnId, editedText) => {
       const committed = commitActiveStream(snapshotRef.current);
       const originalInput = resolveTurnInput(committed, turnId);
@@ -3731,11 +3818,11 @@ function useTrueFoundryAgentMessages({
     },
     [branchFromTurn]
   );
-  const retryLoad = useCallback2(() => {
+  const retryLoad = useCallback3(() => {
     setLoadRetryTrigger((n) => n + 1);
   }, []);
   const hasOlderHistory = snapshot.historyPagination?.hasOlder === true;
-  const loadOlderHistory = useCallback2(async () => {
+  const loadOlderHistory = useCallback3(async () => {
     if (sessionId == null || isMain === false) {
       return;
     }
@@ -3830,7 +3917,7 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
   });
   const takeTurnHeaderTimestampRef = useRef3(draftSpec.takeTurnHeaderTimestamp);
   takeTurnHeaderTimestampRef.current = draftSpec.takeTurnHeaderTimestamp;
-  const getTurnHeaders = useCallback3(async () => {
+  const getTurnHeaders = useCallback4(async () => {
     if (agent.mode !== "draft") {
       return void 0;
     }
@@ -3840,8 +3927,8 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
     }
     return { [DRAFT_SESSION_LAST_UPDATED_AT_HEADER]: updatedAt };
   }, [agent.mode]);
-  const aui = useAui();
-  const initializeSession = useCallback3(
+  const aui = useAui2();
+  const initializeSession = useCallback4(
     () => aui.threadListItem().initialize(),
     [aui]
   );
@@ -3889,7 +3976,7 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
     () => () => sendTurn({ resumeMcpAuth: true }),
     [sendTurn]
   );
-  const downloadSandboxFile = useCallback3(
+  const downloadSandboxFile = useCallback4(
     async ({ turnId, path }) => {
       if (server.downloadSandboxFile == null) {
         throw new Error(
@@ -4086,9 +4173,9 @@ function createTrueFoundryOwnedSessionsThreadListAdapter(options) {
 
 // src/hooks.ts
 import { useMemo as useMemo4 } from "react";
-import { useAui as useAui2, useAuiState as useAuiState2 } from "@assistant-ui/store";
+import { useAui as useAui3, useAuiState as useAuiState2 } from "@assistant-ui/store";
 var useTrueFoundryApprovals = () => {
-  const extras = trueFoundryExtras.use((e) => e, void 0);
+  const extras = useTrueFoundryRuntimeExtras();
   return useMemo4(
     () => ({
       pending: extras?.pendingApprovals ?? [],
@@ -4100,7 +4187,7 @@ var useTrueFoundryApprovals = () => {
   );
 };
 var useTrueFoundryToolResponses = () => {
-  const extras = trueFoundryExtras.use((e) => e, void 0);
+  const extras = useTrueFoundryRuntimeExtras();
   return useMemo4(
     () => ({
       pending: extras?.pendingToolResponses ?? [],
@@ -4112,7 +4199,7 @@ var useTrueFoundryToolResponses = () => {
   );
 };
 var useTrueFoundryMcpAuth = () => {
-  const extras = trueFoundryExtras.use((e) => e, void 0);
+  const extras = useTrueFoundryRuntimeExtras();
   return useMemo4(
     () => ({
       pending: extras?.pendingMcpAuth ?? null,
@@ -4124,23 +4211,23 @@ var useTrueFoundryMcpAuth = () => {
   );
 };
 var useTrueFoundryRespondToToolApproval = () => {
-  const aui = useAui2();
-  return (response) => trueFoundryExtras.get(aui).respondToToolApproval(response);
+  const aui = useAui3();
+  return (response) => getTrueFoundryExtras(aui).respondToToolApproval(response);
 };
 var useTrueFoundryRespondToToolResponse = () => {
-  const aui = useAui2();
-  return (response) => trueFoundryExtras.get(aui).respondToToolResponse(response);
+  const aui = useAui3();
+  return (response) => getTrueFoundryExtras(aui).respondToToolResponse(response);
 };
 var useTrueFoundryResumeMcpAuth = () => {
-  const aui = useAui2();
-  return () => trueFoundryExtras.get(aui).resumeMcpAuth();
+  const aui = useAui3();
+  return () => getTrueFoundryExtras(aui).resumeMcpAuth();
 };
-var useTrueFoundrySandboxId = () => trueFoundryExtras.use((e) => e.sandboxId, void 0);
+var useTrueFoundrySandboxId = () => useTrueFoundryRuntimeExtras()?.sandboxId;
 var useTrueFoundryTurnId = () => useAuiState2(
   (state) => state.message.metadata?.custom?.turnId
 );
 var useTrueFoundryDownloadSandboxFile = () => {
-  const aui = useAui2();
+  const aui = useAui3();
   const turnId = useTrueFoundryTurnId();
   return (path) => {
     if (turnId == null) {
@@ -4148,19 +4235,19 @@ var useTrueFoundryDownloadSandboxFile = () => {
         "Downloading a sandbox file requires a message scope to resolve its turn."
       );
     }
-    return trueFoundryExtras.get(aui).downloadSandboxFile({ turnId, path });
+    return getTrueFoundryExtras(aui).downloadSandboxFile({ turnId, path });
   };
 };
 var useTrueFoundryCancel = () => {
-  const aui = useAui2();
-  return () => trueFoundryExtras.get(aui).cancel();
+  const aui = useAui3();
+  return () => getTrueFoundryExtras(aui).cancel();
 };
 var useTrueFoundryReload = () => {
-  const aui = useAui2();
-  return () => trueFoundryExtras.get(aui).reload();
+  const aui = useAui3();
+  return () => getTrueFoundryExtras(aui).reload();
 };
 var useTrueFoundryHistoryPagination = () => {
-  const extras = trueFoundryExtras.use((e) => e, void 0);
+  const extras = useTrueFoundryRuntimeExtras();
   return useMemo4(
     () => ({
       hasOlderHistory: extras?.hasOlderHistory ?? false,
@@ -4172,29 +4259,29 @@ var useTrueFoundryHistoryPagination = () => {
     [extras]
   );
 };
-var useTrueFoundryResumeUnavailable = () => trueFoundryExtras.use((e) => e.resumeUnavailable, false) ?? false;
+var useTrueFoundryResumeUnavailable = () => useTrueFoundryRuntimeExtras()?.resumeUnavailable ?? false;
 var useTrueFoundryResetFromTurn = () => {
-  const aui = useAui2();
-  return (turnId) => trueFoundryExtras.get(aui).resetFromTurn(turnId);
+  const aui = useAui3();
+  return (turnId) => getTrueFoundryExtras(aui).resetFromTurn(turnId);
 };
 var useTrueFoundryAgentSpec = () => {
-  const extras = trueFoundryExtras.use((e) => e.draft, null);
+  const extras = useTrueFoundryRuntimeExtras()?.draft ?? null;
   return useMemo4(
     () => ({ ...EMPTY_DRAFT_EXTRAS, ...extras }),
     [extras]
   );
 };
 var useTrueFoundryUpdateAgentSpec = () => {
-  const aui = useAui2();
-  return (update) => trueFoundryExtras.get(aui).draft?.updateAgentSpec(update);
+  const aui = useAui3();
+  return (update) => getTrueFoundryExtras(aui).draft?.updateAgentSpec(update);
 };
 var useTrueFoundryFlushAgentSpec = () => {
-  const aui = useAui2();
-  return () => trueFoundryExtras.get(aui).draft?.flushAgentSpec() ?? Promise.resolve();
+  const aui = useAui3();
+  return () => getTrueFoundryExtras(aui).draft?.flushAgentSpec() ?? Promise.resolve();
 };
 var useTrueFoundryAdoptAgentSpec = () => {
-  const aui = useAui2();
-  return (request) => trueFoundryExtras.get(aui).draft?.adoptAgentSpec(request);
+  const aui = useAui3();
+  return (request) => getTrueFoundryExtras(aui).draft?.adoptAgentSpec(request);
 };
 
 // src/attachmentAdapter.ts
@@ -4269,6 +4356,7 @@ export {
   getTfyMcpInitServers,
   getTfyThreadState,
   getTfyUsage,
+  getTrueFoundryExtras,
   getTurnMessageContent,
   isEventDelta,
   isTfyMcpToolInfo,
@@ -4284,6 +4372,7 @@ export {
   toTrueFoundryApprovalInputs,
   trueFoundryAttachmentAdapter,
   trueFoundryExtras,
+  tryGetTrueFoundryExtras,
   useTrueFoundryAdoptAgentSpec,
   useTrueFoundryAgentRuntime,
   useTrueFoundryAgentSpec,