@truefoundry/assistant-ui-runtime 0.1.10-rc.1 → 0.1.12

package/src/plugins/truefoundry-agent-server-adapter/normalizeAgentSpec.test.ts

@@ -29,6 +29,24 @@ describe("normalizeMcpMount", () => {
             name: "deepwiki-mcp",
         });
     });
+
+    it("forwards enableTools / preload / config from catalog rows", () => {
+        expect(
+            normalizeMcpMount({
+                id: "gmail",
+                name: "gmail",
+                enableTools: ["@read-only"],
+                preload: true,
+                config: { locale: "en" },
+            }),
+        ).toEqual({
+            type: "truefoundry-mcp-registry",
+            name: "gmail",
+            enableTools: ["@read-only"],
+            preload: true,
+            config: { locale: "en" },
+        });
+    });
 });
 
 describe("normalizeSkillMount", () => {
@@ -57,6 +75,22 @@ describe("normalizeSkillMount", () => {
             fqn: "agent-skill:truefoundry/skills/web:1",
         });
     });
+
+    it("forwards preload / config from catalog rows, including preload false", () => {
+        expect(
+            normalizeSkillMount({
+                id: "agent-skill:truefoundry/skills/web:1",
+                name: "web",
+                preload: false,
+                config: { timeoutMs: 1000 },
+            }),
+        ).toEqual({
+            type: "truefoundry-skills-registry",
+            fqn: "agent-skill:truefoundry/skills/web:1",
+            preload: false,
+            config: { timeoutMs: 1000 },
+        });
+    });
 });
 
 describe("normalizeAgentSpecForGateway", () => {

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

@@ -20,7 +20,7 @@ export function normalizeMcpMount(
         throw new Error("mcpServers entry must be an object");
     }
     if (raw.type === "truefoundry-mcp-registry" || raw.type === "inline") {
-        return raw as TruefoundryGatewayApi.McpServer;
+        return raw as unknown as TruefoundryGatewayApi.McpServer;
     }
     const name =
         (typeof raw.name === "string" && raw.name !== "" ? raw.name : null) ??
@@ -33,7 +33,13 @@ export function normalizeMcpMount(
             "mcpServers entry needs name, mcpName, or id to mount as registry MCP",
         );
     }
-    return { type: "truefoundry-mcp-registry", name };
+    return {
+        type: "truefoundry-mcp-registry",
+        name,
+        ...(Array.isArray(raw.enableTools) ? { enableTools: raw.enableTools } : {}),
+        ...(typeof raw.preload === "boolean" ? { preload: raw.preload } : {}),
+        ...(raw.config != null ? { config: raw.config } : {}),
+    } as unknown as TruefoundryGatewayApi.McpServer;
 }
 
 export function normalizeSkillMount(
@@ -43,7 +49,7 @@ export function normalizeSkillMount(
         throw new Error("skills entry must be an object");
     }
     if (raw.type === "truefoundry-skills-registry" || raw.type === "git") {
-        return raw as TruefoundryGatewayApi.SkillMount;
+        return raw as unknown as TruefoundryGatewayApi.SkillMount;
     }
     const fqn =
         (typeof raw.fqn === "string" && raw.fqn !== "" ? raw.fqn : null) ??
@@ -56,8 +62,9 @@ export function normalizeSkillMount(
     return {
         type: "truefoundry-skills-registry",
         fqn,
-        ...(raw.preload === true ? { preload: true } : {}),
-    };
+        ...(typeof raw.preload === "boolean" ? { preload: raw.preload } : {}),
+        ...(raw.config != null ? { config: raw.config } : {}),
+    } as unknown as TruefoundryGatewayApi.SkillMount;
 }
 
 /** Rewrite UI `{id,name}` mounts so create/update draft session can serialize. */

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

@@ -1,8 +1,8 @@
 /**
  * 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/src/server/index.ts

@@ -4,11 +4,17 @@ export type {
     ConnectorSelectorEntry,
     AgentSelectorEntry,
     SearchAgentSelectorParams,
+    ModelSelection,
+    AgentSkill,
+    ConnectorState,
+    AgentLibraryEntry,
+    SearchAgentsParams,
     SkillMount,
     McpServerMount,
     ModelParams,
     Model,
     AgentSpec,
+    SaveAgentRequest,
     Session,
     CreateSessionRequest,
     UpdateSessionRequest,
@@ -30,6 +36,7 @@ export type {
     TurnState,
     Turn,
     AgentChatServer,
+    AgentBuilderCapabilitiesResponse,
     AgentBuilderServer,
     ProviderType,
     ModelEntry,
@@ -54,6 +61,8 @@ export type {
     ConnectorCatalogEntry,
     CreateConnectorRequest,
     UpdateConnectorRequest,
+    AuthenticateConnectorRequest,
+    ConnectorAuthenticationResult,
     ConnectorCatalogServer,
     SkillBase,
     RegistrySkill,
@@ -71,9 +80,15 @@ export type {
     SandboxBase,
     CreateSandboxRequest,
     UpdateSandboxRequest,
+    SandboxProviderConfig,
+    SandboxProviderCatalogEntry,
+    SandboxProviderBase,
+    CreateSandboxProviderRequest,
+    UpdateSandboxProviderRequest,
     SandboxCatalogServer,
     CatalogServer,
     AgentUIServerPort,
+    AgentUIServer,
 } from "./types.js";
 
 export type {