@mastra/react 1.12.0-alpha.11 → 1.12.0-alpha.12

package/dist/experiments-B7vRMK5T.js

@@ -0,0 +1,306 @@
+import { n as useMastraClient } from "./mastra-client-context-Bx_g84YB.js";
+import { t as useInView } from "./use-in-view-C9sYPGRm.js";
+import { t as useCombineByIds } from "./combine-by-ids-CcaZKGT9.js";
+import { t as getOrCreate } from "./map-CDv4fbu4.js";
+import { useEffect } from "react";
+import { keepPreviousData, useInfiniteQuery, useQueries, useQuery } from "@tanstack/react-query";
+//#region src/hooks/experiments/usage-accumulator.ts
+const INPUT_TOKEN_METRIC = "mastra_model_total_input_tokens";
+const OUTPUT_TOKEN_METRIC = "mastra_model_total_output_tokens";
+const createAccumulator = () => ({
+	estimatedCost: 0,
+	costUnits: /* @__PURE__ */ new Set(),
+	hasUnknownCostUnit: false
+});
+/** The token field a breakdown group feeds, or undefined when the group is not a token metric. */
+function tokenFieldOf(group) {
+	const metricName = group.dimensions.name;
+	if (metricName === "mastra_model_total_input_tokens") return "inputTokens";
+	if (metricName === "mastra_model_total_output_tokens") return "outputTokens";
+}
+function accumulate(accumulator, tokenField, group) {
+	accumulator[tokenField] = (accumulator[tokenField] ?? 0) + group.value;
+	if (group.estimatedCost == null) return;
+	accumulator.estimatedCost += group.estimatedCost;
+	if (group.costUnit) accumulator.costUnits.add(group.costUnit.toLowerCase());
+	else accumulator.hasUnknownCostUnit = true;
+}
+function toUsage(accumulator) {
+	const isPriced = !accumulator.hasUnknownCostUnit && accumulator.costUnits.size === 1;
+	return {
+		inputTokens: accumulator.inputTokens,
+		outputTokens: accumulator.outputTokens,
+		estimatedCost: isPriced ? accumulator.estimatedCost : void 0,
+		costUnit: isPriced ? [...accumulator.costUnits][0] : void 0
+	};
+}
+//#endregion
+//#region src/hooks/experiments/use-experiment-item-usage.ts
+const POLL_INTERVAL_MS$1 = 2e3;
+/** Builds the experiment totals and the per-trace usage in a single pass over the groups. */
+function toExperimentUsage(response) {
+	const summary = createAccumulator();
+	const accumulators = /* @__PURE__ */ new Map();
+	for (const group of response.groups) {
+		const tokenField = tokenFieldOf(group);
+		if (!tokenField) continue;
+		accumulate(summary, tokenField, group);
+		const traceId = group.dimensions.traceId;
+		if (traceId) accumulate(getOrCreate(accumulators, traceId, createAccumulator), tokenField, group);
+	}
+	const traces = {};
+	for (const [traceId, accumulator] of accumulators) traces[traceId] = toUsage(accumulator);
+	return {
+		summary: toUsage(summary),
+		traces
+	};
+}
+/** How many breakdowns to fetch after completion while the exporter may still be flushing metrics. */
+const MAX_EMPTY_ATTEMPTS = 5;
+const isEmptyUsage = (usage) => !usage || Object.keys(usage.traces).length === 0 && usage.summary.inputTokens === void 0 && usage.summary.outputTokens === void 0;
+/**
+* Token and cost usage of several experiments, keyed by experiment id: each entry holds the
+* experiment totals (`summary`) and every item's usage keyed by trace id (`traces`).
+* Sends one metric breakdown request per experiment, filtered by its id rather than by trace ids,
+* so it does not wait for the experiment results. Usage is only fetched once an experiment has
+* finished; while the breakdown is still empty (metrics flushing), it retries every 2 seconds up to
+* 5 attempts. No request is made when the observability store cannot serve metrics.
+*/
+const useExperimentsItemUsage = ({ experiments, supportsMetrics }) => {
+	const client = useMastraClient();
+	const combine = useCombineByIds(experiments.map((experiment) => experiment.id));
+	return useQueries({
+		queries: experiments.map(({ id, status }) => {
+			const finished = status !== "running" && status !== "pending";
+			return {
+				queryKey: [
+					"experiment-item-usage",
+					id,
+					finished
+				],
+				queryFn: async () => toExperimentUsage(await client.getMetricBreakdown({
+					name: [INPUT_TOKEN_METRIC, OUTPUT_TOKEN_METRIC],
+					aggregation: "sum",
+					groupBy: ["traceId", "name"],
+					filters: { experimentId: id }
+				})),
+				enabled: supportsMetrics && finished,
+				refetchInterval: (query) => isEmptyUsage(query.state.data) && query.state.dataUpdateCount < MAX_EMPTY_ATTEMPTS ? POLL_INTERVAL_MS$1 : false
+			};
+		}),
+		combine
+	});
+};
+//#endregion
+//#region src/hooks/experiments/use-experiment-metrics.ts
+/**
+* Experiment-scoped metrics (tokens, cost, avg agent latency).
+* Filters only by experimentId — no time window, since the experiment already bounds the row set.
+* Polls every 2 seconds while the experiment is running or pending.
+*
+* Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.
+*/
+const useExperimentMetrics = ({ experimentId, experimentStatus, supportsMetrics, queryOptions }) => {
+	const client = useMastraClient();
+	const isEnabled = Boolean(experimentId) && supportsMetrics;
+	const query = useQuery({
+		queryKey: ["experiment-metrics", experimentId],
+		refetchInterval: experimentStatus === "running" || experimentStatus === "pending" ? 2e3 : false,
+		queryFn: async () => {
+			if (!experimentId) throw new Error("experimentId is required");
+			const filters = { experimentId };
+			const [tokens, avgDuration, runs] = await Promise.all([
+				client.getMetricAggregate({
+					name: ["mastra_model_total_input_tokens", "mastra_model_total_output_tokens"],
+					aggregation: "sum",
+					filters
+				}),
+				client.getMetricAggregate({
+					name: ["mastra_agent_duration_ms"],
+					aggregation: "avg",
+					filters
+				}),
+				client.getMetricAggregate({
+					name: ["mastra_agent_duration_ms"],
+					aggregation: "count",
+					filters
+				})
+			]);
+			return {
+				totalTokens: tokens.value ?? null,
+				estimatedCost: tokens.estimatedCost ?? null,
+				costUnit: tokens.costUnit ?? null,
+				avgAgentDurationMs: avgDuration.value ?? null,
+				agentRuns: runs.value ?? null
+			};
+		},
+		...queryOptions
+	});
+	return {
+		data: query.data,
+		isLoading: query.isLoading,
+		isEnabled
+	};
+};
+//#endregion
+//#region src/hooks/experiments/use-experiment-trace.ts
+/**
+* Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.
+*/
+const useExperimentTrace = ({ traceId, queryOptions }) => {
+	const client = useMastraClient();
+	return useQuery({
+		queryKey: ["experiment-trace-light", traceId],
+		queryFn: async () => {
+			if (!traceId) throw new Error("Trace ID is required");
+			return client.getTraceLight(traceId);
+		},
+		...queryOptions
+	});
+};
+//#endregion
+//#region src/hooks/experiments/use-experiments-for-dataset-filter.ts
+/** Explicit page size: server defaults are 20 (global) / 10 (per dataset), which is too small for the list. */
+const EXPERIMENTS_PAGE_SIZE = 100;
+function useExperimentsForDatasetFilter({ datasetId, target, queryOptions }) {
+	const client = useMastraClient();
+	const params = { perPage: 100 };
+	if (target?.targetType) params.targetType = target.targetType;
+	if (target?.targetId) params.targetId = target.targetId;
+	return useQuery({
+		queryKey: datasetId ? [
+			"dataset-experiments",
+			datasetId,
+			params
+		] : ["experiments", params],
+		queryFn: () => datasetId ? client.listDatasetExperiments(datasetId, params) : client.listExperiments(params),
+		...queryOptions
+	});
+}
+//#endregion
+//#region src/hooks/experiments/use-experiments-list-metrics.ts
+const AGENT_DURATION_METRIC = "mastra_agent_duration_ms";
+const EXPERIMENT_IDS_PER_REQUEST = 500;
+const POLL_INTERVAL_MS = 5e3;
+function chunk(ids) {
+	const chunks = [];
+	for (let index = 0; index < ids.length; index += EXPERIMENT_IDS_PER_REQUEST) chunks.push(ids.slice(index, index + EXPERIMENT_IDS_PER_REQUEST));
+	return chunks;
+}
+function settledValues(results) {
+	return results.flatMap((result) => result.status === "fulfilled" ? [result.value] : []);
+}
+/**
+* Builds each requested experiment's metrics. Groups for other experiments are skipped,
+* in case a store ignores the `experimentIds` filter.
+*/
+function toListMetrics(experimentIds, tokenResponses, durationResponses) {
+	const requested = new Set(experimentIds);
+	const accumulators = /* @__PURE__ */ new Map();
+	for (const group of tokenResponses.flatMap((response) => response.groups)) {
+		const experimentId = group.dimensions.experimentId;
+		const tokenField = tokenFieldOf(group);
+		if (!experimentId || !requested.has(experimentId) || !tokenField) continue;
+		accumulate(getOrCreate(accumulators, experimentId, createAccumulator), tokenField, group);
+	}
+	const metrics = /* @__PURE__ */ new Map();
+	for (const [experimentId, accumulator] of accumulators) metrics.set(experimentId, toUsage(accumulator));
+	for (const group of durationResponses.flatMap((response) => response.groups)) {
+		const experimentId = group.dimensions.experimentId;
+		if (!experimentId || !requested.has(experimentId)) continue;
+		getOrCreate(metrics, experimentId, () => ({})).avgDurationMs = group.value;
+	}
+	return metrics;
+}
+/**
+* Token usage, estimated cost and average agent duration of many experiments, keyed by experiment id.
+* Sends two metric breakdowns per 500 experiments (tokens grouped by experiment and metric name,
+* durations grouped by experiment) instead of one request per experiment.
+* Experiments without any metric are absent from the map. Polls while an experiment is running
+* or pending. No request is made when the observability store cannot serve metrics.
+*/
+const useExperimentsListMetrics = ({ experiments, supportsMetrics }) => {
+	const client = useMastraClient();
+	const experimentIds = [...new Set(experiments.map((experiment) => experiment.id))].toSorted();
+	const hasActiveExperiment = experiments.some(({ status }) => status === "running" || status === "pending");
+	return useQuery({
+		queryKey: ["experiments-list-metrics", experimentIds],
+		enabled: supportsMetrics && experimentIds.length > 0,
+		queryFn: async () => {
+			const batches = chunk(experimentIds);
+			const [tokenResults, durationResults] = await Promise.all([Promise.allSettled(batches.map((batch) => client.getMetricBreakdown({
+				name: [INPUT_TOKEN_METRIC, OUTPUT_TOKEN_METRIC],
+				aggregation: "sum",
+				groupBy: ["experimentId", "name"],
+				filters: { experimentIds: batch },
+				limit: batch.length * 2
+			}))), Promise.allSettled(batches.map((batch) => client.getMetricBreakdown({
+				name: [AGENT_DURATION_METRIC],
+				aggregation: "avg",
+				groupBy: ["experimentId"],
+				filters: { experimentIds: batch },
+				limit: batch.length
+			})))]);
+			return toListMetrics(experimentIds, settledValues(tokenResults), settledValues(durationResults));
+		},
+		placeholderData: keepPreviousData,
+		refetchInterval: hasActiveExperiment ? POLL_INTERVAL_MS : false
+	});
+};
+//#endregion
+//#region src/hooks/experiments/use-infinite-experiments.ts
+const EXPERIMENTS_PER_PAGE = 100;
+function useInfiniteExperiments({ datasetId, target, orderBy, queryOptions }) {
+	const client = useMastraClient();
+	const { inView: isEndOfListInView, setRef: setEndOfListElement } = useInView();
+	const targetType = target?.targetType || void 0;
+	const targetId = target?.targetId || void 0;
+	const query = useInfiniteQuery({
+		queryKey: datasetId ? [
+			"dataset-experiments",
+			datasetId,
+			"infinite",
+			{
+				targetType,
+				targetId,
+				orderBy
+			}
+		] : [
+			"experiments",
+			"infinite",
+			{
+				targetType,
+				targetId,
+				orderBy
+			}
+		],
+		queryFn: ({ pageParam }) => {
+			const params = {
+				page: pageParam,
+				perPage: 100
+			};
+			if (targetType) params.targetType = targetType;
+			if (targetId) params.targetId = targetId;
+			if (orderBy) params.orderBy = orderBy;
+			return datasetId ? client.listDatasetExperiments(datasetId, params) : client.listExperiments(params);
+		},
+		initialPageParam: 0,
+		getNextPageParam: (lastPage, _, lastPageParam) => {
+			if (!lastPage?.experiments?.length || !lastPage.pagination?.hasMore) return;
+			return lastPageParam + 1;
+		},
+		select: (data) => data.pages.flatMap((page) => page?.experiments ?? []),
+		...queryOptions
+	});
+	useEffect(() => {
+		if (isEndOfListInView && query.hasNextPage && !query.isFetchingNextPage) query.fetchNextPage();
+	}, [isEndOfListInView, query]);
+	return {
+		...query,
+		setEndOfListElement
+	};
+}
+//#endregion
+export { useExperimentsForDatasetFilter as a, useExperimentsItemUsage as c, EXPERIMENTS_PAGE_SIZE as i, useInfiniteExperiments as n, useExperimentTrace as o, useExperimentsListMetrics as r, useExperimentMetrics as s, EXPERIMENTS_PER_PAGE as t };
+
+//# sourceMappingURL=experiments-B7vRMK5T.js.map

