datadog-mcp 5.3.1 → 5.3.3

package/dist/index.js

@@ -3197,27 +3197,69 @@ function formatSlo(s) {
     timeframe: String(primaryThreshold?.timeframe ?? ""),
     tags: s.tags ?? [],
     status: {
-      // Note: SLI status requires a separate API call to getSLOHistory
       sli: null,
       errorBudgetRemaining: null,
       state: "unknown"
     },
+    overallStatus: [],
     createdAt: s.createdAt ? new Date(s.createdAt * 1e3).toISOString() : "",
     modifiedAt: s.modifiedAt ? new Date(s.modifiedAt * 1e3).toISOString() : ""
   };
 }
+function formatSearchSlo(slo) {
+  const attrs = slo.data?.attributes;
+  const primaryThreshold = attrs?.thresholds?.[0];
+  return {
+    id: slo.data?.id ?? "",
+    name: attrs?.name ?? "",
+    description: attrs?.description ?? null,
+    type: String(attrs?.sloType ?? "unknown"),
+    targetThreshold: primaryThreshold?.target ?? 0,
+    warningThreshold: primaryThreshold?.warning ?? null,
+    timeframe: String(primaryThreshold?.timeframe ?? ""),
+    tags: attrs?.allTags ?? [],
+    status: {
+      sli: attrs?.status?.sli ?? null,
+      errorBudgetRemaining: attrs?.status?.errorBudgetRemaining ?? null,
+      state: String(attrs?.status?.state ?? "unknown")
+    },
+    overallStatus: (attrs?.overallStatus ?? []).map((os) => ({
+      sli: os.status ?? null,
+      errorBudgetRemaining: os.errorBudgetRemaining ?? null,
+      state: String(os.state ?? "unknown"),
+      target: os.target ?? null,
+      timeframe: String(os.timeframe ?? "")
+    })),
+    createdAt: attrs?.createdAt ? new Date(attrs.createdAt * 1e3).toISOString() : "",
+    modifiedAt: attrs?.modifiedAt ? new Date(attrs.modifiedAt * 1e3).toISOString() : ""
+  };
+}
+function buildSearchQuery(query, tags) {
+  const parts = [];
+  if (query) parts.push(query);
+  if (tags?.length) parts.push(...tags);
+  return parts.join(" ");
+}
 async function listSlos(api, params, limits) {
   const effectiveLimit = params.limit ?? limits.defaultLimit;
-  const response = await api.listSLOs({
-    ids: params.ids?.join(","),
-    query: params.query,
-    tagsQuery: params.tags?.join(","),
-    limit: effectiveLimit
+  if (params.ids?.length) {
+    const response2 = await api.listSLOs({
+      ids: params.ids.join(","),
+      limit: effectiveLimit
+    });
+    const slos3 = (response2.data ?? []).map(formatSlo);
+    return { slos: slos3, total: slos3.length };
+  }
+  const searchQuery = buildSearchQuery(params.query, params.tags);
+  const response = await api.searchSLO({
+    query: searchQuery || void 0,
+    pageSize: effectiveLimit
   });
-  const slos2 = (response.data ?? []).map(formatSlo);
+  const searchSlos = response.data?.attributes?.slos ?? [];
+  const slos2 = searchSlos.map(formatSearchSlo);
   return {
     slos: slos2,
-    total: response.data?.length ?? 0
+    total: slos2.length
   };
 }
 async function getSlo(api, id) {
@@ -3309,7 +3351,7 @@ async function getSloHistory(api, id, params) {
 function registerSlosTool(server, api, limits, readOnly = false, _site = "datadoghq.com") {
   server.tool(
     "slos",
-    "Manage Datadog Service Level Objectives. Actions: list, get, create, update, delete, history. SLO types: metric-based, monitor-based. Use for: reliability tracking, error budgets, SLA compliance, performance targets.",
+    "Manage Datadog Service Level Objectives. Actions: list (with SLI status & error budget), get, create, update, delete, history. SLO types: metric-based, monitor-based. Use for: reliability tracking, error budgets, SLA compliance, performance targets.",
     InputSchema8,
     async ({ action, id, ids, query, tags, limit, config, from, to }) => {
       try {
@@ -5565,6 +5607,24 @@ function createServer(config) {
   );
   return server;
 }
+function createServerFactory(config) {
+  const clients = createDatadogClients(config.datadog);
+  return () => {
+    const server = new McpServer({
+      name: config.server.name,
+      version: config.server.version
+    });
+    registerAllTools(
+      server,
+      clients,
+      config.limits,
+      config.features,
+      config.datadog.site,
+      config.datadog
+    );
+    return server;
+  };
+}
 
 // src/transport/stdio.ts
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
@@ -5579,11 +5639,11 @@ import express from "express";
 import { randomUUID } from "crypto";
 import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
 import { isInitializeRequest } from "@modelcontextprotocol/sdk/types.js";
-var transports = {};
-function createExpressApp(server, config) {
+function createExpressApp(createServer2, config) {
   const app = express();
   app.disable("x-powered-by");
   app.use(express.json());
+  const transports = {};
   app.get("/health", (_req, res) => {
     res.json({ status: "ok", name: config.name, version: config.version });
   });
@@ -5606,7 +5666,15 @@ function createExpressApp(server, config) {
           console.error(`[MCP] Session closed: ${transport.sessionId}`);
         }
       };
-      await server.connect(transport);
+      try {
+        const server = createServer2();
+        await server.connect(transport);
+      } catch (error) {
+        if (transport.sessionId) {
+          delete transports[transport.sessionId];
+        }
+        throw error;
+      }
     } else {
       res.status(400).json({
         jsonrpc: "2.0",
@@ -5637,8 +5705,8 @@ function createExpressApp(server, config) {
   });
   return app;
 }
-async function connectHttp(server, config) {
-  const app = createExpressApp(server, config);
+async function connectHttp(createServer2, config) {
+  const app = createExpressApp(createServer2, config);
   app.listen(config.port, config.host, () => {
     console.error(`[MCP] Datadog MCP server running on http://${config.host}:${config.port}/mcp`);
     console.error(`[MCP] Health check available at http://${config.host}:${config.port}/health`);
@@ -5648,10 +5716,11 @@ async function connectHttp(server, config) {
 // src/index.ts
 try {
   const config = loadConfig();
-  const server = createServer(config);
   if (config.server.transport === "http") {
-    await connectHttp(server, config.server);
+    const factory = createServerFactory(config);
+    await connectHttp(factory, config.server);
   } else {
+    const server = createServer(config);
     await connectStdio(server);
   }
 } catch (error) {