@truefoundry/assistant-ui-runtime 0.1.5 → 0.1.6

package/src/useTrueFoundryAgentRuntime.ts

@@ -44,9 +44,8 @@ function useTrueFoundryAgentRuntimeImpl(
     pendingAgentSpecRef: MutableRefObject<AgentSpec | undefined>,
 ) {
     const {
-        client,
+        server,
         agent,
-        privateClient,
         adapters,
         onError,
         listEventsConcurrency,
@@ -54,9 +53,7 @@ function useTrueFoundryAgentRuntimeImpl(
     } = options;
 
     const draftBridgeRef = useRef(
-        agent.mode === "draft" && privateClient != null
-            ? createDraftSessionBridge(privateClient)
-            : null,
+        agent.mode === "draft" ? createDraftSessionBridge(server) : null,
     );
 
     const draftSessionId = useAuiState(
@@ -119,13 +116,12 @@ function useTrueFoundryAgentRuntimeImpl(
         resetFromTurn,
         retryLoad,
     } = useTrueFoundryAgentMessages({
-        client,
+        server,
         sessionId,
         isMain,
         listEventsConcurrency,
         onError,
         initializeSession,
-        privateClient: agent.mode === "draft" ? privateClient : undefined,
         getTurnHeaders: agent.mode === "draft" ? getTurnHeaders : undefined,
     });
 
@@ -151,9 +147,9 @@ function useTrueFoundryAgentRuntimeImpl(
 
     const downloadSandboxFile = useCallback(
         async (path: string) => {
-            if (privateClient == null) {
+            if (server.downloadSandboxFile == null) {
                 const error = new Error(
-                    "Downloading a sandbox file requires a `privateClient` PrivateAgentSessionClient.",
+                    "Downloading a sandbox file requires AgentChatServer.downloadSandboxFile.",
                 );
                 onError?.(error);
                 throw error;
@@ -163,10 +159,9 @@ function useTrueFoundryAgentRuntimeImpl(
                 onError?.(error);
                 throw error;
             }
-            const response = await privateClient.downloadSandboxFile(sandboxId, { path });
-            return await response.blob();
+            return await server.downloadSandboxFile(sandboxId, { path });
         },
-        [privateClient, sandboxId, onError],
+        [server, sandboxId, onError],
     );
 
     const draftExtras = useMemo(() => {
@@ -258,42 +253,30 @@ function useTrueFoundryAgentRuntimeImpl(
 }
 
 export function useTrueFoundryAgentRuntime(options: UseTrueFoundryAgentRuntimeOptions) {
-    // resolveTrueFoundryAgentRuntimeOptions is a cheap pure function; memoizing on the
-    // whole `options` object (which callers typically pass as an inline literal) means
-    // `resolved` — and everything derived from it — would be recreated on every render,
-    // causing `threadListAdapter` to be a new object each render.  We compute `resolved`
-    // unconditionally and stabilize individual downstream values with primitives/refs.
     const resolved = resolveTrueFoundryAgentRuntimeOptions(options);
-    const { client, agent, privateClient } = resolved;
+    const { server, agent } = resolved;
 
     const pendingAgentSpecRef = useRef<AgentSpec | undefined>(
         agent.mode === "draft" ? agent.defaultAgentSpec : undefined,
     );
 
-    // Use stable primitive deps so that threadListAdapter is not recreated every render
-    // when options is an inline object literal (new reference on every parent render).
     const agentMode = agent.mode;
     const namedAgentName = agent.mode === "named" ? agent.agentName : undefined;
     const threadListAdapter = useMemo(() => {
         if (agentMode === "draft") {
-            if (privateClient == null) {
-                throw new Error(
-                    "Draft agent mode requires a `privateClient` PrivateAgentSessionClient.",
-                );
-            }
             const draftAgent = agent as Extract<typeof agent, { mode: "draft" }>;
             return createTrueFoundryDraftThreadListAdapter({
-                privateClient,
+                server,
                 defaultAgentSpec: draftAgent.defaultAgentSpec,
                 getAgentSpec: () => pendingAgentSpecRef.current ?? draftAgent.defaultAgentSpec,
             });
         }
         return createTrueFoundryThreadListAdapter({
-            client,
+            server,
             agentName: namedAgentName!,
         });
         // eslint-disable-next-line react-hooks/exhaustive-deps
-    }, [agentMode, namedAgentName, client, privateClient]);
+    }, [agentMode, namedAgentName, server]);
 
     return useRemoteThreadListRuntime({
         allowNesting: true,

package/src/private/bindDraftAgentSession.test.ts

@@ -1,54 +0,0 @@
-import { describe, expect, it, vi } from "vitest";
-
-import type { PrivateAgentSessionClient } from "truefoundry-gateway-sdk/agents/private";
-
-import { bindDraftAgentSession } from "./bindDraftAgentSession.js";
-
-const draft = {
-    type: "session/draft" as const,
-    id: "draft-1",
-    agentSpec: { model: { name: "anthropic/claude-sonnet-4-6" } },
-    createdBySubject: { type: "user" as const, id: "u1" },
-    createdAt: "2026-06-30T10:00:00.000Z",
-    updatedAt: "2026-06-30T10:00:00.000Z",
-};
-
-describe("bindDraftAgentSession", () => {
-    it("validates the draft and binds turns to the draft session id", async () => {
-        const getDraftSession = vi.fn().mockResolvedValue(draft);
-        const privateClient = {
-            getDraftSession,
-        } as unknown as PrivateAgentSessionClient;
-
-        const session = await bindDraftAgentSession(privateClient, "draft-1");
-
-        expect(getDraftSession).toHaveBeenCalledWith({ draftSessionId: "draft-1" });
-        expect(session.id).toBe("draft-1");
-    });
-});
-
-describe("createDraftSessionBridge", () => {
-    it("syncs agent spec via draftSessions.update and returns updatedAt", async () => {
-        const { createDraftSessionBridge } = await import("./draftSessionBridge.js");
-        const update = vi.fn().mockResolvedValue({ data: draft });
-        const getDraftSession = vi.fn().mockResolvedValue(draft);
-        const privateClient = {
-            getDraftSession,
-            client: {
-                agents: {
-                    private: {
-                        draftSessions: { update },
-                    },
-                },
-            },
-        } as unknown as PrivateAgentSessionClient;
-
-        const bridge = createDraftSessionBridge(privateClient);
-        const updatedAt = await bridge.syncAgentSpec("draft-1", draft.agentSpec);
-
-        expect(update).toHaveBeenCalledWith("draft-1", {
-            agentSpec: draft.agentSpec,
-        });
-        expect(updatedAt).toBe(draft.updatedAt);
-    });
-});

package/src/private/bindDraftAgentSession.ts

@@ -1,28 +0,0 @@
-import type { AgentSession } from "truefoundry-gateway-sdk/agents";
-import type { PrivateAgentSessionClient } from "truefoundry-gateway-sdk/agents/private";
-
-const inflightByDraftId = new Map<string, Promise<AgentSession>>();
-
-/**
- * Binds turn APIs for a draft session via `PrivateAgentSessionClient.getDraftSession`.
- * Returns an `AgentDraftSession` (identical turn surface to `AgentSession`).
- */
-export async function bindDraftAgentSession(
-    privateClient: PrivateAgentSessionClient,
-    draftSessionId: string,
-): Promise<AgentSession> {
-    let inflight = inflightByDraftId.get(draftSessionId);
-    if (inflight == null) {
-        inflight = privateClient
-            .getDraftSession({ draftSessionId })
-            // AgentDraftSession shares the turn API with AgentSession.
-            .then((draft) => draft as unknown as AgentSession)
-            .finally(() => {
-                if (inflightByDraftId.get(draftSessionId) === inflight) {
-                    inflightByDraftId.delete(draftSessionId);
-                }
-            });
-        inflightByDraftId.set(draftSessionId, inflight);
-    }
-    return inflight;
-}

package/src/private/getGatewayFromPrivateClient.ts

@@ -1,13 +0,0 @@
-import type { TrueFoundryGateway } from "truefoundry-gateway-sdk";
-import type { PrivateAgentSessionClient } from "truefoundry-gateway-sdk/agents/private";
-
-/** Reach the underlying gateway for APIs not yet wrapped on PrivateAgentSessionClient (update, sandbox download). */
-export function getGatewayFromPrivateClient(
-    privateClient: PrivateAgentSessionClient,
-): TrueFoundryGateway {
-    const internal = privateClient as unknown as { client: TrueFoundryGateway };
-    if (internal.client == null) {
-        throw new Error("PrivateAgentSessionClient is missing an internal gateway client.");
-    }
-    return internal.client;
-}