@truefoundry/assistant-ui-runtime 0.1.13 → 0.1.14

package/dist/plugins/truefoundry-agent-server-adapter/index.d.ts

@@ -1,6 +1,6 @@
 import { AgentSessionClient } from 'truefoundry-gateway-sdk/agents';
 import { PrivateAgentSessionClient } from 'truefoundry-gateway-sdk/agents/private';
-import { A as AgentSpec, K as CreateSessionRequest, W as ListSessionsParams, S as Session, d as Turn, az as TurnState, a as AgentChatServer, aH as UpdateSessionRequest, i as AgentSelectorEntry, E as ConnectorSelectorEntry, a6 as ModelSelectorEntry, au as SkillSelectorEntry, g as AgentBuilderServer } from '../../types-BE-F3vCD.js';
+import { A as AgentSpec, N as CreateSessionRequest, Y as ListSessionsParams, S as Session, d as Turn, aC as TurnState, a as AgentChatServer, aK as UpdateSessionRequest, k as AgentSelectorEntry, G as ConnectorSelectorEntry, a8 as ModelSelectorEntry, ax as SkillSelectorEntry, g as AgentBuilderServer } from '../../types-VU3Fc17W.js';
 import { TruefoundryGatewayApi } from 'truefoundry-gateway-sdk';
 
 /**
@@ -29,8 +29,7 @@ type RuntimeSkillMount = NonNullable<NonNullable<AgentSpec["skills"]>[number]>;
 type RuntimeMcpServerMount = NonNullable<NonNullable<AgentSpec["mcpServers"]>[number]>;
 type TfySkillMount = RuntimeSkillMount & TruefoundryGatewayApi.SkillMount;
 type TfyMcpServerMount = RuntimeMcpServerMount & TruefoundryGatewayApi.McpServer;
-interface TfyAgentSpec extends AgentSpec<TruefoundryGatewayApi.Model, TfySkillMount, TfyMcpServerMount> {
-    config?: TruefoundryGatewayApi.RuntimeConfig;
+interface TfyAgentSpec extends AgentSpec<TruefoundryGatewayApi.Model, TfySkillMount, TfyMcpServerMount, TruefoundryGatewayApi.RuntimeConfig> {
     responseFormat?: TruefoundryGatewayApi.ResponseFormat;
     messages?: TruefoundryGatewayApi.AgentSpecUserMessage[];
 }

package/dist/plugins/truefoundry-agent-server-adapter/index.js

@@ -7,7 +7,7 @@ import {
   isTfyMcpToolInfo,
   isTfySystemToolInfo,
   isTfyToolInfo
-} from "../../chunk-XYGEKRA2.js";
+} from "../../chunk-S4SH2KT2.js";
 export {
   createTrueFoundryAgentUIServer,
   createTrueFoundryChatServer,

package/dist/server/index.d.ts

@@ -1,5 +1,5 @@
-import { aC as TurnStreamingEvent, aJ as DeltaEvents, T as TurnEvent } from '../types-BE-F3vCD.js';
-export { aK as ActionRequiredEvent, f as AgentBuilderCapabilitiesResponse, g as AgentBuilderServer, a as AgentChatServer, aL as AgentInfo, h as AgentLibraryEntry, aM as AgentParent, i as AgentSelectorEntry, j as AgentSkill, A as AgentSpec, k as AgentUIServer, l as AgentUIServerPort, m as ApprovalDecision, n as AuthenticateConnectorRequest, C as CatalogServer, aN as ChunkDeltaToolCall, o as ConnectorAuth, p as ConnectorAuthApiKey, q as ConnectorAuthNone, r as ConnectorAuthOAuth, s as ConnectorAuthPublic, t as ConnectorAuthPublicApiKey, u as ConnectorAuthPublicNone, v as ConnectorAuthPublicOAuth, w as ConnectorAuthType, x as ConnectorAuthenticationResult, y as ConnectorBase, z as ConnectorCatalogEntry, B as ConnectorCatalogServer, D as ConnectorConfigBase, E as ConnectorSelectorEntry, F as ConnectorState, G as CreateConnectorRequest, H as CreateModelProviderRequest, I as CreateSandboxProviderRequest, J as CreateSandboxRequest, K as CreateSessionRequest, L as CreateSkillRequest, N as CreateSkillRequestBase, O as DefinedSkill, P as GithubSkill, Q as ImportGithubSkillRequest, R as ListResult, V as ListSessionsOrder, W as ListSessionsParams, M as McpAuthRequiredEvent, aO as McpInitializeEvent, aP as McpServerAuthInfo, X as McpServerMount, Y as Model, Z as ModelCatalogServer, _ as ModelEntry, aQ as ModelMessageContentPart, aR as ModelMessageDeltaEvent, $ as ModelMessageEvent, a0 as ModelParams, a1 as ModelProperties, a2 as ModelProviderBase, a3 as ModelProviderCatalogEntry, a4 as ModelProviderConfigBase, a5 as ModelSelection, a6 as ModelSelectorEntry, a7 as PageParams, a8 as PreviousTurnIdInput, a9 as ProviderEntry, aa as ProviderType, ab as RegistrySkill, ac as SandboxBase, ad as SandboxCatalogEntry, ae as SandboxCatalogServer, af as SandboxConfig, ag as SandboxCreatedEvent, ah as SandboxProviderBase, ai as SandboxProviderCatalogEntry, aj as SandboxProviderConfig, ak as SaveAgentRequest, al as SearchAgentSelectorParams, am as SearchAgentsParams, an as SelectRegistrySkillRequest, S as Session, ao as SessionEventItem, ap as SkillBase, aq as SkillCatalogEntry, ar as SkillCatalogServer, as as SkillConfigBase, at as SkillMount, au as SkillSelectorEntry, b as ThreadCreatedEvent, aS as ThreadDoneEvent, av as ToolApprovalRequiredEvent, aw as ToolBase, ax as ToolCall, aT as ToolCallFunction, aU as ToolCallRef, aV as ToolInfo, aW as ToolResponseEvent, ay as ToolResponseRequiredEvent, d as Turn, aX as TurnCreatedEvent, aY as TurnDoneEvent, e as TurnInputItem, az as TurnState, aZ as TurnStateCancelled, aA as TurnStateDone, a_ as TurnStateError, a$ as TurnStateRunning, aB as TurnStreamData, aD as UpdateConnectorRequest, aE as UpdateModelProviderRequest, aF as UpdateSandboxProviderRequest, aG as UpdateSandboxRequest, aH as UpdateSessionRequest, aI as UserMessage, b0 as UserMessageContent, c as UserToolApprovalEvent, U as UserToolResponseEvent } from '../types-BE-F3vCD.js';
+import { aF as TurnStreamingEvent, aM as DeltaEvents, T as TurnEvent } from '../types-VU3Fc17W.js';
+export { aN as ActionRequiredEvent, f as AgentBuilderCapabilitiesResponse, g as AgentBuilderServer, h as AgentCapabilityConfig, a as AgentChatServer, aO as AgentInfo, i as AgentLibraryEntry, aP as AgentParent, j as AgentRuntimeConfig, k as AgentSelectorEntry, l as AgentSkill, A as AgentSpec, m as AgentUIServer, n as AgentUIServerPort, o as ApprovalDecision, p as AuthenticateConnectorRequest, C as CatalogServer, aQ as ChunkDeltaToolCall, q as ConnectorAuth, r as ConnectorAuthApiKey, s as ConnectorAuthNone, t as ConnectorAuthOAuth, u as ConnectorAuthPublic, v as ConnectorAuthPublicApiKey, w as ConnectorAuthPublicNone, x as ConnectorAuthPublicOAuth, y as ConnectorAuthType, z as ConnectorAuthenticationResult, B as ConnectorBase, D as ConnectorCatalogEntry, E as ConnectorCatalogServer, F as ConnectorConfigBase, G as ConnectorSelectorEntry, H as ConnectorState, I as CreateConnectorRequest, J as CreateModelProviderRequest, K as CreateSandboxProviderRequest, L as CreateSandboxRequest, N as CreateSessionRequest, O as CreateSkillRequest, P as CreateSkillRequestBase, Q as DefinedSkill, R as GithubSkill, V as ImportGithubSkillRequest, W as ListResult, X as ListSessionsOrder, Y as ListSessionsParams, M as McpAuthRequiredEvent, aR as McpInitializeEvent, aS as McpServerAuthInfo, Z as McpServerMount, _ as Model, $ as ModelCatalogServer, a0 as ModelEntry, aT as ModelMessageContentPart, aU as ModelMessageDeltaEvent, a1 as ModelMessageEvent, a2 as ModelParams, a3 as ModelProperties, a4 as ModelProviderBase, a5 as ModelProviderCatalogEntry, a6 as ModelProviderConfigBase, a7 as ModelSelection, a8 as ModelSelectorEntry, a9 as PageParams, aa as PreviousTurnIdInput, ab as ProviderEntry, ac as ProviderType, ad as RegistrySkill, ae as SandboxBase, af as SandboxCatalogEntry, ag as SandboxCatalogServer, ah as SandboxConfig, ai as SandboxCreatedEvent, aj as SandboxProviderBase, ak as SandboxProviderCatalogEntry, al as SandboxProviderConfig, am as SaveAgentRequest, an as SaveAgentResult, ao as SearchAgentSelectorParams, ap as SearchAgentsParams, aq as SelectRegistrySkillRequest, S as Session, ar as SessionEventItem, as as SkillBase, at as SkillCatalogEntry, au as SkillCatalogServer, av as SkillConfigBase, aw as SkillMount, ax as SkillSelectorEntry, b as ThreadCreatedEvent, aV as ThreadDoneEvent, ay as ToolApprovalRequiredEvent, az as ToolBase, aA as ToolCall, aW as ToolCallFunction, aX as ToolCallRef, aY as ToolInfo, aZ as ToolResponseEvent, aB as ToolResponseRequiredEvent, d as Turn, a_ as TurnCreatedEvent, a$ as TurnDoneEvent, e as TurnInputItem, aC as TurnState, b0 as TurnStateCancelled, aD as TurnStateDone, b1 as TurnStateError, b2 as TurnStateRunning, aE as TurnStreamData, aG as UpdateConnectorRequest, aH as UpdateModelProviderRequest, aI as UpdateSandboxProviderRequest, aJ as UpdateSandboxRequest, aK as UpdateSessionRequest, aL as UserMessage, b3 as UserMessageContent, c as UserToolApprovalEvent, U as UserToolResponseEvent } from '../types-VU3Fc17W.js';
 
 /**
  * Local implementations of streaming delta helpers.

package/dist/{types-BE-F3vCD.d.ts → types-VU3Fc17W.d.ts}

@@ -294,15 +294,24 @@ interface Model {
     name: string;
     params?: ModelParams;
 }
+interface AgentCapabilityConfig {
+    enabled?: boolean;
+}
+interface AgentRuntimeConfig {
+    generativeUi?: AgentCapabilityConfig;
+    dynamicSubAgents?: AgentCapabilityConfig;
+    askUserQuestions?: AgentCapabilityConfig;
+}
 /**
 * SDK-owned agent definition — fields the FE reads/writes.
-* Host widens `model` / `skills` / `mcpServers` via type params, and adds
-* extra fields via `TSpec extends AgentSpec<...>`.
+* Host widens `model` / `skills` / `mcpServers` / `config` via type params,
+* and adds extra fields via `TSpec extends AgentSpec<...>`.
 */
