@truefoundry/assistant-ui-runtime 0.1.6-rc.0 → 0.1.6

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

@@ -1,6 +1,131 @@
 import { AgentSessionClient } from 'truefoundry-gateway-sdk/agents';
 import { PrivateAgentSessionClient } from 'truefoundry-gateway-sdk/agents/private';
-import { a as AgentChatServer } from '../../types-VUBzoJT2.js';
+import { A as AgentSpec, C as CreateSessionRequest, L as ListSessionsParams, S as Session, T as Turn, a as TurnState, b as AgentChatServer, U as UpdateSessionRequest } from '../../types-BfiFf8O1.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.
+ *
+ * This file builds the concrete TrueFoundry types by:
+ *  - Importing the runtime's AgentSpec as the base to extend.
+ *  - Importing concrete sub-types from the gateway SDK namespace rather than
+ *    re-defining them — they're purely gateway concepts.
+ *  - Composing a TfyAgentSpec that satisfies both the runtime's
+ *    `TSpec extends AgentSpec` constraint and the gateway SDK's AgentSpec shape.
+ */
+
+type TfyRuntimeConfig = TruefoundryGatewayApi.RuntimeConfig;
+type TfyResponseFormat = TruefoundryGatewayApi.ResponseFormat;
+type TfyModelParams = TruefoundryGatewayApi.ModelParams;
+type TfySubject = TruefoundryGatewayApi.Subject;
+type ToolsSelectorTag = "@all" | "@read-only";
+type RequireApprovalToolsSelectorTag = "@all" | "@write" | "@destructive";
+type ToolsSelectorItem = ToolsSelectorTag | string;
+type RequireApprovalToolSelectorItem = RequireApprovalToolsSelectorTag | string;
+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 {
+    model: TruefoundryGatewayApi.Model;
+    skills?: TfySkillMount[];
+    mcpServers?: TfyMcpServerMount[];
+    config?: TruefoundryGatewayApi.RuntimeConfig;
+    responseFormat?: TruefoundryGatewayApi.ResponseFormat;
+    messages?: TruefoundryGatewayApi.AgentSpecUserMessage[];
+}
+type TfyTurnCancelledReason = TruefoundryGatewayApi.TurnStateCancelledReason;
+type TfyTurnStateDoneOutput = TruefoundryGatewayApi.TurnStateDoneOutput;
+type RuntimeTurnStateDone = Extract<TurnState, {
+    status: "done";
+}>;
+type RuntimeTurnStateCancelled = Extract<TurnState, {
+    status: "cancelled";
+}>;
+/**
+ * The runtime types `output` as `unknown` and `reason` as bare `string`. The
+ * gateway sends a model message and one of four reasons — `cancelled-for-next-turn`
+ * in particular is routine and should not render like a failure.
+ */
+type TfyTurnState = Exclude<TurnState, {
+    status: "done" | "cancelled";
+}> | (Omit<RuntimeTurnStateDone, "output"> & {
+    output?: TfyTurnStateDoneOutput;
+}) | (Omit<RuntimeTurnStateCancelled, "reason"> & {
+    reason: TfyTurnCancelledReason;
+});
+interface TfyTurn extends Turn {
+    state: TfyTurnState;
+    createdBySubject: TfySubject;
+}
+interface TfySession<TSpec extends TfyAgentSpec = TfyAgentSpec> extends Session<TSpec> {
+    createdBySubject: TfySubject;
+}
+interface TfyCreateSessionRequest<TSpec extends TfyAgentSpec = TfyAgentSpec> extends CreateSessionRequest<TSpec> {
+    /** Sent as `x-tfy-metadata`, persisted server-side as `request_metadata`. */
+    tfyMetadata?: string;
+}
+interface TfyListSessionsParams extends ListSessionsParams {
+    /** Inclusive upper bound on `createdAt` (ISO-8601). */
+    endTimestamp?: string;
+}
+type TfyToolInfo = TruefoundryGatewayApi.ToolInfo;
+type TfySystemToolInfo = TruefoundryGatewayApi.TrueFoundrySystemToolInfo;
+type TfyMcpToolInfo = TruefoundryGatewayApi.McpToolInfo;
+type TfyModelMessageUsage = TruefoundryGatewayApi.ModelMessageUsage;
+type TfyFinishReason = TruefoundryGatewayApi.FinishReason;
+type TfyThreadState = TruefoundryGatewayApi.ThreadState;
+type TfyMcpServerInitInfo = TruefoundryGatewayApi.McpServerInitInfo;
+
+/**
+ * Point-of-use narrowing for the event half of the gateway protocol.
+ *
+ * `AgentChatServer` hardcodes the runtime's event types on listEvents,
+ * listTurnEvents, subscribeToTurn and prepareAndExecuteTurn — there is no
+ * generic to override them from here. So instead of typing those channels,
+ * hosts call these guards on the values they receive.
+ *
+ * They validate rather than cast: this data comes off the network, and the
+ * runtime types the relevant fields as `unknown` precisely because nothing
+ * has checked them yet.
+ */
+
+/**
+ * Identifies built-in tools such as `ask_user_question` and `create_sub_agent`.
+ * Note `toolInfo` is legitimately absent on streamed deltas, so callers must
+ * keep their `function.name` fallback rather than treating absence as an error.
+ */
+declare function isTfySystemToolInfo(toolInfo: unknown): toolInfo is TfySystemToolInfo;
+/** Carries `serverId` / `serverName`, so the UI can attribute a call to its MCP server. */
+declare function isTfyMcpToolInfo(toolInfo: unknown): toolInfo is TfyMcpToolInfo;
+declare function isTfyToolInfo(toolInfo: unknown): toolInfo is TfyToolInfo;
+/**
+ * Token usage including the TrueFoundry-specific `inputTokensBreakdown`, which
+ * attributes input tokens across harness, skills, instructions, tool
+ * definitions and messages.
+ */
+declare function getTfyUsage(source: {
+    usage?: unknown;
+} | null | undefined): TfyModelMessageUsage | undefined;
+/**
+ * Completion state of a sub-agent thread. The runtime types `thread.done`'s
+ * `state` as `unknown`, so a sub-agent that errored is otherwise
+ * indistinguishable from one that succeeded.
+ */
+declare function getTfyThreadState(event: {
+    state?: unknown;
+} | null | undefined): TfyThreadState | undefined;
+/**
+ * Servers from an `mcp.initialize` event, including each one's `transportType`.
+ * The runtime models this event with an index signature, so the array is
+ * `unknown` until checked.
+ */
+declare function getTfyMcpInitServers(event: {
+    mcpServers?: unknown;
+} | null | undefined): TfyMcpServerInitInfo[] | undefined;
 
 type CreateTrueFoundryChatServerOptions = {
     apiKey: string;
@@ -12,7 +137,14 @@ type CreateTrueFoundryChatServerOptions = {
         sessionId: string;
     }) => Promise<void>;
 };
