@cdot65/prisma-airs-cli 4.3.1 → 4.4.0

package/dist/cli/index.js

@@ -4549,6 +4549,12 @@ function registerAiGatewayInventory(root) {
   );
 }
 
+// src/cli/commands/aigateway/telemetry.ts
+import {
+  AI_GW_GROUP_COLUMNS,
+  AI_GW_GROUP_DIMENSIONS
+} from "@cdot65/prisma-airs-sdk";
+
 // src/cli/commands/aigateway/telemetry-filters.ts
 import { AIGatewayChartFiltersSchema as AIGatewayChartFiltersSchema2 } from "@cdot65/prisma-airs-sdk";
 function addChartFilterOptions(command) {
@@ -4648,6 +4654,25 @@ function registerFilteredMetric(telemetry, name, description) {
     await runDetail(command, opts, (client) => client.telemetry[name](options));
   });
 }
+var groupDimensions = [...AI_GW_GROUP_DIMENSIONS, "status_code", "users"];
+function isGroupDimension(value) {
+  return groupDimensions.some((dimension) => dimension === value);
+}
+function isGroupColumn(value) {
+  return AI_GW_GROUP_COLUMNS.some((column) => column === value);
+}
+function groupOptionsFrom(dimension, opts) {
+  const options = { ...windowFrom(opts), ...chartFiltersFrom(opts) };
+  if (opts.columns !== void 0) {
+    if (dimension === "users") throw new CliUsageError("User grouping does not support --columns");
+    const columns = opts.columns.split(",").map((column) => column.trim());
+    if (!columns.every(isGroupColumn)) {
+      throw new CliUsageError(`Invalid --columns: choose from ${AI_GW_GROUP_COLUMNS.join(", ")}`);
+    }
+    options.columns = columns;
+  }
+  return options;
+}
 function registerAiGatewayTelemetryReads(telemetry) {
   const cache = showHelpOnEmpty(telemetry.command("cache").description("Inspect cache telemetry"));
   const cacheSummary = addWindowOptions(cache.command("summary").description("Get cache totals"));
@@ -4675,19 +4700,30 @@ function registerAiGatewayTelemetryReads(telemetry) {
       (opts) => runDetail(command, opts, (client) => client.telemetry[method](windowFrom(opts)))
     );
   }
-  const groupBy = addWindowOptions(
-    telemetry.command("group-by <dimension>").description("Aggregate telemetry by provider, model, status, or another SDK dimension").option("--columns <names>", "Comma-separated aggregate columns")
-  );
-  groupBy.action(
-    (dimension, opts) => runDetail(
-      groupBy,
-      opts,
-      (client) => client.telemetry.groupBy(dimension, {
-        ...windowFrom(opts),
-        ...opts.columns ? { columns: opts.columns.split(",").map((value) => value.trim()) } : {}
-      })
+  const groupBy = addChartFilterOptions(
+    addWindowOptions(
+      telemetry.command("group-by <dimension>").description(`Aggregate telemetry by ${groupDimensions.join(", ")}`).option("--columns <names>", "Comma-separated aggregate columns (not supported for users)")
     )
   );
+  groupBy.action(async (dimension, opts) => {
+    let options;
+    try {
+      if (!isGroupDimension(dimension)) {
+        throw new CliUsageError(
+          `Invalid grouping dimension: choose from ${groupDimensions.join(", ")}`
+        );
+      }
+      options = groupOptionsFrom(dimension, opts);
+    } catch (error) {
+      failAiGateway(error);
+    }
+    await runDetail(groupBy, opts, (client) => {
+      if (dimension === "users") return client.telemetry.byUser(options);
+      if (dimension === "status_code") return client.telemetry.byStatusCode(options);
+      if (isGroupDimension(dimension)) return client.telemetry.groupBy(dimension, options);
+      throw new CliUsageError("Invalid grouping dimension");
+    });
+  });
   registerFilteredMetric(telemetry, "latency", "Get latency telemetry");
   const logs = showHelpOnEmpty(telemetry.command("logs").description("Inspect request logs"));
   const logsList = addWindowOptions(

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@cdot65/prisma-airs-cli",
   "packageManager": "pnpm@10.6.5",
-  "version": "4.3.1",
+  "version": "4.4.0",
   "description": "CLI and library for Palo Alto Prisma AIRS — guardrail refinement, AI red teaming, model security scanning, profile audits",
   "type": "module",
   "main": "dist/index.js",
@@ -59,7 +59,7 @@
   },
   "license": "MIT",
   "dependencies": {
-    "@cdot65/prisma-airs-sdk": "0.24.0",
+    "@cdot65/prisma-airs-sdk": "0.25.0",
     "@inquirer/prompts": "^8.3.0",
     "chalk": "^5.6.2",
     "commander": "^14.0.3",