@truefoundry/trueforge-ui 0.2.4 → 0.3.0-rc.1

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

@@ -1,195 +1,114 @@
 import {
   createTrueFoundryServer
-} from "../../chunk-CFLVPV5M.js";
+} from "../../chunk-QC42MLAK.js";
+import {
+  drainListPages
+} from "../../chunk-SFZNIO6U.js";
+import {
+  readSessionIsCreateAgent
+} from "../../chunk-RDXBMTP5.js";
 
-// src/plugins/trueforge-agent-server-adapter/catalogs/connectorCatalog.ts
-var DEFAULT_API_KEY_HEADER = "Authorization";
-function toUiAuthPublic(auth) {
-  if (auth === void 0) {
-    return { type: "none" };
-  }
-  if (auth.type === "dcr") {
-    return { type: "dcr" };
-  }
-  const headerName = Object.keys(auth.headers)[0];
-  return {
-    type: "header",
-    ...headerName === void 0 ? {} : { headerName }
-  };
-}
-function toHarnessAuth(auth) {
-  if (auth.type === "none") {
-    return void 0;
-  }
-  if (auth.type === "dcr") {
-    return { type: "dcr" };
-  }
-  const apiKey = auth.apiKey?.trim();
-  if (apiKey === void 0 || apiKey === "") {
-    throw new Error("API key is required for header-authenticated MCP servers");
+// src/plugins/trueforge-agent-server-adapter/client.ts
+import { TrueForge } from "@truefoundry/trueforge-sdk";
+var DEFAULT_BASE_URL = "/";
+function resolveTrueForgeBaseUrl(baseUrl) {
+  if (/^https?:\/\//i.test(baseUrl)) return baseUrl;
+  const origin = globalThis.location?.origin;
+  if (typeof origin === "string" && origin.length > 0 && origin !== "null") {
+    return new URL(baseUrl, origin).href;
   }
-  const trimmedHeader = auth.headerName?.trim();
-  const headerName = trimmedHeader !== void 0 && trimmedHeader !== "" ? trimmedHeader : DEFAULT_API_KEY_HEADER;
-  return { type: "header", headers: { [headerName]: apiKey } };
-}
-function toUiCatalogEntry(server) {
-  return {
-    id: server.name,
-    name: server.name,
-    url: server.url,
-    description: server.description,
-    ...server.logo === void 0 ? {} : { logo: server.logo },
-    auth: toUiAuthPublic(server.auth)
-  };
+  return baseUrl;
 }
-function toUiTool(tool) {
-  const name = typeof tool.name === "string" && tool.name !== "" ? tool.name : "tool";
-  const description = typeof tool.description === "string" ? tool.description : "";
-  return { id: name, name, description };
-}
-function toUiConnector(server) {
-  const auth = toUiAuthPublic(server.manifest.auth);
-  return {
-    id: server.name,
-    name: server.name,
-    description: server.manifest.description,
-    url: server.manifest.url,
-    auth,
-    requiresAuth: server.authStatus.status === "auth_required",
-    authenticated: server.authStatus.status !== "auth_required"
-  };
-}
-function toUiConnectorFromReadEntry(server) {
-  const auth = server.auth?.type ? { type: server.auth.type } : { type: "none" };
-  return {
-    id: server.name,
-    name: server.name,
-    description: server.url,
-    url: server.url,
-    auth,
-    requiresAuth: server.authStatus.status === "auth_required",
-    authenticated: server.authStatus.status !== "auth_required"
-  };
-}
-function toHarnessManifest(req) {
-  const auth = toHarnessAuth(req.auth);
-  const trimmed = req.description?.trim();
-  return {
-    type: "remote",
-    name: req.name,
-    url: req.url,
-    description: trimmed !== void 0 && trimmed !== "" ? trimmed : `${req.name} MCP server`,
-    ...auth === void 0 ? {} : { auth }
-  };
+function createTrueForgeClient(options = {}) {
+  return new TrueForge({
+    baseUrl: resolveTrueForgeBaseUrl(options.baseUrl ?? DEFAULT_BASE_URL),
+    ...options.token === void 0 ? {} : { token: options.token },
+    ...options.fetch === void 0 ? {} : { fetch: options.fetch }
+  });
 }
-function createConnectorCatalog(client) {
-  async function getConfigured(name) {
-    const listed = await client.settings.mcpServers.list();
-    const existing = listed.data.find((server) => server.name === name);
-    if (existing === void 0) {
-      throw new Error(`MCP server "${name}" not found`);
-    }
-    return existing;
+function parseIsoDate(value) {
+  const date = new Date(value);
+  if (Number.isNaN(date.getTime())) {
+    throw new Error(`Invalid ISO timestamp: ${value}`);
   }
-  async function resolveWriteAuth(req) {
-    if (req.auth.type !== "header") {
-      return req.auth;
-    }
-    const apiKey = req.auth.apiKey?.trim();
-    if (apiKey !== void 0 && apiKey !== "") {
-      return req.auth;
-    }
-    if (req.id === void 0) {
-      throw new Error("API key is required for header-authenticated MCP servers");
-    }
-    const existing = await getConfigured(req.id);
-    if (existing.manifest.auth?.type !== "header") {
-      throw new Error(`MCP server "${req.id}" has no stored header credentials to reuse`);
-    }
-    const preferredHeader = req.auth.headerName?.trim();
-    const storedHeaderName = Object.keys(existing.manifest.auth.headers)[0];
-    const headerName = preferredHeader !== void 0 && preferredHeader !== "" ? preferredHeader : storedHeaderName ?? DEFAULT_API_KEY_HEADER;
-    const stored = existing.manifest.auth.headers[headerName] ?? Object.values(existing.manifest.auth.headers)[0];
-    if (stored === void 0) {
-      throw new Error(`MCP server "${req.id}" has no stored header credentials to reuse`);
-    }
-    return { type: "header", apiKey: stored, headerName };
+  return date;
+}
+
+// src/plugins/trueforge-agent-server-adapter/agentMetricsServer.ts
+function toHarnessChartName(name) {
+  if (name === "sessions_over_time" || name === "sessions_cost_over_time" || name === "turns_over_time") {
+    return name;
   }
+  throw new Error(`Unsupported Harness metrics chart: ${name}`);
+}
+function createHarnessAgentMetricsServer(options = {}) {
+  const client = options.client ?? createTrueForgeClient(options);
   return {
-    getConnectorCatalog: async () => {
-      const body = await client.catalogs.mcpServers.list();
-      return body.data.map(toUiCatalogEntry);
+    async getCharts() {
+      const { data } = await client.internal.metrics.listCharts();
+      return data.charts;
     },
-    getConnector: async (req) => {
-      const body = await client.settings.mcpServers.get(req.id);
-      return toUiConnector(body.data);
-    },
-    listConnectors: async (req) => {
-      const body = await client.settings.mcpServers.list();
-      const connectors = body.data.map(toUiConnector);
-      const query = req?.query?.trim().toLowerCase();
-      if (query === void 0 || query === "") {
-        return connectors;
-      }
-      return connectors.filter(
-        (connector) => connector.name.toLowerCase().includes(query) || connector.description.toLowerCase().includes(query) || connector.url.toLowerCase().includes(query)
-      );
-    },
-    getToolsByConnectorId: async ({ id }) => {
-      const body = await client.mcpServers.listTools(id);
-      return body.data.map(toUiTool);
-    },
-    createConnector: async (req) => {
-      const auth = await resolveWriteAuth({ auth: req.auth });
-      const body = await client.settings.mcpServers.create({
-        manifest: toHarnessManifest({
-          name: req.name,
-          url: req.url,
-          auth,
-          description: req.description
-        })
+    async getMeters({ agentId, startTimestamp, endTimestamp }) {
+      const { data } = await client.internal.metrics.getMeters({
+        agentId,
+        startTimestamp: parseIsoDate(startTimestamp),
+        endTimestamp: parseIsoDate(endTimestamp)
       });
-      return toUiConnector(body.data);
+      return data.meters;
     },
-    updateConnector: async (req) => {
-      const auth = await resolveWriteAuth({ id: req.id, auth: req.auth });
-      const body = await client.settings.mcpServers.createOrUpdate({
-        manifest: toHarnessManifest({
-          name: req.id,
-          url: req.url,
-          auth,
-          description: req.description
-        })
+    async getChartData({ agentId, startTimestamp, endTimestamp, chartName }) {
+      const { data } = await client.internal.metrics.getChartData({
+        agentId,
+        startTimestamp: parseIsoDate(startTimestamp),
+        endTimestamp: parseIsoDate(endTimestamp),
+        chartName: toHarnessChartName(chartName)
       });
-      return toUiConnector(body.data);
-    },
-    authenticateConnector: async (req) => {
-      const result = await client.mcpServers.authorize(
-        req.id,
-        req.returnTo === void 0 ? {} : { returnTo: req.returnTo }
-      );
-      return { status: result.status, authorization_endpoint: result.authorizationUrl };
-    },
-    disconnectConnector: async (req) => {
-      const existing = await getConfigured(req.id);
-      if (existing.manifest.auth?.type !== "dcr") {
-        throw new Error(`Disconnect is only supported for OAuth MCP servers`);
-      }
-      const body = await client.mcpServers.deleteAuthorization(req.id);
-      return toUiConnector(body.data);
+      return data;
     }
   };
 }
 
-// src/plugins/trueforge-agent-server-adapter/client.ts
-import { TrueForge } from "@truefoundry/trueforge-sdk";
-var DEFAULT_BASE_URL = "/";
-function createTrueForgeClient(options = {}) {
-  return new TrueForge({
-    baseUrl: options.baseUrl ?? DEFAULT_BASE_URL,
-    ...options.token === void 0 ? {} : { token: options.token },
-    ...options.fetch === void 0 ? {} : { fetch: options.fetch }
-  });
+// src/plugins/trueforge-agent-server-adapter/toUiTurnState.ts
+function toUiTurnDoneMetrics(metrics) {
+  return {
+    totalInputTokens: metrics.totalInputTokens ?? 0,
+    totalOutputTokens: metrics.totalOutputTokens ?? 0,
+    totalTokens: metrics.totalTokens ?? 0,
+    totalCacheReadTokens: metrics.totalCacheReadTokens ?? 0,
+    totalCacheWriteTokens: metrics.totalCacheWriteTokens ?? 0,
+    totalReasoningTokens: metrics.totalReasoningTokens ?? 0,
+    ...metrics.totalCostInUsd == null ? {} : { totalCostInUsd: metrics.totalCostInUsd }
+  };
+}
+function toUiTurnState(state) {
+  return state.status === "running" ? { status: "running" } : toUiTerminalTurnState(state);
+}
+function toUiTerminalTurnState(state) {
+  switch (state.status) {
+    case "cancelled":
+      return { status: "cancelled", reason: state.reason, completedAt: state.completedAt };
+    case "error":
+      return { status: "error", message: state.message, completedAt: state.completedAt };
+    case "done":
+      return {
+        status: "done",
+        completedAt: state.completedAt,
+        requiredActions: state.requiredActions,
+        ...state.output == null ? {} : { output: state.output },
+        ...state.metrics == null ? {} : { metrics: toUiTurnDoneMetrics(state.metrics) }
+      };
+  }
+}
+function toUiSessionEvent(event) {
+  if (event.type !== "turn.done") return { ...event };
+  return { ...event, state: toUiTerminalTurnState(event.state) };
+}
+function toUiStreamingEvent(event) {
+  if (event.type !== "turn.done") return { ...event };
+  return { ...event, state: toUiTerminalTurnState(event.state) };
+}
+function toUiEventItem(item) {
+  return { turnId: item.turnId, event: toUiSessionEvent(item.event) };
 }
 
 // src/plugins/trueforge-agent-server-adapter/chatServer.ts
@@ -228,6 +147,7 @@ function toUiSession(session) {
   return {
     id: session.id,
     isMutable: session.agent.type === "inline",
+    isCreateAgent: readSessionIsCreateAgent(session.metadata),
     createdAt: session.createdAt,
     updatedAt: session.updatedAt,
     ...session.title === null ? {} : { title: session.title },
@@ -237,6 +157,10 @@ function toUiSession(session) {
     ...session.agent.type === "inline" ? { agentSpec: toUiAgentSpec(session.agent.spec) } : {}
   };
 }
+function createSessionMetadata(request) {
+  if (request.metadata === void 0) return void 0;
+  return request.metadata;
+}
 function toUiContent(content) {
   return typeof content === "string" ? content : content.map((part) => ({ ...part }));
 }
@@ -244,16 +168,14 @@ function toUiInput(input) {
   return input.map((item) => item.type === "user.message" ? { ...item, content: toUiContent(item.content) } : item);
 }
 function toUiTurn(turn) {
-  const { previousTurnId, input, ...rest } = turn;
+  const { previousTurnId, input, state, ...rest } = turn;
   return {
     ...rest,
+    state: toUiTurnState(state),
     ...previousTurnId === null ? {} : { previousTurnId },
     ...input === void 0 ? {} : { input: toUiInput(input) }
   };
 }
-function toUiEventItem(item) {
-  return { turnId: item.turnId, event: { ...item.event } };
-}
 function toListResult(page, map) {
   const data = page.data.map(map);
   const token = page.response.pagination.nextPageToken;
@@ -293,15 +215,18 @@ function createHarnessChatServer(options = {}) {
       return response.blob();
     },
     async createSession(request) {
+      const metadata = createSessionMetadata(request);
       if (request.agentName !== void 0 && request.agentName.length > 0) {
         const created = await client.sessions.create({
-          agent: { name: request.agentName }
+          agent: { name: request.agentName },
+          ...metadata === void 0 ? {} : { metadata }
         });
         return toUiSession(created.data);
       }
       if (request.agentSpec !== void 0) {
         const created = await client.sessions.create({
-          agent: { spec: toHarnessAgentSpec(request.agentSpec) }
+          agent: { spec: toHarnessAgentSpec(request.agentSpec) },
+          ...metadata === void 0 ? {} : { metadata }
         });
         return toUiSession(created.data);
       }
@@ -320,6 +245,9 @@ function createHarnessChatServer(options = {}) {
       const response = await client.sessions.get(sessionId);
       return toUiSession(response.data);
     },
+    async deleteSession({ sessionId }) {
+      await client.sessions.delete(sessionId);
+    },
     async updateSession({ sessionId, agentSpec }) {
       const response = await client.sessions.update(sessionId, {
         ...agentSpec === void 0 ? {} : { agent: { spec: toHarnessAgentSpec(agentSpec) } }
@@ -339,7 +267,7 @@ function createHarnessChatServer(options = {}) {
       for await (const item of stream.withMetadata()) {
         yield {
           sequenceNumber: sequenceNumber(item.id, fallbackSequence),
-          event: { ...item.data }
+          event: toUiStreamingEvent(item.data)
         };
         fallbackSequence += 1;
       }
@@ -357,7 +285,7 @@ function createHarnessChatServer(options = {}) {
       for await (const item of stream.withMetadata()) {
         yield {
           sequenceNumber: sequenceNumber(item.id, fallbackSequence),
-          event: { ...item.data }
+          event: toUiStreamingEvent(item.data)
         };
         fallbackSequence += 1;
       }
@@ -387,14 +315,256 @@ function createHarnessChatServer(options = {}) {
   };
 }
 
+// src/plugins/trueforge-agent-server-adapter/agentSessionsServer.ts
+function toSessionListEntry(session) {
+  return {
+    id: session.id,
+    createdAt: session.createdAt,
+    updatedAt: session.updatedAt,
+    lastActivityAt: session.updatedAt,
+    isCreateAgent: readSessionIsCreateAgent(session.metadata),
+    isMutable: session.agent.type === "inline",
+    metrics: {
+      totalTurns: session.metrics.totalTurns,
+      totalCostInUsd: session.metrics.totalCostInUsd,
+      totalDurationMs: session.metrics.totalDurationMs
+    },
+    ...session.title === null ? {} : { title: session.title },
+    ...session.agent.type === "reference" && session.agent.name !== null ? { agentName: session.agent.name } : {},
+    ...session.agent.type === "inline" ? { agentSpec: toUiAgentSpec(session.agent.spec) } : {}
+  };
+}
+function createHarnessAgentSessionsServer(options = {}) {
+  const client = options.client ?? createTrueForgeClient(options);
+  return {
+    async getAgent({ agentId }) {
+      const { data } = await client.agents.get(agentId);
+      return {
+        agentId: data.id,
+        name: data.name,
+        agentSpec: toUiAgentSpec(data.manifest)
+      };
+    },
+    async getCodeSnippets({ agentId }) {
+      const { data } = await client.internal.agents.getCodeSnippets(agentId);
+      return data.snippets;
+    },
+    async listSessions(requestParams = {}) {
+      const page = await client.sessions.list({
+        ...requestParams.limit === void 0 ? {} : { limit: requestParams.limit },
+        ...requestParams.order === void 0 ? {} : { order: requestParams.order },
+        ...requestParams.pageToken === void 0 ? {} : { pageToken: requestParams.pageToken },
+        ...requestParams.startTimestamp === void 0 ? {} : { startTimestamp: parseIsoDate(requestParams.startTimestamp) },
+        ...requestParams.endTimestamp === void 0 ? {} : { endTimestamp: parseIsoDate(requestParams.endTimestamp) },
+        ...requestParams.agentId === void 0 || requestParams.agentId.length === 0 ? {} : { agentId: requestParams.agentId }
+      });
+      return toListResult(page, toSessionListEntry);
+    },
+    async listSessionEvents({ sessionId, pageToken, limit }) {
+      const page = await client.sessions.listEvents(sessionId, {
+        ...pageToken === void 0 ? {} : { pageToken },
+        ...limit === void 0 ? {} : { limit }
+      });
+      return toListResult(page, toUiEventItem);
+    }
+  };
+}
+
+// src/plugins/trueforge-agent-server-adapter/catalogs/connectorCatalog.ts
+var DEFAULT_API_KEY_HEADER = "Authorization";
+function toUiAuthPublic(auth) {
+  if (auth === void 0) {
+    return { type: "none" };
+  }
+  if (auth.type === "dcr") {
+    return { type: "dcr" };
+  }
+  const headerName = Object.keys(auth.headers)[0];
+  return {
+    type: "header",
+    ...headerName === void 0 ? {} : { headerName }
+  };
+}
+function toHarnessAuth(auth) {
+  if (auth.type === "none") {
+    return void 0;
+  }
+  if (auth.type === "dcr") {
+    return { type: "dcr" };
+  }
+  const apiKey = auth.apiKey?.trim();
+  if (apiKey === void 0 || apiKey === "") {
+    throw new Error("API key is required for header-authenticated MCP servers");
+  }
+  const trimmedHeader = auth.headerName?.trim();
+  const headerName = trimmedHeader !== void 0 && trimmedHeader !== "" ? trimmedHeader : DEFAULT_API_KEY_HEADER;
+  return { type: "header", headers: { [headerName]: apiKey } };
+}
+function toUiCatalogEntry(server) {
+  return {
+    id: server.name,
+    name: server.name,
+    url: server.url,
+    description: server.description,
+    ...server.logo === void 0 ? {} : { logo: server.logo },
+    auth: toUiAuthPublic(server.auth)
+  };
+}
+function toUiTool(tool) {
+  const name = typeof tool.name === "string" && tool.name !== "" ? tool.name : "tool";
+  const description = typeof tool.description === "string" ? tool.description : "";
+  return { id: name, name, description };
+}
+function toUiConnector(server) {
+  const auth = toUiAuthPublic(server.manifest.auth);
+  return {
+    id: server.name,
+    name: server.name,
+    description: server.manifest.description,
+    url: server.manifest.url,
+    auth,
+    requiresAuth: server.authStatus.status === "auth_required",
+    authenticated: server.authStatus.status !== "auth_required"
+  };
+}
+function toUiConnectorFromReadEntry(server) {
+  const auth = server.auth?.type ? { type: server.auth.type } : { type: "none" };
+  return {
+    id: server.name,
+    name: server.name,
+    description: server.url,
+    url: server.url,
+    auth,
+    requiresAuth: server.authStatus.status === "auth_required",
+    authenticated: server.authStatus.status !== "auth_required"
+  };
+}
+function toHarnessManifest(req) {
+  const auth = toHarnessAuth(req.auth);
+  const trimmed = req.description?.trim();
+  return {
+    type: "remote",
+    name: req.name,
+    url: req.url,
+    description: trimmed !== void 0 && trimmed !== "" ? trimmed : `${req.name} MCP server`,
+    ...auth === void 0 ? {} : { auth }
+  };
+}
+function createConnectorCatalog(client) {
+  async function getConfigured(name) {
+    const listed = await client.settings.mcpServers.list();
+    const existing = listed.data.find((server) => server.name === name);
+    if (existing === void 0) {
+      throw new Error(`MCP server "${name}" not found`);
+    }
+    return existing;
+  }
+  async function resolveWriteAuth(req) {
+    if (req.auth.type !== "header") {
+      return req.auth;
+    }
+    const apiKey = req.auth.apiKey?.trim();
+    if (apiKey !== void 0 && apiKey !== "") {
+      return req.auth;
+    }
+    if (req.id === void 0) {
+      throw new Error("API key is required for header-authenticated MCP servers");
+    }
+    const existing = await getConfigured(req.id);
+    if (existing.manifest.auth?.type !== "header") {
+      throw new Error(`MCP server "${req.id}" has no stored header credentials to reuse`);
+    }
+    const preferredHeader = req.auth.headerName?.trim();
+    const storedHeaderName = Object.keys(existing.manifest.auth.headers)[0];
+    const headerName = preferredHeader !== void 0 && preferredHeader !== "" ? preferredHeader : storedHeaderName ?? DEFAULT_API_KEY_HEADER;
+    const stored = existing.manifest.auth.headers[headerName] ?? Object.values(existing.manifest.auth.headers)[0];
+    if (stored === void 0) {
+      throw new Error(`MCP server "${req.id}" has no stored header credentials to reuse`);
+    }
+    return { type: "header", apiKey: stored, headerName };
+  }
+  return {
+    getConnectorCatalog: async () => {
+      const body = await client.catalogs.mcpServers.list();
+      return body.data.map(toUiCatalogEntry);
+    },
+    getConnector: async (req) => {
+      const body = await client.settings.mcpServers.get(req.id);
+      return toUiConnector(body.data);
+    },
+    listConnectors: async (req) => {
+      const body = await client.settings.mcpServers.list();
+      const connectors = body.data.map(toUiConnector);
+      const query = req?.query?.trim().toLowerCase();
+      if (query === void 0 || query === "") {
+        return connectors;
+      }
+      return connectors.filter(
+        (connector) => connector.name.toLowerCase().includes(query) || connector.description.toLowerCase().includes(query) || connector.url.toLowerCase().includes(query)
+      );
+    },
+    getToolsByConnectorId: async ({ id }) => {
+      const body = await client.mcpServers.listTools(id);
+      return body.data.map(toUiTool);
+    },
+    createConnector: async (req) => {
+      const auth = await resolveWriteAuth({ auth: req.auth });
+      const body = await client.settings.mcpServers.create({
+        manifest: toHarnessManifest({
+          name: req.name,
+          url: req.url,
+          auth,
+          description: req.description
+        })
+      });
+      return toUiConnector(body.data);
+    },
+    updateConnector: async (req) => {
+      const auth = await resolveWriteAuth({ id: req.id, auth: req.auth });
+      const body = await client.settings.mcpServers.createOrUpdate({
+        manifest: toHarnessManifest({
+          name: req.id,
+          url: req.url,
+          auth,
+          description: req.description
+        })
+      });
+      return toUiConnector(body.data);
+    },
+    authenticateConnector: async (req) => {
+      const result = await client.mcpServers.authorize(
+        req.id,
+        req.returnTo === void 0 ? {} : { returnTo: req.returnTo }
+      );
+      return { status: result.status, authorization_endpoint: result.authorizationUrl };
+    },
+    disconnectConnector: async (req) => {
+      const existing = await getConfigured(req.id);
+      if (existing.manifest.auth?.type !== "dcr") {
+        throw new Error(`Disconnect is only supported for OAuth MCP servers`);
+      }
+      const body = await client.mcpServers.deleteAuthorization(req.id);
+      return toUiConnector(body.data);
+    }
+  };
+}
+
 // src/plugins/trueforge-agent-server-adapter/lists.ts
 async function listModels(client) {
   const body = await client.models.list();
   return body.data;
 }
+async function listConfiguredMcpServersPage(client, req = {}) {
+  const page = await client.mcpServers.list({
+    ...req.limit === void 0 ? {} : { limit: req.limit },
+    ...req.pageToken === void 0 ? {} : { pageToken: req.pageToken }
+  });
+  return toListResult(page, (server) => server);
+}
 async function listConfiguredMcpServers(client) {
-  const body = await client.mcpServers.list();
-  return body.data;
+  return drainListPages({
+    fetchPage: (pageToken) => listConfiguredMcpServersPage(client, { pageToken })
+  });
 }
 async function listSkills(client) {
   const body = await client.skills.list();
@@ -420,7 +590,7 @@ function toModelSelection({
   model,
   logo
 }) {
-  const efforts = model.properties.reasoningEfforts;
+  const { contextLength, maxOutputTokens, reasoningEfforts } = model.properties;
   return {
     id: model.modelId,
     name: model.name,
@@ -429,7 +599,9 @@ function toModelSelection({
       ...logo === void 0 ? {} : { logo }
     },
     properties: {
-      ...efforts !== void 0 && efforts.length > 0 ? { reasoningEfforts: [...efforts] } : {}
+      ...contextLength === void 0 ? {} : { contextLength },
+      ...maxOutputTokens === void 0 ? {} : { maxOutputTokens },
+      ...reasoningEfforts === void 0 ? {} : { reasoningEfforts: [...reasoningEfforts] }
     }
   };
 }
@@ -464,6 +636,19 @@ function createHarnessBuilderServer(options = {}) {
       return skills.map((skill) => ({ id: skill.name, name: skill.name, description: skill.description }));
     },
     getMcp: async () => (await listConfiguredMcpServers(client)).map(toUiConnectorFromReadEntry),
+    listMcp: async (req) => {
+      const page = await listConfiguredMcpServersPage(client, req ?? {});
+      return {
+        data: page.data.map(toUiConnectorFromReadEntry),
+        ...page.nextPageToken === void 0 ? {} : { nextPageToken: page.nextPageToken }
+      };
+    },
+    getMcpTools: async ({ connectorId }) => {
+      const body = await client.mcpServers.listTools(connectorId);
+      return body.data.flatMap(
+        (tool) => typeof tool.name === "string" && tool.name.trim() !== "" ? [toUiTool(tool)] : []
+      );
+    },
     async searchAgents(req) {
       const { data } = await client.agents.list();
       const query = req?.query?.trim().toLowerCase();
@@ -485,6 +670,12 @@ function createHarnessBuilderServer(options = {}) {
       }
       const created = await client.agents.create({ name: agentName, manifest });
       return { agentId: created.data.id };
+    },
+    async deleteAgent({ agentName }) {
+      const { data } = await client.agents.list();
+      const existing = data.find((agent) => agent.name === agentName);
+      if (!existing) return;
+      await client.agents.delete(existing.id);
     }
   };
 }
@@ -814,6 +1005,126 @@ function createSkillCatalog(client) {
   };
 }
 
+// src/plugins/trueforge-agent-server-adapter/schedules/scheduleServer.ts
+var SCHEDULES_PAGE_LIMIT = 25;
+async function loadAgentIndex(client) {
+  const { data } = await client.agents.list();
+  const idToName = /* @__PURE__ */ new Map();
+  const nameToId = /* @__PURE__ */ new Map();
+  for (const agent of data) {
+    idToName.set(agent.id, agent.name);
+    nameToId.set(agent.name, agent.id);
+  }
+  return { idToName, nameToId };
+}
+function resolveAgentName(agentId, index) {
+  const byId = index.idToName.get(agentId);
+  if (byId != null) return byId;
+  if (index.nameToId.has(agentId)) return agentId;
+  throw new Error(`Unknown agent: ${agentId}`);
+}
+function collectAgentIds(req) {
+  const ids = [];
+  if (req?.agentIds != null) {
+    for (const id of req.agentIds) {
+      if (id !== "") ids.push(id);
+    }
+  }
+  if (req?.agentId != null && req.agentId !== "") ids.push(req.agentId);
+  return [...new Set(ids)];
+}
+function wireStatus(manifest) {
+  return manifest.status ?? "active";
+}
+function toUiSchedule(wire, index) {
+  const agentId = index.nameToId.get(wire.agentName) ?? wire.agentName;
+  return {
+    id: wire.id,
+    name: wire.name,
+    agentId,
+    agentName: wire.agentName,
+    task: wire.manifest.task,
+    cron: wire.manifest.cron,
+    timezone: wire.manifest.timezone ?? "UTC",
+    status: wireStatus(wire.manifest),
+    lastRunAt: null
+  };
+}
+function toWireManifest(input) {
+  return {
+    task: input.task,
+    cron: input.cron,
+    timezone: input.timezone ?? "UTC",
+    status: input.status ?? "active"
+  };
+}
+function toIsoInstant(value) {
+  if (value instanceof Date) return value.toISOString();
+  return value;
+}
+function toUiScheduleRun(wire) {
+  return {
+    id: wire.id,
+    scheduleId: wire.scheduleId,
+    name: wire.name,
+    scheduledFor: toIsoInstant(wire.scheduledFor),
+    status: wire.status,
+    triggeredAt: wire.triggeredAt == null ? null : toIsoInstant(wire.triggeredAt),
+    triggeredBy: wire.createdBySubject.subjectDisplayName || wire.createdBySubject.subjectId
+  };
+}
+function createScheduleServer(options) {
+  const { client } = options;
+  return {
+    async listSchedules(req) {
+      const index = await loadAgentIndex(client);
+      const agentIds = collectAgentIds(req);
+      const agentNames = agentIds.length === 0 ? void 0 : agentIds.map((id) => resolveAgentName(id, index)).join(",");
+      const limit = Math.min(Math.max(req?.limit ?? SCHEDULES_PAGE_LIMIT, 1), SCHEDULES_PAGE_LIMIT);
+      const page = await client.schedules.list({
+        limit,
+        ...req?.pageToken === void 0 || req.pageToken === "" ? {} : { pageToken: req.pageToken },
+        ...agentNames === void 0 ? {} : { agentNames }
+      });
+      return toListResult(page, (row) => toUiSchedule(row, index));
+    },
+    async getSchedule({ id }) {
+      const index = await loadAgentIndex(client);
+      const { data } = await client.schedules.get(id);
+      return toUiSchedule(data, index);
+    },
+    async createSchedule(req) {
+      const index = await loadAgentIndex(client);
+      const agentName = resolveAgentName(req.agentId, index);
+      const { data } = await client.schedules.create({
+        agentName,
+        name: req.name,
+        manifest: toWireManifest(req)
+      });
+      return toUiSchedule(data, index);
+    },
+    async updateSchedule(req) {
+      const index = await loadAgentIndex(client);
+      const { data } = await client.schedules.update(req.id, {
+        name: req.name,
+        manifest: toWireManifest(req)
+      });
+      return toUiSchedule(data, index);
+    },
+    async deleteSchedule({ id }) {
+      await client.schedules.delete(id);
+    },
+    async listScheduleRuns({ scheduleId }) {
+      const { data } = await client.schedules.listRuns(scheduleId);
+      return data.map(toUiScheduleRun);
+    },
+    async createScheduleRun(req) {
+      const { data } = await client.schedules.createRun({ scheduleId: req.scheduleId });
+      return toUiScheduleRun(data);
+    }
+  };
+}
+
 // src/plugins/trueforge-agent-server-adapter/index.ts
 function createTrueForgeAgentUIServer(options = {}) {
   const { catalog: catalogOverride, ...clientOptions } = options;
@@ -827,22 +1138,29 @@ function createTrueForgeAgentUIServer(options = {}) {
   return createTrueFoundryServer({
     chatServer: createHarnessChatServer({ client }),
     ...createHarnessBuilderServer({ client }),
-    catalog
+    catalog,
+    sessions: createHarnessAgentSessionsServer({ ...clientOptions, client }),
+    metrics: createHarnessAgentMetricsServer({ ...clientOptions, client }),
+    schedules: createScheduleServer({ client })
   });
 }
 export {
   configFromHarness,
   createConnectorCatalog,
+  createHarnessAgentMetricsServer,
+  createHarnessAgentSessionsServer,
   createHarnessBuilderServer,
   createHarnessChatServer,
   createModelProviderCatalog,
   createSandboxProviderCatalog,
+  createScheduleServer,
   createSkillCatalog,
   createTrueForgeAgentUIServer,
   createTrueForgeClient,
   filterUiSandboxProviders,
   getCapabilities,
   listConfiguredMcpServers,
+  listConfiguredMcpServersPage,
   listModels,
   listSkills,
   modelProviderLogosByName,