package/dist/experiments-B7vRMK5T.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"experiments-B7vRMK5T.js","names":["POLL_INTERVAL_MS"],"sources":["../src/hooks/experiments/usage-accumulator.ts","../src/hooks/experiments/use-experiment-item-usage.ts","../src/hooks/experiments/use-experiment-metrics.ts","../src/hooks/experiments/use-experiment-trace.ts","../src/hooks/experiments/use-experiments-for-dataset-filter.ts","../src/hooks/experiments/use-experiments-list-metrics.ts","../src/hooks/experiments/use-infinite-experiments.ts"],"sourcesContent":["import type { GetMetricBreakdownResponse } from '@mastra/core/storage';\n\nexport const INPUT_TOKEN_METRIC = 'mastra_model_total_input_tokens';\nexport const OUTPUT_TOKEN_METRIC = 'mastra_model_total_output_tokens';\n\nexport interface ExperimentItemUsage {\n  inputTokens?: number;\n  outputTokens?: number;\n  /** Undefined when no group is priced or when groups use different cost units. */\n  estimatedCost?: number;\n  costUnit?: string;\n}\n\nexport type UsageAccumulator = {\n  inputTokens?: number;\n  outputTokens?: number;\n  estimatedCost: number;\n  costUnits: Set<string>;\n  hasUnknownCostUnit: boolean;\n};\n\nexport const createAccumulator = (): UsageAccumulator => ({\n  estimatedCost: 0,\n  costUnits: new Set(),\n  hasUnknownCostUnit: false,\n});\n\ntype BreakdownGroup = GetMetricBreakdownResponse['groups'][number];\n\n/** The token field a breakdown group feeds, or undefined when the group is not a token metric. */\nexport function tokenFieldOf(group: BreakdownGroup): 'inputTokens' | 'outputTokens' | undefined {\n  const metricName = group.dimensions.name;\n  if (metricName === INPUT_TOKEN_METRIC) return 'inputTokens';\n  if (metricName === OUTPUT_TOKEN_METRIC) return 'outputTokens';\n  return undefined;\n}\n\nexport function accumulate(\n  accumulator: UsageAccumulator,\n  tokenField: 'inputTokens' | 'outputTokens',\n  group: BreakdownGroup,\n) {\n  accumulator[tokenField] = (accumulator[tokenField] ?? 0) + group.value;\n\n  if (group.estimatedCost == null) return;\n  accumulator.estimatedCost += group.estimatedCost;\n  if (group.costUnit) {\n    accumulator.costUnits.add(group.costUnit.toLowerCase());\n  } else {\n    accumulator.hasUnknownCostUnit = true;\n  }\n}\n\nexport function toUsage(accumulator: UsageAccumulator): ExperimentItemUsage {\n  // A single known unit implies at least one priced group.\n  const isPriced = !accumulator.hasUnknownCostUnit && accumulator.costUnits.size === 1;\n  return {\n    inputTokens: accumulator.inputTokens,\n    outputTokens: accumulator.outputTokens,\n    estimatedCost: isPriced ? accumulator.estimatedCost : undefined,\n    costUnit: isPriced ? [...accumulator.costUnits][0] : undefined,\n  };\n}\n","import type { GetMetricBreakdownResponse } from '@mastra/core/storage';\nimport { useQueries } from '@tanstack/react-query';\nimport type { Query } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { ExperimentRef } from '../datasets/use-dataset-experiments';\nimport { useCombineByIds } from '../shared/combine-by-ids';\nimport type { ExperimentQueriesResult } from '../shared/combine-by-ids';\nimport { getOrCreate } from '../shared/map';\nimport {\n  INPUT_TOKEN_METRIC,\n  OUTPUT_TOKEN_METRIC,\n  accumulate,\n  createAccumulator,\n  tokenFieldOf,\n  toUsage,\n} from './usage-accumulator';\nimport type { ExperimentItemUsage, UsageAccumulator } from './usage-accumulator';\n\nexport type { ExperimentItemUsage } from './usage-accumulator';\n\nconst POLL_INTERVAL_MS = 2000;\n\n/** Usage of one experiment: its totals, and each item's usage keyed by the trace id of its run. */\nexport interface ExperimentUsage {\n  /** Totals across every item. The cost sums priced groups only. */\n  summary: ExperimentItemUsage;\n  traces: Record<string, ExperimentItemUsage>;\n}\n\n/** Builds the experiment totals and the per-trace usage in a single pass over the groups. */\nfunction toExperimentUsage(response: GetMetricBreakdownResponse): ExperimentUsage {\n  const summary = createAccumulator();\n  const accumulators = new Map<string, UsageAccumulator>();\n\n  for (const group of response.groups) {\n    const tokenField = tokenFieldOf(group);\n    if (!tokenField) continue;\n\n    accumulate(summary, tokenField, group);\n\n    const traceId = group.dimensions.traceId;\n    if (traceId) accumulate(getOrCreate(accumulators, traceId, createAccumulator), tokenField, group);\n  }\n\n  const traces: Record<string, ExperimentItemUsage> = {};\n  for (const [traceId, accumulator] of accumulators) {\n    traces[traceId] = toUsage(accumulator);\n  }\n  return { summary: toUsage(summary), traces };\n}\n\n/** How many breakdowns to fetch after completion while the exporter may still be flushing metrics. */\nconst MAX_EMPTY_ATTEMPTS = 5;\n\nconst isEmptyUsage = (usage: ExperimentUsage | undefined) =>\n  !usage ||\n  (Object.keys(usage.traces).length === 0 &&\n    usage.summary.inputTokens === undefined &&\n    usage.summary.outputTokens === undefined);\n\n/**\n * Token and cost usage of several experiments, keyed by experiment id: each entry holds the\n * experiment totals (`summary`) and every item's usage keyed by trace id (`traces`).\n * Sends one metric breakdown request per experiment, filtered by its id rather than by trace ids,\n * so it does not wait for the experiment results. Usage is only fetched once an experiment has\n * finished; while the breakdown is still empty (metrics flushing), it retries every 2 seconds up to\n * 5 attempts. No request is made when the observability store cannot serve metrics.\n */\nexport const useExperimentsItemUsage = ({\n  experiments,\n  supportsMetrics,\n}: {\n  experiments: ExperimentRef[];\n  supportsMetrics: boolean;\n}): ExperimentQueriesResult<ExperimentUsage> => {\n  const client = useMastraClient();\n  const combine = useCombineByIds<ExperimentUsage>(experiments.map(experiment => experiment.id));\n\n  return useQueries({\n    queries: experiments.map(({ id, status }) => {\n      const finished = status !== 'running' && status !== 'pending';\n      return {\n        queryKey: ['experiment-item-usage', id, finished],\n        queryFn: async () =>\n          toExperimentUsage(\n            await client.getMetricBreakdown({\n              name: [INPUT_TOKEN_METRIC, OUTPUT_TOKEN_METRIC],\n              aggregation: 'sum',\n              groupBy: ['traceId', 'name'],\n              filters: { experimentId: id },\n            }),\n          ),\n        enabled: supportsMetrics && finished,\n        refetchInterval: (query: Query<ExperimentUsage>) =>\n          isEmptyUsage(query.state.data) && query.state.dataUpdateCount < MAX_EMPTY_ATTEMPTS ? POLL_INTERVAL_MS : false,\n      };\n    }),\n    combine,\n  });\n};\n","import type { DatasetExperiment } from '@mastra/client-js';\nimport { useQuery } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraQueryOptions } from '../shared/query-options';\n\nexport interface ExperimentMetrics {\n  totalTokens: number | null;\n  estimatedCost: number | null;\n  costUnit: string | null;\n  avgAgentDurationMs: number | null;\n  agentRuns: number | null;\n}\n\ninterface UseExperimentMetricsArgs {\n  experimentId: string | undefined;\n  experimentStatus: DatasetExperiment['status'] | undefined;\n  /** Whether the observability store can serve metrics; no requests are made when false. */\n  supportsMetrics: boolean;\n  queryOptions?: MastraQueryOptions<ExperimentMetrics>;\n}\n\n/**\n * Experiment-scoped metrics (tokens, cost, avg agent latency).\n * Filters only by experimentId — no time window, since the experiment already bounds the row set.\n * Polls every 2 seconds while the experiment is running or pending.\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useExperimentMetrics = ({\n  experimentId,\n  experimentStatus,\n  supportsMetrics,\n  queryOptions,\n}: UseExperimentMetricsArgs) => {\n  const client = useMastraClient();\n  const isEnabled = Boolean(experimentId) && supportsMetrics;\n  const isActive = experimentStatus === 'running' || experimentStatus === 'pending';\n\n  const query = useQuery({\n    queryKey: ['experiment-metrics', experimentId],\n    refetchInterval: isActive ? 2000 : false,\n    queryFn: async (): Promise<ExperimentMetrics> => {\n      // `enabled` already guards this; the check exists to narrow the type.\n      if (!experimentId) throw new Error('experimentId is required');\n      const filters = { experimentId };\n\n      const [tokens, avgDuration, runs] = await Promise.all([\n        client.getMetricAggregate({\n          name: ['mastra_model_total_input_tokens', 'mastra_model_total_output_tokens'],\n          aggregation: 'sum',\n          filters,\n        }),\n        client.getMetricAggregate({ name: ['mastra_agent_duration_ms'], aggregation: 'avg', filters }),\n        client.getMetricAggregate({ name: ['mastra_agent_duration_ms'], aggregation: 'count', filters }),\n      ]);\n\n      return {\n        totalTokens: tokens.value ?? null,\n        estimatedCost: tokens.estimatedCost ?? null,\n        costUnit: tokens.costUnit ?? null,\n        avgAgentDurationMs: avgDuration.value ?? null,\n        agentRuns: runs.value ?? null,\n      };\n    },\n    ...queryOptions,\n  });\n\n  return { data: query.data, isLoading: query.isLoading, isEnabled };\n};\n","import type { MastraClient } from '@mastra/client-js';\nimport type { UseQueryResult } from '@tanstack/react-query';\nimport { useQuery } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraQueryOptions } from '../shared/query-options';\n\ntype TraceLightResponse = Awaited<ReturnType<MastraClient['getTraceLight']>>;\n\n/**\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useExperimentTrace = <TData = TraceLightResponse>({\n  traceId,\n  queryOptions,\n}: {\n  traceId: string | null | undefined;\n  queryOptions?: MastraQueryOptions<TraceLightResponse, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['experiment-trace-light', traceId],\n    queryFn: async () => {\n      if (!traceId) {\n        throw new Error('Trace ID is required');\n      }\n      return client.getTraceLight(traceId);\n    },\n    ...queryOptions,\n  });\n};\n","import type { ExperimentTargetType, ListExperimentsParams, MastraClient } from '@mastra/client-js';\nimport type { UseQueryResult } from '@tanstack/react-query';\nimport { useQuery } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraQueryOptions } from '../shared/query-options';\n\n/** Explicit page size: server defaults are 20 (global) / 10 (per dataset), which is too small for the list. */\nexport const EXPERIMENTS_PAGE_SIZE = 100;\n\nexport interface ExperimentTargetFilter {\n  targetType?: ExperimentTargetType | '';\n  targetId?: string;\n}\n\n/**\n * Experiments for the list page: the global list, or the dataset-scoped list when a dataset filter is active.\n * A dataset's runs may not be in the first page of the global list, so filtering client-side is not enough.\n * The optional target filter is applied server-side for the same reason.\n */\ntype ListExperimentsResponse = Awaited<ReturnType<MastraClient['listExperiments']>>;\n\nexport function useExperimentsForDatasetFilter<TData = ListExperimentsResponse>({\n  datasetId,\n  target,\n  queryOptions,\n}: {\n  datasetId: string | undefined;\n  target?: ExperimentTargetFilter;\n  queryOptions?: MastraQueryOptions<ListExperimentsResponse, TData>;\n}): UseQueryResult<TData, Error> {\n  const client = useMastraClient();\n  const params: ListExperimentsParams = { perPage: EXPERIMENTS_PAGE_SIZE };\n  if (target?.targetType) params.targetType = target.targetType;\n  if (target?.targetId) params.targetId = target.targetId;\n\n  return useQuery({\n    // Prefixes match the keys invalidated by dataset/experiment mutations.\n    queryKey: datasetId ? ['dataset-experiments', datasetId, params] : ['experiments', params],\n    queryFn: () => (datasetId ? client.listDatasetExperiments(datasetId, params) : client.listExperiments(params)),\n    ...queryOptions,\n  });\n}\n","import type { GetMetricBreakdownResponse } from '@mastra/core/storage';\nimport { keepPreviousData, useQuery } from '@tanstack/react-query';\nimport type { UseQueryResult } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { ExperimentRef } from '../datasets/use-dataset-experiments';\nimport { getOrCreate } from '../shared/map';\nimport {\n  INPUT_TOKEN_METRIC,\n  OUTPUT_TOKEN_METRIC,\n  accumulate,\n  createAccumulator,\n  tokenFieldOf,\n  toUsage,\n} from './usage-accumulator';\nimport type { ExperimentItemUsage, UsageAccumulator } from './usage-accumulator';\n\nconst AGENT_DURATION_METRIC = 'mastra_agent_duration_ms';\n// Each experiment can return two token groups, while the breakdown endpoint allows at most 1,000.\nconst EXPERIMENT_IDS_PER_REQUEST = 500;\nconst POLL_INTERVAL_MS = 5000;\n\n/** Totals of one experiment: token usage, its estimated cost, and the average agent run duration. */\nexport interface ExperimentListMetrics extends ExperimentItemUsage {\n  avgDurationMs?: number;\n}\n\nfunction chunk(ids: readonly string[]): string[][] {\n  const chunks: string[][] = [];\n  for (let index = 0; index < ids.length; index += EXPERIMENT_IDS_PER_REQUEST) {\n    chunks.push(ids.slice(index, index + EXPERIMENT_IDS_PER_REQUEST));\n  }\n  return chunks;\n}\n\nfunction settledValues<T>(results: PromiseSettledResult<T>[]): T[] {\n  return results.flatMap(result => (result.status === 'fulfilled' ? [result.value] : []));\n}\n\n/**\n * Builds each requested experiment's metrics. Groups for other experiments are skipped,\n * in case a store ignores the `experimentIds` filter.\n */\nfunction toListMetrics(\n  experimentIds: readonly string[],\n  tokenResponses: GetMetricBreakdownResponse[],\n  durationResponses: GetMetricBreakdownResponse[],\n): Map<string, ExperimentListMetrics> {\n  const requested = new Set(experimentIds);\n  const accumulators = new Map<string, UsageAccumulator>();\n  for (const group of tokenResponses.flatMap(response => response.groups)) {\n    const experimentId = group.dimensions.experimentId;\n    const tokenField = tokenFieldOf(group);\n    if (!experimentId || !requested.has(experimentId) || !tokenField) continue;\n    accumulate(getOrCreate(accumulators, experimentId, createAccumulator), tokenField, group);\n  }\n\n  const metrics = new Map<string, ExperimentListMetrics>();\n  for (const [experimentId, accumulator] of accumulators) {\n    metrics.set(experimentId, toUsage(accumulator));\n  }\n  for (const group of durationResponses.flatMap(response => response.groups)) {\n    const experimentId = group.dimensions.experimentId;\n    if (!experimentId || !requested.has(experimentId)) continue;\n    getOrCreate(metrics, experimentId, (): ExperimentListMetrics => ({})).avgDurationMs = group.value;\n  }\n  return metrics;\n}\n\n/**\n * Token usage, estimated cost and average agent duration of many experiments, keyed by experiment id.\n * Sends two metric breakdowns per 500 experiments (tokens grouped by experiment and metric name,\n * durations grouped by experiment) instead of one request per experiment.\n * Experiments without any metric are absent from the map. Polls while an experiment is running\n * or pending. No request is made when the observability store cannot serve metrics.\n */\nexport const useExperimentsListMetrics = ({\n  experiments,\n  supportsMetrics,\n}: {\n  experiments: ExperimentRef[];\n  supportsMetrics: boolean;\n}): UseQueryResult<Map<string, ExperimentListMetrics>, Error> => {\n  const client = useMastraClient();\n  const experimentIds = [...new Set(experiments.map(experiment => experiment.id))].toSorted();\n  const hasActiveExperiment = experiments.some(({ status }) => status === 'running' || status === 'pending');\n\n  return useQuery({\n    queryKey: ['experiments-list-metrics', experimentIds],\n    enabled: supportsMetrics && experimentIds.length > 0,\n    queryFn: async () => {\n      const batches = chunk(experimentIds);\n      const [tokenResults, durationResults] = await Promise.all([\n        Promise.allSettled(\n          batches.map(batch =>\n            client.getMetricBreakdown({\n              name: [INPUT_TOKEN_METRIC, OUTPUT_TOKEN_METRIC],\n              aggregation: 'sum',\n              groupBy: ['experimentId', 'name'],\n              filters: { experimentIds: batch },\n              limit: batch.length * 2,\n            }),\n          ),\n        ),\n        Promise.allSettled(\n          batches.map(batch =>\n            client.getMetricBreakdown({\n              name: [AGENT_DURATION_METRIC],\n              aggregation: 'avg',\n              groupBy: ['experimentId'],\n              filters: { experimentIds: batch },\n              limit: batch.length,\n            }),\n          ),\n        ),\n      ]);\n      return toListMetrics(experimentIds, settledValues(tokenResults), settledValues(durationResults));\n    },\n    // Loading another page changes the ids; keep the rows already shown filled meanwhile.\n    placeholderData: keepPreviousData,\n    refetchInterval: hasActiveExperiment ? POLL_INTERVAL_MS : false,\n  });\n};\n","import type { DatasetExperiment, ListExperimentsParams, MastraClient } from '@mastra/client-js';\nimport { useInfiniteQuery } from '@tanstack/react-query';\nimport type { InfiniteData, QueryKey, UseInfiniteQueryResult } from '@tanstack/react-query';\nimport { useEffect } from 'react';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraInfiniteQueryOptions } from '../shared/query-options';\nimport { useInView } from '../shared/use-in-view';\nimport type { ExperimentTargetFilter } from './use-experiments-for-dataset-filter';\n\nexport const EXPERIMENTS_PER_PAGE = 100;\n\n/**\n * Infinite-scroll experiments for the list page: the global list, or the dataset-scoped list when a\n * dataset filter is active (a dataset's runs may not be in the first pages of the global list). The\n * optional target filter is applied server-side for the same reason.\n */\nexport type ExperimentsOrderBy = NonNullable<ListExperimentsParams['orderBy']>;\n\ntype ListExperimentsResponse = Awaited<ReturnType<MastraClient['listExperiments']>>;\n\nexport function useInfiniteExperiments<TData = DatasetExperiment[]>({\n  datasetId,\n  target,\n  orderBy,\n  queryOptions,\n}: {\n  datasetId: string | undefined;\n  target?: ExperimentTargetFilter;\n  orderBy?: ExperimentsOrderBy;\n  queryOptions?: MastraInfiniteQueryOptions<ListExperimentsResponse, TData, QueryKey, number>;\n}): UseInfiniteQueryResult<TData, Error> & {\n  setEndOfListElement: (element: HTMLDivElement | null) => void;\n} {\n  const client = useMastraClient();\n  const { inView: isEndOfListInView, setRef: setEndOfListElement } = useInView();\n  const targetType = target?.targetType || undefined;\n  const targetId = target?.targetId || undefined;\n\n  const query = useInfiniteQuery({\n    // Prefixes match the keys invalidated by dataset/experiment mutations.\n    queryKey: datasetId\n      ? ['dataset-experiments', datasetId, 'infinite', { targetType, targetId, orderBy }]\n      : ['experiments', 'infinite', { targetType, targetId, orderBy }],\n    queryFn: ({ pageParam }) => {\n      const params: ListExperimentsParams = { page: pageParam, perPage: EXPERIMENTS_PER_PAGE };\n      if (targetType) params.targetType = targetType;\n      if (targetId) params.targetId = targetId;\n      if (orderBy) params.orderBy = orderBy;\n      return datasetId ? client.listDatasetExperiments(datasetId, params) : client.listExperiments(params);\n    },\n    initialPageParam: 0,\n    getNextPageParam: (lastPage, _, lastPageParam) => {\n      if (!lastPage?.experiments?.length || !lastPage.pagination?.hasMore) {\n        return undefined;\n      }\n      return lastPageParam + 1;\n    },\n    select: (data: InfiniteData<ListExperimentsResponse, number>) =>\n      data.pages.flatMap(page => page?.experiments ?? []) as TData,\n    ...queryOptions,\n  });\n\n  useEffect(() => {\n    if (isEndOfListInView && query.hasNextPage && !query.isFetchingNextPage) {\n      void query.fetchNextPage();\n    }\n  }, [isEndOfListInView, query]);\n\n  return { ...query, setEndOfListElement };\n}\n"],"mappings":";;;;;;;AAEA,MAAa,qBAAqB;AAClC,MAAa,sBAAsB;AAkBnC,MAAa,2BAA6C;CACxD,eAAe;CACf,2BAAW,IAAI,IAAI;CACnB,oBAAoB;AACtB;;AAKA,SAAgB,aAAa,OAAmE;CAC9F,MAAM,aAAa,MAAM,WAAW;CACpC,IAAI,eAAA,mCAAmC,OAAO;CAC9C,IAAI,eAAA,oCAAoC,OAAO;AAEjD;AAEA,SAAgB,WACd,aACA,YACA,OACA;CACA,YAAY,eAAe,YAAY,eAAe,KAAK,MAAM;CAEjE,IAAI,MAAM,iBAAiB,MAAM;CACjC,YAAY,iBAAiB,MAAM;CACnC,IAAI,MAAM,UACR,YAAY,UAAU,IAAI,MAAM,SAAS,YAAY,CAAC;MAEtD,YAAY,qBAAqB;AAErC;AAEA,SAAgB,QAAQ,aAAoD;CAE1E,MAAM,WAAW,CAAC,YAAY,sBAAsB,YAAY,UAAU,SAAS;CACnF,OAAO;EACL,aAAa,YAAY;EACzB,cAAc,YAAY;EAC1B,eAAe,WAAW,YAAY,gBAAgB,KAAA;EACtD,UAAU,WAAW,CAAC,GAAG,YAAY,SAAS,CAAC,CAAC,KAAK,KAAA;CACvD;AACF;;;AC1CA,MAAMA,qBAAmB;;AAUzB,SAAS,kBAAkB,UAAuD;CAChF,MAAM,UAAU,kBAAkB;CAClC,MAAM,+BAAe,IAAI,IAA8B;CAEvD,KAAK,MAAM,SAAS,SAAS,QAAQ;EACnC,MAAM,aAAa,aAAa,KAAK;EACrC,IAAI,CAAC,YAAY;EAEjB,WAAW,SAAS,YAAY,KAAK;EAErC,MAAM,UAAU,MAAM,WAAW;EACjC,IAAI,SAAS,WAAW,YAAY,cAAc,SAAS,iBAAiB,GAAG,YAAY,KAAK;CAClG;CAEA,MAAM,SAA8C,CAAC;CACrD,KAAK,MAAM,CAAC,SAAS,gBAAgB,cACnC,OAAO,WAAW,QAAQ,WAAW;CAEvC,OAAO;EAAE,SAAS,QAAQ,OAAO;EAAG;CAAO;AAC7C;;AAGA,MAAM,qBAAqB;AAE3B,MAAM,gBAAgB,UACpB,CAAC,SACA,OAAO,KAAK,MAAM,MAAM,CAAC,CAAC,WAAW,KACpC,MAAM,QAAQ,gBAAgB,KAAA,KAC9B,MAAM,QAAQ,iBAAiB,KAAA;;;;;;;;;AAUnC,MAAa,2BAA2B,EACtC,aACA,sBAI8C;CAC9C,MAAM,SAAS,gBAAgB;CAC/B,MAAM,UAAU,gBAAiC,YAAY,KAAI,eAAc,WAAW,EAAE,CAAC;CAE7F,OAAO,WAAW;EAChB,SAAS,YAAY,KAAK,EAAE,IAAI,aAAa;GAC3C,MAAM,WAAW,WAAW,aAAa,WAAW;GACpD,OAAO;IACL,UAAU;KAAC;KAAyB;KAAI;IAAQ;IAChD,SAAS,YACP,kBACE,MAAM,OAAO,mBAAmB;KAC9B,MAAM,CAAC,oBAAoB,mBAAmB;KAC9C,aAAa;KACb,SAAS,CAAC,WAAW,MAAM;KAC3B,SAAS,EAAE,cAAc,GAAG;IAC9B,CAAC,CACH;IACF,SAAS,mBAAmB;IAC5B,kBAAkB,UAChB,aAAa,MAAM,MAAM,IAAI,KAAK,MAAM,MAAM,kBAAkB,qBAAqBA,qBAAmB;GAC5G;EACF,CAAC;EACD;CACF,CAAC;AACH;;;;;;;;;;ACvEA,MAAa,wBAAwB,EACnC,cACA,kBACA,iBACA,mBAC8B;CAC9B,MAAM,SAAS,gBAAgB;CAC/B,MAAM,YAAY,QAAQ,YAAY,KAAK;CAG3C,MAAM,QAAQ,SAAS;EACrB,UAAU,CAAC,sBAAsB,YAAY;EAC7C,iBAJe,qBAAqB,aAAa,qBAAqB,YAI1C,MAAO;EACnC,SAAS,YAAwC;GAE/C,IAAI,CAAC,cAAc,MAAM,IAAI,MAAM,0BAA0B;GAC7D,MAAM,UAAU,EAAE,aAAa;GAE/B,MAAM,CAAC,QAAQ,aAAa,QAAQ,MAAM,QAAQ,IAAI;IACpD,OAAO,mBAAmB;KACxB,MAAM,CAAC,mCAAmC,kCAAkC;KAC5E,aAAa;KACb;IACF,CAAC;IACD,OAAO,mBAAmB;KAAE,MAAM,CAAC,0BAA0B;KAAG,aAAa;KAAO;IAAQ,CAAC;IAC7F,OAAO,mBAAmB;KAAE,MAAM,CAAC,0BAA0B;KAAG,aAAa;KAAS;IAAQ,CAAC;GACjG,CAAC;GAED,OAAO;IACL,aAAa,OAAO,SAAS;IAC7B,eAAe,OAAO,iBAAiB;IACvC,UAAU,OAAO,YAAY;IAC7B,oBAAoB,YAAY,SAAS;IACzC,WAAW,KAAK,SAAS;GAC3B;EACF;EACA,GAAG;CACL,CAAC;CAED,OAAO;EAAE,MAAM,MAAM;EAAM,WAAW,MAAM;EAAW;CAAU;AACnE;;;;;;ACzDA,MAAa,sBAAkD,EAC7D,SACA,mBAIkC;CAClC,MAAM,SAAS,gBAAgB;CAE/B,OAAO,SAAS;EACd,UAAU,CAAC,0BAA0B,OAAO;EAC5C,SAAS,YAAY;GACnB,IAAI,CAAC,SACH,MAAM,IAAI,MAAM,sBAAsB;GAExC,OAAO,OAAO,cAAc,OAAO;EACrC;EACA,GAAG;CACL,CAAC;AACH;;;;ACvBA,MAAa,wBAAwB;AAcrC,SAAgB,+BAAgE,EAC9E,WACA,QACA,gBAK+B;CAC/B,MAAM,SAAS,gBAAgB;CAC/B,MAAM,SAAgC,EAAE,SAAA,IAA+B;CACvE,IAAI,QAAQ,YAAY,OAAO,aAAa,OAAO;CACnD,IAAI,QAAQ,UAAU,OAAO,WAAW,OAAO;CAE/C,OAAO,SAAS;EAEd,UAAU,YAAY;GAAC;GAAuB;GAAW;EAAM,IAAI,CAAC,eAAe,MAAM;EACzF,eAAgB,YAAY,OAAO,uBAAuB,WAAW,MAAM,IAAI,OAAO,gBAAgB,MAAM;EAC5G,GAAG;CACL,CAAC;AACH;;;ACzBA,MAAM,wBAAwB;AAE9B,MAAM,6BAA6B;AACnC,MAAM,mBAAmB;AAOzB,SAAS,MAAM,KAAoC;CACjD,MAAM,SAAqB,CAAC;CAC5B,KAAK,IAAI,QAAQ,GAAG,QAAQ,IAAI,QAAQ,SAAS,4BAC/C,OAAO,KAAK,IAAI,MAAM,OAAO,QAAQ,0BAA0B,CAAC;CAElE,OAAO;AACT;AAEA,SAAS,cAAiB,SAAyC;CACjE,OAAO,QAAQ,SAAQ,WAAW,OAAO,WAAW,cAAc,CAAC,OAAO,KAAK,IAAI,CAAC,CAAE;AACxF;;;;;AAMA,SAAS,cACP,eACA,gBACA,mBACoC;CACpC,MAAM,YAAY,IAAI,IAAI,aAAa;CACvC,MAAM,+BAAe,IAAI,IAA8B;CACvD,KAAK,MAAM,SAAS,eAAe,SAAQ,aAAY,SAAS,MAAM,GAAG;EACvE,MAAM,eAAe,MAAM,WAAW;EACtC,MAAM,aAAa,aAAa,KAAK;EACrC,IAAI,CAAC,gBAAgB,CAAC,UAAU,IAAI,YAAY,KAAK,CAAC,YAAY;EAClE,WAAW,YAAY,cAAc,cAAc,iBAAiB,GAAG,YAAY,KAAK;CAC1F;CAEA,MAAM,0BAAU,IAAI,IAAmC;CACvD,KAAK,MAAM,CAAC,cAAc,gBAAgB,cACxC,QAAQ,IAAI,cAAc,QAAQ,WAAW,CAAC;CAEhD,KAAK,MAAM,SAAS,kBAAkB,SAAQ,aAAY,SAAS,MAAM,GAAG;EAC1E,MAAM,eAAe,MAAM,WAAW;EACtC,IAAI,CAAC,gBAAgB,CAAC,UAAU,IAAI,YAAY,GAAG;EACnD,YAAY,SAAS,qBAA4C,CAAC,EAAE,CAAC,CAAC,gBAAgB,MAAM;CAC9F;CACA,OAAO;AACT;;;;;;;;AASA,MAAa,6BAA6B,EACxC,aACA,sBAI+D;CAC/D,MAAM,SAAS,gBAAgB;CAC/B,MAAM,gBAAgB,CAAC,GAAG,IAAI,IAAI,YAAY,KAAI,eAAc,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;CAC1F,MAAM,sBAAsB,YAAY,MAAM,EAAE,aAAa,WAAW,aAAa,WAAW,SAAS;CAEzG,OAAO,SAAS;EACd,UAAU,CAAC,4BAA4B,aAAa;EACpD,SAAS,mBAAmB,cAAc,SAAS;EACnD,SAAS,YAAY;GACnB,MAAM,UAAU,MAAM,aAAa;GACnC,MAAM,CAAC,cAAc,mBAAmB,MAAM,QAAQ,IAAI,CACxD,QAAQ,WACN,QAAQ,KAAI,UACV,OAAO,mBAAmB;IACxB,MAAM,CAAC,oBAAoB,mBAAmB;IAC9C,aAAa;IACb,SAAS,CAAC,gBAAgB,MAAM;IAChC,SAAS,EAAE,eAAe,MAAM;IAChC,OAAO,MAAM,SAAS;GACxB,CAAC,CACH,CACF,GACA,QAAQ,WACN,QAAQ,KAAI,UACV,OAAO,mBAAmB;IACxB,MAAM,CAAC,qBAAqB;IAC5B,aAAa;IACb,SAAS,CAAC,cAAc;IACxB,SAAS,EAAE,eAAe,MAAM;IAChC,OAAO,MAAM;GACf,CAAC,CACH,CACF,CACF,CAAC;GACD,OAAO,cAAc,eAAe,cAAc,YAAY,GAAG,cAAc,eAAe,CAAC;EACjG;EAEA,iBAAiB;EACjB,iBAAiB,sBAAsB,mBAAmB;CAC5D,CAAC;AACH;;;AChHA,MAAa,uBAAuB;AAWpC,SAAgB,uBAAoD,EAClE,WACA,QACA,SACA,gBAQA;CACA,MAAM,SAAS,gBAAgB;CAC/B,MAAM,EAAE,QAAQ,mBAAmB,QAAQ,wBAAwB,UAAU;CAC7E,MAAM,aAAa,QAAQ,cAAc,KAAA;CACzC,MAAM,WAAW,QAAQ,YAAY,KAAA;CAErC,MAAM,QAAQ,iBAAiB;EAE7B,UAAU,YACN;GAAC;GAAuB;GAAW;GAAY;IAAE;IAAY;IAAU;GAAQ;EAAC,IAChF;GAAC;GAAe;GAAY;IAAE;IAAY;IAAU;GAAQ;EAAC;EACjE,UAAU,EAAE,gBAAgB;GAC1B,MAAM,SAAgC;IAAE,MAAM;IAAW,SAAA;GAA8B;GACvF,IAAI,YAAY,OAAO,aAAa;GACpC,IAAI,UAAU,OAAO,WAAW;GAChC,IAAI,SAAS,OAAO,UAAU;GAC9B,OAAO,YAAY,OAAO,uBAAuB,WAAW,MAAM,IAAI,OAAO,gBAAgB,MAAM;EACrG;EACA,kBAAkB;EAClB,mBAAmB,UAAU,GAAG,kBAAkB;GAChD,IAAI,CAAC,UAAU,aAAa,UAAU,CAAC,SAAS,YAAY,SAC1D;GAEF,OAAO,gBAAgB;EACzB;EACA,SAAS,SACP,KAAK,MAAM,SAAQ,SAAQ,MAAM,eAAe,CAAC,CAAC;EACpD,GAAG;CACL,CAAC;CAED,gBAAgB;EACd,IAAI,qBAAqB,MAAM,eAAe,CAAC,MAAM,oBACnD,MAAW,cAAc;CAE7B,GAAG,CAAC,mBAAmB,KAAK,CAAC;CAE7B,OAAO;EAAE,GAAG;EAAO;CAAoB;AACzC"}