@truefoundry/assistant-ui-runtime 0.1.9 → 0.1.10

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, 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-6yWuWHzK.js';
+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 { TruefoundryGatewayApi } from 'truefoundry-gateway-sdk';
 
 /**

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

@@ -7,7 +7,7 @@ import {
   isTfyMcpToolInfo,
   isTfySystemToolInfo,
   isTfyToolInfo
-} from "../../chunk-CXBZ6WLZ.js";
+} from "../../chunk-6ZOGBD7E.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-6yWuWHzK.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-6yWuWHzK.js';
+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';
 
 /**
  * Local implementations of streaming delta helpers.

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

@@ -245,9 +245,13 @@ interface ConnectorSelectorEntry {
     name: string;
     description?: string;
 }
-/** Agent selector row — UI shows name only. Host extends for metadata. */
+/** Agent selector row. Host extends for metadata; `agentSpec` enables Edit. */
 interface AgentSelectorEntry {
     name: string;
+    /** Stable id when distinct from display `name`. Falls back to `name` when omitted. */
+    agentId?: string;
+    /** Published agent spec — required for Edit; optional for Try-only hosts. */
+    agentSpec?: AgentSpec;
 }
 type SearchAgentSelectorParams = {
     query?: string;
@@ -321,7 +325,8 @@ type PageParams = {
     pageToken?: string;
 };
 interface ListSessionsParams extends PageParams {
-    agentName?: string;
+    /** Host-owned agent identity filter. Hosts that key agents by name pass that name here. */
+    agentId?: string;
     /** Host-specific filter (e.g. TFY startTimestamp). */
     startTimestamp?: string;
 }
@@ -442,7 +447,15 @@ interface AgentChatServer<TSpec extends AgentSpec = AgentSpec, TSession extends
         afterSequenceNumber?: number;
         abortSignal?: AbortSignal;
     }): AsyncIterable<TurnStreamData>;
