@cdot65/prisma-airs-sdk 0.23.0 → 0.25.0

package/dist/index.js

@@ -29,7 +29,7 @@ var MAX_NUMBER_OF_BATCH_SCAN_OBJECTS = 20;
 var MAX_CONNECTION_POOL_SIZE = 100;
 var MAX_NUMBER_OF_RETRIES = 5;
 var HTTP_FORCE_RETRY_STATUS_CODES = [500, 502, 503, 504];
-var SDK_VERSION = "0.23.0";
+var SDK_VERSION = "0.25.0";
 var USER_AGENT = `PAN-AIRS/${SDK_VERSION}-typescript-sdk`;
 var DEFAULT_MGMT_ENDPOINT = "https://api.sase.paloaltonetworks.com/aisec";
 var DEFAULT_TOKEN_ENDPOINT = "https://auth.apps.paloaltonetworks.com/oauth2/access_token";
@@ -16606,37 +16606,63 @@ function toOffsetIso(d) {
   return `${String(d.getFullYear()).padStart(4, "0")}-${pad(d.getMonth() + 1)}-${pad(d.getDate())}T${pad(d.getHours())}:${pad(d.getMinutes())}:${pad(d.getSeconds())}${sign}${pad(offsetMin / 60)}:${pad(offsetMin % 60)}`;
 }
 function serializeWindow(tsgId, opts, schema = telemetryWindowSchema) {
+  return serializeWindowWithOptions(tsgId, opts, schema).params;
+}
+function serializeWindowWithOptions(tsgId, opts, schema) {
   const result = schema.safeParse(opts);
   if (!result.success || typeof tsgId !== "string" || !/^\d+$/.test(tsgId)) invalidWindow();
   const end = result.data.end ?? /* @__PURE__ */ new Date();
   const start = result.data.start ?? new Date(end.getTime() - (result.data.days ?? 7) * 864e5);
   if (start.getTime() > end.getTime()) invalidWindow();
   return {
-    organisationId: tsgId,
-    workspaceSlug: result.data.workspaceSlug,
-    timeOfGenerationMin: toOffsetIso(start),
-    timeOfGenerationMax: toOffsetIso(end)
+    options: result.data,
+    params: {
+      organisationId: tsgId,
+      workspaceSlug: result.data.workspaceSlug,
+      timeOfGenerationMin: toOffsetIso(start),
+      timeOfGenerationMax: toOffsetIso(end)
+    }
   };
 }
 
 // src/ai-gateway/telemetry-client.ts
 var nonBlankString = z54.string().min(1).refine((value2) => value2.trim().length > 0);
-var groupOptionsSchema = telemetryWindowSchema.extend({
-  columns: z54.array(z54.enum(AI_GW_GROUP_COLUMNS)).optional()
-}).strict();
 var logsOptionsSchema = telemetryWindowSchema.extend({
   pageSize: z54.number().int().nonnegative().safe().optional(),
   traceId: nonBlankString.optional(),
   statusCode: z54.number().int().nonnegative().safe().optional()
 }).strict();
