@truefoundry/assistant-ui-runtime 0.1.10 → 0.1.12

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

@@ -1,13 +1,13 @@
 import { AgentSessionClient } from 'truefoundry-gateway-sdk/agents';
 import { PrivateAgentSessionClient } from 'truefoundry-gateway-sdk/agents/private';
-import { A as AgentSpec, x as CreateSessionRequest, B as ListSessionsParams, S as Session, d as Turn, a7 as TurnState, a as AgentChatServer, ae as UpdateSessionRequest, ag as AgentSelectorEntry, ah as ConnectorSelectorEntry, ai as ModelSelectorEntry, aj as SkillSelectorEntry, f as AgentBuilderServer } from '../../types-Bb8Kaf1h.js';
+import { A as AgentSpec, K as CreateSessionRequest, W as ListSessionsParams, S as Session, d as Turn, ax as TurnState, a as AgentChatServer, aF as UpdateSessionRequest, i as AgentSelectorEntry, E as ConnectorSelectorEntry, a5 as ModelSelectorEntry, as as SkillSelectorEntry, g as AgentBuilderServer } from '../../types-CSRQSnHu.js';
 import { TruefoundryGatewayApi } from 'truefoundry-gateway-sdk';
 
 /**
  * TrueFoundry-specific type extensions over the runtime's generic bases.
  *
- * The runtime defines minimal bases (SkillMount = {id,name}, McpServerMount =
- * {id,name}, AgentSpec.config = unknown, etc.) that hosts extend via generics.
+ * The runtime defines opaque mount bases (`SkillMount` / `McpServerMount` =
+ * `object`) that hosts intersect with gateway shapes via generics.
  *
  * This file builds the concrete TrueFoundry types by:
  *  - Importing the runtime's AgentSpec as the base to extend.

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

@@ -7,7 +7,7 @@ import {
   isTfyMcpToolInfo,
   isTfySystemToolInfo,
   isTfyToolInfo
-} from "../../chunk-6ZOGBD7E.js";
+} from "../../chunk-2NLIAJJX.js";
 export {
   createTrueFoundryAgentUIServer,
   createTrueFoundryChatServer,

package/dist/server/index.d.ts

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

package/dist/{types-Bb8Kaf1h.d.ts → types-CSRQSnHu.d.ts}

@@ -232,6 +232,8 @@ interface ModelSelectorEntry {
     name: string;
     provider: string;
     reasoningEfforts?: string[];
+    /** Provider logo URL from the model catalog; selector falls back to a monogram when omitted. */
+    providerLogo?: string;
 }
 /** Skill selector row. Host extends for fqn, preload, etc. */
 interface SkillSelectorEntry {
@@ -244,6 +246,10 @@ interface ConnectorSelectorEntry {
     id: string;
     name: string;
     description?: string;
+    /** When true, the connector must be authenticated before use. Omitted when the host does not report auth. */
+    requiresAuth?: boolean;
+    /** When true, the connector is already authenticated. Omitted when the host does not report auth. */
+    authenticated?: boolean;
 }
 /** Agent selector row. Host extends for metadata; `agentSpec` enables Edit. */
 interface AgentSelectorEntry {
@@ -259,19 +265,19 @@ type SearchAgentSelectorParams = {
     offset?: number;
 };
 /**
- * Mounts written to AgentSpec.skills[] / AgentSpec.mcpServers[].
- *
- * These are opaque to the runtime — it stores and forwards them but never reads
- * a field, and the backend owns the shape (the gateway identifies a skill by
- * `fqn`, with no `id` or `name` anywhere). So the base constrains only that a
- * mount is an object; hosts intersect their concrete mount type over it, as
- * `TfySkillMount` / `TfyMcpServerMount` do in the gateway adapter.
- *
- * Naming a field here would not just be unread, it would be wrong: a base with
- * required fields rejects the backend's own payloads, and one with only optional
- * fields is a weak type, which TypeScript rejects for a source that shares no
- * property with it — the gateway's registry skill shares none.
- */
+* Mounts written to AgentSpec.skills[] / AgentSpec.mcpServers[].
+*
+* These are opaque to the runtime — it stores and forwards them but never reads
+* a field, and the backend owns the shape (the gateway identifies a skill by
+* `fqn`, with no `id` or `name` anywhere). So the base constrains only that a
+* mount is an object; hosts intersect their concrete mount type over it, as
+* `TfySkillMount` / `TfyMcpServerMount` do in the gateway adapter.
+*
+* Naming a field here would not just be unread, it would be wrong: a base with
+* required fields rejects the backend's own payloads, and one with only optional
+* fields is a weak type, which TypeScript rejects for a source that shares no
+* property with it — the gateway's registry skill shares none.
+*/
 type SkillMount = object;
 type McpServerMount = object;
 interface ModelParams {
@@ -283,10 +289,10 @@ interface Model {
     params?: ModelParams;
 }
 /**
- * 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<...>`.
- */
+* 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<...>`.
+*/
 interface AgentSpec<TModel extends Model = Model, TSkill extends SkillMount = SkillMount, TMcp extends McpServerMount = McpServerMount> {
     model: TModel;
     skills?: TSkill[];
@@ -330,7 +336,7 @@ interface ListSessionsParams extends PageParams {
     /** Host-specific filter (e.g. TFY startTimestamp). */
     startTimestamp?: string;
 }
-type PreviousTurnIdInput = "auto" | string;
+type PreviousTurnIdInput = "auto" | "none" | string;
 type UserMessageContent = string | Array<{
     type: "text";
     text: string;
@@ -392,11 +398,11 @@ interface Turn {
     createdAt: string;
 }
 /**
- * Chat / session port — the runtime calls these.
- *
- * All session ops are flat (sessionId param). No gateway client dependency.
- * `createTrueFoundryServer` is one possible implementation (TFY adapter).
- */
+* Chat / session port — the runtime calls these.
+*
+* All session ops are flat (sessionId param). No gateway client dependency.
+* `createTrueFoundryServer` is one possible implementation (TFY adapter).
+*/
 interface AgentChatServer<TSpec extends AgentSpec = AgentSpec, TSession extends Session<TSpec> = Session<TSpec>, TCreate extends CreateSessionRequest<TSpec> = CreateSessionRequest<TSpec>, TList extends ListSessionsParams = ListSessionsParams, TUpdate extends UpdateSessionRequest<TSpec> = UpdateSessionRequest<TSpec>, TTurn extends Turn = Turn> {
     createSession(req: TCreate): Promise<TSession>;
     listSessions(req?: TList): Promise<ListResult<TSession>>;
@@ -459,43 +465,63 @@ interface AgentChatServer<TSpec extends AgentSpec = AgentSpec, TSession extends
         path: string;
     }): Promise<Blob>;
 }
+/** Request body for `AgentBuilderServer.saveAgent`. */
+interface SaveAgentRequest<TSpec extends AgentSpec = AgentSpec> {
+    agentName: string;
+    agentSpec: TSpec;
+}
 /**
- * Builder catalog + persist port — atoms call these.
- * Passed separately from the runtime's chat server.
+ * Builder feature flags — atoms gate sandbox / skill / settings surfaces on these.
  */
-interface AgentBuilderServer<TSpec extends AgentSpec = AgentSpec, TModel extends ModelSelectorEntry = ModelSelectorEntry, TSkill extends SkillSelectorEntry = SkillSelectorEntry, TMcp extends ConnectorSelectorEntry = ConnectorSelectorEntry, TAgent extends AgentSelectorEntry = AgentSelectorEntry, TSave = unknown> {
+interface AgentBuilderCapabilitiesResponse {
+    data: {
+        sandbox: {
+            enabled: boolean;
+        };
+        skill: {
+            enabled: boolean;
+            reason?: string;
+        };
+        settings?: {
+            enabled: boolean;
+        };
+    };
+}
+/**
+* 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> {
+    getCapabilities(): Promise<TCapabilities>;
     getModels(): Promise<TModel[]>;
     getSkills(): Promise<TSkill[]>;
     getMcp(): Promise<TMcp[]>;
     searchAgents(req?: SearchAgentSelectorParams): Promise<TAgent[]>;
-    saveAgent(req: {
-        agentName: string;
-        agentSpec: TSpec;
-    }): Promise<TSave>;
+    saveAgent(req: SaveAgentRequest<TSpec>): Promise<TSave>;
     deleteAgent?(req: {
         agentName: string;
     }): Promise<void>;
 }
 /**
- * Provider type id. Reserved literal: `"custom"` for user-defined providers;
- * any other string is a builtin (e.g. `"openai"`, `"anthropic"`).
- *
- * Note: `string | "custom"` is useless in TypeScript (`"custom"` ⊆ `string`),
- * so this stays `string` and `"custom"` is a documented convention.
- */
+* Provider type id. Reserved literal: `"custom"` for user-defined providers;
+* any other string is a builtin (e.g. `"openai"`, `"anthropic"`).
+*
+* Note: `string | "custom"` is useless in TypeScript (`"custom"` ⊆ `string`),
+* so this stays `string` and `"custom"` is a documented convention.
+*/
 type ProviderType = string;
 /**
- * Model row — form "Model ID" + "Display name".
- * Host extends for properties, etc.
- */
+* Model row — form "Model ID" + "Display name".
+* Host extends for properties, etc.
+*/
 interface ModelEntry {
     id: string;
     name: string;
 }
 /**
- * Write config for create/update (custom form + catalog "Save key").
- * Host extends. `baseUrl` present iff `type === "custom"`.
- */
+* Write config for create/update (custom form + catalog "Save key").
+* Host extends. `baseUrl` present iff `type === "custom"`.
+*/
 interface ModelProviderConfigBase<TModel extends ModelEntry = ModelEntry> {
     type: ProviderType;
     name: string;
@@ -505,9 +531,9 @@ interface ModelProviderConfigBase<TModel extends ModelEntry = ModelEntry> {
     models: TModel[];
 }
 /**
- * Configured provider card (list/read). No raw `apiKey`.
- * Host extends for apiKeySet, timestamps, etc.
- */
+* Configured provider card (list/read). No raw `apiKey`.
+* Host extends for apiKeySet, timestamps, etc.
+*/
 interface ModelProviderBase<TModel extends ModelEntry = ModelEntry> {
     id: string;
     type: ProviderType;
@@ -517,14 +543,16 @@ interface ModelProviderBase<TModel extends ModelEntry = ModelEntry> {
     models: TModel[];
 }
 /**
- * Discovery-only catalog provider (AVAILABLE list).
- * `type` must not be `"custom"` — custom providers use the custom form.
- * Host extends for richer model rows.
- */
+* Discovery-only catalog provider (AVAILABLE list).
+* `type` must not be `"custom"` — custom providers use the custom form.
+* Host extends for richer model rows.
+*/
 interface ModelProviderCatalogEntry<TModel extends ModelEntry = ModelEntry> {
     type: ProviderType;
     name: string;
     models: TModel[];
+    supportedReasoningEfforts?: string[];
+    logo?: string;
 }
 /** Create — no `id`; server assigns it. Catalog path = entry + apiKey. */
 type CreateModelProviderRequest<TModel extends ModelEntry = ModelEntry> = ModelProviderConfigBase<TModel>;
@@ -565,7 +593,7 @@ type ConnectorAuthNone = {
 type ConnectorAuth = ConnectorAuthOAuth | ConnectorAuthApiKey | ConnectorAuthNone;
 type ConnectorAuthPublicOAuth = {
     type: "dcr";
-    authUrl: string;
+    authUrl?: string;
 };
 type ConnectorAuthPublicApiKey = {
     type: "header";
@@ -576,18 +604,19 @@ type ConnectorAuthPublicNone = {
 };
 type ConnectorAuthPublic = ConnectorAuthPublicOAuth | ConnectorAuthPublicApiKey | ConnectorAuthPublicNone;
 /**
- * MCP / connector create-edit config. Host extends for extra fields, etc.
- */
+* MCP / connector create-edit config. Host extends for extra fields, etc.
+*/
 interface ConnectorConfigBase<TAuth extends ConnectorAuth = ConnectorAuth> {
     name: string;
     url: string;
     auth: TAuth;
 }
 /**
- * Connected connector row (settings/connectors). No raw `apiKey`.
+ * Connected connector row (settings/connectors). No raw `apiKey` or tools.
+ * Tools are fetched separately with `getToolsByConnectorId`.
  * Host extends.
  */
-interface ConnectorBase<TTool extends ToolBase = ToolBase, TAuth extends ConnectorAuthPublic = ConnectorAuthPublic> {
+interface ConnectorBase<TAuth extends ConnectorAuthPublic = ConnectorAuthPublic> {
     id: string;
     name: string;
     description: string;
@@ -596,7 +625,6 @@ interface ConnectorBase<TTool extends ToolBase = ToolBase, TAuth extends Connect
     /** When true, UI should not show Disconnect. */
     requiresAuth: boolean;
     authenticated: boolean;
-    tools: TTool[];
 }
 /** Discovery catalog entry for "+ Add MCP server". Host extends. */
 interface ConnectorCatalogEntry<TAuth extends ConnectorAuthPublic = ConnectorAuthPublic> {
@@ -605,6 +633,7 @@ interface ConnectorCatalogEntry<TAuth extends ConnectorAuthPublic = ConnectorAut
     description?: string;
     url: string;
     auth: TAuth;
+    logo?: string;
 }
 /** Create connector — no `id`; server assigns it. Host extends. */
 type CreateConnectorRequest<TAuth extends ConnectorAuth = ConnectorAuth> = ConnectorConfigBase<TAuth>;
@@ -612,21 +641,36 @@ type CreateConnectorRequest<TAuth extends ConnectorAuth = ConnectorAuth> = Conne
 type UpdateConnectorRequest<TAuth extends ConnectorAuth = ConnectorAuth> = ConnectorConfigBase<TAuth> & {
     id: string;
 };
-interface ConnectorCatalogServer<TTool extends ToolBase = ToolBase, TAuthWrite extends ConnectorAuth = ConnectorAuth, TAuthPublic extends ConnectorAuthPublic = ConnectorAuthPublic, TConnector extends ConnectorBase<TTool, TAuthPublic> = ConnectorBase<TTool, TAuthPublic>, TCatalogEntry extends ConnectorCatalogEntry<TAuthPublic> = ConnectorCatalogEntry<TAuthPublic>, TCreate extends CreateConnectorRequest<TAuthWrite> = CreateConnectorRequest<TAuthWrite>, TUpdate extends UpdateConnectorRequest<TAuthWrite> = UpdateConnectorRequest<TAuthWrite>> {
+interface AuthenticateConnectorRequest {
+    id: string;
+    /** OAuth callback page owned by the host application. */
+    redirectURL?: string;
+}
+interface ConnectorAuthenticationResult<TConnector extends ConnectorBase = ConnectorBase> {
+    connector?: TConnector;
+    status?: string;
+    authorization_endpoint?: string | undefined;
+}
+interface ConnectorCatalogServer<TTool extends ToolBase = ToolBase, TAuthWrite extends ConnectorAuth = ConnectorAuth, TAuthPublic extends ConnectorAuthPublic = ConnectorAuthPublic, TConnector extends ConnectorBase<TAuthPublic> = ConnectorBase<TAuthPublic>, TCatalogEntry extends ConnectorCatalogEntry<TAuthPublic> = ConnectorCatalogEntry<TAuthPublic>, TCreate extends CreateConnectorRequest<TAuthWrite> = CreateConnectorRequest<TAuthWrite>, TUpdate extends UpdateConnectorRequest<TAuthWrite> = UpdateConnectorRequest<TAuthWrite>> {
     getConnectorCatalog(): Promise<TCatalogEntry[]>;
+    getConnector(req: {
+        id: string;
+    }): Promise<TConnector>;
     listConnectors(req?: {
         query?: string;
     }): Promise<TConnector[]>;
+    getToolsByConnectorId(req: {
+        id: string;
+    }): Promise<TTool[]>;
     createConnector(req: TCreate): Promise<TConnector>;
     /** Full replace update keyed by connector `id`. */
     updateConnector(req: TUpdate): Promise<TConnector>;
     /**
      * Start connector auth (e.g. OAuth).
-     * For oauth, the returned connector's `auth.authUrl` is the authorize URL.
+     * May return a connector (already authenticated / with `auth.authUrl`) or a
+     * result carrying `authorization_endpoint` for the popup flow.
      */
-    authenticateConnector(req: {
-        id: string;
-    }): Promise<TConnector>;
+    authenticateConnector(req: AuthenticateConnectorRequest): Promise<TConnector | ConnectorAuthenticationResult<TConnector>>;
     /** Clear connector auth. */
     disconnectConnector(req: {
         id: string;
@@ -679,7 +723,7 @@ interface SkillCatalogServer<TSkill extends SkillBase = SkillBase, TCatalogEntry
         id: string;
     }): Promise<void>;
 }
-/** Mutable sandbox settings shared by catalog rows, create, and update. */
+/** Mutable sandbox provider settings shared by catalog rows, create, and update. */
 interface SandboxConfig {
     snapshotName: string;
     execTimeoutMs: number;
@@ -692,14 +736,18 @@ interface SandboxCatalogEntry extends SandboxConfig {
     name: string;
     type: string;
 }
-interface SandboxBase {
+/**
+ * Connected sandbox provider row (settings/sandboxes). No raw `apiKey`.
+ * Includes last-saved config so update forms can show previous values.
+ */
+interface SandboxBase extends SandboxConfig {
     id: string;
     name: string;
     catalogId: string;
     isConnected: boolean;
 }
 interface CreateSandboxRequest extends SandboxConfig {
-    /** `SandboxCatalogEntry.id` used to create this sandbox. */
+    /** `SandboxCatalogEntry.id` used to create this sandbox provider. */
     catalogId: string;
     name: string;
     type: string;
@@ -707,24 +755,37 @@ interface CreateSandboxRequest extends SandboxConfig {
 }
 interface UpdateSandboxRequest extends SandboxConfig {
     id: string;
-    apiKey: string;
+    /** Omit to keep the existing key; send a value to rotate. */
+    apiKey?: string;
 }
-interface SandboxCatalogServer<TSandbox extends SandboxBase = SandboxBase, TCatalogEntry extends SandboxCatalogEntry = SandboxCatalogEntry, TCreate extends CreateSandboxRequest = CreateSandboxRequest, TUpdate extends UpdateSandboxRequest = UpdateSandboxRequest> {
-    getSandboxCatalog(): Promise<TCatalogEntry[]>;
-    listSandboxes(req?: {
+/** Host-facing aliases (trueforge-ui public names). */
+type SandboxProviderConfig = SandboxConfig;
+type SandboxProviderCatalogEntry = SandboxCatalogEntry;
+type SandboxProviderBase = SandboxBase;
+type CreateSandboxProviderRequest = CreateSandboxRequest;
+type UpdateSandboxProviderRequest = UpdateSandboxRequest;
+interface SandboxCatalogServer<TProvider extends SandboxBase = SandboxBase, TCatalogEntry extends SandboxCatalogEntry = SandboxCatalogEntry, TCreate extends CreateSandboxRequest = CreateSandboxRequest, TUpdate extends UpdateSandboxRequest = UpdateSandboxRequest> {
+    getSandboxProviderCatalog(): Promise<TCatalogEntry[]>;
+    listSandboxProviders(req?: {
         query?: string;
-    }): Promise<TSandbox[]>;
-    createSandbox(req: TCreate): Promise<TSandbox>;
-    updateSandbox(req: TUpdate): Promise<TSandbox>;
-    deleteSandbox(req: {
+    }): Promise<TProvider[]>;
+    createSandboxProvider(req: TCreate): Promise<TProvider>;
+    updateSandboxProvider(req: TUpdate): Promise<TProvider>;
+    deleteSandboxProvider?(req: {
         id: string;
     }): Promise<void>;
 }
+/** Host-facing selector / compose aliases (trueforge-ui public names). */
+type ModelSelection = ModelSelectorEntry;
+type AgentSkill = SkillSelectorEntry;
+type ConnectorState = ConnectorSelectorEntry;
+type AgentLibraryEntry = AgentSelectorEntry;
+type SearchAgentsParams = SearchAgentSelectorParams;
 /**
- * Settings management aggregate — modelCatalog + connectorCatalog + optional
- * skill and sandbox catalogs.
- * Hosts may pass the whole object to an app shell, or a focused sub-port to a page.
- */
+* Settings management aggregate — modelCatalog + connectorCatalog + optional
+* skill and sandbox catalogs.
+* Hosts may pass the whole object to an app shell, or a focused sub-port to a page.
+*/
 interface CatalogServer<TModelCatalog extends ModelCatalogServer = ModelCatalogServer, TConnectorCatalog extends ConnectorCatalogServer = ConnectorCatalogServer, TSkillCatalog extends SkillCatalogServer = SkillCatalogServer, TSandboxCatalog extends SandboxCatalogServer = SandboxCatalogServer> {
     modelCatalog: TModelCatalog;
     connectorCatalog: TConnectorCatalog;
@@ -735,15 +796,17 @@ interface CatalogServer<TModelCatalog extends ModelCatalogServer = ModelCatalogS
 }
 /**
  * Composed host port: chat + builder + optional settings catalog.
- * Agent-ui's `AgentUIServer` mirrors this shape; named differently here to
- * avoid colliding with that package's local type name.
  *
  * `catalog` is optional — if the host passes it, settings UI can call
  * `useCatalogServer()` / show modelCatalog, connectorCatalog, and skillCatalog;
  * if omitted, those surfaces stay hidden.
+ *
+ * trueforge-ui re-exports this as `AgentUIServer`.
  */
 type AgentUIServerPort<TChat extends AgentChatServer = AgentChatServer, TBuilder extends AgentBuilderServer = AgentBuilderServer, TCatalog extends CatalogServer = CatalogServer> = TChat & TBuilder & {
     catalog?: TCatalog;
 };
+/** Host-facing alias used by trueforge-ui. */
+type AgentUIServer = AgentUIServerPort;
 
-export type { SkillBase as $, AgentSpec as A, ListSessionsParams as B, CatalogServer as C, DefinedSkill as D, ModelCatalogServer as E, ModelEntry as F, GithubSkill as G, ModelMessageEvent as H, ImportGithubSkillRequest as I, ModelProviderBase as J, ModelProviderCatalogEntry as K, ListResult as L, McpAuthRequiredEvent as M, ModelProviderConfigBase as N, ProviderType as O, PreviousTurnIdInput as P, SandboxBase as Q, RegistrySkill as R, Session as S, TurnEvent as T, UserToolResponseEvent as U, SandboxCatalogEntry as V, SandboxCatalogServer as W, SandboxConfig as X, SandboxCreatedEvent as Y, SelectRegistrySkillRequest as Z, SessionEventItem as _, AgentChatServer as a, SkillCatalogEntry as a0, SkillCatalogServer as a1, SkillConfigBase as a2, ToolApprovalRequiredEvent as a3, ToolBase as a4, ToolCall as a5, ToolResponseRequiredEvent as a6, TurnState as a7, TurnStateDone as a8, TurnStreamData as a9, SkillMount as aA, ThreadDoneEvent as aB, ToolCallFunction as aC, ToolCallRef as aD, ToolInfo as aE, ToolResponseEvent as aF, TurnCreatedEvent as aG, TurnDoneEvent as aH, TurnStateCancelled as aI, TurnStateError as aJ, TurnStateRunning as aK, UserMessageContent as aL, TurnStreamingEvent as aa, UpdateConnectorRequest as ab, UpdateModelProviderRequest as ac, UpdateSandboxRequest as ad, UpdateSessionRequest as ae, UserMessage as af, AgentSelectorEntry as ag, ConnectorSelectorEntry as ah, ModelSelectorEntry as ai, SkillSelectorEntry as aj, DeltaEvents as ak, ActionRequiredEvent as al, AgentInfo as am, AgentParent as an, ApprovalDecision as ao, ChunkDeltaToolCall as ap, ListSessionsOrder as aq, McpInitializeEvent as ar, McpServerAuthInfo as as, McpServerMount as at, Model as au, ModelMessageContentPart as av, ModelMessageDeltaEvent as aw, ModelParams as ax, PageParams as ay, SearchAgentSelectorParams as az, ThreadCreatedEvent as b, UserToolApprovalEvent as c, Turn as d, TurnInputItem as e, AgentBuilderServer as f, AgentUIServerPort as g, ConnectorAuth as h, ConnectorAuthApiKey as i, ConnectorAuthNone as j, ConnectorAuthOAuth as k, ConnectorAuthPublic as l, ConnectorAuthPublicApiKey as m, ConnectorAuthPublicNone as n, ConnectorAuthPublicOAuth as o, ConnectorAuthType as p, ConnectorBase as q, ConnectorCatalogEntry as r, ConnectorCatalogServer as s, ConnectorConfigBase as t, CreateConnectorRequest as u, CreateModelProviderRequest as v, CreateSandboxRequest as w, CreateSessionRequest as x, CreateSkillRequest as y, CreateSkillRequestBase as z };
+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, ModelParams as a0, ModelProviderBase as a1, ModelProviderCatalogEntry as a2, ModelProviderConfigBase as a3, ModelSelection as a4, ModelSelectorEntry as a5, PageParams as a6, PreviousTurnIdInput as a7, ProviderType as a8, RegistrySkill as a9, TurnStreamingEvent as aA, UpdateConnectorRequest as aB, UpdateModelProviderRequest as aC, UpdateSandboxProviderRequest as aD, UpdateSandboxRequest as aE, UpdateSessionRequest as aF, UserMessage as aG, DeltaEvents as aH, ActionRequiredEvent as aI, AgentInfo as aJ, AgentParent as aK, ChunkDeltaToolCall as aL, McpInitializeEvent as aM, McpServerAuthInfo as aN, ModelMessageContentPart as aO, ModelMessageDeltaEvent as aP, ThreadDoneEvent as aQ, ToolCallFunction as aR, ToolCallRef as aS, ToolInfo as aT, ToolResponseEvent as aU, TurnCreatedEvent as aV, TurnDoneEvent as aW, TurnStateCancelled as aX, TurnStateError as aY, TurnStateRunning as aZ, UserMessageContent as a_, SandboxBase as aa, SandboxCatalogEntry as ab, SandboxCatalogServer as ac, SandboxConfig as ad, SandboxCreatedEvent as ae, SandboxProviderBase as af, SandboxProviderCatalogEntry as ag, SandboxProviderConfig as ah, SaveAgentRequest as ai, SearchAgentSelectorParams as aj, SearchAgentsParams as ak, SelectRegistrySkillRequest as al, SessionEventItem as am, SkillBase as an, SkillCatalogEntry as ao, SkillCatalogServer as ap, SkillConfigBase as aq, SkillMount as ar, SkillSelectorEntry as as, ToolApprovalRequiredEvent as at, ToolBase as au, ToolCall as av, ToolResponseRequiredEvent as aw, TurnState as ax, TurnStateDone as ay, TurnStreamData as az, ThreadCreatedEvent as b, 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 };

package/package.json

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

package/src/convertTurnMessages.test.ts

@@ -472,6 +472,7 @@ describe("convertTurnMessages", () => {
         it("advances the turn id when a continuation reuses the existing message", () => {
             const existing = turnStreamUpdateToAssistantMessage("turn-1", {
                 content: [{ type: "text", text: "first" }],
+                metadata: { custom: { sandboxId: "sbx-123" } },
             });
             const updated = turnStreamUpdateToAssistantMessage(
                 "turn-2",
@@ -480,6 +481,7 @@ describe("convertTurnMessages", () => {
             );
             expect(updated.id).toBe("turn-1-assistant");
             expect(updated.metadata.custom.turnId).toBe("turn-2");
+            expect(updated.metadata.custom.sandboxId).toBe("sbx-123");
         });
     });
 
@@ -625,6 +627,127 @@ describe("convertTurnMessages", () => {
             expect(assistant.metadata.custom.turnId).toBe("turn-2");
         });
 
+        it("keeps the sandboxId when an ask-user continuation is merged", async () => {
+            const firstTurn = mockTurn({
+                id: "turn-1",
+                createdAt,
+                input: [{ type: "user.message", content: "create a file" }],
+                events: [
+                    sandboxCreated({ id: "sandbox-evt", sandboxId: "sbx-123" }),
+                    modelMessage({
+                        id: "m1",
+                        threadId: ROOT_THREAD_ID,
+                        content: "Which format?",
+                    }),
+                ],
+            });
+            const continuationTurn = mockTurn({
+                id: "turn-2",
+                createdAt,
+                input: [
+                    {
+                        type: "user.tool_response",
+                        threadId: ROOT_THREAD_ID,
+                        toolCallId: "ask-1",
+                        content: "Markdown",
+                    },
+                ],
+                events: [
+                    modelMessage({
+                        id: "m2",
+                        threadId: ROOT_THREAD_ID,
+                        content: "Done",
+                    }),
+                ],
+            });
+
+            const result = await convertTurnsToThreadMessages(
+                mockServerWithTurns([continuationTurn, firstTurn]),
+                SESSION_ID,
+            );
+
+            expect(result.messages[1]).toMatchObject({
+                role: "assistant",
+                metadata: {
+                    custom: {
+                        sandboxId: "sbx-123",
+                        turnId: "turn-2",
+                    },
+                },
+            });
+        });
+
+        it("clears pendingMcpAuth when an MCP auth continuation is merged", async () => {
+            const mcpServers = [
+                {
+                    id: "github-auth",
+                    name: "github",
+                    authUrl: "https://example.com/auth",
+                },
+            ];
+            const pausedTurn = mockTurn({
+                id: "turn-1",
+                createdAt,
+                input: [{ type: "user.message", content: "use github" }],
+                state: {
+                    status: "done",
+                    output: undefined,
+                    requiredActions: [
+                        {
+                            type: "mcp.auth_required",
+                            id: "mcp-auth",
+                            createdAt,
+                            mcpServers,
+                        },
+                    ] as never,
+                    completedAt: createdAt,
+                },
+                events: [
+                    sandboxCreated({ id: "sandbox-evt", sandboxId: "sbx-123" }),
+                    modelMessage({
+                        id: "m1",
+                        threadId: ROOT_THREAD_ID,
+                        content: "Need auth",
+                    }),
+                    {
+                        type: "mcp.auth_required",
+                        id: "mcp-auth",
+                        createdAt,
+                        mcpServers,
+                    },
+                ],
+            });
+            const continuationTurn = mockTurn({
+                id: "turn-2",
+                createdAt,
+                input: [],
+                events: [
+                    modelMessage({
+                        id: "m2",
+                        threadId: ROOT_THREAD_ID,
+                        content: "Authorized",
+                    }),
+                ],
+            });
+
+            const result = await convertTurnsToThreadMessages(
+                mockServerWithTurns([continuationTurn, pausedTurn]),
+                SESSION_ID,
+            );
+
+            const assistant = result.messages[1];
+            expect(assistant?.role).toBe("assistant");
+            if (assistant?.role !== "assistant") {
+                return;
+            }
+            expect(assistant.status).toEqual({ type: "complete", reason: "stop" });
+            expect(assistant.metadata.custom).toEqual({
+                sandboxId: "sbx-123",
+                turnId: "turn-2",
+            });
+            expect(assistant.metadata.custom.pendingMcpAuth).toBeUndefined();
+        });
+
         it("preserves requires-action on reload when a turn ends awaiting tool approval", async () => {
             const pausedTurn = mockTurn({
                 id: "turn-approval",

package/src/convertTurnMessages.ts

@@ -1029,10 +1029,12 @@ function projectHistoryTurns(
     const messages: ThreadMessage[] = [];
     let lastAssistantIndex: number | undefined;
     let groupRootIds: string[] = [];
+    let sandboxId: string | undefined;
 
     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];
@@ -1080,7 +1082,7 @@ function projectHistoryTurns(
         const custom = {
             ...baseCustom,
             turnId: record.id,
-            ...(record.sandboxId != null ? { sandboxId: record.sandboxId } : {}),
+            ...(sandboxId != null ? { sandboxId } : {}),
         };
 
         if (record.userText) {
@@ -1738,7 +1740,11 @@ export function turnStreamUpdateToAssistantMessage(
             unstable_annotations: [],
             unstable_data: [],
             steps: [],
-            custom: { ...update.metadata?.custom, turnId },
+            custom: {
+                ...(existing?.role === "assistant" ? existing.metadata.custom : {}),
+                ...update.metadata?.custom,
+                turnId,
+            },
         },
     };
 }