-interface AgentSpec<TModel extends Model = Model, TSkill extends SkillMount = SkillMount, TMcp extends McpServerMount = McpServerMount> {
+interface AgentSpec<TModel extends Model = Model, TSkill extends SkillMount = SkillMount, TMcp extends McpServerMount = McpServerMount, TConfig extends AgentRuntimeConfig = AgentRuntimeConfig> {
     model: TModel;
     skills?: TSkill[];
     mcpServers?: TMcp[];
+    config?: TConfig;
     instructions?: string;
     variables?: Record<string, string>;
 }
@@ -475,6 +484,15 @@ interface AgentChatServer<TSpec extends AgentSpec = AgentSpec, TSession extends
 interface SaveAgentRequest<TSpec extends AgentSpec = AgentSpec> {
     agentName: string;
     agentSpec: TSpec;
+    intent: "create" | "update";
+    /** Current mutable session to update atomically with the named agent. */
+    sessionId?: string;
+}
+interface SaveAgentResult {
+    /** Immutable id allocated by the host registry. */
+    agentId?: string;
+    /** Timestamp returned when the active mutable session was updated. */
+    sessionUpdatedAt?: string;
 }
 /**
  * Builder feature flags — atoms gate sandbox / skill / settings surfaces on these.
@@ -497,7 +515,7 @@ interface AgentBuilderCapabilitiesResponse {
 * Builder catalog + persist port — atoms call these.
 * Passed separately from the runtime's chat server.
 */
-interface AgentBuilderServer<TSpec extends AgentSpec = AgentSpec, TModel extends ModelSelectorEntry = ModelSelectorEntry, TSkill extends SkillSelectorEntry = SkillSelectorEntry, TMcp extends ConnectorSelectorEntry = ConnectorSelectorEntry, TAgent extends AgentSelectorEntry = AgentSelectorEntry, TSave = unknown, TCapabilities extends AgentBuilderCapabilitiesResponse = AgentBuilderCapabilitiesResponse> {
+interface AgentBuilderServer<TSpec extends AgentSpec = AgentSpec, TModel extends ModelSelectorEntry = ModelSelectorEntry, TSkill extends SkillSelectorEntry = SkillSelectorEntry, TMcp extends ConnectorSelectorEntry = ConnectorSelectorEntry, TAgent extends AgentSelectorEntry = AgentSelectorEntry, TSave = SaveAgentResult, TCapabilities extends AgentBuilderCapabilitiesResponse = AgentBuilderCapabilitiesResponse> {
     getCapabilities(): Promise<TCapabilities>;
     getModels(): Promise<TModel[]>;
     getSkills(): Promise<TSkill[]>;
@@ -815,4 +833,4 @@ type AgentUIServerPort<TChat extends AgentChatServer = AgentChatServer, TBuilder
 /** Host-facing alias used by trueforge-ui. */
 type AgentUIServer = AgentUIServerPort;
 
-export type { ModelMessageEvent as $, AgentSpec as A, ConnectorCatalogServer as B, CatalogServer as C, ConnectorConfigBase as D, ConnectorSelectorEntry as E, ConnectorState as F, CreateConnectorRequest as G, CreateModelProviderRequest as H, CreateSandboxProviderRequest as I, CreateSandboxRequest as J, CreateSessionRequest as K, CreateSkillRequest as L, McpAuthRequiredEvent as M, CreateSkillRequestBase as N, DefinedSkill as O, GithubSkill as P, ImportGithubSkillRequest as Q, ListResult as R, Session as S, TurnEvent as T, UserToolResponseEvent as U, ListSessionsOrder as V, ListSessionsParams as W, McpServerMount as X, Model as Y, ModelCatalogServer as Z, ModelEntry as _, AgentChatServer as a, TurnStateRunning as a$, ModelParams as a0, ModelProperties as a1, ModelProviderBase as a2, ModelProviderCatalogEntry as a3, ModelProviderConfigBase as a4, ModelSelection as a5, ModelSelectorEntry as a6, PageParams as a7, PreviousTurnIdInput as a8, ProviderEntry as a9, TurnStateDone as aA, TurnStreamData as aB, TurnStreamingEvent as aC, UpdateConnectorRequest as aD, UpdateModelProviderRequest as aE, UpdateSandboxProviderRequest as aF, UpdateSandboxRequest as aG, UpdateSessionRequest as aH, UserMessage as aI, DeltaEvents as aJ, ActionRequiredEvent as aK, AgentInfo as aL, AgentParent as aM, ChunkDeltaToolCall as aN, McpInitializeEvent as aO, McpServerAuthInfo as aP, ModelMessageContentPart as aQ, ModelMessageDeltaEvent as aR, ThreadDoneEvent as aS, ToolCallFunction as aT, ToolCallRef as aU, ToolInfo as aV, ToolResponseEvent as aW, TurnCreatedEvent as aX, TurnDoneEvent as aY, TurnStateCancelled as aZ, TurnStateError as a_, ProviderType as aa, RegistrySkill as ab, SandboxBase as ac, SandboxCatalogEntry as ad, SandboxCatalogServer as ae, SandboxConfig as af, SandboxCreatedEvent as ag, SandboxProviderBase as ah, SandboxProviderCatalogEntry as ai, SandboxProviderConfig as aj, SaveAgentRequest as ak, SearchAgentSelectorParams as al, SearchAgentsParams as am, SelectRegistrySkillRequest as an, SessionEventItem as ao, SkillBase as ap, SkillCatalogEntry as aq, SkillCatalogServer as ar, SkillConfigBase as as, SkillMount as at, SkillSelectorEntry as au, ToolApprovalRequiredEvent as av, ToolBase as aw, ToolCall as ax, ToolResponseRequiredEvent as ay, TurnState as az, ThreadCreatedEvent as b, UserMessageContent as b0, UserToolApprovalEvent as c, Turn as d, TurnInputItem as e, AgentBuilderCapabilitiesResponse as f, AgentBuilderServer as g, AgentLibraryEntry as h, AgentSelectorEntry as i, AgentSkill as j, AgentUIServer as k, AgentUIServerPort as l, ApprovalDecision as m, AuthenticateConnectorRequest as n, ConnectorAuth as o, ConnectorAuthApiKey as p, ConnectorAuthNone as q, ConnectorAuthOAuth as r, ConnectorAuthPublic as s, ConnectorAuthPublicApiKey as t, ConnectorAuthPublicNone as u, ConnectorAuthPublicOAuth as v, ConnectorAuthType as w, ConnectorAuthenticationResult as x, ConnectorBase as y, ConnectorCatalogEntry as z };
+export type { ModelCatalogServer as $, AgentSpec as A, ConnectorBase as B, CatalogServer as C, ConnectorCatalogEntry as D, ConnectorCatalogServer as E, ConnectorConfigBase as F, ConnectorSelectorEntry as G, ConnectorState as H, CreateConnectorRequest as I, CreateModelProviderRequest as J, CreateSandboxProviderRequest as K, CreateSandboxRequest as L, McpAuthRequiredEvent as M, CreateSessionRequest as N, CreateSkillRequest as O, CreateSkillRequestBase as P, DefinedSkill as Q, GithubSkill as R, Session as S, TurnEvent as T, UserToolResponseEvent as U, ImportGithubSkillRequest as V, ListResult as W, ListSessionsOrder as X, ListSessionsParams as Y, McpServerMount as Z, Model as _, AgentChatServer as a, TurnDoneEvent as a$, ModelEntry as a0, ModelMessageEvent as a1, ModelParams as a2, ModelProperties as a3, ModelProviderBase as a4, ModelProviderCatalogEntry as a5, ModelProviderConfigBase as a6, ModelSelection as a7, ModelSelectorEntry as a8, PageParams as a9, ToolCall as aA, ToolResponseRequiredEvent as aB, TurnState as aC, TurnStateDone as aD, TurnStreamData as aE, TurnStreamingEvent as aF, UpdateConnectorRequest as aG, UpdateModelProviderRequest as aH, UpdateSandboxProviderRequest as aI, UpdateSandboxRequest as aJ, UpdateSessionRequest as aK, UserMessage as aL, DeltaEvents as aM, ActionRequiredEvent as aN, AgentInfo as aO, AgentParent as aP, ChunkDeltaToolCall as aQ, McpInitializeEvent as aR, McpServerAuthInfo as aS, ModelMessageContentPart as aT, ModelMessageDeltaEvent as aU, ThreadDoneEvent as aV, ToolCallFunction as aW, ToolCallRef as aX, ToolInfo as aY, ToolResponseEvent as aZ, TurnCreatedEvent as a_, PreviousTurnIdInput as aa, ProviderEntry as ab, ProviderType as ac, RegistrySkill as ad, SandboxBase as ae, SandboxCatalogEntry as af, SandboxCatalogServer as ag, SandboxConfig as ah, SandboxCreatedEvent as ai, SandboxProviderBase as aj, SandboxProviderCatalogEntry as ak, SandboxProviderConfig as al, SaveAgentRequest as am, SaveAgentResult as an, SearchAgentSelectorParams as ao, SearchAgentsParams as ap, SelectRegistrySkillRequest as aq, SessionEventItem as ar, SkillBase as as, SkillCatalogEntry as at, SkillCatalogServer as au, SkillConfigBase as av, SkillMount as aw, SkillSelectorEntry as ax, ToolApprovalRequiredEvent as ay, ToolBase as az, ThreadCreatedEvent as b, TurnStateCancelled as b0, TurnStateError as b1, TurnStateRunning as b2, UserMessageContent as b3, UserToolApprovalEvent as c, Turn as d, TurnInputItem as e, AgentBuilderCapabilitiesResponse as f, AgentBuilderServer as g, AgentCapabilityConfig as h, AgentLibraryEntry as i, AgentRuntimeConfig as j, AgentSelectorEntry as k, AgentSkill as l, AgentUIServer as m, AgentUIServerPort as n, ApprovalDecision as o, AuthenticateConnectorRequest as p, ConnectorAuth as q, ConnectorAuthApiKey as r, ConnectorAuthNone as s, ConnectorAuthOAuth as t, ConnectorAuthPublic as u, ConnectorAuthPublicApiKey as v, ConnectorAuthPublicNone as w, ConnectorAuthPublicOAuth as x, ConnectorAuthType as y, ConnectorAuthenticationResult as z };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@truefoundry/assistant-ui-runtime",
-  "version": "0.1.13",
+  "version": "0.1.14",
   "description": "TrueFoundry Gateway agent runtime adapter for assistant-ui",
   "license": "Apache-2.0",
   "repository": {

package/src/draft/useDraftAgentSpec.test.tsx

@@ -150,4 +150,111 @@ describe("useDraftAgentSpec", () => {
         expect(syncAgentSpec).toHaveBeenCalledOnce();
         expect(result.current.isSpecSyncing).toBe(false);
     });
+
+    it("flushes pending work before a coordinated save", async () => {
+        const syncAgentSpec = vi
+            .fn()
+            .mockResolvedValue("2026-06-30T15:00:00.000Z");
+        const draftBridge: DraftSessionBridge = {
+            getDraftAgentSpec: vi.fn().mockResolvedValue(defaultAgentSpec),
+            syncAgentSpec,
+        };
+        const { result } = renderHook(() =>
+            useDraftAgentSpec({
+                draftSessionId: "draft-1",
+                draftBridge,
+                defaultAgentSpec,
+            }),
+        );
+        await flushMicrotasks();
+
+        act(() => {
+            result.current.updateAgentSpec({ instructions: "latest draft" });
+        });
+        await act(async () => {
+            await result.current.flushAgentSpec();
+        });
+
+        expect(syncAgentSpec).toHaveBeenCalledWith("draft-1", {
+            model: defaultAgentSpec.model,
+            instructions: "latest draft",
+        });
+    });
+
+    it("adopts an atomically persisted spec without scheduling another sync", async () => {
+        const syncAgentSpec = vi.fn().mockResolvedValue("unused");
+        const draftBridge: DraftSessionBridge = {
+            getDraftAgentSpec: vi.fn().mockResolvedValue(defaultAgentSpec),
+            syncAgentSpec,
+        };
+        const { result } = renderHook(() =>
+            useDraftAgentSpec({
+                draftSessionId: "draft-1",
+                draftBridge,
+                defaultAgentSpec,
+            }),
+        );
+        await flushMicrotasks();
+
+        act(() => {
+            result.current.adoptAgentSpec({
+                agentSpec: {
+                    model: { name: "openai/gpt-5" },
+                    config: { generativeUi: { enabled: false } },
+                },
+                updatedAt: "2026-06-30T16:00:00.000Z",
+            });
+        });
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(400);
+        });
+
+        expect(result.current.agentSpec).toEqual({
+            model: { name: "openai/gpt-5" },
+            config: { generativeUi: { enabled: false } },
+        });
+        expect(syncAgentSpec).not.toHaveBeenCalled();
+        await expect(result.current.takeTurnHeaderTimestamp()).resolves.toBe(
+            "2026-06-30T16:00:00.000Z",
+        );
+    });
+
+    it("preserves the last sync timestamp when adopt omits updatedAt", async () => {
+        const syncAgentSpec = vi
+            .fn()
+            .mockResolvedValue("2026-06-30T17:00:00.000Z");
+        const draftBridge: DraftSessionBridge = {
+            getDraftAgentSpec: vi.fn().mockResolvedValue(defaultAgentSpec),
+            syncAgentSpec,
+        };
+        const { result } = renderHook(() =>
+            useDraftAgentSpec({
+                draftSessionId: "draft-1",
+                draftBridge,
+                defaultAgentSpec,
+            }),
+        );
+        await flushMicrotasks();
+
+        act(() => {
+            result.current.updateAgentSpec({ instructions: "synced" });
+        });
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(400);
+        });
+        await flushMicrotasks();
+        expect(syncAgentSpec).toHaveBeenCalledOnce();
+
+        act(() => {
+            result.current.adoptAgentSpec({
+                agentSpec: {
+                    model: { name: "openai/gpt-5" },
+                },
+            });
+        });
+
+        await expect(result.current.takeTurnHeaderTimestamp()).resolves.toBe(
+            "2026-06-30T17:00:00.000Z",
+        );
+    });
 });

