@rustrak/client 0.14.3 → 0.14.4

package/dist/index.cjs

@@ -59,8 +59,40 @@ var agentTraceSummarySchema = zod.z.object({
   duration_ms: zod.z.number().nullable(),
   total_tokens: zod.z.number(),
   tool_call_count: zod.z.number().int(),
+  /** `ai_client` spans: how many times a model was actually called. */
+  llm_call_count: zod.z.number().int(),
+  /** AI spans in this trace whose status is set and not `ok`. */
+  error_count: zod.z.number().int(),
   started_at: dateTimeSchema
 });
+var agentSummarySchema = zod.z.object({
+  agent_runs: zod.z.number().int(),
+  llm_calls: zod.z.number().int(),
+  tool_calls: zod.z.number().int(),
+  error_count: zod.z.number().int(),
+  total_tokens: zod.z.number(),
+  avg_duration_ms: zod.z.number(),
+  p95_duration_ms: zod.z.number()
+});
+var agentModelRowSchema = zod.z.object({
+  model: zod.z.string(),
+  requests: zod.z.number().int(),
+  errors: zod.z.number().int(),
+  avg_ms: zod.z.number(),
+  p95_ms: zod.z.number(),
+  input_tokens: zod.z.number(),
+  cached_input_tokens: zod.z.number(),
+  output_tokens: zod.z.number(),
+  reasoning_output_tokens: zod.z.number(),
+  total_tokens: zod.z.number()
+});
+var agentToolRowSchema = zod.z.object({
+  tool: zod.z.string(),
+  calls: zod.z.number().int(),
+  errors: zod.z.number().int(),
+  avg_ms: zod.z.number(),
+  p95_ms: zod.z.number()
+});
 var providerTypeSchema = zod.z.enum(["webhook", "email", "slack"]);
 var alertTypeSchema = zod.z.enum(["new_issue", "regression", "unmute"]);
 var alertStatusSchema = zod.z.enum([
@@ -598,7 +630,7 @@ var spanSchema = zod.z.object({
   exclusive_time_ms: zod.z.number().nullable(),
   is_segment: zod.z.boolean(),
   segment_id: zod.z.string().nullable(),
-  /** Only ever set for standalone spans — a transaction-embedded span has no parent row to inherit these from. */
+  /** Reported by the span itself, or stamped from its parent transaction. Null only for rows ingested before that stamping existed. */
   platform: zod.z.string().nullable(),
   release: zod.z.string().nullable(),
   environment: zod.z.string().nullable(),
@@ -613,6 +645,11 @@ var spanSchema = zod.z.object({
   gen_ai_usage_output_tokens: zod.z.number().nullable(),
   gen_ai_usage_total_tokens: zod.z.number().nullable()
 });
+var spanDetailSchema = spanSchema.extend({
+  attributes: zod.z.record(zod.z.string(), zod.z.any()),
+  /** Always absent for a Spans Protocol v2 span: v2 has no tags concept. */
+  tags: zod.z.record(zod.z.string(), zod.z.any()).nullish()
+});
 var transactionStatsSchema = zod.z.object({
   transaction_name: zod.z.string(),
   op: zod.z.string().nullable(),
@@ -1005,6 +1042,9 @@ function timeseriesSearchParams(options) {
   if (options?.interval_hours) {
     searchParams.interval_hours = String(options.interval_hours);
   }
+  if (options?.environment) {
+    searchParams.environment = options.environment;
+  }
   return searchParams;
 }
 function breakdownSearchParams(options) {
@@ -1015,6 +1055,9 @@ function breakdownSearchParams(options) {
   if (options?.limit) {
     searchParams.limit = String(options.limit);
   }
+  if (options?.environment) {
+    searchParams.environment = options.environment;
+  }
   return searchParams;
 }
 var AgentsResource = class extends BaseResource {
@@ -1085,6 +1128,12 @@ var AgentsResource = class extends BaseResource {
     if (options?.per_page) {
       searchParams.per_page = String(options.per_page);
     }
+    if (options?.period_hours) {
+      searchParams.period_hours = String(options.period_hours);
+    }
+    if (options?.environment) {
+      searchParams.environment = options.environment;
+    }
     return this.request(
       () => this.http.get(`api/projects/${projectId}/agents/traces`, {
         searchParams
@@ -1092,6 +1141,50 @@ var AgentsResource = class extends BaseResource {
       offsetPaginatedResponseSchema(agentTraceSummarySchema)
     );
   }
+  /**
+   * Headline totals over the selected window — the numbers the charts show
+   * the shape of but cannot state.
+   */
+  async getSummary(projectId, options) {
+    return this.request(
+      () => this.http.get(`api/projects/${projectId}/agents/summary`, {
+        searchParams: breakdownSearchParams(options)
+      }),
+      agentSummarySchema
+    );
+  }
+  /**
+   * Per-model table: volume, failures, latency and the full token split.
+   */
+  async getModelsTable(projectId, options) {
+    return this.request(
+      () => this.http.get(`api/projects/${projectId}/agents/models`, {
+        searchParams: breakdownSearchParams(options)
+      }),
+      zod.z.array(agentModelRowSchema)
+    );
+  }
+  /**
+   * Per-tool table: call volume, failures and latency.
+   */
+  async getToolsTable(projectId, options) {
+    return this.request(
+      () => this.http.get(`api/projects/${projectId}/agents/tools/stats`, {
+        searchParams: breakdownSearchParams(options)
+      }),
+      zod.z.array(agentToolRowSchema)
+    );
+  }
+  /**
+   * Environments present in this project's AI spans — the options the filter
+   * offers, read from the data rather than hardcoded.
+   */
+  async getEnvironments(projectId) {
+    return this.request(
+      () => this.http.get(`api/projects/${projectId}/agents/environments`),
+      zod.z.array(zod.z.string())
+    );
+  }
 };
 var AlertIntegrationsResource = class extends BaseResource {
   /**
@@ -2024,6 +2117,20 @@ var SpansResource = class extends BaseResource {
       offsetPaginatedResponseSchema(spanSchema)
     );
   }
+  /**
+   * Fetch one span together with its raw attribute bag — the `gen_ai.*`
+   * payload (prompts, responses, tool arguments and results) that `list()`
+   * omits so a trace's worth of spans stays cheap to load.
+   *
+   * `spanId` is the row id from a list item's `id`, not the SDK's 16-hex
+   * `span_id`.
+   */
+  async get(projectId, spanId) {
+    return this.request(
+      () => this.http.get(`api/projects/${projectId}/spans/${spanId}`),
+      spanDetailSchema
+    );
+  }
 };
 
 // src/resources/stats.ts