-    downloadSandboxFile?(sandboxId: string, req: {
+    /**
+     * Reads a file the agent wrote inside its sandbox. Hosts whose download route is scoped to a
+     * turn resolve the sandbox from `turnId` and ignore `sandboxId`; hosts addressing sandboxes
+     * directly use `sandboxId`.
+     */
+    downloadSandboxFile?(req: {
+        sessionId: string;
+        turnId: string;
+        sandboxId: string;
         path: string;
     }): Promise<Blob>;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@truefoundry/assistant-ui-runtime",
-  "version": "0.1.9",
+  "version": "0.1.10",
   "description": "TrueFoundry Gateway agent runtime adapter for assistant-ui",
   "license": "Apache-2.0",
   "repository": {
@@ -54,8 +54,8 @@
     "typecheck": "tsc --noEmit"
   },
   "dependencies": {
-    "@assistant-ui/core": "^0.2.19",
-    "@assistant-ui/store": "^0.2.19"
+    "@assistant-ui/core": "^0.2.22",
+    "@assistant-ui/store": "^0.2.21"
   },
   "peerDependencies": {
     "@types/react": "*",

package/src/convertTurnMessages.test.ts

@@ -461,6 +461,26 @@ describe("convertTurnMessages", () => {
             expect(updated.id).toBe(existing.id);
             expect(updated.content).toEqual([{ type: "text", text: "second" }]);
         });
+
+        it("stamps the streaming turn id onto the message metadata", () => {
+            const message = turnStreamUpdateToAssistantMessage("turn-1", {
+                content: [{ type: "text", text: "hi" }],
+            });
+            expect(message.metadata.custom.turnId).toBe("turn-1");
+        });
+
+        it("advances the turn id when a continuation reuses the existing message", () => {
+            const existing = turnStreamUpdateToAssistantMessage("turn-1", {
+                content: [{ type: "text", text: "first" }],
+            });
+            const updated = turnStreamUpdateToAssistantMessage(
+                "turn-2",
+                { content: [{ type: "text", text: "second" }] },
+                existing,
+            );
+            expect(updated.id).toBe("turn-1-assistant");
+            expect(updated.metadata.custom.turnId).toBe("turn-2");
+        });
     });
 
     describe("repositoryItemsFromMessages", () => {
@@ -546,7 +566,7 @@ describe("convertTurnMessages", () => {
 
             expect(result.messages[1]).toMatchObject({
                 role: "assistant",
-                metadata: { custom: { sandboxId: "sbx-123" } },
+                metadata: { custom: { sandboxId: "sbx-123", turnId: "turn-1" } },
             });
         });
 
@@ -601,6 +621,8 @@ describe("convertTurnMessages", () => {
                 { type: "text", text: "first chunk" },
                 { type: "text", text: " after approval" },
             ]);
+            // Artifacts on a merged message belong to the turn that wrote them last.
+            expect(assistant.metadata.custom.turnId).toBe("turn-2");
         });
 
         it("preserves requires-action on reload when a turn ends awaiting tool approval", async () => {

package/src/convertTurnMessages.ts

@@ -1074,8 +1074,14 @@ function projectHistoryTurns(
             content,
             record,
         );
-        const custom =
-            record.sandboxId != null ? { ...baseCustom, sandboxId: record.sandboxId } : baseCustom;
+        // Continuation turns fold into the previous assistant message and overwrite this custom,
+        // so a folded message reports the latest turn that contributed to it — the one whose
+        // sandbox wrote the newest artifacts.
+        const custom = {
+            ...baseCustom,
+            turnId: record.id,
+            ...(record.sandboxId != null ? { sandboxId: record.sandboxId } : {}),
+        };
 
         if (record.userText) {
             messages.push(
@@ -1732,7 +1738,7 @@ export function turnStreamUpdateToAssistantMessage(
             unstable_annotations: [],
             unstable_data: [],
             steps: [],
-            custom: update.metadata?.custom ?? {},
+            custom: { ...update.metadata?.custom, turnId },
         },
     };
 }

package/src/draft/truefoundryDraftThreadListAdapter.test.ts

@@ -10,7 +10,12 @@ const defaultAgentSpec: AgentSpec = {
     instructions: "You are helpful.",
 };
 
-function mockDraft(id: string, title: string | undefined, updatedAt: string): Session {
+function mockDraft(
+    id: string,
+    title: string | undefined,
+    updatedAt: string,
+    extras?: Partial<Session>,
+): Session {
     return {
         id,
         agentSpec: defaultAgentSpec,
@@ -18,6 +23,7 @@ function mockDraft(id: string, title: string | undefined, updatedAt: string): Se
         createdAt: updatedAt,
         updatedAt,
         isMutable: true,
+        ...extras,
     };
 }
 
@@ -33,10 +39,22 @@ function mockServer(partial: Partial<AgentChatServer>): AgentChatServer {
 }
 
 describe("createTrueFoundryDraftThreadListAdapter", () => {
-    it("lists draft sessions with pagination cursor", async () => {
+    it("lists sessions with pagination cursor and does not filter isMutable client-side", async () => {
         const listSessions = vi.fn().mockResolvedValue(
             mockDraftListPage(
-                [mockDraft("d1", "My draft", "2026-06-30T10:00:00.000Z")],
+                [
+                    mockDraft("d1", "My draft", "2026-06-30T10:00:00.000Z"),
+                    mockDraft("s1", "Named", "2026-06-30T09:00:00.000Z", {
+                        isMutable: false,
+                        agentName: "from-sdk",
+                        agentSpec: undefined,
+                    }),
+                    mockDraft("s2", undefined, "2026-06-30T08:00:00.000Z", {
+                        isMutable: false,
+                        agentName: "untitled-agent",
+                        agentSpec: undefined,
+                    }),
+                ],
                 "page-2",
             ),
         );
@@ -61,7 +79,7 @@ describe("createTrueFoundryDraftThreadListAdapter", () => {
             }),
         );
         expect(listSessions).toHaveBeenCalledWith(
-            expect.not.objectContaining({ agentName: expect.anything() }),
+            expect.not.objectContaining({ agentId: expect.anything() }),
         );
         expect(result.threads).toEqual([
             {
@@ -70,10 +88,45 @@ describe("createTrueFoundryDraftThreadListAdapter", () => {
                 title: "My draft",
                 lastMessageAt: new Date("2026-06-30T10:00:00.000Z"),
             },
+            {
+                status: "regular",
+                remoteId: "s1",
+                title: "Named",
+                lastMessageAt: new Date("2026-06-30T09:00:00.000Z"),
+                custom: { agentName: "from-sdk" },
+            },
+            {
+                status: "regular",
+                remoteId: "s2",
+                title: "untitled-agent",
+                lastMessageAt: new Date("2026-06-30T08:00:00.000Z"),
+                custom: { agentName: "untitled-agent" },
+            },
         ]);
         expect(result.nextCursor).toBe("page-2");
     });
 
+    it("forwards listSessionsAgentId as agentId", async () => {
+        const listSessions = vi.fn().mockResolvedValue(mockDraftListPage([]));
+        const server = mockServer({
+            listSessions,
+            createSession: vi.fn(),
+            getSession: vi.fn(),
+        });
+
+        const adapter = createTrueFoundryDraftThreadListAdapter({
+            server,
+            defaultAgentSpec,
+            listSessionsAgentId: "agent-x",
+        });
+
+        await adapter.list();
+
+        expect(listSessions).toHaveBeenCalledWith(
+            expect.objectContaining({ agentId: "agent-x" }),
+        );
+    });
+
     it("creates a draft session on initialize", async () => {
         const createSession = vi.fn().mockResolvedValue(
             mockDraft("d-new", undefined, "2026-06-30T12:00:00.000Z"),

package/src/draft/truefoundryDraftThreadListAdapter.ts

@@ -2,8 +2,11 @@ import type { RemoteThreadListAdapter } from "@assistant-ui/core";
 
 import type { AgentChatServer } from "../server/types.js";
 import type { AgentSpec } from "../server/types.js";
-import { draftSessionTitle } from "./agentSpec.js";
 import { sessionListStartTimestamp } from "../sessionListStartTimestamp.js";
+import {
+    sessionDisplayTitle,
+    sessionToThreadMetadata,
+} from "../sessionThreadMetadata.js";
 
 const THREAD_LIST_PAGE_SIZE = 20;
 
@@ -11,27 +14,25 @@ export function createTrueFoundryDraftThreadListAdapter(options: {
     server: AgentChatServer;
     defaultAgentSpec: AgentSpec;
     getAgentSpec?: () => AgentSpec;
+    /** When set, filters `listSessions` by this agent id. Omit for all chats. */
+    listSessionsAgentId?: string;
 }): RemoteThreadListAdapter {
-    const { server, defaultAgentSpec, getAgentSpec } = options;
+    const { server, defaultAgentSpec, getAgentSpec, listSessionsAgentId } = options;
 
     return {
         async list({ after } = {}) {
             const page = await server.listSessions({
+                ...(listSessionsAgentId != null ? { agentId: listSessionsAgentId } : {}),
                 limit: THREAD_LIST_PAGE_SIZE,
                 pageToken: after,
                 startTimestamp: sessionListStartTimestamp(),
             });
-            const threads = page.data
-                .filter((session) => session.isMutable)
-                .map((draft) => ({
-                    status: "regular" as const,
-                    remoteId: draft.id,
-                    title: draftSessionTitle({
-                        title: draft.title,
-                        agentSpec: draft.agentSpec ?? defaultAgentSpec,
-                    }),
-                    lastMessageAt: new Date(draft.updatedAt),
-                }));
+            const threads = page.data.map((session) =>
+                sessionToThreadMetadata(
+                    session,
+                    sessionDisplayTitle(session, defaultAgentSpec),
+                ),
+            );
             return {
                 threads,
                 nextCursor: page.nextPageToken ?? undefined,
@@ -47,15 +48,10 @@ export function createTrueFoundryDraftThreadListAdapter(options: {
 
         async fetch(remoteId) {
             const draft = await server.getSession({ sessionId: remoteId });
-            return {
-                status: "regular" as const,
-                remoteId: draft.id,
-                title: draftSessionTitle({
-                    title: draft.title,
-                    agentSpec: draft.agentSpec ?? defaultAgentSpec,
-                }),
-                lastMessageAt: new Date(draft.updatedAt),
-            };
+            return sessionToThreadMetadata(
+                draft,
+                sessionDisplayTitle(draft, defaultAgentSpec),
+            );
         },
 
         async rename() {},

package/src/hooks.ts

@@ -1,8 +1,9 @@
 "use client";
 
 import { useMemo } from "react";
-import { useAui } from "@assistant-ui/store";
+import { useAui, useAuiState } from "@assistant-ui/store";
 
+import type { TrueFoundryMessageCustomMetadata } from "./messageCustomMetadata.js";
 import {
     EMPTY_DRAFT_EXTRAS,
     trueFoundryExtras,
@@ -86,10 +87,29 @@ export const useTrueFoundryResumeMcpAuth = () => {
 export const useTrueFoundrySandboxId = (): string | undefined =>
     trueFoundryExtras.use((e) => e.sandboxId, undefined);
 
-/** Returns a function to download a sandbox file by path from any render context. */
+/** Turn that produced the message being rendered. Only defined inside a message scope. */
+export const useTrueFoundryTurnId = (): string | undefined =>
+    useAuiState(
+        (state) =>
+            (state.message.metadata?.custom as TrueFoundryMessageCustomMetadata | undefined)
+                ?.turnId,
+    );
+
+/**
+ * Returns a function to download a file the current turn wrote to its sandbox. Must be called
+ * from a message scope, since the artifact belongs to the turn that rendered it.
+ */
 export const useTrueFoundryDownloadSandboxFile = () => {
     const aui = useAui();
-    return (path: string) => trueFoundryExtras.get(aui).downloadSandboxFile(path);
+    const turnId = useTrueFoundryTurnId();
+    return (path: string) => {
+        if (turnId == null) {
+            throw new Error(
+                "Downloading a sandbox file requires a message scope to resolve its turn.",
+            );
+        }
+        return trueFoundryExtras.get(aui).downloadSandboxFile({ turnId, path });
+    };
 };
 
 /** Returns a function to cancel the current run from any render context. */

package/src/index.ts

@@ -49,6 +49,7 @@ export {
     useTrueFoundryRespondToToolResponse,
     useTrueFoundryResumeMcpAuth,
     useTrueFoundrySandboxId,
+    useTrueFoundryTurnId,
     useTrueFoundryDownloadSandboxFile,
     useTrueFoundryCancel,
     useTrueFoundryResetFromTurn,

package/src/messageCustomMetadata.ts

@@ -10,6 +10,8 @@ export type TrueFoundryMessageCustomMetadata = {
     pendingMcpAuth?: true;
     mcpServers?: McpAuthRequiredEvent["mcpServers"];
     sandboxId?: string;
+    /** Turn that produced this message. Scopes artifact downloads to their own turn. */
+    turnId?: string;
     [TOOL_APPROVAL_THREAD_ID_CUSTOM_KEY]?: string;
     [TOOL_RESPONSE_THREAD_ID_CUSTOM_KEY]?: string;
 };

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

@@ -227,7 +227,7 @@ export function createTrueFoundryChatServer<
                 pageToken: req?.pageToken,
                 startTimestamp: req?.startTimestamp,
                 endTimestamp: req?.endTimestamp,
-                ...(req?.agentName != null ? { agentName: req.agentName } : {}),
+                ...(req?.agentId != null ? { agentName: req.agentId } : {}),
             });
             const result = await toListResult(page, (s) => toSession<TSpec>(s));
             for (const session of result.data) {
@@ -350,11 +350,12 @@ export function createTrueFoundryChatServer<
             ) as AsyncIterable<TurnStreamData>;
         },
 
-        async downloadSandboxFile(sandboxId, req) {
-            const response = await privateClient.downloadSandboxFile(
-                sandboxId,
-                req,
-            );
+        // The gateway addresses sandboxes directly, so the turn is carried for hosts that need it
+        // but is not part of this route.
+        async downloadSandboxFile({ sandboxId, path }) {
+            const response = await privateClient.downloadSandboxFile(sandboxId, {
+                path,
+            });
             return await response.blob();
         },
 

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

@@ -1,6 +1,7 @@
 import { afterEach, describe, expect, it, vi } from "vitest";
 
 import {
+    agentSpecFromCpManifest,
     buildSaveAgentManifest,
     normalizeAgents,
     normalizeAgentSkills,
@@ -10,6 +11,7 @@ import {
     saveAgent,
     SAVE_AGENT_COLLABORATORS,
     SAVE_AGENT_METADATA_TAGS,
+    toCamelCaseDeep,
     toSnakeCaseDeep,
 } from "./cp.js";
 
@@ -273,16 +275,175 @@ describe("normalizeMcpServers", () => {
 });
 
 describe("normalizeAgents", () => {
-    it("maps name only", () => {
+    it("maps name, agentId, and agentSpec from latestVersionDetails.manifest", () => {
         expect(
             normalizeAgents({
                 data: [
-                    { name: "ask-ai-agent", id: "ag_1" },
+                    {
+                        id: "ag_1",
+                        name: "ask-ai-agent",
+                        latestVersionDetails: {
+                            manifest: {
+                                type: "truefoundry-agent",
+                                name: "ask-ai-agent",
+                                model: {
+                                    name: "openai-main/gpt-4.1",
+                                    params: {
+                                        max_tokens: 8192,
+                                        reasoning_effort: "medium",
+                                    },
+                                },
+                                instructions: "Be helpful",
+                                config: {
+                                    iteration_limit: 50,
+                                    ask_user_questions: { enabled: true },
+                                },
+                                skills: [
+                                    {
+                                        type: "truefoundry-skills-registry",
+                                        fqn: "agent-skill:tfy/skills/web:1",
+                                        preload: true,
+                                        config: { timeout_ms: 1000 },
+                                    },
+                                ],
+                                mcp_servers: [
+                                    {
+                                        name: "gmail",
+                                        enable_tools: ["@read-only"],
+                                        preload: true,
+                                        config: { locale: "en" },
+                                    },
+                                ],
+                            },
+                        },
+                    },
                     { name: "" },
                     {},
+                    {
+                        name: "try-only",
+                        // no manifest → Try still works; Edit hidden
+                    },
                 ],
             }),
-        ).toEqual([{ name: "ask-ai-agent" }]);
+        ).toEqual([
+            {
+                name: "ask-ai-agent",
+                agentId: "ag_1",
+                agentSpec: {
+                    model: {
+                        name: "openai-main/gpt-4.1",
+                        params: { maxTokens: 8192, reasoningEffort: "medium" },
+                    },
+                    instructions: "Be helpful",
+                    config: {
+                        iterationLimit: 50,
+                        askUserQuestions: { enabled: true },
+                    },
+                    skills: [
+                        {
+                            id: "agent-skill:tfy/skills/web:1",
+                            name: "agent-skill:tfy/skills/web:1",
+                            preload: true,
+                            config: { timeoutMs: 1000 },
+                        },
+                    ],
+                    mcpServers: [
+                        {
+                            id: "gmail",
+                            name: "gmail",
+                            enableTools: ["@read-only"],
+                            preload: true,
+                            config: { locale: "en" },
+                        },
+                    ],
+                },
+            },
+            { name: "try-only", agentId: "try-only" },
+        ]);
+    });
+});
+
+describe("toCamelCaseDeep", () => {
+    it("converts nested snake_case keys", () => {
+        expect(
+            toCamelCaseDeep({
+                max_tokens: 8192,
+                reasoning_effort: "medium",
+                nested: { iteration_limit: 50 },
+            }),
+        ).toEqual({
+            maxTokens: 8192,
+            reasoningEffort: "medium",
+            nested: { iterationLimit: 50 },
+        });
+    });
+});
+
+describe("agentSpecFromCpManifest", () => {
+    it("returns undefined when model.name is missing", () => {
+        expect(agentSpecFromCpManifest({ type: "truefoundry-agent" })).toBeUndefined();
+        expect(agentSpecFromCpManifest(null)).toBeUndefined();
+    });
+
+    it("preserves mount runtime fields and agent config for Edit → Save", () => {
+        const spec = agentSpecFromCpManifest({
+            type: "truefoundry-agent",
+            name: "saved",
+            model: { name: "openai-main/gpt-4.1" },
+            config: { iteration_limit: 25 },
+            skills: [
+                {
+                    type: "truefoundry-skills-registry",
+                    fqn: "agent-skill:tfy/skills/web:1",
+                    preload: false,
+                },
+            ],
+            mcp_servers: [
+                {
+                    name: "gmail",
+                    enable_tools: ["@read-only"],
+                    preload: true,
+                },
+            ],
+        });
+
+        expect(spec).toEqual({
+            model: { name: "openai-main/gpt-4.1" },
+            config: { iterationLimit: 25 },
+            skills: [
+                {
+                    id: "agent-skill:tfy/skills/web:1",
+                    name: "agent-skill:tfy/skills/web:1",
+                    preload: false,
+                },
+            ],
+            mcpServers: [
+                {
+                    id: "gmail",
+                    name: "gmail",
+                    enableTools: ["@read-only"],
+                    preload: true,
+                },
+            ],
+        });
+
+        const saved = buildSaveAgentManifest("saved", spec!);
+        expect(saved.config).toEqual({ iteration_limit: 25 });
+        expect(saved.mcp_servers).toEqual([
+            {
+                type: "truefoundry-mcp-registry",
+                name: "gmail",
+                enable_tools: ["@read-only"],
+                preload: true,
+            },
+        ]);
+        expect(saved.skills).toEqual([
+            {
+                type: "truefoundry-skills-registry",
+                fqn: "agent-skill:tfy/skills/web:1",
+                preload: false,
+            },
+        ]);
     });
 });
 
@@ -432,9 +593,11 @@ describe("saveAgent", () => {
                 }),
             }),
         );
-        const body = JSON.parse(
-            (fetchMock.mock.calls[0]?.[1] as RequestInit).body as string,
-        );
+        const [, init] = fetchMock.mock.calls[0] as unknown as [
+            string,
+            RequestInit,
+        ];
+        const body = JSON.parse(String(init.body));
         expect(body.manifest.type).toBe("truefoundry-agent");
         expect(body.manifest.name).toBe("my-agent");
         expect(body.manifest.model).toEqual({ name: "openai-main/gpt-4.1" });