package/src/draft/useDraftAgentSpec.ts

@@ -23,6 +23,11 @@ export type UseDraftAgentSpecResult = {
     isSpecSyncing: boolean;
     specError: unknown | null;
     updateAgentSpec: (update: AgentSpecUpdate) => void;
+    flushAgentSpec: () => Promise<void>;
+    adoptAgentSpec: (request: {
+        agentSpec: AgentSpec;
+        updatedAt?: string;
+    }) => void;
     takeTurnHeaderTimestamp: () => Promise<string | undefined>;
 };
 
@@ -206,6 +211,35 @@ export function useDraftAgentSpec({
         }
     }, []);
 
+    const adoptAgentSpec = useCallback(
+        ({
+            agentSpec: persistedSpec,
+            updatedAt,
+        }: {
+            agentSpec: AgentSpec;
+            updatedAt?: string;
+        }) => {
+            if (syncTimeoutRef.current != null) {
+                clearTimeout(syncTimeoutRef.current);
+                syncTimeoutRef.current = undefined;
+            }
+            syncGenerationRef.current++;
+            pendingFlushRef.current = undefined;
+            inFlightFlushRef.current = undefined;
+            localDirtyRef.current = false;
+            // Keep the last successful sync timestamp when the caller omits
+            // updatedAt (e.g. a save that did not return sessionUpdatedAt).
+            if (updatedAt !== undefined) {
+                lastUpdatedAtRef.current = updatedAt;
+            }
+            agentSpecRef.current = persistedSpec;
+            setAgentSpec(persistedSpec);
+            setSpecError(null);
+            setIsSpecSyncing(false);
+        },
+        [],
+    );
+
     const takeTurnHeaderTimestamp = useCallback(async () => {
         await flushPendingSpecSyncNow();
         const updatedAt = lastUpdatedAtRef.current;
@@ -245,6 +279,8 @@ export function useDraftAgentSpec({
             isSpecSyncing: enabled ? isSpecSyncing : false,
             specError: enabled ? specError : null,
             updateAgentSpec,
+            flushAgentSpec: flushPendingSpecSyncNow,
+            adoptAgentSpec,
             takeTurnHeaderTimestamp,
         }),
         [
@@ -254,6 +290,8 @@ export function useDraftAgentSpec({
             isSpecLoading,
             isSpecSyncing,
             specError,
+            flushPendingSpecSyncNow,
+            adoptAgentSpec,
             takeTurnHeaderTimestamp,
             updateAgentSpec,
         ],

package/src/hooks.ts

@@ -142,6 +142,13 @@ export const useTrueFoundryHistoryPagination = () => {
     );
 };
 
+/**
+ * True while a turn runs that this server cannot stream, so its result will not
+ * arrive in this client. Falls back to `false` on runtimes that predate the flag.
+ */
+export const useTrueFoundryResumeUnavailable = () =>
+    trueFoundryExtras.use((e) => e.resumeUnavailable, false) ?? false;
+
 /** Returns a function to reset (re-submit) a user turn from any render context. */
 export const useTrueFoundryResetFromTurn = () => {
     const aui = useAui();
@@ -165,3 +172,18 @@ export const useTrueFoundryUpdateAgentSpec = () => {
         trueFoundryExtras.get(aui).draft?.updateAgentSpec(update);
 };
 
+/** Flushes any pending draft-spec synchronization before a coordinated write. */
+export const useTrueFoundryFlushAgentSpec = () => {
+    const aui = useAui();
+    return () =>
+        trueFoundryExtras.get(aui).draft?.flushAgentSpec() ?? Promise.resolve();
+};
+
+/** Adopts a spec already persisted by another server operation without syncing again. */
+export const useTrueFoundryAdoptAgentSpec = () => {
+    const aui = useAui();
+    return (
+        request: Parameters<TrueFoundryDraftRuntimeExtras["adoptAgentSpec"]>[0],
+    ) => trueFoundryExtras.get(aui).draft?.adoptAgentSpec(request);
+};
+

package/src/index.ts

@@ -25,7 +25,9 @@ export { createTrueFoundryOwnedSessionsThreadListAdapter } from "./truefoundryOw
 export { createDraftSessionBridge } from "./draft/draftSessionBridge.js";
 export type { DraftSessionBridge } from "./draft/draftSessionBridge.js";
 export {
+    useTrueFoundryAdoptAgentSpec,
     useTrueFoundryAgentSpec,
+    useTrueFoundryFlushAgentSpec,
     useTrueFoundryUpdateAgentSpec,
 } from "./hooks.js";
 export type { TrueFoundryDraftRuntimeExtras } from "./truefoundryExtras.js";
@@ -55,6 +57,7 @@ export {
     useTrueFoundryResetFromTurn,
     useTrueFoundryReload,
     useTrueFoundryHistoryPagination,
+    useTrueFoundryResumeUnavailable,
 } from "./hooks.js";
 export {
     collectApprovalInputs,
@@ -79,7 +82,10 @@ export type {
     AgentChatServer,
     AgentBuilderCapabilitiesResponse,
     AgentBuilderServer,
+    AgentCapabilityConfig,
+    AgentRuntimeConfig,
     SaveAgentRequest,
+    SaveAgentResult,
     Session,
     Turn,
     TurnState,

package/src/plugins/truefoundry-agent-server-adapter/cp.ts

@@ -9,6 +9,7 @@ import type {
     AgentSelectorEntry,
     ConnectorSelectorEntry,
     ModelSelectorEntry,
+    SaveAgentRequest,
     SearchAgentSelectorParams,
     SkillSelectorEntry,
 } from "../../server/types.js";
@@ -595,7 +596,7 @@ export function buildSaveAgentManifest(
  */
 export async function saveAgent(
     opts: CpCredentials,
-    req: { agentName: string; agentSpec: TfyAgentSpec },
+    req: SaveAgentRequest<TfyAgentSpec>,
 ): Promise<unknown> {
     const manifest = buildSaveAgentManifest(req.agentName, req.agentSpec);
     return cpFetch(opts, "/api/svc/v1/agents", {

package/src/plugins/truefoundry-agent-server-adapter/types.ts

@@ -67,9 +67,9 @@ export interface TfyAgentSpec
     extends AgentSpec<
         TruefoundryGatewayApi.Model,
         TfySkillMount,
-        TfyMcpServerMount
+        TfyMcpServerMount,
+        TruefoundryGatewayApi.RuntimeConfig
     > {
-    config?: TruefoundryGatewayApi.RuntimeConfig;
     responseFormat?: TruefoundryGatewayApi.ResponseFormat;
     messages?: TruefoundryGatewayApi.AgentSpecUserMessage[];
 }

package/src/server/index.ts

@@ -15,8 +15,11 @@ export type {
     McpServerMount,
     ModelParams,
     Model,
+    AgentCapabilityConfig,
+    AgentRuntimeConfig,
     AgentSpec,
     SaveAgentRequest,
+    SaveAgentResult,
     Session,
     CreateSessionRequest,
     UpdateSessionRequest,

package/src/server/types.ts

@@ -105,19 +105,31 @@ export interface Model {
   params?: ModelParams;
 }
 
+export interface AgentCapabilityConfig {
+  enabled?: boolean;
+}
+
+export interface AgentRuntimeConfig {
+  generativeUi?: AgentCapabilityConfig;
+  dynamicSubAgents?: AgentCapabilityConfig;
+  askUserQuestions?: AgentCapabilityConfig;
+}
+
 /**
 * SDK-owned agent definition — fields the FE reads/writes.
-* Host widens `model` / `skills` / `mcpServers` via type params, and adds
-* extra fields via `TSpec extends AgentSpec<...>`.
+* Host widens `model` / `skills` / `mcpServers` / `config` via type params,
+* and adds extra fields via `TSpec extends AgentSpec<...>`.
 */
 export interface AgentSpec<
   TModel extends Model = Model,
   TSkill extends SkillMount = SkillMount,
   TMcp extends McpServerMount = McpServerMount,
+  TConfig extends AgentRuntimeConfig = AgentRuntimeConfig,
 > {
   model: TModel;
   skills?: TSkill[];
   mcpServers?: TMcp[];
+  config?: TConfig;
   instructions?: string;
   variables?: Record<string, string>;
 }
@@ -329,6 +341,16 @@ export interface AgentChatServer<
 export interface SaveAgentRequest<TSpec extends AgentSpec = AgentSpec> {
   agentName: string;
   agentSpec: TSpec;
+  intent: "create" | "update";
+  /** Current mutable session to update atomically with the named agent. */
+  sessionId?: string;
+}
+
+export interface SaveAgentResult {
+  /** Immutable id allocated by the host registry. */
+  agentId?: string;
+  /** Timestamp returned when the active mutable session was updated. */
+  sessionUpdatedAt?: string;
 }
 
 /**
@@ -352,7 +374,7 @@ export interface AgentBuilderServer<
   TSkill extends SkillSelectorEntry = SkillSelectorEntry,
   TMcp extends ConnectorSelectorEntry = ConnectorSelectorEntry,
   TAgent extends AgentSelectorEntry = AgentSelectorEntry,
-  TSave = unknown,
+  TSave = SaveAgentResult,
   TCapabilities extends AgentBuilderCapabilitiesResponse = AgentBuilderCapabilitiesResponse,
 > {
   getCapabilities(): Promise<TCapabilities>;

package/src/streamTurn.test.ts

@@ -357,6 +357,24 @@ describe("streamTurn", () => {
             expect(updates).toEqual([{ content: [{ type: "text", text: "resumed" }] }]);
         });
 
+        it("yields nothing when the server omits subscribeToTurn", async () => {
+            const cancelSession = vi.fn().mockResolvedValue(undefined);
+            const server = mockServer({ cancelSession });
+
+            const updates = await collectUpdates(
+                resumeTurnStream(
+                    server,
+                    SESSION_ID,
+                    "turn-1",
+                    new PeerThreadFoldState(),
+                    new AbortController().signal,
+                ),
+            );
+
+            expect(updates).toEqual([]);
+            expect(cancelSession).not.toHaveBeenCalled();
+        });
+
         it("returns early when aborted before streaming starts", async () => {
             const subscribeToTurn = vi.fn(async function* () {});
             const cancelSession = vi.fn().mockResolvedValue(undefined);

package/src/streamTurn.ts

@@ -116,10 +116,10 @@ export async function* resumeTurnStream(
     afterSequenceNumber?: number,
     groupRootBaseline?: readonly string[],
 ): AsyncGenerator<TurnStreamUpdate> {
+    // Optional on custom backends. Callers detect the gap and report it, so an
+    // empty stream here is safer than throwing mid-render.
     if (server.subscribeToTurn == null) {
-        throw new Error(
-            "resumeTurnStream requires AgentChatServer.subscribeToTurn",
-        );
+        return;
     }
 
     const onAbort = () => {

package/src/truefoundryExtras.ts

@@ -16,12 +16,18 @@ export type TrueFoundryDraftRuntimeExtras = {
     isSpecSyncing: boolean;
     specError: unknown | null;
     updateAgentSpec: (update: AgentSpecUpdate) => void;
+    flushAgentSpec: () => Promise<void>;
+    adoptAgentSpec: (request: {
+        agentSpec: AgentSpec;
+        updatedAt?: string;
+    }) => void;
 };
 
 export type TrueFoundryRuntimeExtras = {
     pendingApprovals: PendingApproval[];
     pendingToolResponses: PendingToolResponse[];
     pendingMcpAuth: { mcpServers: McpAuthRequiredEvent["mcpServers"] } | null;
+    resumeUnavailable: boolean;
     sandboxId: string | undefined;
     respondToToolApproval: (response: RespondToToolApprovalOptions) => void;
     respondToToolResponse: (response: RespondToToolResponseOptions) => void;
@@ -49,4 +55,10 @@ export const EMPTY_DRAFT_EXTRAS: TrueFoundryDraftRuntimeExtras = {
     updateAgentSpec: () => {
         throw new Error("Draft agent extras are only available in draft mode.");
     },
+    flushAgentSpec: async () => {
+        throw new Error("Draft agent extras are only available in draft mode.");
+    },
+    adoptAgentSpec: () => {
+        throw new Error("Draft agent extras are only available in draft mode.");
+    },
 };