-type TrueFoundryChatServer = AgentChatServer & {
+/**
+ * Only the spec is generic. Session/Turn/list-params are the concrete Tfy*
+ * types because the adapter builds them as fixed object literals — a generic
+ * there would type fields that nothing ever populates. The spec is safe: the
+ * gateway SDK serializes with `unrecognizedObjectKeys: "passthrough"`, so
+ * host-added spec fields survive the round trip.
+ */
+type TrueFoundryChatServer<TSpec extends TfyAgentSpec = TfyAgentSpec> = AgentChatServer<TSpec, TfySession<TSpec>, TfyCreateSessionRequest<TSpec>, TfyListSessionsParams, UpdateSessionRequest<TSpec>, TfyTurn> & {
     /** Escape hatch for hosts that still need raw gateway clients. */
     getGatewayClients(): {
         client: AgentSessionClient;
@@ -25,6 +157,6 @@ type TrueFoundryChatServer = AgentChatServer & {
  * (populated by createSession/listSessions) determines which gateway client
  * to call. No try/catch fallback, no double network calls.
  */
-declare function createTrueFoundryChatServer(opts: CreateTrueFoundryChatServerOptions): TrueFoundryChatServer;
+declare function createTrueFoundryChatServer<TSpec extends TfyAgentSpec = TfyAgentSpec>(opts: CreateTrueFoundryChatServerOptions): TrueFoundryChatServer<TSpec>;
 
-export { type CreateTrueFoundryChatServerOptions, type TrueFoundryChatServer, createTrueFoundryChatServer };
+export { type CreateTrueFoundryChatServerOptions, type RequireApprovalToolSelectorItem, type RequireApprovalToolsSelectorTag, type TfyAgentSpec, type TfyCreateSessionRequest, type TfyFinishReason, type TfyListSessionsParams, type TfyMcpServerInitInfo, type TfyMcpServerMount, type TfyMcpToolInfo, type TfyModelMessageUsage, type TfyModelParams, type TfyResponseFormat, type TfyRuntimeConfig, type TfySession, type TfySkillMount, type TfySubject, type TfySystemToolInfo, type TfyThreadState, type TfyToolInfo, type TfyTurn, type TfyTurnCancelledReason, type TfyTurnState, type TfyTurnStateDoneOutput, type ToolsSelectorItem, type ToolsSelectorTag, type TrueFoundryChatServer, createTrueFoundryChatServer, getTfyMcpInitServers, getTfyThreadState, getTfyUsage, isTfyMcpToolInfo, isTfySystemToolInfo, isTfyToolInfo };

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

@@ -1,198 +1,19 @@
-// src/plugins/truefoundry-agent-server-adapter/index.ts
-import { AgentSessionClient } from "truefoundry-gateway-sdk/agents";
-import { PrivateAgentSessionClient } from "truefoundry-gateway-sdk/agents/private";
-function isDraft(session) {
-  return session.type === "session/draft";
-}
-function toSession(raw) {
-  const mutable = isDraft(raw);
-  return {
-    id: raw.id,
-    title: raw.title,
-    agentName: raw.agentName,
-    ...mutable ? { agentSpec: raw.agentSpec } : {},
-    isMutable: mutable,
-    createdAt: raw.createdAt,
-    updatedAt: raw.updatedAt
-  };
-}
-function toTurn(raw) {
-  return {
-    id: raw.id,
-    sessionId: raw.sessionId,
-    previousTurnId: raw.previousTurnId,
-    input: raw.input,
-    state: raw.state,
-    createdAt: raw.createdAt
-  };
-}
-async function toListResult(page, map) {
-  const nextPageToken = page.response?.pagination?.nextPageToken;
-  return {
-    data: page.data.map(map),
-    ...nextPageToken != null && nextPageToken !== "" ? { nextPageToken } : {}
-  };
-}
-function createTrueFoundryChatServer(opts) {
-  const gatewayOpts = { apiKey: opts.apiKey, baseUrl: opts.baseUrl };
-  const client = opts.client ?? new AgentSessionClient(gatewayOpts);
-  const privateClient = opts.privateClient ?? new PrivateAgentSessionClient(gatewayOpts);
-  const sessionTypeCache = /* @__PURE__ */ new Map();
-  function cacheSessionType(session) {
-    sessionTypeCache.set(session.id, session.isMutable);
-  }
-  function getSessionObj(sessionId) {
-    const isMutable = sessionTypeCache.get(sessionId);
-    if (isMutable === true) {
-      return privateClient.getDraftSession({ draftSessionId: sessionId });
-    }
-    if (isMutable === false) {
-      return client.getSession({ sessionId });
-    }
-    throw new Error(
-      `Cannot resolve session "${sessionId}": session type not cached. Ensure createSession or listSessions was called first.`
-    );
-  }
-  const server = {
-    async createSession(req) {
-      if (req.agentSpec != null) {
-        const draft = await privateClient.createDraftSession({
-          agentSpec: req.agentSpec,
-          ...req.agentName != null ? { agentName: req.agentName } : {}
-        });
-        const session = toSession(draft);
-        cacheSessionType(session);
-        return session;
-      }
-      if (req.agentName != null) {
-        const named = await client.createSession({
-          agentName: req.agentName
-        });
-        const session = toSession(named);
-        cacheSessionType(session);
-        return session;
-      }
-      throw new Error("createSession requires agentName and/or agentSpec");
-    },
-    async listSessions(req) {
-      const page = await privateClient.listOwnedSessions({
-        limit: req?.limit,
-        order: req?.order,
-        pageToken: req?.pageToken,
-        startTimestamp: req?.startTimestamp,
-        ...req?.agentName != null ? { agentName: req.agentName } : {}
-      });
-      const result = await toListResult(page, toSession);
-      for (const session of result.data) {
-        cacheSessionType(session);
-      }
-      return result;
-    },
-    async getSession({ sessionId }) {
-      const raw = await getSessionObj(sessionId);
-      const session = toSession(raw);
-      cacheSessionType(session);
-      return session;
-    },
-    async updateSession(req) {
-      const raw = await getSessionObj(req.sessionId);
-      if (!isDraft(raw)) {
-        throw new Error(
-          "updateSession: session is not mutable (isMutable=false)"
-        );
-      }
-      if (req.agentSpec != null) {
-        await raw.update({ agentSpec: req.agentSpec });
-      }
-      return toSession(raw);
-    },
-    prepareAndExecuteTurn(req) {
-      return (async function* () {
-        const session = await getSessionObj(req.sessionId);
-        const prepared = session.prepareTurn({
-          input: req.input,
-          previousTurnId: req.previousTurnId ?? "auto"
-        });
-        yield* prepared.execute(
-          { stream: true },
-          {
-            ...req.abortSignal != null ? { abortSignal: req.abortSignal } : {},
-            ...req.headers != null ? { headers: req.headers } : {}
-          }
-        );
-      })();
-    },
-    async cancelSession({ sessionId }) {
-      await (await getSessionObj(sessionId)).cancel();
-    },
-    async deleteSession({ sessionId }) {
-      if (opts.deleteSession == null) {
-        throw new Error(
-          "deleteSession is not on the gateway SDK. Pass deleteSession to createTrueFoundryChatServer."
-        );
-      }
-      await opts.deleteSession({ sessionId });
-    },
-    async listTurns({ sessionId, limit, pageToken, order }) {
-      const raw = await getSessionObj(sessionId);
-      const page = await raw.listTurns({
-        ...limit != null ? { limit } : {},
-        ...pageToken != null ? { pageToken } : {},
-        ...order != null ? { order } : {}
-      });
-      return toListResult(page, (turn) => toTurn(turn));
-    },
-    async getTurn({ sessionId, turnId }) {
-      const raw = await getSessionObj(sessionId);
-      return toTurn(await raw.getTurn({ turnId }));
-    },
-    async listEvents({ sessionId, pageToken, lastTurnId, limit }) {
-      const raw = await getSessionObj(sessionId);
-      const page = await raw.listEvents({
-        ...limit != null ? { limit } : {},
-        ...pageToken != null ? { pageToken } : {},
-        ...lastTurnId != null ? { lastTurnId } : {}
-      });
-      return toListResult(
-        page,
-        (item) => item
-      );
-    },
-    async listTurnEvents({ sessionId, turnId, limit, pageToken, order }) {
-      const raw = await getSessionObj(sessionId);
-      const turn = await raw.getTurn({ turnId });
-      const page = await turn.listEvents({
-        ...limit != null ? { limit } : {},
-        ...pageToken != null ? { pageToken } : {},
-        ...order != null ? { order } : {}
-      });
-      return toListResult(page, (event) => event);
-    },
-    async *subscribeToTurn({
-      sessionId,
-      turnId,
-      afterSequenceNumber,
-      abortSignal
-    }) {
-      const raw = await getSessionObj(sessionId);
-      const turn = await raw.getTurn({ turnId });
-      yield* turn.stream(
-        afterSequenceNumber != null ? { afterSequenceNumber } : {},
-        abortSignal != null ? { abortSignal } : {}
-      );
-    },
-    async downloadSandboxFile(sandboxId, req) {
-      const response = await privateClient.downloadSandboxFile(
-        sandboxId,
-        req
-      );
-      return await response.blob();
-    },
-    getGatewayClients: () => ({ client, privateClient })
-  };
-  return server;
-}
+import {
+  createTrueFoundryChatServer,
+  getTfyMcpInitServers,
+  getTfyThreadState,
+  getTfyUsage,
+  isTfyMcpToolInfo,
+  isTfySystemToolInfo,
+  isTfyToolInfo
+} from "../../chunk-Q2SHKMLM.js";
 export {
-  createTrueFoundryChatServer
+  createTrueFoundryChatServer,
+  getTfyMcpInitServers,
+  getTfyThreadState,
+  getTfyUsage,
+  isTfyMcpToolInfo,
+  isTfySystemToolInfo,
+  isTfyToolInfo
 };
 //# sourceMappingURL=index.js.map

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/plugins/truefoundry-agent-server-adapter/index.ts"],"sourcesContent":["import { AgentSessionClient } from \"truefoundry-gateway-sdk/agents\";\nimport type { AgentSession } from \"truefoundry-gateway-sdk/agents\";\nimport { PrivateAgentSessionClient } from \"truefoundry-gateway-sdk/agents/private\";\nimport type { AgentDraftSession } from \"truefoundry-gateway-sdk/agents/private\";\nimport type {\n    AgentChatServer,\n    AgentSpec,\n    ListResult,\n    Session,\n    Turn,\n    TurnInputItem,\n    PreviousTurnIdInput,\n} from \"../../server/types.js\";\nimport type {\n    TurnEvent,\n    TurnStreamData,\n    SessionEventItem,\n} from \"../../server/events.js\";\n\ntype GwSession = AgentSession | AgentDraftSession;\n\nexport type CreateTrueFoundryChatServerOptions = {\n    apiKey: string;\n    baseUrl: string;\n    /** Optional override — otherwise constructed from apiKey/baseUrl. */\n    client?: AgentSessionClient;\n    privateClient?: PrivateAgentSessionClient;\n    deleteSession?: (req: { sessionId: string }) => Promise<void>;\n};\n\nexport type TrueFoundryChatServer = AgentChatServer & {\n    /** Escape hatch for hosts that still need raw gateway clients. */\n    getGatewayClients(): {\n        client: AgentSessionClient;\n        privateClient: PrivateAgentSessionClient;\n    };\n};\n\nfunction isDraft(session: GwSession): session is AgentDraftSession {\n    return (session as AgentDraftSession).type === \"session/draft\";\n}\n\nfunction toSession(raw: GwSession): Session {\n    const mutable = isDraft(raw);\n    return {\n        id: raw.id,\n        title: raw.title,\n        agentName: raw.agentName,\n        ...(mutable ? { agentSpec: raw.agentSpec as AgentSpec } : {}),\n        isMutable: mutable,\n        createdAt: raw.createdAt,\n        updatedAt: raw.updatedAt,\n    };\n}\n\nfunction toTurn(raw: {\n    id: string;\n    sessionId: string;\n    previousTurnId?: string | null;\n    input?: TurnInputItem[];\n    state: Turn[\"state\"];\n    createdAt: string;\n}): Turn {\n    return {\n        id: raw.id,\n        sessionId: raw.sessionId,\n        previousTurnId: raw.previousTurnId,\n        input: raw.input as TurnInputItem[] | undefined,\n        state: raw.state as Turn[\"state\"],\n        createdAt: raw.createdAt,\n    };\n}\n\nasync function toListResult<TIn, TOut>(\n    page: {\n        data: TIn[];\n        response?: { pagination?: { nextPageToken?: string } };\n        hasNextPage?: () => boolean;\n    },\n    map: (item: TIn) => TOut,\n): Promise<ListResult<TOut>> {\n    const nextPageToken = page.response?.pagination?.nextPageToken;\n    return {\n        data: page.data.map(map),\n        ...(nextPageToken != null && nextPageToken !== \"\"\n            ? { nextPageToken }\n            : {}),\n    };\n}\n\n/**\n * Wraps TrueFoundry gateway clients into a flat `AgentChatServer`.\n * Named vs draft routing is fully internal — an in-memory session-type cache\n * (populated by createSession/listSessions) determines which gateway client\n * to call. No try/catch fallback, no double network calls.\n */\nexport function createTrueFoundryChatServer(\n    opts: CreateTrueFoundryChatServerOptions,\n): TrueFoundryChatServer {\n    const gatewayOpts = { apiKey: opts.apiKey, baseUrl: opts.baseUrl };\n    const client = opts.client ?? new AgentSessionClient(gatewayOpts);\n    const privateClient =\n        opts.privateClient ?? new PrivateAgentSessionClient(gatewayOpts);\n\n    const sessionTypeCache = new Map<string, boolean>();\n\n    function cacheSessionType(session: Session): void {\n        sessionTypeCache.set(session.id, session.isMutable);\n    }\n\n    function getSessionObj(sessionId: string): Promise<GwSession> {\n        const isMutable = sessionTypeCache.get(sessionId);\n        if (isMutable === true) {\n            return privateClient.getDraftSession({ draftSessionId: sessionId });\n        }\n        if (isMutable === false) {\n            return client.getSession({ sessionId });\n        }\n        throw new Error(\n            `Cannot resolve session \"${sessionId}\": session type not cached. ` +\n                `Ensure createSession or listSessions was called first.`,\n        );\n    }\n\n    const server: TrueFoundryChatServer = {\n        async createSession(req) {\n            if (req.agentSpec != null) {\n                const draft = await privateClient.createDraftSession({\n                    agentSpec: req.agentSpec as never,\n                    ...(req.agentName != null ? { agentName: req.agentName } : {}),\n                });\n                const session = toSession(draft);\n                cacheSessionType(session);\n                return session;\n            }\n            if (req.agentName != null) {\n                const named = await client.createSession({\n                    agentName: req.agentName,\n                });\n                const session = toSession(named);\n                cacheSessionType(session);\n                return session;\n            }\n            throw new Error(\"createSession requires agentName and/or agentSpec\");\n        },\n\n        async listSessions(req) {\n            const page = await privateClient.listOwnedSessions({\n                limit: req?.limit,\n                order: req?.order,\n                pageToken: req?.pageToken,\n                startTimestamp: req?.startTimestamp,\n                ...(req?.agentName != null ? { agentName: req.agentName } : {}),\n            });\n            const result = await toListResult(page, toSession);\n            for (const session of result.data) {\n                cacheSessionType(session);\n            }\n            return result;\n        },\n\n        async getSession({ sessionId }) {\n            const raw = await getSessionObj(sessionId);\n            const session = toSession(raw);\n            cacheSessionType(session);\n            return session;\n        },\n\n        async updateSession(req) {\n            const raw = await getSessionObj(req.sessionId);\n            if (!isDraft(raw)) {\n                throw new Error(\n                    \"updateSession: session is not mutable (isMutable=false)\",\n                );\n            }\n            if (req.agentSpec != null) {\n                await raw.update({ agentSpec: req.agentSpec as never });\n            }\n            return toSession(raw);\n        },\n\n        prepareAndExecuteTurn(req: {\n            sessionId: string;\n            input?: TurnInputItem[];\n            previousTurnId?: PreviousTurnIdInput;\n            abortSignal?: AbortSignal;\n            headers?: Record<string, string>;\n        }): AsyncIterable<TurnStreamData> {\n            return (async function* () {\n                const session = await getSessionObj(req.sessionId);\n                const prepared = session.prepareTurn({\n                    input: req.input,\n                    previousTurnId: req.previousTurnId ?? \"auto\",\n                });\n                yield* prepared.execute(\n                    { stream: true },\n                    {\n                        ...(req.abortSignal != null\n                            ? { abortSignal: req.abortSignal }\n                            : {}),\n                        ...(req.headers != null ? { headers: req.headers } : {}),\n                    },\n                ) as AsyncIterable<TurnStreamData>;\n            })();\n        },\n\n        async cancelSession({ sessionId }) {\n            await (await getSessionObj(sessionId)).cancel();\n        },\n\n        async deleteSession({ sessionId }) {\n            if (opts.deleteSession == null) {\n                throw new Error(\n                    \"deleteSession is not on the gateway SDK. Pass deleteSession to createTrueFoundryChatServer.\",\n                );\n            }\n            await opts.deleteSession({ sessionId });\n        },\n\n        async listTurns({ sessionId, limit, pageToken, order }) {\n            const raw = await getSessionObj(sessionId);\n            const page = await raw.listTurns({\n                ...(limit != null ? { limit } : {}),\n                ...(pageToken != null ? { pageToken } : {}),\n                ...(order != null ? { order } : {}),\n            });\n            return toListResult(page, (turn) => toTurn(turn));\n        },\n\n        async getTurn({ sessionId, turnId }) {\n            const raw = await getSessionObj(sessionId);\n            return toTurn(await raw.getTurn({ turnId }));\n        },\n\n        async listEvents({ sessionId, pageToken, lastTurnId, limit }) {\n            const raw = await getSessionObj(sessionId);\n            const page = await raw.listEvents({\n                ...(limit != null ? { limit } : {}),\n                ...(pageToken != null ? { pageToken } : {}),\n                ...(lastTurnId != null ? { lastTurnId } : {}),\n            });\n            return toListResult(\n                page,\n                (item) => item as SessionEventItem,\n            );\n        },\n\n        async listTurnEvents({ sessionId, turnId, limit, pageToken, order }) {\n            const raw = await getSessionObj(sessionId);\n            const turn = await raw.getTurn({ turnId });\n            const page = await turn.listEvents({\n                ...(limit != null ? { limit } : {}),\n                ...(pageToken != null ? { pageToken } : {}),\n                ...(order != null ? { order } : {}),\n            });\n            return toListResult(page, (event) => event as TurnEvent);\n        },\n\n        async *subscribeToTurn({\n            sessionId,\n            turnId,\n            afterSequenceNumber,\n            abortSignal,\n        }) {\n            const raw = await getSessionObj(sessionId);\n            const turn = await raw.getTurn({ turnId });\n            yield* turn.stream(\n                afterSequenceNumber != null ? { afterSequenceNumber } : {},\n                abortSignal != null ? { abortSignal } : {},\n            ) as AsyncIterable<TurnStreamData>;\n        },\n\n        async downloadSandboxFile(sandboxId, req) {\n            const response = await privateClient.downloadSandboxFile(\n                sandboxId,\n                req,\n            );\n            return await response.blob();\n        },\n\n        getGatewayClients: () => ({ client, privateClient }),\n    };\n\n    return server;\n}\n"],"mappings":";AAAA,SAAS,0BAA0B;AAEnC,SAAS,iCAAiC;AAoC1C,SAAS,QAAQ,SAAkD;AAC/D,SAAQ,QAA8B,SAAS;AACnD;AAEA,SAAS,UAAU,KAAyB;AACxC,QAAM,UAAU,QAAQ,GAAG;AAC3B,SAAO;AAAA,IACH,IAAI,IAAI;AAAA,IACR,OAAO,IAAI;AAAA,IACX,WAAW,IAAI;AAAA,IACf,GAAI,UAAU,EAAE,WAAW,IAAI,UAAuB,IAAI,CAAC;AAAA,IAC3D,WAAW;AAAA,IACX,WAAW,IAAI;AAAA,IACf,WAAW,IAAI;AAAA,EACnB;AACJ;AAEA,SAAS,OAAO,KAOP;AACL,SAAO;AAAA,IACH,IAAI,IAAI;AAAA,IACR,WAAW,IAAI;AAAA,IACf,gBAAgB,IAAI;AAAA,IACpB,OAAO,IAAI;AAAA,IACX,OAAO,IAAI;AAAA,IACX,WAAW,IAAI;AAAA,EACnB;AACJ;AAEA,eAAe,aACX,MAKA,KACyB;AACzB,QAAM,gBAAgB,KAAK,UAAU,YAAY;AACjD,SAAO;AAAA,IACH,MAAM,KAAK,KAAK,IAAI,GAAG;AAAA,IACvB,GAAI,iBAAiB,QAAQ,kBAAkB,KACzC,EAAE,cAAc,IAChB,CAAC;AAAA,EACX;AACJ;AAQO,SAAS,4BACZ,MACqB;AACrB,QAAM,cAAc,EAAE,QAAQ,KAAK,QAAQ,SAAS,KAAK,QAAQ;AACjE,QAAM,SAAS,KAAK,UAAU,IAAI,mBAAmB,WAAW;AAChE,QAAM,gBACF,KAAK,iBAAiB,IAAI,0BAA0B,WAAW;AAEnE,QAAM,mBAAmB,oBAAI,IAAqB;AAElD,WAAS,iBAAiB,SAAwB;AAC9C,qBAAiB,IAAI,QAAQ,IAAI,QAAQ,SAAS;AAAA,EACtD;AAEA,WAAS,cAAc,WAAuC;AAC1D,UAAM,YAAY,iBAAiB,IAAI,SAAS;AAChD,QAAI,cAAc,MAAM;AACpB,aAAO,cAAc,gBAAgB,EAAE,gBAAgB,UAAU,CAAC;AAAA,IACtE;AACA,QAAI,cAAc,OAAO;AACrB,aAAO,OAAO,WAAW,EAAE,UAAU,CAAC;AAAA,IAC1C;AACA,UAAM,IAAI;AAAA,MACN,2BAA2B,SAAS;AAAA,IAExC;AAAA,EACJ;AAEA,QAAM,SAAgC;AAAA,IAClC,MAAM,cAAc,KAAK;AACrB,UAAI,IAAI,aAAa,MAAM;AACvB,cAAM,QAAQ,MAAM,cAAc,mBAAmB;AAAA,UACjD,WAAW,IAAI;AAAA,UACf,GAAI,IAAI,aAAa,OAAO,EAAE,WAAW,IAAI,UAAU,IAAI,CAAC;AAAA,QAChE,CAAC;AACD,cAAM,UAAU,UAAU,KAAK;AAC/B,yBAAiB,OAAO;AACxB,eAAO;AAAA,MACX;AACA,UAAI,IAAI,aAAa,MAAM;AACvB,cAAM,QAAQ,MAAM,OAAO,cAAc;AAAA,UACrC,WAAW,IAAI;AAAA,QACnB,CAAC;AACD,cAAM,UAAU,UAAU,KAAK;AAC/B,yBAAiB,OAAO;AACxB,eAAO;AAAA,MACX;AACA,YAAM,IAAI,MAAM,mDAAmD;AAAA,IACvE;AAAA,IAEA,MAAM,aAAa,KAAK;AACpB,YAAM,OAAO,MAAM,cAAc,kBAAkB;AAAA,QAC/C,OAAO,KAAK;AAAA,QACZ,OAAO,KAAK;AAAA,QACZ,WAAW,KAAK;AAAA,QAChB,gBAAgB,KAAK;AAAA,QACrB,GAAI,KAAK,aAAa,OAAO,EAAE,WAAW,IAAI,UAAU,IAAI,CAAC;AAAA,MACjE,CAAC;AACD,YAAM,SAAS,MAAM,aAAa,MAAM,SAAS;AACjD,iBAAW,WAAW,OAAO,MAAM;AAC/B,yBAAiB,OAAO;AAAA,MAC5B;AACA,aAAO;AAAA,IACX;AAAA,IAEA,MAAM,WAAW,EAAE,UAAU,GAAG;AAC5B,YAAM,MAAM,MAAM,cAAc,SAAS;AACzC,YAAM,UAAU,UAAU,GAAG;AAC7B,uBAAiB,OAAO;AACxB,aAAO;AAAA,IACX;AAAA,IAEA,MAAM,cAAc,KAAK;AACrB,YAAM,MAAM,MAAM,cAAc,IAAI,SAAS;AAC7C,UAAI,CAAC,QAAQ,GAAG,GAAG;AACf,cAAM,IAAI;AAAA,UACN;AAAA,QACJ;AAAA,MACJ;AACA,UAAI,IAAI,aAAa,MAAM;AACvB,cAAM,IAAI,OAAO,EAAE,WAAW,IAAI,UAAmB,CAAC;AAAA,MAC1D;AACA,aAAO,UAAU,GAAG;AAAA,IACxB;AAAA,IAEA,sBAAsB,KAMY;AAC9B,cAAQ,mBAAmB;AACvB,cAAM,UAAU,MAAM,cAAc,IAAI,SAAS;AACjD,cAAM,WAAW,QAAQ,YAAY;AAAA,UACjC,OAAO,IAAI;AAAA,UACX,gBAAgB,IAAI,kBAAkB;AAAA,QAC1C,CAAC;AACD,eAAO,SAAS;AAAA,UACZ,EAAE,QAAQ,KAAK;AAAA,UACf;AAAA,YACI,GAAI,IAAI,eAAe,OACjB,EAAE,aAAa,IAAI,YAAY,IAC/B,CAAC;AAAA,YACP,GAAI,IAAI,WAAW,OAAO,EAAE,SAAS,IAAI,QAAQ,IAAI,CAAC;AAAA,UAC1D;AAAA,QACJ;AAAA,MACJ,GAAG;AAAA,IACP;AAAA,IAEA,MAAM,cAAc,EAAE,UAAU,GAAG;AAC/B,aAAO,MAAM,cAAc,SAAS,GAAG,OAAO;AAAA,IAClD;AAAA,IAEA,MAAM,cAAc,EAAE,UAAU,GAAG;AAC/B,UAAI,KAAK,iBAAiB,MAAM;AAC5B,cAAM,IAAI;AAAA,UACN;AAAA,QACJ;AAAA,MACJ;AACA,YAAM,KAAK,cAAc,EAAE,UAAU,CAAC;AAAA,IAC1C;AAAA,IAEA,MAAM,UAAU,EAAE,WAAW,OAAO,WAAW,MAAM,GAAG;AACpD,YAAM,MAAM,MAAM,cAAc,SAAS;AACzC,YAAM,OAAO,MAAM,IAAI,UAAU;AAAA,QAC7B,GAAI,SAAS,OAAO,EAAE,MAAM,IAAI,CAAC;AAAA,QACjC,GAAI,aAAa,OAAO,EAAE,UAAU,IAAI,CAAC;AAAA,QACzC,GAAI,SAAS,OAAO,EAAE,MAAM,IAAI,CAAC;AAAA,MACrC,CAAC;AACD,aAAO,aAAa,MAAM,CAAC,SAAS,OAAO,IAAI,CAAC;AAAA,IACpD;AAAA,IAEA,MAAM,QAAQ,EAAE,WAAW,OAAO,GAAG;AACjC,YAAM,MAAM,MAAM,cAAc,SAAS;AACzC,aAAO,OAAO,MAAM,IAAI,QAAQ,EAAE,OAAO,CAAC,CAAC;AAAA,IAC/C;AAAA,IAEA,MAAM,WAAW,EAAE,WAAW,WAAW,YAAY,MAAM,GAAG;AAC1D,YAAM,MAAM,MAAM,cAAc,SAAS;AACzC,YAAM,OAAO,MAAM,IAAI,WAAW;AAAA,QAC9B,GAAI,SAAS,OAAO,EAAE,MAAM,IAAI,CAAC;AAAA,QACjC,GAAI,aAAa,OAAO,EAAE,UAAU,IAAI,CAAC;AAAA,QACzC,GAAI,cAAc,OAAO,EAAE,WAAW,IAAI,CAAC;AAAA,MAC/C,CAAC;AACD,aAAO;AAAA,QACH;AAAA,QACA,CAAC,SAAS;AAAA,MACd;AAAA,IACJ;AAAA,IAEA,MAAM,eAAe,EAAE,WAAW,QAAQ,OAAO,WAAW,MAAM,GAAG;AACjE,YAAM,MAAM,MAAM,cAAc,SAAS;AACzC,YAAM,OAAO,MAAM,IAAI,QAAQ,EAAE,OAAO,CAAC;AACzC,YAAM,OAAO,MAAM,KAAK,WAAW;AAAA,QAC/B,GAAI,SAAS,OAAO,EAAE,MAAM,IAAI,CAAC;AAAA,QACjC,GAAI,aAAa,OAAO,EAAE,UAAU,IAAI,CAAC;AAAA,QACzC,GAAI,SAAS,OAAO,EAAE,MAAM,IAAI,CAAC;AAAA,MACrC,CAAC;AACD,aAAO,aAAa,MAAM,CAAC,UAAU,KAAkB;AAAA,IAC3D;AAAA,IAEA,OAAO,gBAAgB;AAAA,MACnB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACJ,GAAG;AACC,YAAM,MAAM,MAAM,cAAc,SAAS;AACzC,YAAM,OAAO,MAAM,IAAI,QAAQ,EAAE,OAAO,CAAC;AACzC,aAAO,KAAK;AAAA,QACR,uBAAuB,OAAO,EAAE,oBAAoB,IAAI,CAAC;AAAA,QACzD,eAAe,OAAO,EAAE,YAAY,IAAI,CAAC;AAAA,MAC7C;AAAA,IACJ;AAAA,IAEA,MAAM,oBAAoB,WAAW,KAAK;AACtC,YAAM,WAAW,MAAM,cAAc;AAAA,QACjC;AAAA,QACA;AAAA,MACJ;AACA,aAAO,MAAM,SAAS,KAAK;AAAA,IAC/B;AAAA,IAEA,mBAAmB,OAAO,EAAE,QAAQ,cAAc;AAAA,EACtD;AAEA,SAAO;AACX;","names":[]}
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/server/index.d.ts

@@ -0,0 +1,17 @@
+import { r as TurnStreamingEvent, D as DeltaEvents, c as TurnEvent } from '../types-BfiFf8O1.js';
+export { t as ActionRequiredEvent, h as AgentBuilderServer, b as AgentChatServer, u as AgentInfo, v as AgentParent, w as AgentSelectorEntry, A as AgentSpec, x as ApprovalDecision, y as ChunkDeltaToolCall, z as ConnectorSelectorEntry, C as CreateSessionRequest, i as ListResult, B as ListSessionsOrder, L as ListSessionsParams, M as McpAuthRequiredEvent, E as McpInitializeEvent, F as McpServerAuthInfo, G as McpServerMount, H as Model, I as ModelMessageContentPart, J as ModelMessageDeltaEvent, j as ModelMessageEvent, K as ModelParams, N as ModelSelectorEntry, O as PageParams, P as PreviousTurnIdInput, k as SandboxCreatedEvent, Q as SearchAgentSelectorParams, S as Session, l as SessionEventItem, R as SkillMount, V as SkillSelectorEntry, d as ThreadCreatedEvent, W as ThreadDoneEvent, m as ToolApprovalRequiredEvent, n as ToolCall, X as ToolCallFunction, Y as ToolCallRef, Z as ToolInfo, _ as ToolResponseEvent, o as ToolResponseRequiredEvent, T as Turn, $ as TurnCreatedEvent, a0 as TurnDoneEvent, g as TurnInputItem, a as TurnState, a1 as TurnStateCancelled, p as TurnStateDone, a2 as TurnStateError, a3 as TurnStateRunning, q as TurnStreamData, U as UpdateSessionRequest, s as UserMessage, a4 as UserMessageContent, f as UserToolApprovalEvent, e as UserToolResponseEvent } from '../types-BfiFf8O1.js';
+
+/**
+ * Local implementations of streaming delta helpers.
+ * Formerly imported from truefoundry-gateway-sdk/agents.
+ */
+
+/** True for `.delta` streaming events. */
+declare function isEventDelta(event: TurnStreamingEvent): event is DeltaEvents;
+/**
+ * Merge `delta` into `base` in place (same `id` required).
+ * Currently handles `model.message.delta` → `model.message`.
+ */
+declare function mergeEventDelta(base: TurnEvent, delta: DeltaEvents): void;
+
+export { DeltaEvents, TurnEvent, TurnStreamingEvent, isEventDelta, mergeEventDelta };

package/dist/server/index.js

@@ -0,0 +1,9 @@
+import {
+  isEventDelta,
+  mergeEventDelta
+} from "../chunk-3A2EPLQG.js";
+export {
+  isEventDelta,
+  mergeEventDelta
+};
+//# sourceMappingURL=index.js.map

package/dist/server/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/{types-VUBzoJT2.d.ts → types-BfiFf8O1.d.ts}

@@ -253,16 +253,22 @@ type SearchAgentSelectorParams = {
     limit?: number;
     offset?: number;
 };
-/** Skill mount base written to AgentSpec.skills[]. Host extends for fqn, preload, etc. */
-interface SkillMount {
-    id: string;
-    name: string;
-}
-/** MCP server mount base written to AgentSpec.mcpServers[]. Host extends for type, enableTools, etc. */
-interface McpServerMount {
-    id: string;
-    name: string;
-}
+/**
+ * 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 {
     maxTokens?: number;
     reasoningEffort?: string;
@@ -459,4 +465,4 @@ interface AgentBuilderServer<TSpec extends AgentSpec = AgentSpec, TModel extends
     }): Promise<void>;
 }
 
-export type { AgentSpec as A, CreateSessionRequest as C, DeltaEvents as D, ListResult as L, McpAuthRequiredEvent as M, PreviousTurnIdInput as P, Session as S, TurnStreamingEvent as T, UserToolResponseEvent as U, AgentChatServer as a, TurnEvent as b, ThreadCreatedEvent as c, UserToolApprovalEvent as d, Turn as e, TurnInputItem as f, AgentBuilderServer as g, ListSessionsParams as h, ModelMessageEvent as i, SandboxCreatedEvent as j, SessionEventItem as k, ToolApprovalRequiredEvent as l, ToolCall as m, ToolResponseRequiredEvent as n, TurnState as o, TurnStateDone as p, TurnStreamData as q, UpdateSessionRequest as r, UserMessage as s };
+export type { TurnCreatedEvent as $, AgentSpec as A, ListSessionsOrder as B, CreateSessionRequest as C, DeltaEvents as D, McpInitializeEvent as E, McpServerAuthInfo as F, McpServerMount as G, Model as H, ModelMessageContentPart as I, ModelMessageDeltaEvent as J, ModelParams as K, ListSessionsParams as L, McpAuthRequiredEvent as M, ModelSelectorEntry as N, PageParams as O, PreviousTurnIdInput as P, SearchAgentSelectorParams as Q, SkillMount as R, Session as S, Turn as T, UpdateSessionRequest as U, SkillSelectorEntry as V, ThreadDoneEvent as W, ToolCallFunction as X, ToolCallRef as Y, ToolInfo as Z, ToolResponseEvent as _, TurnState as a, TurnDoneEvent as a0, TurnStateCancelled as a1, TurnStateError as a2, TurnStateRunning as a3, UserMessageContent as a4, AgentChatServer as b, TurnEvent as c, ThreadCreatedEvent as d, UserToolResponseEvent as e, UserToolApprovalEvent as f, TurnInputItem as g, AgentBuilderServer as h, ListResult as i, ModelMessageEvent as j, SandboxCreatedEvent as k, SessionEventItem as l, ToolApprovalRequiredEvent as m, ToolCall as n, ToolResponseRequiredEvent as o, TurnStateDone as p, TurnStreamData as q, TurnStreamingEvent as r, UserMessage as s, ActionRequiredEvent as t, AgentInfo as u, AgentParent as v, AgentSelectorEntry as w, ApprovalDecision as x, ChunkDeltaToolCall as y, ConnectorSelectorEntry as z };

package/package.json

@@ -1,11 +1,11 @@
 {
   "name": "@truefoundry/assistant-ui-runtime",
-  "version": "0.1.6-rc.0",
+  "version": "0.1.6",
   "description": "TrueFoundry Gateway agent runtime adapter for assistant-ui",
   "license": "Apache-2.0",
   "repository": {
     "type": "git",
-    "url": "https://github.com/truefoundry/truefoundry-agents-assistant-ui-runtime"
+    "url": "git+https://github.com/truefoundry/truefoundry-agents-assistant-ui-runtime.git"
   },
   "homepage": "https://github.com/truefoundry/truefoundry-agents-assistant-ui-runtime#readme",
   "keywords": [
@@ -27,6 +27,11 @@
       "import": "./dist/index.js",
       "default": "./dist/index.js"
     },
+    "./server": {
+      "types": "./dist/server/index.d.ts",
+      "import": "./dist/server/index.js",
+      "default": "./dist/server/index.js"
+    },
     "./plugins/truefoundry-agent-server-adapter": {
       "types": "./dist/plugins/truefoundry-agent-server-adapter/index.d.ts",
       "import": "./dist/plugins/truefoundry-agent-server-adapter/index.js",
@@ -55,7 +60,7 @@
   "peerDependencies": {
     "@types/react": "*",
     "react": "^18 || ^19",
-    "truefoundry-gateway-sdk": "^0.4.0-rc.5"
+    "truefoundry-gateway-sdk": "^0.4.0-rc.6"
   },
   "peerDependenciesMeta": {
     "@types/react": {
@@ -71,9 +76,9 @@
     "@types/react": "^19.2.17",
     "jsdom": "^29.1.1",
     "react": "^19.2.4",
-    "truefoundry-gateway-sdk": "0.4.0-rc.5",
+    "truefoundry-gateway-sdk": "0.4.0-rc.6",
     "tsup": "^8.5.0",
     "typescript": "^5.9.3",
     "vitest": "4.1.9"
   }
-}
+}

package/src/index.ts

@@ -106,3 +106,44 @@ export type {
     ToolResponseRequiredEvent,
 } from "./server/index.js";
 export { isEventDelta, mergeEventDelta } from "./server/index.js";
+
+// ---------------------------------------------------------------------------
+// Plugin: truefoundry-agent-server-adapter
+// ---------------------------------------------------------------------------
+
+export {
+    createTrueFoundryChatServer,
+    type CreateTrueFoundryChatServerOptions,
+    type TrueFoundryChatServer,
+    type TfyAgentSpec,
+    type TfySkillMount,
+    type TfyMcpServerMount,
+    type TfyModelParams,
+    type TfyRuntimeConfig,
+    type TfyResponseFormat,
+    type TfySubject,
+    type ToolsSelectorItem,
+    type ToolsSelectorTag,
+    type RequireApprovalToolSelectorItem,
+    type RequireApprovalToolsSelectorTag,
+    type TfyTurn,
+    type TfyTurnState,
+    type TfyTurnCancelledReason,
+    type TfyTurnStateDoneOutput,
+    type TfySession,
+    type TfyCreateSessionRequest,
+    type TfyListSessionsParams,
+    type TfyToolInfo,
+    type TfySystemToolInfo,
+    type TfyMcpToolInfo,
+    type TfyModelMessageUsage,
+    type TfyFinishReason,
+    type TfyThreadState,
+    type TfyMcpServerInitInfo,
+    isTfyToolInfo,
+    isTfySystemToolInfo,
+    isTfyMcpToolInfo,
+    getTfyUsage,
+    getTfyThreadState,
+    getTfyMcpInitServers,
+} from "./plugins/truefoundry-agent-server-adapter/index.js";