-var filteredChartOptionsSchema = telemetryWindowSchema.extend({
+var chartFilterFields = {
   traceId: nonBlankString.optional(),
-  metadata: GatewayJsonObjectSchema.pipe(z54.record(z54.string())).optional()
-}).strict();
-function serializeChartOptions(tsgId, opts) {
-  const params = serializeWindow(tsgId, opts, filteredChartOptionsSchema);
+  metadata: GatewayJsonObjectSchema.pipe(z54.record(z54.string())).optional(),
+  statusCodes: z54.array(z54.number().int().nonnegative().safe()).nonempty().optional(),
+  apiKeyIds: z54.array(z54.string().uuid()).nonempty().optional(),
+  aiOrgModels: z54.array(z54.string().regex(/^[^,\s]+__[^,\s]+$/)).nonempty().optional(),
+  totalUnitsMin: z54.number().int().nonnegative().safe().optional(),
+  totalUnitsMax: z54.number().int().nonnegative().safe().optional(),
+  costMin: z54.number().finite().nonnegative().optional(),
+  costMax: z54.number().finite().nonnegative().optional()
+};
+function validChartBounds(opts) {
+  return (opts.totalUnitsMin === void 0 || opts.totalUnitsMax === void 0 || opts.totalUnitsMin <= opts.totalUnitsMax) && (opts.costMin === void 0 || opts.costMax === void 0 || opts.costMin <= opts.costMax);
+}
+var AIGatewayChartFiltersSchema = z54.object(chartFilterFields).strict().refine(validChartBounds);
+var filteredChartOptionsSchema = telemetryWindowSchema.extend(chartFilterFields).strict().refine(validChartBounds);
+var groupOptionsSchema = telemetryWindowSchema.extend({
+  ...chartFilterFields,
+  columns: z54.array(z54.enum(AI_GW_GROUP_COLUMNS)).optional()
+}).strict().refine(validChartBounds);
+function serializeFilteredOptions(tsgId, input, schema = filteredChartOptionsSchema) {
+  const { params, options: opts } = serializeWindowWithOptions(tsgId, input, schema);
   if (opts.traceId !== void 0) params.traceId = opts.traceId;
   if (opts.metadata !== void 0) params.metadata = JSON.stringify(opts.metadata);
+  if (opts.statusCodes !== void 0) params.statusCode = opts.statusCodes.join(",");
+  if (opts.apiKeyIds !== void 0) params.apiKeyIds = opts.apiKeyIds.join(",");
+  if (opts.aiOrgModels !== void 0) params.aiOrgModel = opts.aiOrgModels.join(",");
+  for (const key of ["totalUnitsMin", "totalUnitsMax", "costMin", "costMax"]) {
+    if (opts[key] !== void 0) params[key] = String(opts[key]);
+  }
+  if (opts.columns?.length) params.columns = opts.columns.join(",");
   return params;
 }
 var AIGatewayTelemetryClient = class {
@@ -16664,7 +16690,7 @@ var AIGatewayTelemetryClient = class {
   }
   /**
    * Total and per-day spend. **Values are in cents.**
-   * @param opts - Workspace slug, time window and optional verified trace/metadata filters.
+   * @param opts - Workspace slug, time window and verified trace, metadata, list and range filters.
    * @returns Cost series plus the period total, in cents.
    * @example
    * ```ts
@@ -16680,12 +16706,12 @@ var AIGatewayTelemetryClient = class {
       "cost",
       opts,
       CostChartResponseSchema,
-      serializeChartOptions(this.tsgId, opts)
+      serializeFilteredOptions(this.tsgId, opts)
     );
   }
   /**
    * Per-day request counts.
-   * @param opts - Workspace slug, time window and optional verified trace/metadata filters.
+   * @param opts - Workspace slug, time window and verified trace, metadata, list and range filters.
    * @returns Request-count series plus the filtered period total.
    * @example
    * ```ts
@@ -16701,12 +16727,12 @@ var AIGatewayTelemetryClient = class {
       "requests",
       opts,
       CountChartResponseSchema,
-      serializeChartOptions(this.tsgId, opts)
+      serializeFilteredOptions(this.tsgId, opts)
     );
   }
   /**
    * Latency in milliseconds. Percentiles are returned per-bucket and for the period.
-   * @param opts - Workspace slug, time window and optional verified trace/metadata filters.
+   * @param opts - Workspace slug, time window and verified trace, metadata, list and range filters.
    * @returns Latency series with p50/p90/p99; `data.total` is the period mean, not a sum.
    * The period mean and percentiles are null when the cohort has no matching requests.
    * @example
@@ -16723,12 +16749,12 @@ var AIGatewayTelemetryClient = class {
       "latency",
       opts,
       LatencyChartResponseSchema,
-      serializeChartOptions(this.tsgId, opts)
+      serializeFilteredOptions(this.tsgId, opts)
     );
   }
   /**
    * Token usage, split into request and response units.
-   * @param opts - Workspace slug, time window and optional verified trace/metadata filters.
+   * @param opts - Workspace slug, time window and verified trace, metadata, list and range filters.
    * @returns Token series plus request/response unit totals.
    * @example
    * ```ts
@@ -16744,7 +16770,7 @@ var AIGatewayTelemetryClient = class {
       "tokens",
       opts,
       TokensChartResponseSchema,
-      serializeChartOptions(this.tsgId, opts)
+      serializeFilteredOptions(this.tsgId, opts)
     );
   }
   /**
@@ -16928,7 +16954,7 @@ var AIGatewayTelemetryClient = class {
   /**
    * Aggregate requests by a dimension.
    * @param dimension - One of {@link AI_GW_GROUP_DIMENSIONS}. Underscore names only.
-   * @param opts - Window plus optional extra columns.
+   * @param opts - Window, verified trace/metadata/list/range filters, and optional extra columns.
    * @returns One row per distinct dimension value.
    * @example
    * ```ts
@@ -16948,8 +16974,7 @@ var AIGatewayTelemetryClient = class {
         "Invalid telemetry grouping dimension",
         "AISEC_USER_REQUEST_PAYLOAD_ERROR" /* USER_REQUEST_PAYLOAD_ERROR */
       );
-    const params = serializeWindow(this.tsgId, opts, groupOptionsSchema);
-    if (opts.columns?.length) params.columns = opts.columns.join(",");
+    const params = serializeFilteredOptions(this.tsgId, opts, groupOptionsSchema);
     return request({
       method: "GET",
       baseUrl: this.baseUrl,
@@ -16962,7 +16987,7 @@ var AIGatewayTelemetryClient = class {
   }
   /**
    * Requests and cost per end user.
-   * @param opts - Workspace slug and time window.
+   * @param opts - Workspace slug, time window and verified trace/metadata/list/range filters.
    * @returns One record per user; `_user: ''` means calls with no end-user id. Costs in cents.
    * @example
    * ```ts
@@ -16978,7 +17003,7 @@ var AIGatewayTelemetryClient = class {
       method: "GET",
       baseUrl: this.baseUrl,
       path: `${AI_GW_GROUPS_PATH}/users`,
-      params: serializeWindow(this.tsgId, opts),
+      params: serializeFilteredOptions(this.tsgId, opts),
       responseSchema: UserGroupResponseSchema,
       auth: this.auth,
       numRetries: this.numRetries
@@ -16986,7 +17011,7 @@ var AIGatewayTelemetryClient = class {
   }
   /**
    * Requests grouped by HTTP status code.
-   * @param opts - Window plus optional extra columns.
+   * @param opts - Window, verified trace/metadata/list/range filters, and optional extra columns.
    * @returns One row per status. **446 = AIRS security block** (cost 0, never reached the LLM).
    * @example
    * ```ts
@@ -17001,8 +17026,7 @@ var AIGatewayTelemetryClient = class {
    * ```
    */
   async byStatusCode(opts) {
-    const params = serializeWindow(this.tsgId, opts, groupOptionsSchema);
-    if (opts.columns?.length) params.columns = opts.columns.join(",");
+    const params = serializeFilteredOptions(this.tsgId, opts, groupOptionsSchema);
     return request({
       method: "GET",
       baseUrl: this.baseUrl,
@@ -17037,10 +17061,14 @@ var AIGatewayTelemetryClient = class {
    * ```
    */
   async logs(opts) {
-    const params = serializeWindow(this.tsgId, opts, logsOptionsSchema);
-    if (opts.pageSize !== void 0) params.pageSize = String(opts.pageSize);
-    if (opts.traceId !== void 0) params.traceId = opts.traceId;
-    if (opts.statusCode !== void 0) params.statusCode = String(opts.statusCode);
+    const { params, options: validated } = serializeWindowWithOptions(
+      this.tsgId,
+      opts,
+      logsOptionsSchema
+    );
+    if (validated.pageSize !== void 0) params.pageSize = String(validated.pageSize);
+    if (validated.traceId !== void 0) params.traceId = validated.traceId;
+    if (validated.statusCode !== void 0) params.statusCode = String(validated.statusCode);
     return request({
       method: "GET",
       baseUrl: this.baseUrl,
@@ -21298,6 +21326,7 @@ function setDottedValue(input, path, value2) {
 export {
   AIGatewayApiKeysClient,
   AIGatewayAuditLogsClient,
+  AIGatewayChartFiltersSchema,
   AIGatewayClient,
   AIGatewayConfigsClient,
   AIGatewayDeploymentsClient,