@mastra/react 1.11.1-alpha.6 → 1.12.0-alpha.11

package/dist/hooks/datasets/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":["PER_PAGE","tanstackQueryOptions"],"sources":["../../../src/hooks/datasets/use-dataset-items.ts","../../../src/hooks/datasets/use-dataset-mutations.ts","../../../src/hooks/datasets/use-datasets.ts","../../../src/hooks/datasets/use-compare-experiments.ts","../../../src/hooks/datasets/use-dataset-experiments.ts","../../../src/hooks/datasets/use-dataset-item-versions.ts","../../../src/hooks/datasets/use-dataset-versions.ts","../../../src/hooks/datasets/use-experiments.ts","../../../src/hooks/datasets/use-workflow-schema.ts"],"sourcesContent":["import type { MastraClient } from '@mastra/client-js';\nimport { useQuery, useInfiniteQuery } from '@tanstack/react-query';\nimport type { InfiniteData, QueryKey, UseInfiniteQueryResult, UseQueryResult } from '@tanstack/react-query';\nimport { useEffect } from 'react';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraInfiniteQueryOptions, MastraQueryOptions } from '../shared/query-options';\nimport { useInView } from '../shared/use-in-view';\n\n/**\n * Hook to fetch a single dataset item by ID\n */\ntype DatasetItemResponse = Awaited<ReturnType<MastraClient['getDatasetItem']>>;\n\n/**\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetItem = <TData = DatasetItemResponse>({\n  datasetId,\n  itemId,\n  queryOptions,\n}: {\n  datasetId: string;\n  itemId: string;\n  queryOptions?: MastraQueryOptions<DatasetItemResponse, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['dataset-item', datasetId, itemId],\n    queryFn: () => client.getDatasetItem(datasetId, itemId),\n    retry: false, // Don't retry 404s for deleted items\n    ...queryOptions,\n  });\n};\n\nconst PER_PAGE = 10;\n\ntype ListDatasetItemsResponse = Awaited<ReturnType<MastraClient['listDatasetItems']>>;\n\nexport type UseDatasetItemsResult = Omit<\n  UseInfiniteQueryResult<InfiniteData<ListDatasetItemsResponse>, Error>,\n  'data'\n> & {\n  data: ListDatasetItemsResponse['items'];\n  total: number | undefined;\n  setEndOfListElement: (element: HTMLDivElement | null) => void;\n};\n\n/**\n * Hook to list items in a dataset with infinite scroll pagination and optional search\n * @param version - Optional version timestamp to view historical snapshot\n */\nexport type DatasetItemsOrderBy = NonNullable<NonNullable<Parameters<MastraClient['listDatasetItems']>[1]>['orderBy']>;\n\n/**\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetItems = ({\n  datasetId,\n  search,\n  version,\n  orderBy,\n  queryOptions,\n}: {\n  datasetId: string;\n  search?: string;\n  version?: number | null;\n  orderBy?: DatasetItemsOrderBy;\n  queryOptions?: MastraInfiniteQueryOptions<\n    ListDatasetItemsResponse,\n    InfiniteData<ListDatasetItemsResponse, number>,\n    QueryKey,\n    number\n  >;\n}): UseDatasetItemsResult => {\n  const client = useMastraClient();\n  const { inView: isEndOfListInView, setRef: setEndOfListElement } = useInView();\n\n  const query = useInfiniteQuery({\n    queryKey: ['dataset-items', datasetId, search, version, orderBy],\n    queryFn: async ({ pageParam }) => {\n      const res = await client.listDatasetItems(datasetId, {\n        page: pageParam,\n        perPage: PER_PAGE,\n        search: search || undefined,\n        version: version || undefined,\n        orderBy,\n      });\n      return res;\n    },\n    initialPageParam: 0,\n    getNextPageParam: (lastPage, _, lastPageParam) => {\n      if (!lastPage?.items?.length) {\n        return undefined;\n      }\n      const totalFetched = (lastPageParam + 1) * PER_PAGE;\n      const total = lastPage?.pagination?.total ?? 0;\n      if (totalFetched >= total) {\n        return undefined;\n      }\n      return lastPageParam + 1;\n    },\n    retry: false,\n    ...queryOptions,\n  });\n\n  const items = query.data?.pages.flatMap(page => page?.items ?? []) ?? [];\n  const total = query.data?.pages[0]?.pagination?.total;\n\n  useEffect(() => {\n    if (isEndOfListInView && query.hasNextPage && !query.isFetchingNextPage) {\n      void query.fetchNextPage();\n    }\n  }, [isEndOfListInView, query]);\n\n  return { ...query, data: items, total, setEndOfListElement };\n};\n","import type {\n  CreateDatasetParams,\n  UpdateDatasetParams,\n  AddDatasetItemParams,\n  UpdateDatasetItemParams,\n  TriggerDatasetExperimentParams,\n  UpdateDatasetExperimentParams,\n  UpdateExperimentResultParams,\n  BatchInsertDatasetItemsParams,\n  BatchDeleteDatasetItemsParams,\n  MastraClient,\n} from '@mastra/client-js';\nimport { useMutation, useQueryClient } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraMutationOptions } from '../shared/query-options';\n\ntype DatasetItemMutationVariables = {\n  datasetId: string;\n  itemId: string;\n};\n\ntype ClientResult<K extends keyof MastraClient> = MastraClient[K] extends (...args: any[]) => Promise<infer R>\n  ? R\n  : never;\n\nexport interface DatasetMutationsQueryOptions {\n  createDataset?: MastraMutationOptions<ClientResult<'createDataset'>, CreateDatasetParams>;\n  updateDataset?: MastraMutationOptions<ClientResult<'updateDataset'>, UpdateDatasetParams>;\n  deleteDataset?: MastraMutationOptions<ClientResult<'deleteDataset'>, string>;\n  addItem?: MastraMutationOptions<ClientResult<'addDatasetItem'>, AddDatasetItemParams>;\n  updateItem?: MastraMutationOptions<ClientResult<'updateDatasetItem'>, UpdateDatasetItemParams>;\n  deleteItem?: MastraMutationOptions<ClientResult<'deleteDatasetItem'>, DatasetItemMutationVariables>;\n  purgeItem?: MastraMutationOptions<ClientResult<'purgeDatasetItem'>, DatasetItemMutationVariables>;\n  deleteItems?: MastraMutationOptions<\n    ClientResult<'batchDeleteDatasetItems'>,\n    { datasetId: string; itemIds: string[] }\n  >;\n  batchInsertItems?: MastraMutationOptions<ClientResult<'batchInsertDatasetItems'>, BatchInsertDatasetItemsParams>;\n  batchDeleteItems?: MastraMutationOptions<ClientResult<'batchDeleteDatasetItems'>, BatchDeleteDatasetItemsParams>;\n  triggerExperiment?: MastraMutationOptions<ClientResult<'triggerDatasetExperiment'>, TriggerDatasetExperimentParams>;\n  deleteExperiment?: MastraMutationOptions<ClientResult<'deleteExperiment'>, string>;\n  updateExperiment?: MastraMutationOptions<ClientResult<'updateDatasetExperiment'>, UpdateDatasetExperimentParams>;\n  updateExperimentResult?: MastraMutationOptions<\n    ClientResult<'updateDatasetExperimentResult'>,\n    UpdateExperimentResultParams\n  >;\n}\n\n/**\n * Hook providing mutation functions for datasets, items, and runs\n * All mutations invalidate relevant query caches on success.\n * `queryOptions` is keyed by returned mutation name and spread last; a user `onSuccess` replaces the internal invalidation.\n */\nexport const useDatasetMutations = ({ queryOptions }: { queryOptions?: DatasetMutationsQueryOptions } = {}) => {\n  const client = useMastraClient();\n  const queryClient = useQueryClient();\n\n  const createDataset = useMutation({\n    mutationFn: (params: CreateDatasetParams) => client.createDataset(params),\n    onSuccess: () => {\n      void queryClient.invalidateQueries({ queryKey: ['datasets'] });\n    },\n    ...queryOptions?.createDataset,\n  });\n\n  const updateDataset = useMutation({\n    mutationFn: (params: UpdateDatasetParams) => client.updateDataset(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['datasets'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset', variables.datasetId] });\n    },\n    ...queryOptions?.updateDataset,\n  });\n\n  const deleteDataset = useMutation({\n    mutationFn: (datasetId: string) => client.deleteDataset(datasetId),\n    onSuccess: () => {\n      void queryClient.invalidateQueries({ queryKey: ['datasets'] });\n    },\n    ...queryOptions?.deleteDataset,\n  });\n\n  const addItem = useMutation({\n    mutationFn: (params: AddDatasetItemParams) => client.addDatasetItem(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-versions', variables.datasetId] });\n    },\n    ...queryOptions?.addItem,\n  });\n\n  const updateItem = useMutation({\n    mutationFn: (params: UpdateDatasetItemParams) => client.updateDatasetItem(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-item', variables.datasetId, variables.itemId] });\n      void queryClient.invalidateQueries({\n        queryKey: ['dataset-item-versions', variables.datasetId, variables.itemId],\n      });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-versions', variables.datasetId] });\n    },\n    ...queryOptions?.updateItem,\n  });\n\n  const deleteItem = useMutation({\n    mutationFn: ({ datasetId, itemId }: DatasetItemMutationVariables) => client.deleteDatasetItem(datasetId, itemId),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-versions', variables.datasetId] });\n    },\n    ...queryOptions?.deleteItem,\n  });\n\n  const purgeItem = useMutation({\n    mutationFn: ({ datasetId, itemId }: DatasetItemMutationVariables) => client.purgeDatasetItem(datasetId, itemId),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-item', variables.datasetId, variables.itemId] });\n      void queryClient.invalidateQueries({\n        queryKey: ['dataset-item-versions', variables.datasetId, variables.itemId],\n      });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiment-results'] });\n      void queryClient.invalidateQueries({ queryKey: ['experiment-results'] });\n      void queryClient.invalidateQueries({ queryKey: ['review-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-review-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-completed-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['experiment-review-summary'] });\n    },\n    ...queryOptions?.purgeItem,\n  });\n\n  // Batch insert items using the batch endpoint\n  const batchInsertItems = useMutation({\n    mutationFn: (params: BatchInsertDatasetItemsParams) => client.batchInsertDatasetItems(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-versions', variables.datasetId] });\n    },\n    ...queryOptions?.batchInsertItems,\n  });\n\n  // Batch delete items using the batch endpoint\n  const batchDeleteItems = useMutation({\n    mutationFn: (params: BatchDeleteDatasetItemsParams) => client.batchDeleteDatasetItems(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-versions', variables.datasetId] });\n    },\n    ...queryOptions?.batchDeleteItems,\n  });\n\n  // @deprecated - use batchDeleteItems mutation instead\n  const deleteItems = useMutation({\n    mutationFn: async ({ datasetId, itemIds }: { datasetId: string; itemIds: string[] }) => {\n      return client.batchDeleteDatasetItems({ datasetId, itemIds });\n    },\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-versions', variables.datasetId] });\n    },\n    ...queryOptions?.deleteItems,\n  });\n\n  const triggerExperiment = useMutation({\n    mutationFn: (params: TriggerDatasetExperimentParams) => client.triggerDatasetExperiment(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiments', variables.datasetId] });\n    },\n    ...queryOptions?.triggerExperiment,\n  });\n\n  const deleteExperiment = useMutation({\n    mutationFn: (experimentId: string) => client.deleteExperiment(experimentId),\n    onSuccess: () => {\n      void queryClient.invalidateQueries({ queryKey: ['experiments'] });\n      void queryClient.invalidateQueries({ queryKey: ['experiment'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiments'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiment'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiment-results'] });\n      void queryClient.invalidateQueries({ queryKey: ['review-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['completed-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['experiment-review-summary'] });\n    },\n    ...queryOptions?.deleteExperiment,\n  });\n\n  const updateExperiment = useMutation({\n    mutationFn: (params: UpdateDatasetExperimentParams) => client.updateDatasetExperiment(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['experiments'] });\n      void queryClient.invalidateQueries({ queryKey: ['experiment', variables.experimentId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiments', variables.datasetId] });\n      void queryClient.invalidateQueries({\n        queryKey: ['dataset-experiment', variables.datasetId, variables.experimentId],\n      });\n    },\n    ...queryOptions?.updateExperiment,\n  });\n\n  const updateExperimentResult = useMutation({\n    mutationFn: (params: UpdateExperimentResultParams) => client.updateDatasetExperimentResult(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['experiment-results', variables.experimentId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiment-results'] });\n      void queryClient.invalidateQueries({ queryKey: ['review-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-review-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-completed-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['experiment-review-summary'] });\n    },\n    ...queryOptions?.updateExperimentResult,\n  });\n\n  return {\n    createDataset,\n    updateDataset,\n    deleteDataset,\n    addItem,\n    updateItem,\n    deleteItem,\n    purgeItem,\n    deleteItems,\n    batchInsertItems,\n    batchDeleteItems,\n    triggerExperiment,\n    deleteExperiment,\n    updateExperiment,\n    updateExperimentResult,\n  };\n};\n","import type { ExperimentTargetType, ListDatasetsParams, MastraClient } from '@mastra/client-js';\nimport { useInfiniteQuery, useQuery } from '@tanstack/react-query';\nimport type { InfiniteData, QueryKey, UseInfiniteQueryResult, UseQueryResult } from '@tanstack/react-query';\nimport { useEffect } from 'react';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraInfiniteQueryOptions, MastraQueryOptions } from '../shared/query-options';\nimport { useInView } from '../shared/use-in-view';\n\ntype ListDatasetsResponse = Awaited<ReturnType<MastraClient['listDatasets']>>;\ntype DatasetResponse = Awaited<ReturnType<MastraClient['getDataset']>>;\n\n/**\n * Hook to list all datasets with optional pagination\n */\nexport const useDatasets = <TData = ListDatasetsResponse>({\n  pagination,\n  queryOptions,\n}: {\n  pagination?: { page?: number; perPage?: number };\n  queryOptions?: MastraQueryOptions<ListDatasetsResponse, TData>;\n} = {}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['datasets', pagination],\n    queryFn: () => client.listDatasets(pagination),\n    placeholderData: previousData => previousData,\n    ...queryOptions,\n  });\n};\n\nconst DATASETS_PER_PAGE = 20;\n\nexport interface DatasetTargetFilter {\n  targetType?: ExperimentTargetType | '';\n  targetId?: string;\n}\n\nexport type DatasetsOrderBy = NonNullable<ListDatasetsParams['orderBy']>;\n\ntype DatasetRecord = ListDatasetsResponse['datasets'][number];\n\nexport type UseInfiniteDatasetsResult<TData = DatasetRecord[]> = UseInfiniteQueryResult<TData, Error> & {\n  setEndOfListElement: (element: HTMLDivElement | null) => void;\n};\n\n/**\n * Hook to list datasets with infinite scroll pagination, optionally scoped server-side to a target.\n * `data` defaults to the flattened dataset list; a user `select` replaces that flattening.\n */\nexport const useInfiniteDatasets = <TData = DatasetRecord[]>({\n  filter,\n  orderBy,\n  queryOptions,\n}: {\n  filter?: DatasetTargetFilter;\n  orderBy?: DatasetsOrderBy;\n  queryOptions?: MastraInfiniteQueryOptions<ListDatasetsResponse, TData, QueryKey, number>;\n} = {}): UseInfiniteDatasetsResult<TData> => {\n  const client = useMastraClient();\n  const { inView: isEndOfListInView, setRef: setEndOfListElement } = useInView();\n  const targetType = filter?.targetType || undefined;\n  const targetIds = filter?.targetId ? [filter.targetId] : undefined;\n\n  const query = useInfiniteQuery({\n    queryKey: ['datasets', 'infinite', { targetType, targetIds, orderBy }],\n    queryFn: ({ pageParam }) =>\n      client.listDatasets({ page: pageParam, perPage: DATASETS_PER_PAGE, targetType, targetIds, orderBy }),\n    initialPageParam: 0,\n    getNextPageParam: (lastPage, _, lastPageParam) => {\n      if (!lastPage?.datasets?.length || !lastPage.pagination?.hasMore) {\n        return undefined;\n      }\n      return lastPageParam + 1;\n    },\n    select: (data: InfiniteData<ListDatasetsResponse, number>) =>\n      data.pages.flatMap(page => page?.datasets ?? []) 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\n/**\n * Hook to fetch a single dataset by ID\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDataset = <TData = DatasetResponse>({\n  datasetId,\n  queryOptions,\n}: {\n  datasetId: string;\n  queryOptions?: MastraQueryOptions<DatasetResponse, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['dataset', datasetId],\n    queryFn: () => client.getDataset(datasetId),\n    ...queryOptions,\n  });\n};\n","import type { CompareExperimentsResponse } 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/**\n * Hook to compare two to ten dataset experiments item by item against a baseline.\n * @param datasetId - ID of the dataset\n * @param experimentIds - IDs of the experiments to compare\n * @param baselineId - ID of the baseline experiment; must be one of `experimentIds` (defaults to the first)\n * @param queryOptions - TanStack Query options, spread last to override defaults\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useCompareExperiments = <TData = CompareExperimentsResponse>({\n  datasetId,\n  experimentIds,\n  baselineId,\n  queryOptions,\n}: {\n  datasetId: string;\n  experimentIds: string[];\n  baselineId?: string;\n  queryOptions?: MastraQueryOptions<CompareExperimentsResponse, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['compare-experiments', datasetId, experimentIds, baselineId],\n    queryFn: () => client.compareExperiments({ datasetId, experimentIds, baselineId }),\n    ...queryOptions,\n  });\n};\n","import type { ClientScoreRowData, DatasetExperimentResult, MastraClient } from '@mastra/client-js';\nimport type { ExperimentStatus } from '@mastra/core/storage';\nimport { queryOptions as tanstackQueryOptions, useInfiniteQuery, useQueries, useQuery } from '@tanstack/react-query';\nimport type {\n  InfiniteData,\n  QueryKey,\n  UseInfiniteQueryResult,\n  UseQueryOptions,\n  UseQueryResult,\n} from '@tanstack/react-query';\nimport { useEffect } from 'react';\nimport { useMastraClient } from '../../mastra-client-context';\nimport { useCombineByIds } from '../shared/combine-by-ids';\nimport type { ExperimentQueriesResult } from '../shared/combine-by-ids';\nimport type { MastraInfiniteQueryOptions, MastraQueryOptions } from '../shared/query-options';\n\nexport type { ExperimentQueriesResult } from '../shared/combine-by-ids';\nimport { useInView } from '../shared/use-in-view';\n\ntype DatasetExperimentResponse = Awaited<ReturnType<MastraClient['getDatasetExperiment']>>;\ntype ExperimentResultsPage = Awaited<ReturnType<MastraClient['listDatasetExperimentResults']>>;\ntype ExperimentScoresByEntity = Record<string, ClientScoreRowData[]>;\ntype DatasetExperimentQueryKey = string[];\ntype ExperimentStatusQueryKey = (string | undefined)[];\n\nconst POLL_INTERVAL_MS = 2000;\nconst RESULTS_PER_PAGE = 100;\nconst SCORES_PER_PAGE = 100;\n\nconst isInProgress = (status: string | undefined) => status === 'running' || status === 'pending';\n\n/**\n * Key, fetcher and polling policy of the `dataset-experiment` query, shared by the single and\n * multi-experiment hooks so they hit the same cache entry.\n * Polls every 2 seconds while the experiment is 'running' or 'pending'.\n */\nexport const datasetExperimentQueryOptions = (\n  client: MastraClient,\n  datasetId: string,\n  experimentId: string,\n): UseQueryOptions<DatasetExperimentResponse, Error, DatasetExperimentResponse, DatasetExperimentQueryKey> =>\n  tanstackQueryOptions({\n    queryKey: ['dataset-experiment', datasetId, experimentId],\n    queryFn: () => client.getDatasetExperiment(datasetId, experimentId),\n    gcTime: 0,\n    staleTime: 0,\n    refetchInterval: query => (isInProgress(query.state.data?.status) ? POLL_INTERVAL_MS : false),\n  });\n\n/**\n * Key and fetcher of the `dataset-experiment-scores` query: every score of an experiment, across all\n * pages, grouped by dataset item id. Polls every 2 seconds while `experimentStatus` is in progress.\n */\nexport const experimentScoresQueryOptions = (\n  client: MastraClient,\n  experimentId: string,\n  experimentStatus: ExperimentStatus | undefined,\n): UseQueryOptions<ExperimentScoresByEntity, Error, ExperimentScoresByEntity, ExperimentStatusQueryKey> =>\n  tanstackQueryOptions({\n    queryKey: ['dataset-experiment-scores', experimentId, experimentStatus],\n    queryFn: async (): Promise<ExperimentScoresByEntity> => {\n      const allScores: ClientScoreRowData[] = [];\n      let page = 0;\n\n      while (true) {\n        const response = await client.listScoresByRunId({ runId: experimentId, page, perPage: SCORES_PER_PAGE });\n        allScores.push(...response.scores);\n        if (!response.pagination.hasMore) break;\n        page++;\n      }\n\n      const grouped: ExperimentScoresByEntity = {};\n      for (const row of allScores) {\n        (grouped[row.entityId] ??= []).push(row);\n      }\n      return grouped;\n    },\n    refetchInterval: isInProgress(experimentStatus) ? POLL_INTERVAL_MS : false,\n  });\n\n/**\n * Key and fetcher of the `dataset-experiment-results-all` query: every result of an experiment, across\n * all pages. Polls every 2 seconds while `experimentStatus` is in progress.\n */\nexport const allDatasetExperimentResultsQueryOptions = (\n  client: MastraClient,\n  datasetId: string,\n  experimentId: string,\n  experimentStatus: ExperimentStatus | undefined,\n): UseQueryOptions<DatasetExperimentResult[], Error, DatasetExperimentResult[], ExperimentStatusQueryKey> =>\n  tanstackQueryOptions({\n    queryKey: ['dataset-experiment-results-all', datasetId, experimentId, experimentStatus],\n    queryFn: async (): Promise<DatasetExperimentResult[]> => {\n      const allResults: DatasetExperimentResult[] = [];\n      let page = 0;\n\n      while (true) {\n        const response = await client.listDatasetExperimentResults(datasetId, experimentId, {\n          page,\n          perPage: RESULTS_PER_PAGE,\n        });\n        allResults.push(...response.results);\n        if (!response.pagination.hasMore || response.results.length === 0) break;\n        page++;\n      }\n\n      return allResults;\n    },\n    refetchInterval: isInProgress(experimentStatus) ? POLL_INTERVAL_MS : false,\n  });\n\n/** An experiment id plus its status, which drives polling of its dependent queries. */\nexport interface ExperimentRef {\n  id: string;\n  status?: ExperimentStatus;\n}\n\n/**\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetExperiment = <TData = DatasetExperimentResponse>({\n  datasetId,\n  experimentId,\n  queryOptions,\n}: {\n  datasetId: string;\n  experimentId: string;\n  queryOptions?: MastraQueryOptions<DatasetExperimentResponse, TData, DatasetExperimentQueryKey>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  const { queryKey, queryFn, gcTime, staleTime, refetchInterval } = datasetExperimentQueryOptions(\n    client,\n    datasetId,\n    experimentId,\n  );\n  return useQuery<DatasetExperimentResponse, Error, TData, DatasetExperimentQueryKey>({\n    queryKey,\n    queryFn,\n    gcTime,\n    staleTime,\n    refetchInterval,\n    ...queryOptions,\n  });\n};\n\n/**\n * Fetches several experiments of one dataset in parallel, each polled while it runs.\n * The server 404s an experiment that does not belong to `datasetId`, which surfaces as `error`.\n */\nexport const useDatasetExperimentsQueries = ({\n  datasetId,\n  experimentIds,\n}: {\n  datasetId: string;\n  experimentIds: string[];\n}): ExperimentQueriesResult<DatasetExperimentResponse> => {\n  const client = useMastraClient();\n  const combine = useCombineByIds<DatasetExperimentResponse>(experimentIds);\n  return useQueries({\n    queries: experimentIds.map(id => datasetExperimentQueryOptions(client, datasetId, id)),\n    combine,\n  });\n};\n\nexport type ExperimentResultsOrderBy = NonNullable<\n  NonNullable<Parameters<MastraClient['listDatasetExperimentResults']>[2]>['orderBy']\n>;\n\ninterface UseDatasetExperimentResultsParams<TData = DatasetExperimentResult[]> {\n  datasetId: string;\n  experimentId: string;\n  experimentStatus?: ExperimentStatus;\n  orderBy?: ExperimentResultsOrderBy;\n  queryOptions?: MastraInfiniteQueryOptions<ExperimentResultsPage, TData, QueryKey, number>;\n}\n\n/**\n * Hook to list results for a dataset experiment with infinite scroll pagination.\n * Polls every 2 seconds while experiment status is 'pending' or 'running'.\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetExperimentResults = <TData = DatasetExperimentResult[]>({\n  datasetId,\n  experimentId,\n  experimentStatus,\n  orderBy,\n  queryOptions,\n}: UseDatasetExperimentResultsParams<TData>): UseInfiniteQueryResult<TData, Error> & {\n  setEndOfListElement: (element: HTMLDivElement | null) => void;\n} => {\n  const client = useMastraClient();\n  const { inView: isEndOfListInView, setRef: setEndOfListElement } = useInView();\n\n  const query = useInfiniteQuery({\n    queryKey: ['dataset-experiment-results', datasetId, experimentId, experimentStatus, orderBy],\n    queryFn: async ({ pageParam }) => {\n      return client.listDatasetExperimentResults(datasetId, experimentId, {\n        page: pageParam,\n        perPage: RESULTS_PER_PAGE,\n        orderBy,\n      });\n    },\n    initialPageParam: 0,\n    getNextPageParam: (lastPage, _, lastPageParam) => {\n      if (!lastPage?.results?.length) {\n        return undefined;\n      }\n      const totalFetched = (lastPageParam + 1) * RESULTS_PER_PAGE;\n      const total = lastPage?.pagination?.total ?? 0;\n      if (totalFetched >= total) {\n        return undefined;\n      }\n      return lastPageParam + 1;\n    },\n    refetchInterval: isInProgress(experimentStatus) ? POLL_INTERVAL_MS : false,\n    select: (data: InfiniteData<ExperimentResultsPage, number>) => {\n      return data.pages.flatMap(page => page?.results ?? []) as TData;\n    },\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\n/**\n * Fetches every result (all pages) of several experiments of one dataset in parallel.\n * Each experiment is polled while its `status` is in progress.\n */\nexport const useAllDatasetExperimentResultsQueries = ({\n  datasetId,\n  experiments,\n}: {\n  datasetId: string;\n  experiments: ExperimentRef[];\n}): ExperimentQueriesResult<DatasetExperimentResult[]> => {\n  const client = useMastraClient();\n  const combine = useCombineByIds<DatasetExperimentResult[]>(experiments.map(experiment => experiment.id));\n  return useQueries({\n    queries: experiments.map(experiment =>\n      allDatasetExperimentResultsQueryOptions(client, datasetId, experiment.id, experiment.status),\n    ),\n    combine,\n  });\n};\n\n/**\n * Hook to fetch all scores for an experiment, transformed to Record<entityId, ClientScoreRowData[]>\n * Paginates through all pages to ensure no scores are silently dropped.\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useScoresByExperimentId = <TData = ExperimentScoresByEntity>({\n  experimentId,\n  experimentStatus,\n  queryOptions,\n}: {\n  experimentId: string;\n  experimentStatus?: ExperimentStatus;\n  queryOptions?: MastraQueryOptions<ExperimentScoresByEntity, TData, ExperimentStatusQueryKey>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  const { queryKey, queryFn, refetchInterval } = experimentScoresQueryOptions(client, experimentId, experimentStatus);\n  return useQuery<ExperimentScoresByEntity, Error, TData, ExperimentStatusQueryKey>({\n    queryKey,\n    queryFn,\n    refetchInterval,\n    ...queryOptions,\n  });\n};\n\n/**\n * Fetches every score of several experiments in parallel, each grouped by dataset item id.\n * Each experiment is polled while its `status` is in progress.\n */\nexport const useScoresByExperimentIdsQueries = ({\n  experiments,\n}: {\n  experiments: ExperimentRef[];\n}): ExperimentQueriesResult<ExperimentScoresByEntity> => {\n  const client = useMastraClient();\n  const combine = useCombineByIds<ExperimentScoresByEntity>(experiments.map(experiment => experiment.id));\n  return useQueries({\n    queries: experiments.map(experiment => experimentScoresQueryOptions(client, experiment.id, experiment.status)),\n    combine,\n  });\n};\n","import type { DatasetItemVersionResponse } 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\nexport type DatasetItemVersion = DatasetItemVersionResponse & {\n  validTo: number | null;\n  isDeleted: boolean;\n  isLatest: boolean;\n};\n\n/**\n * Hook to fetch full item history (SCD-2 rows).\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetItemVersions = <TData = DatasetItemVersion[]>({\n  datasetId,\n  itemId,\n  queryOptions,\n}: {\n  datasetId: string;\n  itemId: string;\n  queryOptions?: MastraQueryOptions<DatasetItemVersion[], TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['dataset-item-versions', datasetId, itemId],\n    queryFn: async (): Promise<DatasetItemVersion[]> => {\n      const res = await client.getItemHistory(datasetId, itemId);\n\n      return (res?.history ?? []).map(\n        (version, index): DatasetItemVersion => ({\n          id: version.id,\n          datasetId: version.datasetId,\n          datasetVersion: version.datasetVersion,\n          input: version.input,\n          groundTruth: version.groundTruth,\n          expectedTrajectory: version.expectedTrajectory,\n          toolMocks: version.toolMocks ?? undefined,\n          scorerIds: version.scorerIds ?? undefined,\n          requestContext: version.requestContext ?? undefined,\n          metadata: version.metadata ?? undefined,\n          validTo: version.validTo,\n          isDeleted: version.isDeleted,\n          createdAt: version.createdAt,\n          updatedAt: version.updatedAt,\n          isLatest: index === 0,\n        }),\n      );\n    },\n    ...queryOptions,\n  });\n};\n\n/**\n * Hook to fetch a specific version of a dataset item.\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetItemVersion = <TData = DatasetItemVersion>({\n  datasetId,\n  itemId,\n  datasetVersion,\n  latestVersion,\n  queryOptions,\n}: {\n  datasetId: string;\n  itemId: string;\n  datasetVersion: number;\n  latestVersion?: number;\n  queryOptions?: MastraQueryOptions<DatasetItemVersion, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['dataset-item-version', datasetId, itemId, datasetVersion],\n    queryFn: async (): Promise<DatasetItemVersion> => {\n      const v = await client.getDatasetItemVersion(datasetId, itemId, datasetVersion);\n\n      return {\n        id: v.id,\n        datasetId: v.datasetId,\n        datasetVersion: v.datasetVersion,\n        input: v.input,\n        groundTruth: v.groundTruth,\n        expectedTrajectory: v.expectedTrajectory,\n        toolMocks: v.toolMocks ?? undefined,\n        scorerIds: v.scorerIds ?? undefined,\n        requestContext: v.requestContext ?? undefined,\n        metadata: v.metadata ?? undefined,\n        validTo: null,\n        isDeleted: false,\n        createdAt: v.createdAt,\n        updatedAt: v.updatedAt,\n        isLatest: latestVersion != null ? datasetVersion === latestVersion : false,\n      };\n    },\n    ...queryOptions,\n  });\n};\n","import type { MastraClient } from '@mastra/client-js';\nimport { useInfiniteQuery } from '@tanstack/react-query';\nimport type { InfiniteData, QueryKey } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraInfiniteQueryOptions } from '../shared/query-options';\n\nexport interface DatasetVersion {\n  id?: string;\n  datasetId?: string;\n  version: number;\n  createdAt?: Date | string;\n  isCurrent: boolean;\n}\n\ntype DatasetVersionsPage = Awaited<ReturnType<MastraClient['listDatasetVersions']>>;\n\nconst PER_PAGE = 10;\n\n/**\n * Hook to fetch dataset versions from the API with infinite pagination.\n * `data` defaults to the flattened `DatasetVersion[]`; a user `select` replaces that.\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetVersions = <TData = DatasetVersion[]>({\n  datasetId,\n  queryOptions,\n}: {\n  datasetId: string;\n  queryOptions?: MastraInfiniteQueryOptions<DatasetVersionsPage, TData, QueryKey, number>;\n}) => {\n  const client = useMastraClient();\n\n  return useInfiniteQuery({\n    queryKey: ['dataset-versions', datasetId],\n    queryFn: async ({ pageParam }) => {\n      return client.listDatasetVersions(datasetId, { page: pageParam, perPage: PER_PAGE });\n    },\n    initialPageParam: 0,\n    getNextPageParam: (lastPage, _, lastPageParam) => {\n      if (lastPage?.pagination?.hasMore) {\n        return lastPageParam + 1;\n      }\n      return undefined;\n    },\n    select: (data: InfiniteData<DatasetVersionsPage, number>) => {\n      return data.pages\n        .flatMap(page => page?.versions ?? [])\n        .map(\n          (v, index): DatasetVersion => ({\n            id: v.id,\n            datasetId: v.datasetId,\n            version: v.version,\n            createdAt: v.createdAt,\n            isCurrent: index === 0,\n          }),\n        ) as TData;\n    },\n    ...queryOptions,\n  });\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 ListExperimentsResponse = Awaited<ReturnType<MastraClient['listExperiments']>>;\ntype GetExperimentResponse = Awaited<ReturnType<MastraClient['getExperiment']>>;\n\n/**\n * Hook to list all experiments across all datasets with optional pagination\n */\nexport const useExperiments = <TData = ListExperimentsResponse>({\n  pagination,\n  queryOptions,\n}: {\n  pagination?: { page?: number; perPage?: number };\n  queryOptions?: MastraQueryOptions<ListExperimentsResponse, TData>;\n} = {}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['experiments', pagination],\n    queryFn: () => client.listExperiments(pagination),\n    ...queryOptions,\n  });\n};\n\n/**\n * Hook to fetch a single experiment by ID without knowing its dataset\n */\nexport const useExperiment = <TData = GetExperimentResponse>({\n  experimentId,\n  queryOptions,\n}: {\n  experimentId: string;\n  queryOptions?: MastraQueryOptions<GetExperimentResponse, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['experiment', experimentId],\n    queryFn: () => client.getExperiment(experimentId),\n    ...queryOptions,\n  });\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 WorkflowSchemaResponse = Awaited<ReturnType<ReturnType<MastraClient['getWorkflow']>['getSchema']>>;\n\n/**\n * Hook to fetch workflow input/output schema by workflow ID.\n * Returns { inputSchema, outputSchema } where each is Record<string, unknown> | null.\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport function useWorkflowSchema<TData = WorkflowSchemaResponse>({\n  workflowId,\n  queryOptions,\n}: {\n  workflowId: string | null;\n  queryOptions?: MastraQueryOptions<WorkflowSchemaResponse, TData>;\n}): UseQueryResult<TData, Error> {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['workflow-schema', workflowId],\n    queryFn: async () => {\n      if (!workflowId) throw new Error('No workflow selected');\n      return client.getWorkflow(workflowId).getSchema();\n    },\n    staleTime: 5 * 60 * 1000, // Cache for 5 minutes\n    ...queryOptions,\n  });\n}\n"],"mappings":";;;;;;;;;AAgBA,MAAa,kBAA+C,EAC1D,WACA,QACA,mBAKkC;CAClC,MAAM,SAAS,gBAAgB;CAC/B,OAAO,SAAS;EACd,UAAU;GAAC;GAAgB;GAAW;EAAM;EAC5C,eAAe,OAAO,eAAe,WAAW,MAAM;EACtD,OAAO;EACP,GAAG;CACL,CAAC;AACH;AAEA,MAAMA,aAAW;;;;AAsBjB,MAAa,mBAAmB,EAC9B,WACA,QACA,SACA,SACA,mBAY2B;CAC3B,MAAM,SAAS,gBAAgB;CAC/B,MAAM,EAAE,QAAQ,mBAAmB,QAAQ,wBAAwB,UAAU;CAE7E,MAAM,QAAQ,iBAAiB;EAC7B,UAAU;GAAC;GAAiB;GAAW;GAAQ;GAAS;EAAO;EAC/D,SAAS,OAAO,EAAE,gBAAgB;GAQhC,OAAO,MAPW,OAAO,iBAAiB,WAAW;IACnD,MAAM;IACN,SAASA;IACT,QAAQ,UAAU,KAAA;IAClB,SAAS,WAAW,KAAA;IACpB;GACF,CAAC;EAEH;EACA,kBAAkB;EAClB,mBAAmB,UAAU,GAAG,kBAAkB;GAChD,IAAI,CAAC,UAAU,OAAO,QACpB;GAIF,KAFsB,gBAAgB,KAAKA,eAC7B,UAAU,YAAY,SAAS,IAE3C;GAEF,OAAO,gBAAgB;EACzB;EACA,OAAO;EACP,GAAG;CACL,CAAC;CAED,MAAM,QAAQ,MAAM,MAAM,MAAM,SAAQ,SAAQ,MAAM,SAAS,CAAC,CAAC,KAAK,CAAC;CACvE,MAAM,QAAQ,MAAM,MAAM,MAAM,EAAE,EAAE,YAAY;CAEhD,gBAAgB;EACd,IAAI,qBAAqB,MAAM,eAAe,CAAC,MAAM,oBACnD,MAAW,cAAc;CAE7B,GAAG,CAAC,mBAAmB,KAAK,CAAC;CAE7B,OAAO;EAAE,GAAG;EAAO,MAAM;EAAO;EAAO;CAAoB;AAC7D;;;;;;;;AC9DA,MAAa,uBAAuB,EAAE,iBAAkE,CAAC,MAAM;CAC7G,MAAM,SAAS,gBAAgB;CAC/B,MAAM,cAAc,eAAe;CAEnC,MAAM,gBAAgB,YAAY;EAChC,aAAa,WAAgC,OAAO,cAAc,MAAM;EACxE,iBAAiB;GACf,YAAiB,kBAAkB,EAAE,UAAU,CAAC,UAAU,EAAE,CAAC;EAC/D;EACA,GAAG,cAAc;CACnB,CAAC;CAED,MAAM,gBAAgB,YAAY;EAChC,aAAa,WAAgC,OAAO,cAAc,MAAM;EACxE,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,UAAU,EAAE,CAAC;GAC7D,YAAiB,kBAAkB,EAAE,UAAU,CAAC,WAAW,UAAU,SAAS,EAAE,CAAC;EACnF;EACA,GAAG,cAAc;CACnB,CAAC;CAED,MAAM,gBAAgB,YAAY;EAChC,aAAa,cAAsB,OAAO,cAAc,SAAS;EACjE,iBAAiB;GACf,YAAiB,kBAAkB,EAAE,UAAU,CAAC,UAAU,EAAE,CAAC;EAC/D;EACA,GAAG,cAAc;CACnB,CAAC;CAED,MAAM,UAAU,YAAY;EAC1B,aAAa,WAAiC,OAAO,eAAe,MAAM;EAC1E,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;GACvF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,WAAW,UAAU,SAAS,EAAE,CAAC;GACjF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,UAAU,SAAS,EAAE,CAAC;EAC5F;EACA,GAAG,cAAc;CACnB,CAAC;CAED,MAAM,aAAa,YAAY;EAC7B,aAAa,WAAoC,OAAO,kBAAkB,MAAM;EAChF,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;GACvF,YAAiB,kBAAkB,EAAE,UAAU;IAAC;IAAgB,UAAU;IAAW,UAAU;GAAM,EAAE,CAAC;GACxG,YAAiB,kBAAkB,EACjC,UAAU;IAAC;IAAyB,UAAU;IAAW,UAAU;GAAM,EAC3E,CAAC;GACD,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,UAAU,SAAS,EAAE,CAAC;EAC5F;EACA,GAAG,cAAc;CACnB,CAAC;CAED,MAAM,aAAa,YAAY;EAC7B,aAAa,EAAE,WAAW,aAA2C,OAAO,kBAAkB,WAAW,MAAM;EAC/G,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;GACvF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,WAAW,UAAU,SAAS,EAAE,CAAC;GACjF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,UAAU,SAAS,EAAE,CAAC;EAC5F;EACA,GAAG,cAAc;CACnB,CAAC;CAED,MAAM,YAAY,YAAY;EAC5B,aAAa,EAAE,WAAW,aAA2C,OAAO,iBAAiB,WAAW,MAAM;EAC9G,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;GACvF,YAAiB,kBAAkB,EAAE,UAAU;IAAC;IAAgB,UAAU;IAAW,UAAU;GAAM,EAAE,CAAC;GACxG,YAAiB,kBAAkB,EACjC,UAAU;IAAC;IAAyB,UAAU;IAAW,UAAU;GAAM,EAC3E,CAAC;GACD,YAAiB,kBAAkB,EAAE,UAAU,CAAC,4BAA4B,EAAE,CAAC;GAC/E,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,EAAE,CAAC;GACvE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,cAAc,EAAE,CAAC;GACjE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,sBAAsB,EAAE,CAAC;GACzE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,yBAAyB,EAAE,CAAC;GAC5E,YAAiB,kBAAkB,EAAE,UAAU,CAAC,2BAA2B,EAAE,CAAC;EAChF;EACA,GAAG,cAAc;CACnB,CAAC;CAGD,MAAM,mBAAmB,YAAY;EACnC,aAAa,WAA0C,OAAO,wBAAwB,MAAM;EAC5F,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;GACvF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,WAAW,UAAU,SAAS,EAAE,CAAC;GACjF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,UAAU,SAAS,EAAE,CAAC;EAC5F;EACA,GAAG,cAAc;CACnB,CAAC;CAGD,MAAM,mBAAmB,YAAY;EACnC,aAAa,WAA0C,OAAO,wBAAwB,MAAM;EAC5F,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;GACvF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,WAAW,UAAU,SAAS,EAAE,CAAC;GACjF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,UAAU,SAAS,EAAE,CAAC;EAC5F;EACA,GAAG,cAAc;CACnB,CAAC;CAgED,OAAO;EACL;EACA;EACA;EACA;EACA;EACA;EACA;EACA,aArEkB,YAAY;GAC9B,YAAY,OAAO,EAAE,WAAW,cAAwD;IACtF,OAAO,OAAO,wBAAwB;KAAE;KAAW;IAAQ,CAAC;GAC9D;GACA,YAAY,GAAG,cAAc;IAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;IACvF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,WAAW,UAAU,SAAS,EAAE,CAAC;IACjF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,UAAU,SAAS,EAAE,CAAC;GAC5F;GACA,GAAG,cAAc;EACnB,CA2DY;EACV;EACA;EACA,mBA5DwB,YAAY;GACpC,aAAa,WAA2C,OAAO,yBAAyB,MAAM;GAC9F,YAAY,GAAG,cAAc;IAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,uBAAuB,UAAU,SAAS,EAAE,CAAC;GAC/F;GACA,GAAG,cAAc;EACnB,CAsDkB;EAChB,kBArDuB,YAAY;GACnC,aAAa,iBAAyB,OAAO,iBAAiB,YAAY;GAC1E,iBAAiB;IACf,YAAiB,kBAAkB,EAAE,UAAU,CAAC,aAAa,EAAE,CAAC;IAChE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,YAAY,EAAE,CAAC;IAC/D,YAAiB,kBAAkB,EAAE,UAAU,CAAC,qBAAqB,EAAE,CAAC;IACxE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,EAAE,CAAC;IACvE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,4BAA4B,EAAE,CAAC;IAC/E,YAAiB,kBAAkB,EAAE,UAAU,CAAC,cAAc,EAAE,CAAC;IACjE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,EAAE,CAAC;IACpE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,2BAA2B,EAAE,CAAC;GAChF;GACA,GAAG,cAAc;EACnB,CAwCiB;EACf,kBAvCuB,YAAY;GACnC,aAAa,WAA0C,OAAO,wBAAwB,MAAM;GAC5F,YAAY,GAAG,cAAc;IAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,aAAa,EAAE,CAAC;IAChE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,cAAc,UAAU,YAAY,EAAE,CAAC;IACvF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,uBAAuB,UAAU,SAAS,EAAE,CAAC;IAC7F,YAAiB,kBAAkB,EACjC,UAAU;KAAC;KAAsB,UAAU;KAAW,UAAU;IAAY,EAC9E,CAAC;GACH;GACA,GAAG,cAAc;EACnB,CA4BiB;EACf,wBA3B6B,YAAY;GACzC,aAAa,WAAyC,OAAO,8BAA8B,MAAM;GACjG,YAAY,GAAG,cAAc;IAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,sBAAsB,UAAU,YAAY,EAAE,CAAC;IAC/F,YAAiB,kBAAkB,EAAE,UAAU,CAAC,4BAA4B,EAAE,CAAC;IAC/E,YAAiB,kBAAkB,EAAE,UAAU,CAAC,cAAc,EAAE,CAAC;IACjE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,sBAAsB,EAAE,CAAC;IACzE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,yBAAyB,EAAE,CAAC;IAC5E,YAAiB,kBAAkB,EAAE,UAAU,CAAC,2BAA2B,EAAE,CAAC;GAChF;GACA,GAAG,cAAc;EACnB,CAgBuB;CACvB;AACF;;;;;;AC3NA,MAAa,eAA6C,EACxD,YACA,iBAIE,CAAC,MAAoC;CACvC,MAAM,SAAS,gBAAgB;CAC/B,OAAO,SAAS;EACd,UAAU,CAAC,YAAY,UAAU;EACjC,eAAe,OAAO,aAAa,UAAU;EAC7C,kBAAiB,iBAAgB;EACjC,GAAG;CACL,CAAC;AACH;AAEA,MAAM,oBAAoB;;;;;AAmB1B,MAAa,uBAAgD,EAC3D,QACA,SACA,iBAKE,CAAC,MAAwC;CAC3C,MAAM,SAAS,gBAAgB;CAC/B,MAAM,EAAE,QAAQ,mBAAmB,QAAQ,wBAAwB,UAAU;CAC7E,MAAM,aAAa,QAAQ,cAAc,KAAA;CACzC,MAAM,YAAY,QAAQ,WAAW,CAAC,OAAO,QAAQ,IAAI,KAAA;CAEzD,MAAM,QAAQ,iBAAiB;EAC7B,UAAU;GAAC;GAAY;GAAY;IAAE;IAAY;IAAW;GAAQ;EAAC;EACrE,UAAU,EAAE,gBACV,OAAO,aAAa;GAAE,MAAM;GAAW,SAAS;GAAmB;GAAY;GAAW;EAAQ,CAAC;EACrG,kBAAkB;EAClB,mBAAmB,UAAU,GAAG,kBAAkB;GAChD,IAAI,CAAC,UAAU,UAAU,UAAU,CAAC,SAAS,YAAY,SACvD;GAEF,OAAO,gBAAgB;EACzB;EACA,SAAS,SACP,KAAK,MAAM,SAAQ,SAAQ,MAAM,YAAY,CAAC,CAAC;EACjD,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;;;;;;AAOA,MAAa,cAAuC,EAClD,WACA,mBAIkC;CAClC,MAAM,SAAS,gBAAgB;CAC/B,OAAO,SAAS;EACd,UAAU,CAAC,WAAW,SAAS;EAC/B,eAAe,OAAO,WAAW,SAAS;EAC1C,GAAG;CACL,CAAC;AACH;;;;;;;;;;;;AC3FA,MAAa,yBAA6D,EACxE,WACA,eACA,YACA,mBAMkC;CAClC,MAAM,SAAS,gBAAgB;CAC/B,OAAO,SAAS;EACd,UAAU;GAAC;GAAuB;GAAW;GAAe;EAAU;EACtE,eAAe,OAAO,mBAAmB;GAAE;GAAW;GAAe;EAAW,CAAC;EACjF,GAAG;CACL,CAAC;AACH;;;ACPA,MAAM,mBAAmB;AACzB,MAAM,mBAAmB;AACzB,MAAM,kBAAkB;AAExB,MAAM,gBAAgB,WAA+B,WAAW,aAAa,WAAW;;;;;;AAOxF,MAAa,iCACX,QACA,WACA,iBAEAC,aAAqB;CACnB,UAAU;EAAC;EAAsB;EAAW;CAAY;CACxD,eAAe,OAAO,qBAAqB,WAAW,YAAY;CAClE,QAAQ;CACR,WAAW;CACX,kBAAiB,UAAU,aAAa,MAAM,MAAM,MAAM,MAAM,IAAI,mBAAmB;AACzF,CAAC;;;;;AAMH,MAAa,gCACX,QACA,cACA,qBAEAA,aAAqB;CACnB,UAAU;EAAC;EAA6B;EAAc;CAAgB;CACtE,SAAS,YAA+C;EACtD,MAAM,YAAkC,CAAC;EACzC,IAAI,OAAO;EAEX,OAAO,MAAM;GACX,MAAM,WAAW,MAAM,OAAO,kBAAkB;IAAE,OAAO;IAAc;IAAM,SAAS;GAAgB,CAAC;GACvG,UAAU,KAAK,GAAG,SAAS,MAAM;GACjC,IAAI,CAAC,SAAS,WAAW,SAAS;GAClC;EACF;EAEA,MAAM,UAAoC,CAAC;EAC3C,KAAK,MAAM,OAAO,WAChB,CAAC,QAAQ,IAAI,cAAc,CAAC,EAAA,CAAG,KAAK,GAAG;EAEzC,OAAO;CACT;CACA,iBAAiB,aAAa,gBAAgB,IAAI,mBAAmB;AACvE,CAAC;;;;;AAMH,MAAa,2CACX,QACA,WACA,cACA,qBAEAA,aAAqB;CACnB,UAAU;EAAC;EAAkC;EAAW;EAAc;CAAgB;CACtF,SAAS,YAAgD;EACvD,MAAM,aAAwC,CAAC;EAC/C,IAAI,OAAO;EAEX,OAAO,MAAM;GACX,MAAM,WAAW,MAAM,OAAO,6BAA6B,WAAW,cAAc;IAClF;IACA,SAAS;GACX,CAAC;GACD,WAAW,KAAK,GAAG,SAAS,OAAO;GACnC,IAAI,CAAC,SAAS,WAAW,WAAW,SAAS,QAAQ,WAAW,GAAG;GACnE;EACF;EAEA,OAAO;CACT;CACA,iBAAiB,aAAa,gBAAgB,IAAI,mBAAmB;AACvE,CAAC;;;;AAWH,MAAa,wBAA2D,EACtE,WACA,cACA,mBAKkC;CAClC,MAAM,SAAS,gBAAgB;CAC/B,MAAM,EAAE,UAAU,SAAS,QAAQ,WAAW,oBAAoB,8BAChE,QACA,WACA,YACF;CACA,OAAO,SAA6E;EAClF;EACA;EACA;EACA;EACA;EACA,GAAG;CACL,CAAC;AACH;;;;;AAMA,MAAa,gCAAgC,EAC3C,WACA,oBAIwD;CACxD,MAAM,SAAS,gBAAgB;CAC/B,MAAM,UAAU,gBAA2C,aAAa;CACxE,OAAO,WAAW;EAChB,SAAS,cAAc,KAAI,OAAM,8BAA8B,QAAQ,WAAW,EAAE,CAAC;EACrF;CACF,CAAC;AACH;;;;;;;AAoBA,MAAa,+BAAkE,EAC7E,WACA,cACA,kBACA,SACA,mBAGG;CACH,MAAM,SAAS,gBAAgB;CAC/B,MAAM,EAAE,QAAQ,mBAAmB,QAAQ,wBAAwB,UAAU;CAE7E,MAAM,QAAQ,iBAAiB;EAC7B,UAAU;GAAC;GAA8B;GAAW;GAAc;GAAkB;EAAO;EAC3F,SAAS,OAAO,EAAE,gBAAgB;GAChC,OAAO,OAAO,6BAA6B,WAAW,cAAc;IAClE,MAAM;IACN,SAAS;IACT;GACF,CAAC;EACH;EACA,kBAAkB;EAClB,mBAAmB,UAAU,GAAG,kBAAkB;GAChD,IAAI,CAAC,UAAU,SAAS,QACtB;GAIF,KAFsB,gBAAgB,KAAK,qBAC7B,UAAU,YAAY,SAAS,IAE3C;GAEF,OAAO,gBAAgB;EACzB;EACA,iBAAiB,aAAa,gBAAgB,IAAI,mBAAmB;EACrE,SAAS,SAAsD;GAC7D,OAAO,KAAK,MAAM,SAAQ,SAAQ,MAAM,WAAW,CAAC,CAAC;EACvD;EACA,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;;;;;AAMA,MAAa,yCAAyC,EACpD,WACA,kBAIwD;CACxD,MAAM,SAAS,gBAAgB;CAC/B,MAAM,UAAU,gBAA2C,YAAY,KAAI,eAAc,WAAW,EAAE,CAAC;CACvG,OAAO,WAAW;EAChB,SAAS,YAAY,KAAI,eACvB,wCAAwC,QAAQ,WAAW,WAAW,IAAI,WAAW,MAAM,CAC7F;EACA;CACF,CAAC;AACH;;;;;;;AAQA,MAAa,2BAA6D,EACxE,cACA,kBACA,mBAKkC;CAClC,MAAM,SAAS,gBAAgB;CAC/B,MAAM,EAAE,UAAU,SAAS,oBAAoB,6BAA6B,QAAQ,cAAc,gBAAgB;CAClH,OAAO,SAA2E;EAChF;EACA;EACA;EACA,GAAG;CACL,CAAC;AACH;;;;;AAMA,MAAa,mCAAmC,EAC9C,kBAGuD;CACvD,MAAM,SAAS,gBAAgB;CAC/B,MAAM,UAAU,gBAA0C,YAAY,KAAI,eAAc,WAAW,EAAE,CAAC;CACtG,OAAO,WAAW;EAChB,SAAS,YAAY,KAAI,eAAc,6BAA6B,QAAQ,WAAW,IAAI,WAAW,MAAM,CAAC;EAC7G;CACF,CAAC;AACH;;;;;;;;ACnRA,MAAa,0BAAwD,EACnE,WACA,QACA,mBAKkC;CAClC,MAAM,SAAS,gBAAgB;CAE/B,OAAO,SAAS;EACd,UAAU;GAAC;GAAyB;GAAW;EAAM;EACrD,SAAS,YAA2C;GAGlD,SAAQ,MAFU,OAAO,eAAe,WAAW,MAAM,EAAA,EAE5C,WAAW,CAAC,EAAA,CAAG,KACzB,SAAS,WAA+B;IACvC,IAAI,QAAQ;IACZ,WAAW,QAAQ;IACnB,gBAAgB,QAAQ;IACxB,OAAO,QAAQ;IACf,aAAa,QAAQ;IACrB,oBAAoB,QAAQ;IAC5B,WAAW,QAAQ,aAAa,KAAA;IAChC,WAAW,QAAQ,aAAa,KAAA;IAChC,gBAAgB,QAAQ,kBAAkB,KAAA;IAC1C,UAAU,QAAQ,YAAY,KAAA;IAC9B,SAAS,QAAQ;IACjB,WAAW,QAAQ;IACnB,WAAW,QAAQ;IACnB,WAAW,QAAQ;IACnB,UAAU,UAAU;GACtB,EACF;EACF;EACA,GAAG;CACL,CAAC;AACH;;;;;;AAOA,MAAa,yBAAqD,EAChE,WACA,QACA,gBACA,eACA,mBAOkC;CAClC,MAAM,SAAS,gBAAgB;CAE/B,OAAO,SAAS;EACd,UAAU;GAAC;GAAwB;GAAW;GAAQ;EAAc;EACpE,SAAS,YAAyC;GAChD,MAAM,IAAI,MAAM,OAAO,sBAAsB,WAAW,QAAQ,cAAc;GAE9E,OAAO;IACL,IAAI,EAAE;IACN,WAAW,EAAE;IACb,gBAAgB,EAAE;IAClB,OAAO,EAAE;IACT,aAAa,EAAE;IACf,oBAAoB,EAAE;IACtB,WAAW,EAAE,aAAa,KAAA;IAC1B,WAAW,EAAE,aAAa,KAAA;IAC1B,gBAAgB,EAAE,kBAAkB,KAAA;IACpC,UAAU,EAAE,YAAY,KAAA;IACxB,SAAS;IACT,WAAW;IACX,WAAW,EAAE;IACb,WAAW,EAAE;IACb,UAAU,iBAAiB,OAAO,mBAAmB,gBAAgB;GACvE;EACF;EACA,GAAG;CACL,CAAC;AACH;;;ACtFA,MAAM,WAAW;;;;;;;AAQjB,MAAa,sBAAgD,EAC3D,WACA,mBAII;CACJ,MAAM,SAAS,gBAAgB;CAE/B,OAAO,iBAAiB;EACtB,UAAU,CAAC,oBAAoB,SAAS;EACxC,SAAS,OAAO,EAAE,gBAAgB;GAChC,OAAO,OAAO,oBAAoB,WAAW;IAAE,MAAM;IAAW,SAAS;GAAS,CAAC;EACrF;EACA,kBAAkB;EAClB,mBAAmB,UAAU,GAAG,kBAAkB;GAChD,IAAI,UAAU,YAAY,SACxB,OAAO,gBAAgB;EAG3B;EACA,SAAS,SAAoD;GAC3D,OAAO,KAAK,MACT,SAAQ,SAAQ,MAAM,YAAY,CAAC,CAAC,CAAC,CACrC,KACE,GAAG,WAA2B;IAC7B,IAAI,EAAE;IACN,WAAW,EAAE;IACb,SAAS,EAAE;IACX,WAAW,EAAE;IACb,WAAW,UAAU;GACvB,EACF;EACJ;EACA,GAAG;CACL,CAAC;AACH;;;;;;AChDA,MAAa,kBAAmD,EAC9D,YACA,iBAIE,CAAC,MAAoC;CACvC,MAAM,SAAS,gBAAgB;CAC/B,OAAO,SAAS;EACd,UAAU,CAAC,eAAe,UAAU;EACpC,eAAe,OAAO,gBAAgB,UAAU;EAChD,GAAG;CACL,CAAC;AACH;;;;AAKA,MAAa,iBAAgD,EAC3D,cACA,mBAIkC;CAClC,MAAM,SAAS,gBAAgB;CAC/B,OAAO,SAAS;EACd,UAAU,CAAC,cAAc,YAAY;EACrC,eAAe,OAAO,cAAc,YAAY;EAChD,GAAG;CACL,CAAC;AACH;;;;;;;;;AC7BA,SAAgB,kBAAkD,EAChE,YACA,gBAI+B;CAC/B,MAAM,SAAS,gBAAgB;CAE/B,OAAO,SAAS;EACd,UAAU,CAAC,mBAAmB,UAAU;EACxC,SAAS,YAAY;GACnB,IAAI,CAAC,YAAY,MAAM,IAAI,MAAM,sBAAsB;GACvD,OAAO,OAAO,YAAY,UAAU,CAAC,CAAC,UAAU;EAClD;EACA,WAAW,MAAS;EACpB,GAAG;CACL,CAAC;AACH"}
+{"version":3,"file":"index.js","names":["PER_PAGE","tanstackQueryOptions"],"sources":["../../../src/hooks/datasets/use-dataset-items.ts","../../../src/hooks/datasets/use-dataset-mutations.ts","../../../src/hooks/datasets/use-datasets.ts","../../../src/hooks/datasets/use-compare-experiments.ts","../../../src/hooks/datasets/use-dataset-experiments.ts","../../../src/hooks/datasets/use-dataset-item-versions.ts","../../../src/hooks/datasets/use-dataset-versions.ts","../../../src/hooks/datasets/use-experiments.ts","../../../src/hooks/datasets/use-workflow-schema.ts"],"sourcesContent":["import type { MastraClient } from '@mastra/client-js';\nimport { useQuery, useInfiniteQuery } from '@tanstack/react-query';\nimport type { InfiniteData, QueryKey, UseInfiniteQueryResult, UseQueryResult } from '@tanstack/react-query';\nimport { useEffect } from 'react';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraInfiniteQueryOptions, MastraQueryOptions } from '../shared/query-options';\nimport { useInView } from '../shared/use-in-view';\n\n/**\n * Hook to fetch a single dataset item by ID\n */\ntype DatasetItemResponse = Awaited<ReturnType<MastraClient['getDatasetItem']>>;\n\n/**\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetItem = <TData = DatasetItemResponse>({\n  datasetId,\n  itemId,\n  queryOptions,\n}: {\n  datasetId: string;\n  itemId: string;\n  queryOptions?: MastraQueryOptions<DatasetItemResponse, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['dataset-item', datasetId, itemId],\n    queryFn: () => client.getDatasetItem(datasetId, itemId),\n    retry: false, // Don't retry 404s for deleted items\n    ...queryOptions,\n  });\n};\n\nconst PER_PAGE = 10;\n\ntype ListDatasetItemsResponse = Awaited<ReturnType<MastraClient['listDatasetItems']>>;\n\nexport type UseDatasetItemsResult = Omit<\n  UseInfiniteQueryResult<InfiniteData<ListDatasetItemsResponse>, Error>,\n  'data'\n> & {\n  data: ListDatasetItemsResponse['items'];\n  total: number | undefined;\n  setEndOfListElement: (element: HTMLDivElement | null) => void;\n};\n\n/**\n * Hook to list items in a dataset with infinite scroll pagination and optional search\n * @param version - Optional version timestamp to view historical snapshot\n */\nexport type DatasetItemsOrderBy = NonNullable<NonNullable<Parameters<MastraClient['listDatasetItems']>[1]>['orderBy']>;\n\n/**\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetItems = ({\n  datasetId,\n  search,\n  version,\n  orderBy,\n  queryOptions,\n}: {\n  datasetId: string;\n  search?: string;\n  version?: number | null;\n  orderBy?: DatasetItemsOrderBy;\n  queryOptions?: MastraInfiniteQueryOptions<\n    ListDatasetItemsResponse,\n    InfiniteData<ListDatasetItemsResponse, number>,\n    QueryKey,\n    number\n  >;\n}): UseDatasetItemsResult => {\n  const client = useMastraClient();\n  const { inView: isEndOfListInView, setRef: setEndOfListElement } = useInView();\n\n  const query = useInfiniteQuery({\n    queryKey: ['dataset-items', datasetId, search, version, orderBy],\n    queryFn: async ({ pageParam }) => {\n      const res = await client.listDatasetItems(datasetId, {\n        page: pageParam,\n        perPage: PER_PAGE,\n        search: search || undefined,\n        version: version || undefined,\n        orderBy,\n      });\n      return res;\n    },\n    initialPageParam: 0,\n    getNextPageParam: (lastPage, _, lastPageParam) => {\n      if (!lastPage?.items?.length) {\n        return undefined;\n      }\n      const totalFetched = (lastPageParam + 1) * PER_PAGE;\n      const total = lastPage?.pagination?.total ?? 0;\n      if (totalFetched >= total) {\n        return undefined;\n      }\n      return lastPageParam + 1;\n    },\n    retry: false,\n    ...queryOptions,\n  });\n\n  const items = query.data?.pages.flatMap(page => page?.items ?? []) ?? [];\n  const total = query.data?.pages[0]?.pagination?.total;\n\n  useEffect(() => {\n    if (isEndOfListInView && query.hasNextPage && !query.isFetchingNextPage) {\n      void query.fetchNextPage();\n    }\n  }, [isEndOfListInView, query]);\n\n  return { ...query, data: items, total, setEndOfListElement };\n};\n","import type {\n  CreateDatasetParams,\n  UpdateDatasetParams,\n  AddDatasetItemParams,\n  UpdateDatasetItemParams,\n  TriggerDatasetExperimentParams,\n  UpdateDatasetExperimentParams,\n  UpdateExperimentResultParams,\n  BatchInsertDatasetItemsParams,\n  BatchDeleteDatasetItemsParams,\n  MastraClient,\n} from '@mastra/client-js';\nimport { useMutation, useQueryClient } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraMutationOptions } from '../shared/query-options';\n\ntype DatasetItemMutationVariables = {\n  datasetId: string;\n  itemId: string;\n};\n\ntype ClientResult<K extends keyof MastraClient> = MastraClient[K] extends (...args: any[]) => Promise<infer R>\n  ? R\n  : never;\n\nexport interface DatasetMutationsQueryOptions {\n  createDataset?: MastraMutationOptions<ClientResult<'createDataset'>, CreateDatasetParams>;\n  updateDataset?: MastraMutationOptions<ClientResult<'updateDataset'>, UpdateDatasetParams>;\n  deleteDataset?: MastraMutationOptions<ClientResult<'deleteDataset'>, string>;\n  addItem?: MastraMutationOptions<ClientResult<'addDatasetItem'>, AddDatasetItemParams>;\n  updateItem?: MastraMutationOptions<ClientResult<'updateDatasetItem'>, UpdateDatasetItemParams>;\n  deleteItem?: MastraMutationOptions<ClientResult<'deleteDatasetItem'>, DatasetItemMutationVariables>;\n  purgeItem?: MastraMutationOptions<ClientResult<'purgeDatasetItem'>, DatasetItemMutationVariables>;\n  deleteItems?: MastraMutationOptions<\n    ClientResult<'batchDeleteDatasetItems'>,\n    { datasetId: string; itemIds: string[] }\n  >;\n  batchInsertItems?: MastraMutationOptions<ClientResult<'batchInsertDatasetItems'>, BatchInsertDatasetItemsParams>;\n  batchDeleteItems?: MastraMutationOptions<ClientResult<'batchDeleteDatasetItems'>, BatchDeleteDatasetItemsParams>;\n  triggerExperiment?: MastraMutationOptions<ClientResult<'triggerDatasetExperiment'>, TriggerDatasetExperimentParams>;\n  deleteExperiment?: MastraMutationOptions<ClientResult<'deleteExperiment'>, string>;\n  updateExperiment?: MastraMutationOptions<ClientResult<'updateDatasetExperiment'>, UpdateDatasetExperimentParams>;\n  updateExperimentResult?: MastraMutationOptions<\n    ClientResult<'updateDatasetExperimentResult'>,\n    UpdateExperimentResultParams\n  >;\n}\n\n/**\n * Hook providing mutation functions for datasets, items, and runs\n * All mutations invalidate relevant query caches on success.\n * `queryOptions` is keyed by returned mutation name and spread last; a user `onSuccess` replaces the internal invalidation.\n */\nexport const useDatasetMutations = ({ queryOptions }: { queryOptions?: DatasetMutationsQueryOptions } = {}) => {\n  const client = useMastraClient();\n  const queryClient = useQueryClient();\n\n  const createDataset = useMutation({\n    mutationFn: (params: CreateDatasetParams) => client.createDataset(params),\n    onSuccess: () => {\n      void queryClient.invalidateQueries({ queryKey: ['datasets'] });\n    },\n    ...queryOptions?.createDataset,\n  });\n\n  const updateDataset = useMutation({\n    mutationFn: (params: UpdateDatasetParams) => client.updateDataset(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['datasets'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset', variables.datasetId] });\n    },\n    ...queryOptions?.updateDataset,\n  });\n\n  const deleteDataset = useMutation({\n    mutationFn: (datasetId: string) => client.deleteDataset(datasetId),\n    onSuccess: () => {\n      void queryClient.invalidateQueries({ queryKey: ['datasets'] });\n    },\n    ...queryOptions?.deleteDataset,\n  });\n\n  const addItem = useMutation({\n    mutationFn: (params: AddDatasetItemParams) => client.addDatasetItem(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-versions', variables.datasetId] });\n    },\n    ...queryOptions?.addItem,\n  });\n\n  const updateItem = useMutation({\n    mutationFn: (params: UpdateDatasetItemParams) => client.updateDatasetItem(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-item', variables.datasetId, variables.itemId] });\n      void queryClient.invalidateQueries({\n        queryKey: ['dataset-item-versions', variables.datasetId, variables.itemId],\n      });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-versions', variables.datasetId] });\n    },\n    ...queryOptions?.updateItem,\n  });\n\n  const deleteItem = useMutation({\n    mutationFn: ({ datasetId, itemId }: DatasetItemMutationVariables) => client.deleteDatasetItem(datasetId, itemId),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-versions', variables.datasetId] });\n    },\n    ...queryOptions?.deleteItem,\n  });\n\n  const purgeItem = useMutation({\n    mutationFn: ({ datasetId, itemId }: DatasetItemMutationVariables) => client.purgeDatasetItem(datasetId, itemId),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-item', variables.datasetId, variables.itemId] });\n      void queryClient.invalidateQueries({\n        queryKey: ['dataset-item-versions', variables.datasetId, variables.itemId],\n      });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiment-results'] });\n      void queryClient.invalidateQueries({ queryKey: ['experiment-results'] });\n      void queryClient.invalidateQueries({ queryKey: ['review-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-review-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-completed-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['experiment-review-summary'] });\n    },\n    ...queryOptions?.purgeItem,\n  });\n\n  // Batch insert items using the batch endpoint\n  const batchInsertItems = useMutation({\n    mutationFn: (params: BatchInsertDatasetItemsParams) => client.batchInsertDatasetItems(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-versions', variables.datasetId] });\n    },\n    ...queryOptions?.batchInsertItems,\n  });\n\n  // Batch delete items using the batch endpoint\n  const batchDeleteItems = useMutation({\n    mutationFn: (params: BatchDeleteDatasetItemsParams) => client.batchDeleteDatasetItems(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-versions', variables.datasetId] });\n    },\n    ...queryOptions?.batchDeleteItems,\n  });\n\n  // @deprecated - use batchDeleteItems mutation instead\n  const deleteItems = useMutation({\n    mutationFn: async ({ datasetId, itemIds }: { datasetId: string; itemIds: string[] }) => {\n      return client.batchDeleteDatasetItems({ datasetId, itemIds });\n    },\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-items', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset', variables.datasetId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-versions', variables.datasetId] });\n    },\n    ...queryOptions?.deleteItems,\n  });\n\n  const triggerExperiment = useMutation({\n    mutationFn: (params: TriggerDatasetExperimentParams) => client.triggerDatasetExperiment(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiments', variables.datasetId] });\n    },\n    ...queryOptions?.triggerExperiment,\n  });\n\n  const deleteExperiment = useMutation({\n    mutationFn: (experimentId: string) => client.deleteExperiment(experimentId),\n    onSuccess: () => {\n      void queryClient.invalidateQueries({ queryKey: ['experiments'] });\n      void queryClient.invalidateQueries({ queryKey: ['experiment'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiments'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiment'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiment-results'] });\n      void queryClient.invalidateQueries({ queryKey: ['review-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['completed-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['experiment-review-summary'] });\n    },\n    ...queryOptions?.deleteExperiment,\n  });\n\n  const updateExperiment = useMutation({\n    mutationFn: (params: UpdateDatasetExperimentParams) => client.updateDatasetExperiment(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['experiments'] });\n      void queryClient.invalidateQueries({ queryKey: ['experiment', variables.experimentId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiments', variables.datasetId] });\n      void queryClient.invalidateQueries({\n        queryKey: ['dataset-experiment', variables.datasetId, variables.experimentId],\n      });\n    },\n    ...queryOptions?.updateExperiment,\n  });\n\n  const updateExperimentResult = useMutation({\n    mutationFn: (params: UpdateExperimentResultParams) => client.updateDatasetExperimentResult(params),\n    onSuccess: (_, variables) => {\n      void queryClient.invalidateQueries({ queryKey: ['experiment-results', variables.experimentId] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-experiment-results'] });\n      void queryClient.invalidateQueries({ queryKey: ['review-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-review-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['dataset-completed-items'] });\n      void queryClient.invalidateQueries({ queryKey: ['experiment-review-summary'] });\n    },\n    ...queryOptions?.updateExperimentResult,\n  });\n\n  return {\n    createDataset,\n    updateDataset,\n    deleteDataset,\n    addItem,\n    updateItem,\n    deleteItem,\n    purgeItem,\n    deleteItems,\n    batchInsertItems,\n    batchDeleteItems,\n    triggerExperiment,\n    deleteExperiment,\n    updateExperiment,\n    updateExperimentResult,\n  };\n};\n","import type { ExperimentTargetType, ListDatasetsParams, MastraClient } from '@mastra/client-js';\nimport { useInfiniteQuery, useQuery } from '@tanstack/react-query';\nimport type { InfiniteData, QueryKey, UseInfiniteQueryResult, UseQueryResult } from '@tanstack/react-query';\nimport { useEffect } from 'react';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraInfiniteQueryOptions, MastraQueryOptions } from '../shared/query-options';\nimport { useInView } from '../shared/use-in-view';\n\ntype ListDatasetsResponse = Awaited<ReturnType<MastraClient['listDatasets']>>;\ntype DatasetResponse = Awaited<ReturnType<MastraClient['getDataset']>>;\n\n/**\n * Hook to list all datasets with optional pagination\n */\nexport const useDatasets = <TData = ListDatasetsResponse>({\n  pagination,\n  queryOptions,\n}: {\n  pagination?: { page?: number; perPage?: number };\n  queryOptions?: MastraQueryOptions<ListDatasetsResponse, TData>;\n} = {}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['datasets', pagination],\n    queryFn: () => client.listDatasets(pagination),\n    placeholderData: previousData => previousData,\n    ...queryOptions,\n  });\n};\n\nconst DATASETS_PER_PAGE = 20;\n\nexport interface DatasetTargetFilter {\n  targetType?: ExperimentTargetType | '';\n  targetId?: string;\n}\n\nexport type DatasetsOrderBy = NonNullable<ListDatasetsParams['orderBy']>;\n\ntype DatasetRecord = ListDatasetsResponse['datasets'][number];\n\nexport type UseInfiniteDatasetsResult<TData = DatasetRecord[]> = UseInfiniteQueryResult<TData, Error> & {\n  setEndOfListElement: (element: HTMLDivElement | null) => void;\n};\n\n/**\n * Hook to list datasets with infinite scroll pagination, optionally scoped server-side to a target.\n * `data` defaults to the flattened dataset list; a user `select` replaces that flattening.\n */\nexport const useInfiniteDatasets = <TData = DatasetRecord[]>({\n  filter,\n  orderBy,\n  queryOptions,\n}: {\n  filter?: DatasetTargetFilter;\n  orderBy?: DatasetsOrderBy;\n  queryOptions?: MastraInfiniteQueryOptions<ListDatasetsResponse, TData, QueryKey, number>;\n} = {}): UseInfiniteDatasetsResult<TData> => {\n  const client = useMastraClient();\n  const { inView: isEndOfListInView, setRef: setEndOfListElement } = useInView();\n  const targetType = filter?.targetType || undefined;\n  const targetIds = filter?.targetId ? [filter.targetId] : undefined;\n\n  const query = useInfiniteQuery({\n    queryKey: ['datasets', 'infinite', { targetType, targetIds, orderBy }],\n    queryFn: ({ pageParam }) =>\n      client.listDatasets({ page: pageParam, perPage: DATASETS_PER_PAGE, targetType, targetIds, orderBy }),\n    initialPageParam: 0,\n    getNextPageParam: (lastPage, _, lastPageParam) => {\n      if (!lastPage?.datasets?.length || !lastPage.pagination?.hasMore) {\n        return undefined;\n      }\n      return lastPageParam + 1;\n    },\n    select: (data: InfiniteData<ListDatasetsResponse, number>) =>\n      data.pages.flatMap(page => page?.datasets ?? []) 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\n/**\n * Hook to fetch a single dataset by ID\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDataset = <TData = DatasetResponse>({\n  datasetId,\n  queryOptions,\n}: {\n  datasetId: string;\n  queryOptions?: MastraQueryOptions<DatasetResponse, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['dataset', datasetId],\n    queryFn: () => client.getDataset(datasetId),\n    ...queryOptions,\n  });\n};\n","import type { CompareExperimentsResponse } 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/**\n * Hook to compare two to ten dataset experiments item by item against a baseline.\n * @param datasetId - ID of the dataset\n * @param experimentIds - IDs of the experiments to compare\n * @param baselineId - ID of the baseline experiment; must be one of `experimentIds` (defaults to the first)\n * @param queryOptions - TanStack Query options, spread last to override defaults\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useCompareExperiments = <TData = CompareExperimentsResponse>({\n  datasetId,\n  experimentIds,\n  baselineId,\n  queryOptions,\n}: {\n  datasetId: string;\n  experimentIds: string[];\n  baselineId?: string;\n  queryOptions?: MastraQueryOptions<CompareExperimentsResponse, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['compare-experiments', datasetId, experimentIds, baselineId],\n    queryFn: () => client.compareExperiments({ datasetId, experimentIds, baselineId }),\n    ...queryOptions,\n  });\n};\n","import type { ClientScoreRowData, DatasetExperimentResult, MastraClient } from '@mastra/client-js';\nimport type { ExperimentStatus } from '@mastra/core/storage';\nimport { queryOptions as tanstackQueryOptions, useInfiniteQuery, useQueries, useQuery } from '@tanstack/react-query';\nimport type {\n  InfiniteData,\n  QueryKey,\n  UseInfiniteQueryResult,\n  UseQueryOptions,\n  UseQueryResult,\n} from '@tanstack/react-query';\nimport { useEffect } from 'react';\nimport { useMastraClient } from '../../mastra-client-context';\nimport { useCombineByIds } from '../shared/combine-by-ids';\nimport type { ExperimentQueriesResult } from '../shared/combine-by-ids';\nimport type { MastraInfiniteQueryOptions, MastraQueryOptions } from '../shared/query-options';\n\nexport type { ExperimentQueriesResult } from '../shared/combine-by-ids';\nimport { useInView } from '../shared/use-in-view';\n\ntype DatasetExperimentResponse = Awaited<ReturnType<MastraClient['getDatasetExperiment']>>;\ntype ExperimentResultsPage = Awaited<ReturnType<MastraClient['listDatasetExperimentResults']>>;\ntype ExperimentScoresByEntity = Record<string, ClientScoreRowData[]>;\ntype DatasetExperimentQueryKey = string[];\ntype ExperimentStatusQueryKey = (string | undefined)[];\n\nconst POLL_INTERVAL_MS = 2000;\nconst RESULTS_PER_PAGE = 100;\nconst SCORES_PER_PAGE = 100;\n\nconst isInProgress = (status: string | undefined) => status === 'running' || status === 'pending';\n\n/**\n * Key, fetcher and polling policy of the `dataset-experiment` query, shared by the single and\n * multi-experiment hooks so they hit the same cache entry.\n * Polls every 2 seconds while the experiment is 'running' or 'pending'.\n */\nexport const datasetExperimentQueryOptions = (\n  client: MastraClient,\n  datasetId: string,\n  experimentId: string,\n): UseQueryOptions<DatasetExperimentResponse, Error, DatasetExperimentResponse, DatasetExperimentQueryKey> =>\n  tanstackQueryOptions({\n    queryKey: ['dataset-experiment', datasetId, experimentId],\n    queryFn: () => client.getDatasetExperiment(datasetId, experimentId),\n    gcTime: 0,\n    staleTime: 0,\n    refetchInterval: query => (isInProgress(query.state.data?.status) ? POLL_INTERVAL_MS : false),\n  });\n\n/**\n * Key and fetcher of the `dataset-experiment-scores` query: every score of an experiment, across all\n * pages, grouped by dataset item id. Polls every 2 seconds while `experimentStatus` is in progress.\n */\nexport const experimentScoresQueryOptions = (\n  client: MastraClient,\n  experimentId: string,\n  experimentStatus: ExperimentStatus | undefined,\n): UseQueryOptions<ExperimentScoresByEntity, Error, ExperimentScoresByEntity, ExperimentStatusQueryKey> =>\n  tanstackQueryOptions({\n    queryKey: ['dataset-experiment-scores', experimentId, experimentStatus],\n    queryFn: async (): Promise<ExperimentScoresByEntity> => {\n      const allScores: ClientScoreRowData[] = [];\n      let page = 0;\n\n      while (true) {\n        const response = await client.listScoresByRunId({ runId: experimentId, page, perPage: SCORES_PER_PAGE });\n        allScores.push(...response.scores);\n        if (!response.pagination.hasMore) break;\n        page++;\n      }\n\n      const grouped: ExperimentScoresByEntity = {};\n      for (const row of allScores) {\n        (grouped[row.entityId] ??= []).push(row);\n      }\n      return grouped;\n    },\n    refetchInterval: isInProgress(experimentStatus) ? POLL_INTERVAL_MS : false,\n  });\n\n/**\n * Key and fetcher of the `dataset-experiment-results-all` query: every result of an experiment, across\n * all pages. Polls every 2 seconds while `experimentStatus` is in progress.\n */\nexport const allDatasetExperimentResultsQueryOptions = (\n  client: MastraClient,\n  datasetId: string,\n  experimentId: string,\n  experimentStatus: ExperimentStatus | undefined,\n): UseQueryOptions<DatasetExperimentResult[], Error, DatasetExperimentResult[], ExperimentStatusQueryKey> =>\n  tanstackQueryOptions({\n    queryKey: ['dataset-experiment-results-all', datasetId, experimentId, experimentStatus],\n    queryFn: async (): Promise<DatasetExperimentResult[]> => {\n      const allResults: DatasetExperimentResult[] = [];\n      let page = 0;\n\n      while (true) {\n        const response = await client.listDatasetExperimentResults(datasetId, experimentId, {\n          page,\n          perPage: RESULTS_PER_PAGE,\n        });\n        allResults.push(...response.results);\n        if (!response.pagination.hasMore || response.results.length === 0) break;\n        page++;\n      }\n\n      return allResults;\n    },\n    refetchInterval: isInProgress(experimentStatus) ? POLL_INTERVAL_MS : false,\n  });\n\n/** An experiment id plus its status, which drives polling of its dependent queries. */\nexport interface ExperimentRef {\n  id: string;\n  status?: ExperimentStatus;\n}\n\n/**\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetExperiment = <TData = DatasetExperimentResponse>({\n  datasetId,\n  experimentId,\n  queryOptions,\n}: {\n  datasetId: string;\n  experimentId: string;\n  queryOptions?: MastraQueryOptions<DatasetExperimentResponse, TData, DatasetExperimentQueryKey>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  const { queryKey, queryFn, gcTime, staleTime, refetchInterval } = datasetExperimentQueryOptions(\n    client,\n    datasetId,\n    experimentId,\n  );\n  return useQuery<DatasetExperimentResponse, Error, TData, DatasetExperimentQueryKey>({\n    queryKey,\n    queryFn,\n    gcTime,\n    staleTime,\n    refetchInterval,\n    ...queryOptions,\n  });\n};\n\n/**\n * Fetches several experiments of one dataset in parallel, each polled while it runs.\n * The server 404s an experiment that does not belong to `datasetId`, which surfaces as `error`.\n */\nexport const useDatasetExperimentsQueries = ({\n  datasetId,\n  experimentIds,\n}: {\n  datasetId: string;\n  experimentIds: string[];\n}): ExperimentQueriesResult<DatasetExperimentResponse> => {\n  const client = useMastraClient();\n  const combine = useCombineByIds<DatasetExperimentResponse>(experimentIds);\n  return useQueries({\n    queries: experimentIds.map(id => datasetExperimentQueryOptions(client, datasetId, id)),\n    combine,\n  });\n};\n\nexport type ExperimentResultsOrderBy = NonNullable<\n  NonNullable<Parameters<MastraClient['listDatasetExperimentResults']>[2]>['orderBy']\n>;\n\ninterface UseDatasetExperimentResultsParams<TData = DatasetExperimentResult[]> {\n  datasetId: string;\n  experimentId: string;\n  experimentStatus?: ExperimentStatus;\n  orderBy?: ExperimentResultsOrderBy;\n  queryOptions?: MastraInfiniteQueryOptions<ExperimentResultsPage, TData, QueryKey, number>;\n}\n\n/**\n * Hook to list results for a dataset experiment with infinite scroll pagination.\n * Polls every 2 seconds while experiment status is 'pending' or 'running'.\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetExperimentResults = <TData = DatasetExperimentResult[]>({\n  datasetId,\n  experimentId,\n  experimentStatus,\n  orderBy,\n  queryOptions,\n}: UseDatasetExperimentResultsParams<TData>): UseInfiniteQueryResult<TData, Error> & {\n  setEndOfListElement: (element: HTMLDivElement | null) => void;\n} => {\n  const client = useMastraClient();\n  const { inView: isEndOfListInView, setRef: setEndOfListElement } = useInView();\n\n  const query = useInfiniteQuery({\n    queryKey: ['dataset-experiment-results', datasetId, experimentId, experimentStatus, orderBy],\n    queryFn: async ({ pageParam }) => {\n      return client.listDatasetExperimentResults(datasetId, experimentId, {\n        page: pageParam,\n        perPage: RESULTS_PER_PAGE,\n        orderBy,\n      });\n    },\n    initialPageParam: 0,\n    getNextPageParam: (lastPage, _, lastPageParam) => {\n      if (!lastPage?.results?.length) {\n        return undefined;\n      }\n      const totalFetched = (lastPageParam + 1) * RESULTS_PER_PAGE;\n      const total = lastPage?.pagination?.total ?? 0;\n      if (totalFetched >= total) {\n        return undefined;\n      }\n      return lastPageParam + 1;\n    },\n    refetchInterval: isInProgress(experimentStatus) ? POLL_INTERVAL_MS : false,\n    select: (data: InfiniteData<ExperimentResultsPage, number>) => {\n      return data.pages.flatMap(page => page?.results ?? []) as TData;\n    },\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\n/**\n * Fetches every result (all pages) of several experiments of one dataset in parallel.\n * Each experiment is polled while its `status` is in progress.\n */\nexport const useAllDatasetExperimentResultsQueries = ({\n  datasetId,\n  experiments,\n}: {\n  datasetId: string;\n  experiments: ExperimentRef[];\n}): ExperimentQueriesResult<DatasetExperimentResult[]> => {\n  const client = useMastraClient();\n  const combine = useCombineByIds<DatasetExperimentResult[]>(experiments.map(experiment => experiment.id));\n  return useQueries({\n    queries: experiments.map(experiment =>\n      allDatasetExperimentResultsQueryOptions(client, datasetId, experiment.id, experiment.status),\n    ),\n    combine,\n  });\n};\n\n/**\n * Hook to fetch all scores for an experiment, transformed to Record<entityId, ClientScoreRowData[]>\n * Paginates through all pages to ensure no scores are silently dropped.\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useScoresByExperimentId = <TData = ExperimentScoresByEntity>({\n  experimentId,\n  experimentStatus,\n  queryOptions,\n}: {\n  experimentId: string;\n  experimentStatus?: ExperimentStatus;\n  queryOptions?: MastraQueryOptions<ExperimentScoresByEntity, TData, ExperimentStatusQueryKey>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  const { queryKey, queryFn, refetchInterval } = experimentScoresQueryOptions(client, experimentId, experimentStatus);\n  return useQuery<ExperimentScoresByEntity, Error, TData, ExperimentStatusQueryKey>({\n    queryKey,\n    queryFn,\n    refetchInterval,\n    ...queryOptions,\n  });\n};\n\n/**\n * Fetches every score of several experiments in parallel, each grouped by dataset item id.\n * Each experiment is polled while its `status` is in progress.\n */\nexport const useScoresByExperimentIdsQueries = ({\n  experiments,\n}: {\n  experiments: ExperimentRef[];\n}): ExperimentQueriesResult<ExperimentScoresByEntity> => {\n  const client = useMastraClient();\n  const combine = useCombineByIds<ExperimentScoresByEntity>(experiments.map(experiment => experiment.id));\n  return useQueries({\n    queries: experiments.map(experiment => experimentScoresQueryOptions(client, experiment.id, experiment.status)),\n    combine,\n  });\n};\n","import type { DatasetItemVersionResponse } 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\nexport type DatasetItemVersion = DatasetItemVersionResponse & {\n  validTo: number | null;\n  isDeleted: boolean;\n  isLatest: boolean;\n};\n\n/**\n * Hook to fetch full item history (SCD-2 rows).\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetItemVersions = <TData = DatasetItemVersion[]>({\n  datasetId,\n  itemId,\n  queryOptions,\n}: {\n  datasetId: string;\n  itemId: string;\n  queryOptions?: MastraQueryOptions<DatasetItemVersion[], TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['dataset-item-versions', datasetId, itemId],\n    queryFn: async (): Promise<DatasetItemVersion[]> => {\n      const res = await client.getItemHistory(datasetId, itemId);\n\n      return (res?.history ?? []).map(\n        (version, index): DatasetItemVersion => ({\n          id: version.id,\n          datasetId: version.datasetId,\n          datasetVersion: version.datasetVersion,\n          input: version.input,\n          groundTruth: version.groundTruth,\n          expectedTrajectory: version.expectedTrajectory,\n          toolMocks: version.toolMocks ?? undefined,\n          scorerIds: version.scorerIds ?? undefined,\n          requestContext: version.requestContext ?? undefined,\n          metadata: version.metadata ?? undefined,\n          validTo: version.validTo,\n          isDeleted: version.isDeleted,\n          createdAt: version.createdAt,\n          updatedAt: version.updatedAt,\n          isLatest: index === 0,\n        }),\n      );\n    },\n    ...queryOptions,\n  });\n};\n\n/**\n * Hook to fetch a specific version of a dataset item.\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetItemVersion = <TData = DatasetItemVersion>({\n  datasetId,\n  itemId,\n  datasetVersion,\n  latestVersion,\n  queryOptions,\n}: {\n  datasetId: string;\n  itemId: string;\n  datasetVersion: number;\n  latestVersion?: number;\n  queryOptions?: MastraQueryOptions<DatasetItemVersion, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['dataset-item-version', datasetId, itemId, datasetVersion],\n    queryFn: async (): Promise<DatasetItemVersion> => {\n      const v = await client.getDatasetItemVersion(datasetId, itemId, datasetVersion);\n\n      return {\n        id: v.id,\n        datasetId: v.datasetId,\n        datasetVersion: v.datasetVersion,\n        input: v.input,\n        groundTruth: v.groundTruth,\n        expectedTrajectory: v.expectedTrajectory,\n        toolMocks: v.toolMocks ?? undefined,\n        scorerIds: v.scorerIds ?? undefined,\n        requestContext: v.requestContext ?? undefined,\n        metadata: v.metadata ?? undefined,\n        validTo: null,\n        isDeleted: false,\n        createdAt: v.createdAt,\n        updatedAt: v.updatedAt,\n        isLatest: latestVersion != null ? datasetVersion === latestVersion : false,\n      };\n    },\n    ...queryOptions,\n  });\n};\n","import type { MastraClient } from '@mastra/client-js';\nimport { useInfiniteQuery } from '@tanstack/react-query';\nimport type { InfiniteData, QueryKey } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraInfiniteQueryOptions } from '../shared/query-options';\n\nexport interface DatasetVersion {\n  id?: string;\n  datasetId?: string;\n  version: number;\n  createdAt?: Date | string;\n  isCurrent: boolean;\n}\n\ntype DatasetVersionsPage = Awaited<ReturnType<MastraClient['listDatasetVersions']>>;\n\nconst PER_PAGE = 10;\n\n/**\n * Hook to fetch dataset versions from the API with infinite pagination.\n * `data` defaults to the flattened `DatasetVersion[]`; a user `select` replaces that.\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useDatasetVersions = <TData = DatasetVersion[]>({\n  datasetId,\n  queryOptions,\n}: {\n  datasetId: string;\n  queryOptions?: MastraInfiniteQueryOptions<DatasetVersionsPage, TData, QueryKey, number>;\n}) => {\n  const client = useMastraClient();\n\n  return useInfiniteQuery({\n    queryKey: ['dataset-versions', datasetId],\n    queryFn: async ({ pageParam }) => {\n      return client.listDatasetVersions(datasetId, { page: pageParam, perPage: PER_PAGE });\n    },\n    initialPageParam: 0,\n    getNextPageParam: (lastPage, _, lastPageParam) => {\n      if (lastPage?.pagination?.hasMore) {\n        return lastPageParam + 1;\n      }\n      return undefined;\n    },\n    select: (data: InfiniteData<DatasetVersionsPage, number>) => {\n      return data.pages\n        .flatMap(page => page?.versions ?? [])\n        .map(\n          (v, index): DatasetVersion => ({\n            id: v.id,\n            datasetId: v.datasetId,\n            version: v.version,\n            createdAt: v.createdAt,\n            isCurrent: index === 0,\n          }),\n        ) as TData;\n    },\n    ...queryOptions,\n  });\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 ListExperimentsResponse = Awaited<ReturnType<MastraClient['listExperiments']>>;\ntype GetExperimentResponse = Awaited<ReturnType<MastraClient['getExperiment']>>;\n\n/**\n * Hook to list all experiments across all datasets with optional pagination\n */\nexport const useExperiments = <TData = ListExperimentsResponse>({\n  pagination,\n  queryOptions,\n}: {\n  pagination?: { page?: number; perPage?: number };\n  queryOptions?: MastraQueryOptions<ListExperimentsResponse, TData>;\n} = {}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['experiments', pagination],\n    queryFn: () => client.listExperiments(pagination),\n    ...queryOptions,\n  });\n};\n\n/**\n * Hook to fetch a single experiment by ID without knowing its dataset\n */\nexport const useExperiment = <TData = GetExperimentResponse>({\n  experimentId,\n  queryOptions,\n}: {\n  experimentId: string;\n  queryOptions?: MastraQueryOptions<GetExperimentResponse, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['experiment', experimentId],\n    queryFn: () => client.getExperiment(experimentId),\n    ...queryOptions,\n  });\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 WorkflowSchemaResponse = Awaited<ReturnType<ReturnType<MastraClient['getWorkflow']>['getSchema']>>;\n\n/**\n * Hook to fetch workflow input/output schema by workflow ID.\n * Returns { inputSchema, outputSchema } where each is Record<string, unknown> | null.\n *\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport function useWorkflowSchema<TData = WorkflowSchemaResponse>({\n  workflowId,\n  queryOptions,\n}: {\n  workflowId: string | null;\n  queryOptions?: MastraQueryOptions<WorkflowSchemaResponse, TData>;\n}): UseQueryResult<TData, Error> {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['workflow-schema', workflowId],\n    queryFn: async () => {\n      if (!workflowId) throw new Error('No workflow selected');\n      return client.getWorkflow(workflowId).getSchema();\n    },\n    staleTime: 5 * 60 * 1000, // Cache for 5 minutes\n    ...queryOptions,\n  });\n}\n"],"mappings":";;;;;;;;;AAgBA,MAAa,kBAA+C,EAC1D,WACA,QACA,mBAKkC;CAClC,MAAM,SAAS,gBAAgB;CAC/B,OAAO,SAAS;EACd,UAAU;GAAC;GAAgB;GAAW;EAAM;EAC5C,eAAe,OAAO,eAAe,WAAW,MAAM;EACtD,OAAO;EACP,GAAG;CACL,CAAC;AACH;AAEA,MAAMA,aAAW;;;;AAsBjB,MAAa,mBAAmB,EAC9B,WACA,QACA,SACA,SACA,mBAY2B;CAC3B,MAAM,SAAS,gBAAgB;CAC/B,MAAM,EAAE,QAAQ,mBAAmB,QAAQ,wBAAwB,UAAU;CAE7E,MAAM,QAAQ,iBAAiB;EAC7B,UAAU;GAAC;GAAiB;GAAW;GAAQ;GAAS;EAAO;EAC/D,SAAS,OAAO,EAAE,gBAAgB;GAQhC,OAAO,MAPW,OAAO,iBAAiB,WAAW;IACnD,MAAM;IACN,SAASA;IACT,QAAQ,UAAU,KAAA;IAClB,SAAS,WAAW,KAAA;IACpB;GACF,CAAC;EAEH;EACA,kBAAkB;EAClB,mBAAmB,UAAU,GAAG,kBAAkB;GAChD,IAAI,CAAC,UAAU,OAAO,QACpB;GAIF,KAFsB,gBAAgB,KAAKA,eAC7B,UAAU,YAAY,SAAS,IAE3C;GAEF,OAAO,gBAAgB;EACzB;EACA,OAAO;EACP,GAAG;CACL,CAAC;CAED,MAAM,QAAQ,MAAM,MAAM,MAAM,SAAQ,SAAQ,MAAM,SAAS,CAAC,CAAC,KAAK,CAAC;CACvE,MAAM,QAAQ,MAAM,MAAM,MAAM,EAAE,EAAE,YAAY;CAEhD,gBAAgB;EACd,IAAI,qBAAqB,MAAM,eAAe,CAAC,MAAM,oBACnD,MAAW,cAAc;CAE7B,GAAG,CAAC,mBAAmB,KAAK,CAAC;CAE7B,OAAO;EAAE,GAAG;EAAO,MAAM;EAAO;EAAO;CAAoB;AAC7D;;;;;;;;AC9DA,MAAa,uBAAuB,EAAE,iBAAkE,CAAC,MAAM;CAC7G,MAAM,SAAS,gBAAgB;CAC/B,MAAM,cAAc,eAAe;CAEnC,MAAM,gBAAgB,YAAY;EAChC,aAAa,WAAgC,OAAO,cAAc,MAAM;EACxE,iBAAiB;GACf,YAAiB,kBAAkB,EAAE,UAAU,CAAC,UAAU,EAAE,CAAC;EAC/D;EACA,GAAG,cAAc;CACnB,CAAC;CAED,MAAM,gBAAgB,YAAY;EAChC,aAAa,WAAgC,OAAO,cAAc,MAAM;EACxE,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,UAAU,EAAE,CAAC;GAC7D,YAAiB,kBAAkB,EAAE,UAAU,CAAC,WAAW,UAAU,SAAS,EAAE,CAAC;EACnF;EACA,GAAG,cAAc;CACnB,CAAC;CAED,MAAM,gBAAgB,YAAY;EAChC,aAAa,cAAsB,OAAO,cAAc,SAAS;EACjE,iBAAiB;GACf,YAAiB,kBAAkB,EAAE,UAAU,CAAC,UAAU,EAAE,CAAC;EAC/D;EACA,GAAG,cAAc;CACnB,CAAC;CAED,MAAM,UAAU,YAAY;EAC1B,aAAa,WAAiC,OAAO,eAAe,MAAM;EAC1E,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;GACvF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,WAAW,UAAU,SAAS,EAAE,CAAC;GACjF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,UAAU,SAAS,EAAE,CAAC;EAC5F;EACA,GAAG,cAAc;CACnB,CAAC;CAED,MAAM,aAAa,YAAY;EAC7B,aAAa,WAAoC,OAAO,kBAAkB,MAAM;EAChF,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;GACvF,YAAiB,kBAAkB,EAAE,UAAU;IAAC;IAAgB,UAAU;IAAW,UAAU;GAAM,EAAE,CAAC;GACxG,YAAiB,kBAAkB,EACjC,UAAU;IAAC;IAAyB,UAAU;IAAW,UAAU;GAAM,EAC3E,CAAC;GACD,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,UAAU,SAAS,EAAE,CAAC;EAC5F;EACA,GAAG,cAAc;CACnB,CAAC;CAED,MAAM,aAAa,YAAY;EAC7B,aAAa,EAAE,WAAW,aAA2C,OAAO,kBAAkB,WAAW,MAAM;EAC/G,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;GACvF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,WAAW,UAAU,SAAS,EAAE,CAAC;GACjF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,UAAU,SAAS,EAAE,CAAC;EAC5F;EACA,GAAG,cAAc;CACnB,CAAC;CAED,MAAM,YAAY,YAAY;EAC5B,aAAa,EAAE,WAAW,aAA2C,OAAO,iBAAiB,WAAW,MAAM;EAC9G,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;GACvF,YAAiB,kBAAkB,EAAE,UAAU;IAAC;IAAgB,UAAU;IAAW,UAAU;GAAM,EAAE,CAAC;GACxG,YAAiB,kBAAkB,EACjC,UAAU;IAAC;IAAyB,UAAU;IAAW,UAAU;GAAM,EAC3E,CAAC;GACD,YAAiB,kBAAkB,EAAE,UAAU,CAAC,4BAA4B,EAAE,CAAC;GAC/E,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,EAAE,CAAC;GACvE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,cAAc,EAAE,CAAC;GACjE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,sBAAsB,EAAE,CAAC;GACzE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,yBAAyB,EAAE,CAAC;GAC5E,YAAiB,kBAAkB,EAAE,UAAU,CAAC,2BAA2B,EAAE,CAAC;EAChF;EACA,GAAG,cAAc;CACnB,CAAC;CAGD,MAAM,mBAAmB,YAAY;EACnC,aAAa,WAA0C,OAAO,wBAAwB,MAAM;EAC5F,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;GACvF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,WAAW,UAAU,SAAS,EAAE,CAAC;GACjF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,UAAU,SAAS,EAAE,CAAC;EAC5F;EACA,GAAG,cAAc;CACnB,CAAC;CAGD,MAAM,mBAAmB,YAAY;EACnC,aAAa,WAA0C,OAAO,wBAAwB,MAAM;EAC5F,YAAY,GAAG,cAAc;GAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;GACvF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,WAAW,UAAU,SAAS,EAAE,CAAC;GACjF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,UAAU,SAAS,EAAE,CAAC;EAC5F;EACA,GAAG,cAAc;CACnB,CAAC;CAgED,OAAO;EACL;EACA;EACA;EACA;EACA;EACA;EACA;EACA,aArEkB,YAAY;GAC9B,YAAY,OAAO,EAAE,WAAW,cAAwD;IACtF,OAAO,OAAO,wBAAwB;KAAE;KAAW;IAAQ,CAAC;GAC9D;GACA,YAAY,GAAG,cAAc;IAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,UAAU,SAAS,EAAE,CAAC;IACvF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,WAAW,UAAU,SAAS,EAAE,CAAC;IACjF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,UAAU,SAAS,EAAE,CAAC;GAC5F;GACA,GAAG,cAAc;EACnB,CA2DY;EACV;EACA;EACA,mBA5DwB,YAAY;GACpC,aAAa,WAA2C,OAAO,yBAAyB,MAAM;GAC9F,YAAY,GAAG,cAAc;IAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,uBAAuB,UAAU,SAAS,EAAE,CAAC;GAC/F;GACA,GAAG,cAAc;EACnB,CAsDkB;EAChB,kBArDuB,YAAY;GACnC,aAAa,iBAAyB,OAAO,iBAAiB,YAAY;GAC1E,iBAAiB;IACf,YAAiB,kBAAkB,EAAE,UAAU,CAAC,aAAa,EAAE,CAAC;IAChE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,YAAY,EAAE,CAAC;IAC/D,YAAiB,kBAAkB,EAAE,UAAU,CAAC,qBAAqB,EAAE,CAAC;IACxE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,oBAAoB,EAAE,CAAC;IACvE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,4BAA4B,EAAE,CAAC;IAC/E,YAAiB,kBAAkB,EAAE,UAAU,CAAC,cAAc,EAAE,CAAC;IACjE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,iBAAiB,EAAE,CAAC;IACpE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,2BAA2B,EAAE,CAAC;GAChF;GACA,GAAG,cAAc;EACnB,CAwCiB;EACf,kBAvCuB,YAAY;GACnC,aAAa,WAA0C,OAAO,wBAAwB,MAAM;GAC5F,YAAY,GAAG,cAAc;IAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,aAAa,EAAE,CAAC;IAChE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,cAAc,UAAU,YAAY,EAAE,CAAC;IACvF,YAAiB,kBAAkB,EAAE,UAAU,CAAC,uBAAuB,UAAU,SAAS,EAAE,CAAC;IAC7F,YAAiB,kBAAkB,EACjC,UAAU;KAAC;KAAsB,UAAU;KAAW,UAAU;IAAY,EAC9E,CAAC;GACH;GACA,GAAG,cAAc;EACnB,CA4BiB;EACf,wBA3B6B,YAAY;GACzC,aAAa,WAAyC,OAAO,8BAA8B,MAAM;GACjG,YAAY,GAAG,cAAc;IAC3B,YAAiB,kBAAkB,EAAE,UAAU,CAAC,sBAAsB,UAAU,YAAY,EAAE,CAAC;IAC/F,YAAiB,kBAAkB,EAAE,UAAU,CAAC,4BAA4B,EAAE,CAAC;IAC/E,YAAiB,kBAAkB,EAAE,UAAU,CAAC,cAAc,EAAE,CAAC;IACjE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,sBAAsB,EAAE,CAAC;IACzE,YAAiB,kBAAkB,EAAE,UAAU,CAAC,yBAAyB,EAAE,CAAC;IAC5E,YAAiB,kBAAkB,EAAE,UAAU,CAAC,2BAA2B,EAAE,CAAC;GAChF;GACA,GAAG,cAAc;EACnB,CAgBuB;CACvB;AACF;;;;;;AC3NA,MAAa,eAA6C,EACxD,YACA,iBAIE,CAAC,MAAoC;CACvC,MAAM,SAAS,gBAAgB;CAC/B,OAAO,SAAS;EACd,UAAU,CAAC,YAAY,UAAU;EACjC,eAAe,OAAO,aAAa,UAAU;EAC7C,kBAAiB,iBAAgB;EACjC,GAAG;CACL,CAAC;AACH;AAEA,MAAM,oBAAoB;;;;;AAmB1B,MAAa,uBAAgD,EAC3D,QACA,SACA,iBAKE,CAAC,MAAwC;CAC3C,MAAM,SAAS,gBAAgB;CAC/B,MAAM,EAAE,QAAQ,mBAAmB,QAAQ,wBAAwB,UAAU;CAC7E,MAAM,aAAa,QAAQ,cAAc,KAAA;CACzC,MAAM,YAAY,QAAQ,WAAW,CAAC,OAAO,QAAQ,IAAI,KAAA;CAEzD,MAAM,QAAQ,iBAAiB;EAC7B,UAAU;GAAC;GAAY;GAAY;IAAE;IAAY;IAAW;GAAQ;EAAC;EACrE,UAAU,EAAE,gBACV,OAAO,aAAa;GAAE,MAAM;GAAW,SAAS;GAAmB;GAAY;GAAW;EAAQ,CAAC;EACrG,kBAAkB;EAClB,mBAAmB,UAAU,GAAG,kBAAkB;GAChD,IAAI,CAAC,UAAU,UAAU,UAAU,CAAC,SAAS,YAAY,SACvD;GAEF,OAAO,gBAAgB;EACzB;EACA,SAAS,SACP,KAAK,MAAM,SAAQ,SAAQ,MAAM,YAAY,CAAC,CAAC;EACjD,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;;;;;;AAOA,MAAa,cAAuC,EAClD,WACA,mBAIkC;CAClC,MAAM,SAAS,gBAAgB;CAC/B,OAAO,SAAS;EACd,UAAU,CAAC,WAAW,SAAS;EAC/B,eAAe,OAAO,WAAW,SAAS;EAC1C,GAAG;CACL,CAAC;AACH;;;;;;;;;;;;AC3FA,MAAa,yBAA6D,EACxE,WACA,eACA,YACA,mBAMkC;CAClC,MAAM,SAAS,gBAAgB;CAC/B,OAAO,SAAS;EACd,UAAU;GAAC;GAAuB;GAAW;GAAe;EAAU;EACtE,eAAe,OAAO,mBAAmB;GAAE;GAAW;GAAe;EAAW,CAAC;EACjF,GAAG;CACL,CAAC;AACH;;;ACPA,MAAM,mBAAmB;AACzB,MAAM,mBAAmB;AACzB,MAAM,kBAAkB;AAExB,MAAM,gBAAgB,WAA+B,WAAW,aAAa,WAAW;;;;;;AAOxF,MAAa,iCACX,QACA,WACA,iBAEAC,aAAqB;CACnB,UAAU;EAAC;EAAsB;EAAW;CAAY;CACxD,eAAe,OAAO,qBAAqB,WAAW,YAAY;CAClE,QAAQ;CACR,WAAW;CACX,kBAAiB,UAAU,aAAa,MAAM,MAAM,MAAM,MAAM,IAAI,mBAAmB;AACzF,CAAC;;;;;AAMH,MAAa,gCACX,QACA,cACA,qBAEAA,aAAqB;CACnB,UAAU;EAAC;EAA6B;EAAc;CAAgB;CACtE,SAAS,YAA+C;EACtD,MAAM,YAAkC,CAAC;EACzC,IAAI,OAAO;EAEX,OAAO,MAAM;GACX,MAAM,WAAW,MAAM,OAAO,kBAAkB;IAAE,OAAO;IAAc;IAAM,SAAS;GAAgB,CAAC;GACvG,UAAU,KAAK,GAAG,SAAS,MAAM;GACjC,IAAI,CAAC,SAAS,WAAW,SAAS;GAClC;EACF;EAEA,MAAM,UAAoC,CAAC;EAC3C,KAAK,MAAM,OAAO,WAChB,CAAC,QAAQ,IAAI,cAAc,CAAC,EAAA,CAAG,KAAK,GAAG;EAEzC,OAAO;CACT;CACA,iBAAiB,aAAa,gBAAgB,IAAI,mBAAmB;AACvE,CAAC;;;;;AAMH,MAAa,2CACX,QACA,WACA,cACA,qBAEAA,aAAqB;CACnB,UAAU;EAAC;EAAkC;EAAW;EAAc;CAAgB;CACtF,SAAS,YAAgD;EACvD,MAAM,aAAwC,CAAC;EAC/C,IAAI,OAAO;EAEX,OAAO,MAAM;GACX,MAAM,WAAW,MAAM,OAAO,6BAA6B,WAAW,cAAc;IAClF;IACA,SAAS;GACX,CAAC;GACD,WAAW,KAAK,GAAG,SAAS,OAAO;GACnC,IAAI,CAAC,SAAS,WAAW,WAAW,SAAS,QAAQ,WAAW,GAAG;GACnE;EACF;EAEA,OAAO;CACT;CACA,iBAAiB,aAAa,gBAAgB,IAAI,mBAAmB;AACvE,CAAC;;;;AAWH,MAAa,wBAA2D,EACtE,WACA,cACA,mBAKkC;CAClC,MAAM,SAAS,gBAAgB;CAC/B,MAAM,EAAE,UAAU,SAAS,QAAQ,WAAW,oBAAoB,8BAChE,QACA,WACA,YACF;CACA,OAAO,SAA6E;EAClF;EACA;EACA;EACA;EACA;EACA,GAAG;CACL,CAAC;AACH;;;;;AAMA,MAAa,gCAAgC,EAC3C,WACA,oBAIwD;CACxD,MAAM,SAAS,gBAAgB;CAC/B,MAAM,UAAU,gBAA2C,aAAa;CACxE,OAAO,WAAW;EAChB,SAAS,cAAc,KAAI,OAAM,8BAA8B,QAAQ,WAAW,EAAE,CAAC;EACrF;CACF,CAAC;AACH;;;;;;;AAoBA,MAAa,+BAAkE,EAC7E,WACA,cACA,kBACA,SACA,mBAGG;CACH,MAAM,SAAS,gBAAgB;CAC/B,MAAM,EAAE,QAAQ,mBAAmB,QAAQ,wBAAwB,UAAU;CAE7E,MAAM,QAAQ,iBAAiB;EAC7B,UAAU;GAAC;GAA8B;GAAW;GAAc;GAAkB;EAAO;EAC3F,SAAS,OAAO,EAAE,gBAAgB;GAChC,OAAO,OAAO,6BAA6B,WAAW,cAAc;IAClE,MAAM;IACN,SAAS;IACT;GACF,CAAC;EACH;EACA,kBAAkB;EAClB,mBAAmB,UAAU,GAAG,kBAAkB;GAChD,IAAI,CAAC,UAAU,SAAS,QACtB;GAIF,KAFsB,gBAAgB,KAAK,qBAC7B,UAAU,YAAY,SAAS,IAE3C;GAEF,OAAO,gBAAgB;EACzB;EACA,iBAAiB,aAAa,gBAAgB,IAAI,mBAAmB;EACrE,SAAS,SAAsD;GAC7D,OAAO,KAAK,MAAM,SAAQ,SAAQ,MAAM,WAAW,CAAC,CAAC;EACvD;EACA,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;;;;;AAMA,MAAa,yCAAyC,EACpD,WACA,kBAIwD;CACxD,MAAM,SAAS,gBAAgB;CAC/B,MAAM,UAAU,gBAA2C,YAAY,KAAI,eAAc,WAAW,EAAE,CAAC;CACvG,OAAO,WAAW;EAChB,SAAS,YAAY,KAAI,eACvB,wCAAwC,QAAQ,WAAW,WAAW,IAAI,WAAW,MAAM,CAC7F;EACA;CACF,CAAC;AACH;;;;;;;AAQA,MAAa,2BAA6D,EACxE,cACA,kBACA,mBAKkC;CAClC,MAAM,SAAS,gBAAgB;CAC/B,MAAM,EAAE,UAAU,SAAS,oBAAoB,6BAA6B,QAAQ,cAAc,gBAAgB;CAClH,OAAO,SAA2E;EAChF;EACA;EACA;EACA,GAAG;CACL,CAAC;AACH;;;;;AAMA,MAAa,mCAAmC,EAC9C,kBAGuD;CACvD,MAAM,SAAS,gBAAgB;CAC/B,MAAM,UAAU,gBAA0C,YAAY,KAAI,eAAc,WAAW,EAAE,CAAC;CACtG,OAAO,WAAW;EAChB,SAAS,YAAY,KAAI,eAAc,6BAA6B,QAAQ,WAAW,IAAI,WAAW,MAAM,CAAC;EAC7G;CACF,CAAC;AACH;;;;;;;;ACnRA,MAAa,0BAAwD,EACnE,WACA,QACA,mBAKkC;CAClC,MAAM,SAAS,gBAAgB;CAE/B,OAAO,SAAS;EACd,UAAU;GAAC;GAAyB;GAAW;EAAM;EACrD,SAAS,YAA2C;GAGlD,SAAQ,MAFU,OAAO,eAAe,WAAW,MAAM,EAAA,EAE5C,WAAW,CAAC,EAAA,CAAG,KACzB,SAAS,WAA+B;IACvC,IAAI,QAAQ;IACZ,WAAW,QAAQ;IACnB,gBAAgB,QAAQ;IACxB,OAAO,QAAQ;IACf,aAAa,QAAQ;IACrB,oBAAoB,QAAQ;IAC5B,WAAW,QAAQ,aAAa,KAAA;IAChC,WAAW,QAAQ,aAAa,KAAA;IAChC,gBAAgB,QAAQ,kBAAkB,KAAA;IAC1C,UAAU,QAAQ,YAAY,KAAA;IAC9B,SAAS,QAAQ;IACjB,WAAW,QAAQ;IACnB,WAAW,QAAQ;IACnB,WAAW,QAAQ;IACnB,UAAU,UAAU;GACtB,EACF;EACF;EACA,GAAG;CACL,CAAC;AACH;;;;;;AAOA,MAAa,yBAAqD,EAChE,WACA,QACA,gBACA,eACA,mBAOkC;CAClC,MAAM,SAAS,gBAAgB;CAE/B,OAAO,SAAS;EACd,UAAU;GAAC;GAAwB;GAAW;GAAQ;EAAc;EACpE,SAAS,YAAyC;GAChD,MAAM,IAAI,MAAM,OAAO,sBAAsB,WAAW,QAAQ,cAAc;GAE9E,OAAO;IACL,IAAI,EAAE;IACN,WAAW,EAAE;IACb,gBAAgB,EAAE;IAClB,OAAO,EAAE;IACT,aAAa,EAAE;IACf,oBAAoB,EAAE;IACtB,WAAW,EAAE,aAAa,KAAA;IAC1B,WAAW,EAAE,aAAa,KAAA;IAC1B,gBAAgB,EAAE,kBAAkB,KAAA;IACpC,UAAU,EAAE,YAAY,KAAA;IACxB,SAAS;IACT,WAAW;IACX,WAAW,EAAE;IACb,WAAW,EAAE;IACb,UAAU,iBAAiB,OAAO,mBAAmB,gBAAgB;GACvE;EACF;EACA,GAAG;CACL,CAAC;AACH;;;ACtFA,MAAM,WAAW;;;;;;;AAQjB,MAAa,sBAAgD,EAC3D,WACA,mBAII;CACJ,MAAM,SAAS,gBAAgB;CAE/B,OAAO,iBAAiB;EACtB,UAAU,CAAC,oBAAoB,SAAS;EACxC,SAAS,OAAO,EAAE,gBAAgB;GAChC,OAAO,OAAO,oBAAoB,WAAW;IAAE,MAAM;IAAW,SAAS;GAAS,CAAC;EACrF;EACA,kBAAkB;EAClB,mBAAmB,UAAU,GAAG,kBAAkB;GAChD,IAAI,UAAU,YAAY,SACxB,OAAO,gBAAgB;EAG3B;EACA,SAAS,SAAoD;GAC3D,OAAO,KAAK,MACT,SAAQ,SAAQ,MAAM,YAAY,CAAC,CAAC,CAAC,CACrC,KACE,GAAG,WAA2B;IAC7B,IAAI,EAAE;IACN,WAAW,EAAE;IACb,SAAS,EAAE;IACX,WAAW,EAAE;IACb,WAAW,UAAU;GACvB,EACF;EACJ;EACA,GAAG;CACL,CAAC;AACH;;;;;;AChDA,MAAa,kBAAmD,EAC9D,YACA,iBAIE,CAAC,MAAoC;CACvC,MAAM,SAAS,gBAAgB;CAC/B,OAAO,SAAS;EACd,UAAU,CAAC,eAAe,UAAU;EACpC,eAAe,OAAO,gBAAgB,UAAU;EAChD,GAAG;CACL,CAAC;AACH;;;;AAKA,MAAa,iBAAgD,EAC3D,cACA,mBAIkC;CAClC,MAAM,SAAS,gBAAgB;CAC/B,OAAO,SAAS;EACd,UAAU,CAAC,cAAc,YAAY;EACrC,eAAe,OAAO,cAAc,YAAY;EAChD,GAAG;CACL,CAAC;AACH;;;;;;;;;AC7BA,SAAgB,kBAAkD,EAChE,YACA,gBAI+B;CAC/B,MAAM,SAAS,gBAAgB;CAE/B,OAAO,SAAS;EACd,UAAU,CAAC,mBAAmB,UAAU;EACxC,SAAS,YAAY;GACnB,IAAI,CAAC,YAAY,MAAM,IAAI,MAAM,sBAAsB;GACvD,OAAO,OAAO,YAAY,UAAU,CAAC,CAAC,UAAU;EAClD;EACA,WAAW;EACX,GAAG;CACL,CAAC;AACH"}

package/dist/hooks/embedders/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","names":["useMastraClient"],"sources":["../../../src/hooks/embedders/use-embedders.ts"],"sourcesContent":["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\nexport interface Embedder {\n  id: string;\n  provider: string;\n  name: string;\n  description: string;\n  dimensions: number;\n  maxInputTokens: number;\n}\n\ntype ListEmbeddersResponse = Awaited<ReturnType<MastraClient['listEmbedders']>>;\n\nexport function useEmbedders<TData = ListEmbeddersResponse>({\n  queryOptions,\n}: { queryOptions?: MastraQueryOptions<ListEmbeddersResponse, TData> } = {}): UseQueryResult<TData, Error> {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['embedders'],\n    queryFn: async () => {\n      const data = await client.listEmbedders();\n      return data;\n    },\n    staleTime: 30000, // Cache for 30 seconds\n    ...queryOptions,\n  });\n}\n"],"mappings":";;;;AAiBA,SAAgB,aAA4C,EAC1D,iBACuE,CAAC,GAAiC;CACzG,MAAM,SAASA,8BAAAA,gBAAgB;CAE/B,QAAA,GAAA,sBAAA,SAAA,CAAgB;EACd,UAAU,CAAC,WAAW;EACtB,SAAS,YAAY;GAEnB,OAAO,MADY,OAAO,cAAc;EAE1C;EACA,WAAW;EACX,GAAG;CACL,CAAC;AACH"}
+{"version":3,"file":"index.cjs","names":["useMastraClient","useQuery"],"sources":["../../../src/hooks/embedders/use-embedders.ts"],"sourcesContent":["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\nexport interface Embedder {\n  id: string;\n  provider: string;\n  name: string;\n  description: string;\n  dimensions: number;\n  maxInputTokens: number;\n}\n\ntype ListEmbeddersResponse = Awaited<ReturnType<MastraClient['listEmbedders']>>;\n\nexport function useEmbedders<TData = ListEmbeddersResponse>({\n  queryOptions,\n}: { queryOptions?: MastraQueryOptions<ListEmbeddersResponse, TData> } = {}): UseQueryResult<TData, Error> {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['embedders'],\n    queryFn: async () => {\n      const data = await client.listEmbedders();\n      return data;\n    },\n    staleTime: 30000, // Cache for 30 seconds\n    ...queryOptions,\n  });\n}\n"],"mappings":";;;;AAiBA,SAAgB,aAA4C,EAC1D,iBACuE,CAAC,GAAiC;CACzG,MAAM,SAASA,8BAAAA,gBAAgB;CAE/B,QAAA,GAAOC,sBAAAA,SAAAA,CAAS;EACd,UAAU,CAAC,WAAW;EACtB,SAAS,YAAY;GAEnB,OAAO,MADY,OAAO,cAAc;EAE1C;EACA,WAAW;EACX,GAAG;CACL,CAAC;AACH"}

package/dist/hooks/experiments/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","names":["getOrCreate","useMastraClient","useCombineByIds","useMastraClient","useMastraClient","useMastraClient","useMastraClient","useInView"],"sources":["../../../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-infinite-experiments.ts"],"sourcesContent":["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';\n\nconst INPUT_TOKEN_METRIC = 'mastra_model_total_input_tokens';\nconst OUTPUT_TOKEN_METRIC = 'mastra_model_total_output_tokens';\nconst POLL_INTERVAL_MS = 2000;\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\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\ntype UsageAccumulator = {\n  inputTokens?: number;\n  outputTokens?: number;\n  estimatedCost: number;\n  costUnits: Set<string>;\n  hasUnknownCostUnit: boolean;\n};\n\nconst createAccumulator = (): UsageAccumulator => ({\n  estimatedCost: 0,\n  costUnits: new Set(),\n  hasUnknownCostUnit: false,\n});\n\ntype BreakdownGroup = GetMetricBreakdownResponse['groups'][number];\n\nfunction accumulate(accumulator: UsageAccumulator, tokenField: 'inputTokens' | 'outputTokens', group: BreakdownGroup) {\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\nfunction 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\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 metricName = group.dimensions.name;\n    if (metricName !== INPUT_TOKEN_METRIC && metricName !== OUTPUT_TOKEN_METRIC) continue;\n\n    const tokenField = metricName === INPUT_TOKEN_METRIC ? 'inputTokens' : 'outputTokens';\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 { 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":";;;;;;;;AASA,MAAM,qBAAqB;AAC3B,MAAM,sBAAsB;AAC5B,MAAM,mBAAmB;AAyBzB,MAAM,2BAA6C;CACjD,eAAe;CACf,2BAAW,IAAI,IAAI;CACnB,oBAAoB;AACtB;AAIA,SAAS,WAAW,aAA+B,YAA4C,OAAuB;CACpH,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,SAAS,QAAQ,aAAoD;CAEnE,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;;AAGA,SAAS,kBAAkB,UAAuD;CAChF,MAAM,UAAU,kBAAkB;CAClC,MAAM,+BAAe,IAAI,IAA8B;CAEvD,KAAK,MAAM,SAAS,SAAS,QAAQ;EACnC,MAAM,aAAa,MAAM,WAAW;EACpC,IAAI,eAAe,sBAAsB,eAAe,qBAAqB;EAE7E,MAAM,aAAa,eAAe,qBAAqB,gBAAgB;EACvE,WAAW,SAAS,YAAY,KAAK;EAErC,MAAM,UAAU,MAAM,WAAW;EACjC,IAAI,SAAS,WAAWA,YAAAA,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,SAASC,8BAAAA,gBAAgB;CAC/B,MAAM,UAAUC,uBAAAA,gBAAiC,YAAY,KAAI,eAAc,WAAW,EAAE,CAAC;CAE7F,QAAA,GAAA,sBAAA,WAAA,CAAkB;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,qBAAqB,mBAAmB;GAC5G;EACF,CAAC;EACD;CACF,CAAC;AACH;;;;;;;;;;AC9GA,MAAa,wBAAwB,EACnC,cACA,kBACA,iBACA,mBAC8B;CAC9B,MAAM,SAASC,8BAAAA,gBAAgB;CAC/B,MAAM,YAAY,QAAQ,YAAY,KAAK;CAG3C,MAAM,SAAA,GAAA,sBAAA,SAAA,CAAiB;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,SAASC,8BAAAA,gBAAgB;CAE/B,QAAA,GAAA,sBAAA,SAAA,CAAgB;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,SAASC,8BAAAA,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,QAAA,GAAA,sBAAA,SAAA,CAAgB;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;;;AChCA,MAAa,uBAAuB;AAWpC,SAAgB,uBAAoD,EAClE,WACA,QACA,SACA,gBAQA;CACA,MAAM,SAASC,8BAAAA,gBAAgB;CAC/B,MAAM,EAAE,QAAQ,mBAAmB,QAAQ,wBAAwBC,oBAAAA,UAAU;CAC7E,MAAM,aAAa,QAAQ,cAAc,KAAA;CACzC,MAAM,WAAW,QAAQ,YAAY,KAAA;CAErC,MAAM,SAAA,GAAA,sBAAA,iBAAA,CAAyB;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,CAAA,GAAA,MAAA,UAAA,OAAgB;EACd,IAAI,qBAAqB,MAAM,eAAe,CAAC,MAAM,oBACnD,MAAW,cAAc;CAE7B,GAAG,CAAC,mBAAmB,KAAK,CAAC;CAE7B,OAAO;EAAE,GAAG;EAAO;CAAoB;AACzC"}
+{"version":3,"file":"index.cjs","names":["getOrCreate","useMastraClient","useCombineByIds","useQueries","useMastraClient","useQuery","useMastraClient","useQuery","useMastraClient","useQuery","useMastraClient","useInView","useInfiniteQuery"],"sources":["../../../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-infinite-experiments.ts"],"sourcesContent":["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';\n\nconst INPUT_TOKEN_METRIC = 'mastra_model_total_input_tokens';\nconst OUTPUT_TOKEN_METRIC = 'mastra_model_total_output_tokens';\nconst POLL_INTERVAL_MS = 2000;\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\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\ntype UsageAccumulator = {\n  inputTokens?: number;\n  outputTokens?: number;\n  estimatedCost: number;\n  costUnits: Set<string>;\n  hasUnknownCostUnit: boolean;\n};\n\nconst createAccumulator = (): UsageAccumulator => ({\n  estimatedCost: 0,\n  costUnits: new Set(),\n  hasUnknownCostUnit: false,\n});\n\ntype BreakdownGroup = GetMetricBreakdownResponse['groups'][number];\n\nfunction accumulate(accumulator: UsageAccumulator, tokenField: 'inputTokens' | 'outputTokens', group: BreakdownGroup) {\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\nfunction 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\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 metricName = group.dimensions.name;\n    if (metricName !== INPUT_TOKEN_METRIC && metricName !== OUTPUT_TOKEN_METRIC) continue;\n\n    const tokenField = metricName === INPUT_TOKEN_METRIC ? 'inputTokens' : 'outputTokens';\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 { 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":";;;;;;;;AASA,MAAM,qBAAqB;AAC3B,MAAM,sBAAsB;AAC5B,MAAM,mBAAmB;AAyBzB,MAAM,2BAA6C;CACjD,eAAe;CACf,2BAAW,IAAI,IAAI;CACnB,oBAAoB;AACtB;AAIA,SAAS,WAAW,aAA+B,YAA4C,OAAuB;CACpH,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,SAAS,QAAQ,aAAoD;CAEnE,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;;AAGA,SAAS,kBAAkB,UAAuD;CAChF,MAAM,UAAU,kBAAkB;CAClC,MAAM,+BAAe,IAAI,IAA8B;CAEvD,KAAK,MAAM,SAAS,SAAS,QAAQ;EACnC,MAAM,aAAa,MAAM,WAAW;EACpC,IAAI,eAAe,sBAAsB,eAAe,qBAAqB;EAE7E,MAAM,aAAa,eAAe,qBAAqB,gBAAgB;EACvE,WAAW,SAAS,YAAY,KAAK;EAErC,MAAM,UAAU,MAAM,WAAW;EACjC,IAAI,SAAS,WAAWA,YAAAA,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,SAASC,8BAAAA,gBAAgB;CAC/B,MAAM,UAAUC,uBAAAA,gBAAiC,YAAY,KAAI,eAAc,WAAW,EAAE,CAAC;CAE7F,QAAA,GAAOC,sBAAAA,WAAAA,CAAW;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,qBAAqB,mBAAmB;GAC5G;EACF,CAAC;EACD;CACF,CAAC;AACH;;;;;;;;;;AC9GA,MAAa,wBAAwB,EACnC,cACA,kBACA,iBACA,mBAC8B;CAC9B,MAAM,SAASC,8BAAAA,gBAAgB;CAC/B,MAAM,YAAY,QAAQ,YAAY,KAAK;CAG3C,MAAM,SAAA,GAAQC,sBAAAA,SAAAA,CAAS;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,SAASC,8BAAAA,gBAAgB;CAE/B,QAAA,GAAOC,sBAAAA,SAAAA,CAAS;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,SAASC,8BAAAA,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,QAAA,GAAOC,sBAAAA,SAAAA,CAAS;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;;;AChCA,MAAa,uBAAuB;AAWpC,SAAgB,uBAAoD,EAClE,WACA,QACA,SACA,gBAQA;CACA,MAAM,SAASC,8BAAAA,gBAAgB;CAC/B,MAAM,EAAE,QAAQ,mBAAmB,QAAQ,wBAAwBC,oBAAAA,UAAU;CAC7E,MAAM,aAAa,QAAQ,cAAc,KAAA;CACzC,MAAM,WAAW,QAAQ,YAAY,KAAA;CAErC,MAAM,SAAA,GAAQC,sBAAAA,iBAAAA,CAAiB;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,CAAA,GAAA,MAAA,UAAA,OAAgB;EACd,IAAI,qBAAqB,MAAM,eAAe,CAAC,MAAM,oBACnD,MAAW,cAAc;CAE7B,GAAG,CAAC,mBAAmB,KAAK,CAAC;CAE7B,OAAO;EAAE,GAAG;EAAO;CAAoB;AACzC"}

package/dist/hooks/feedback/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","names":["useMastraClient"],"sources":["../../../src/hooks/feedback/use-feedback.ts"],"sourcesContent":["import type { MastraClient } from '@mastra/client-js';\nimport { useMutation, useQueryClient } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraMutationOptions } from '../shared/query-options';\n\nexport type FeedbackReviewStatus = 'needs-review' | 'reviewed';\n\ntype UpdateFeedbackReviewStatusVariables = { feedbackId: string; reviewStatus: FeedbackReviewStatus };\ntype UpdateFeedbackReviewStatusResponse = Awaited<ReturnType<MastraClient['updateFeedbackReviewStatus']>>;\n\nexport function useUpdateFeedbackReviewStatus({\n  queryOptions,\n}: {\n  queryOptions?: MastraMutationOptions<UpdateFeedbackReviewStatusResponse, UpdateFeedbackReviewStatusVariables>;\n} = {}) {\n  const client = useMastraClient();\n  const queryClient = useQueryClient();\n\n  return useMutation({\n    mutationFn: ({ feedbackId, reviewStatus }: UpdateFeedbackReviewStatusVariables) =>\n      client.updateFeedbackReviewStatus({ feedbackId, reviewStatus }),\n    onSuccess: () =>\n      Promise.all([\n        queryClient.invalidateQueries({ queryKey: ['trace-feedback'] }),\n        queryClient.invalidateQueries({ queryKey: ['span-feedback'] }),\n      ]),\n    ...queryOptions,\n  });\n}\n"],"mappings":";;;;;AAUA,SAAgB,8BAA8B,EAC5C,iBAGE,CAAC,GAAG;CACN,MAAM,SAASA,8BAAAA,gBAAgB;CAC/B,MAAM,eAAA,GAAA,sBAAA,eAAA,CAA6B;CAEnC,QAAA,GAAA,sBAAA,YAAA,CAAmB;EACjB,aAAa,EAAE,YAAY,mBACzB,OAAO,2BAA2B;GAAE;GAAY;EAAa,CAAC;EAChE,iBACE,QAAQ,IAAI,CACV,YAAY,kBAAkB,EAAE,UAAU,CAAC,gBAAgB,EAAE,CAAC,GAC9D,YAAY,kBAAkB,EAAE,UAAU,CAAC,eAAe,EAAE,CAAC,CAC/D,CAAC;EACH,GAAG;CACL,CAAC;AACH"}
+{"version":3,"file":"index.cjs","names":["useMastraClient","useQueryClient","useMutation"],"sources":["../../../src/hooks/feedback/use-feedback.ts"],"sourcesContent":["import type { MastraClient } from '@mastra/client-js';\nimport { useMutation, useQueryClient } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraMutationOptions } from '../shared/query-options';\n\nexport type FeedbackReviewStatus = 'needs-review' | 'reviewed';\n\ntype UpdateFeedbackReviewStatusVariables = { feedbackId: string; reviewStatus: FeedbackReviewStatus };\ntype UpdateFeedbackReviewStatusResponse = Awaited<ReturnType<MastraClient['updateFeedbackReviewStatus']>>;\n\nexport function useUpdateFeedbackReviewStatus({\n  queryOptions,\n}: {\n  queryOptions?: MastraMutationOptions<UpdateFeedbackReviewStatusResponse, UpdateFeedbackReviewStatusVariables>;\n} = {}) {\n  const client = useMastraClient();\n  const queryClient = useQueryClient();\n\n  return useMutation({\n    mutationFn: ({ feedbackId, reviewStatus }: UpdateFeedbackReviewStatusVariables) =>\n      client.updateFeedbackReviewStatus({ feedbackId, reviewStatus }),\n    onSuccess: () =>\n      Promise.all([\n        queryClient.invalidateQueries({ queryKey: ['trace-feedback'] }),\n        queryClient.invalidateQueries({ queryKey: ['span-feedback'] }),\n      ]),\n    ...queryOptions,\n  });\n}\n"],"mappings":";;;;;AAUA,SAAgB,8BAA8B,EAC5C,iBAGE,CAAC,GAAG;CACN,MAAM,SAASA,8BAAAA,gBAAgB;CAC/B,MAAM,eAAA,GAAcC,sBAAAA,eAAAA,CAAe;CAEnC,QAAA,GAAOC,sBAAAA,YAAAA,CAAY;EACjB,aAAa,EAAE,YAAY,mBACzB,OAAO,2BAA2B;GAAE;GAAY;EAAa,CAAC;EAChE,iBACE,QAAQ,IAAI,CACV,YAAY,kBAAkB,EAAE,UAAU,CAAC,gBAAgB,EAAE,CAAC,GAC9D,YAAY,kBAAkB,EAAE,UAAU,CAAC,eAAe,EAAE,CAAC,CAC/D,CAAC;EACH,GAAG;CACL,CAAC;AACH"}

package/dist/hooks/feedback/use-feedback.d.ts

@@ -19,13 +19,13 @@ export declare function useUpdateFeedbackReviewStatus({ queryOptions, }?: {
     value: number | string;
     comment?: (string | null) | undefined;
     feedbackUserId?: (string | null) | undefined;
-    entityType?: (("agent" | "input_processor" | "input_step_processor" | "mcp_server" | "memory" | "output_processor" | "output_step_processor" | "rag_ingestion" | "scorer" | "tool" | "tool_result_processor" | "trajectory" | "workflow_run" | "workflow_step") | null) | undefined;
+    entityType?: (("agent" | "classifier" | "input_processor" | "input_step_processor" | "mcp_server" | "memory" | "output_processor" | "output_step_processor" | "rag_ingestion" | "scorer" | "tool" | "tool_result_processor" | "trajectory" | "workflow_run" | "workflow_step") | null) | undefined;
     entityId?: (string | null) | undefined;
     entityName?: (string | null) | undefined;
-    parentEntityType?: (("agent" | "input_processor" | "input_step_processor" | "mcp_server" | "memory" | "output_processor" | "output_step_processor" | "rag_ingestion" | "scorer" | "tool" | "tool_result_processor" | "trajectory" | "workflow_run" | "workflow_step") | null) | undefined;
+    parentEntityType?: (("agent" | "classifier" | "input_processor" | "input_step_processor" | "mcp_server" | "memory" | "output_processor" | "output_step_processor" | "rag_ingestion" | "scorer" | "tool" | "tool_result_processor" | "trajectory" | "workflow_run" | "workflow_step") | null) | undefined;
     parentEntityId?: (string | null) | undefined;
     parentEntityName?: (string | null) | undefined;
-    rootEntityType?: (("agent" | "input_processor" | "input_step_processor" | "mcp_server" | "memory" | "output_processor" | "output_step_processor" | "rag_ingestion" | "scorer" | "tool" | "tool_result_processor" | "trajectory" | "workflow_run" | "workflow_step") | null) | undefined;
+    rootEntityType?: (("agent" | "classifier" | "input_processor" | "input_step_processor" | "mcp_server" | "memory" | "output_processor" | "output_step_processor" | "rag_ingestion" | "scorer" | "tool" | "tool_result_processor" | "trajectory" | "workflow_run" | "workflow_step") | null) | undefined;
     rootEntityId?: (string | null) | undefined;
     rootEntityName?: (string | null) | undefined;
     userId?: (string | null) | undefined;

package/dist/hooks/llm/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","names":["useMastraClient"],"sources":["../../../src/hooks/llm/use-llm-providers.ts"],"sourcesContent":["import type { ListAgentsModelProvidersResponse } 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\nexport const useLLMProviders = <TData = ListAgentsModelProvidersResponse>({\n  queryOptions,\n}: { queryOptions?: MastraQueryOptions<ListAgentsModelProvidersResponse, TData> } = {}): UseQueryResult<\n  TData,\n  Error\n> => {\n  const client = useMastraClient();\n\n  return useQuery<ListAgentsModelProvidersResponse, Error, TData>({\n    queryKey: ['llm-providers'],\n    queryFn: async () => client.listAgentsModelProviders(),\n    retry: false,\n    ...queryOptions,\n  });\n};\n"],"mappings":";;;;AAMA,MAAa,mBAA6D,EACxE,iBACkF,CAAC,MAGhF;CACH,MAAM,SAASA,8BAAAA,gBAAgB;CAE/B,QAAA,GAAA,sBAAA,SAAA,CAAgE;EAC9D,UAAU,CAAC,eAAe;EAC1B,SAAS,YAAY,OAAO,yBAAyB;EACrD,OAAO;EACP,GAAG;CACL,CAAC;AACH"}
+{"version":3,"file":"index.cjs","names":["useMastraClient","useQuery"],"sources":["../../../src/hooks/llm/use-llm-providers.ts"],"sourcesContent":["import type { ListAgentsModelProvidersResponse } 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\nexport const useLLMProviders = <TData = ListAgentsModelProvidersResponse>({\n  queryOptions,\n}: { queryOptions?: MastraQueryOptions<ListAgentsModelProvidersResponse, TData> } = {}): UseQueryResult<\n  TData,\n  Error\n> => {\n  const client = useMastraClient();\n\n  return useQuery<ListAgentsModelProvidersResponse, Error, TData>({\n    queryKey: ['llm-providers'],\n    queryFn: async () => client.listAgentsModelProviders(),\n    retry: false,\n    ...queryOptions,\n  });\n};\n"],"mappings":";;;;AAMA,MAAa,mBAA6D,EACxE,iBACkF,CAAC,MAGhF;CACH,MAAM,SAASA,8BAAAA,gBAAgB;CAE/B,QAAA,GAAOC,sBAAAA,SAAAA,CAAyD;EAC9D,UAAU,CAAC,eAAe;EAC1B,SAAS,YAAY,OAAO,yBAAyB;EACrD,OAAO;EACP,GAAG;CACL,CAAC;AACH"}

package/dist/hooks/logs/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","names":["isUnsupportedObservabilityOperationError","isObservabilityUnavailableError","useMastraClient","useInView"],"sources":["../../../src/hooks/logs/use-logs.ts"],"sourcesContent":["import type { MastraClient } from '@mastra/client-js';\nimport { useInfiniteQuery } from '@tanstack/react-query';\nimport { useEffect } from 'react';\nimport { useMastraClient } from '../../mastra-client-context';\nimport { isObservabilityUnavailableError, isUnsupportedObservabilityOperationError } from '../query/query-utils';\nimport type { MastraInfiniteQueryOptions } from '../shared/query-options';\n\nimport { useInView } from '../shared/use-in-view';\nimport type { LogRecord } from './types';\n\ntype ListLogsArgs = NonNullable<Parameters<MastraClient['listLogsVNext']>[0]>;\ntype ListLogsResponse = Awaited<ReturnType<MastraClient['listLogsVNext']>>;\n\ninterface UseLogsReturn {\n  data: LogRecord[] | undefined;\n  hasNextPage: boolean;\n  isFetchingNextPage: boolean;\n  fetchNextPage: () => void;\n  isLoading: boolean;\n  isError: boolean;\n  error: Error | null;\n  setEndOfListElement: (node: HTMLDivElement | null) => void;\n}\n\nconst LOGS_PER_PAGE = 20;\nconst LOGS_REFETCH_INTERVAL_MS = 10000;\n\nexport type LogsOrderBy = NonNullable<ListLogsArgs['orderBy']>;\n\nconst DEFAULT_LOGS_ORDER_BY: LogsOrderBy = { field: 'timestamp', direction: 'DESC' };\n\nexport interface LogsFilters {\n  filters?: ListLogsArgs['filters'];\n  orderBy?: LogsOrderBy;\n  /** TanStack overrides spread last. The hook returns the selected log list, so `select` must keep that shape. */\n  queryOptions?: MastraInfiniteQueryOptions<\n    ListLogsResponse,\n    ReturnType<typeof selectLogs>,\n    readonly unknown[],\n    number\n  >;\n}\n\nfunction getNextPageParam(lastPage: ListLogsResponse | undefined, _allPages: unknown, lastPageParam: number) {\n  if (lastPage?.pagination?.hasMore) {\n    return lastPageParam + 1;\n  }\n  return undefined;\n}\n\n/** Deduplicates logs across loaded pages. Offset-based pagination duplicates rows at the page\n *  boundary whenever new logs are inserted between sequential `fetchNextPage` calls — without\n *  this filter the duplicates produce duplicate React keys and chaotic virtualizer offsets.\n *  Falls back to the full serialized record when `logId` is absent so logs that happen to\n *  share `(timestamp, message, data)` but differ in other fields aren't collapsed. */\nfunction selectLogs(data: { pages: ListLogsResponse[] }) {\n  const seen = new Set<string>();\n  const result = [];\n  for (const page of data.pages) {\n    for (const log of page.logs ?? []) {\n      const key = log.logId ?? JSON.stringify(log);\n      if (seen.has(key)) continue;\n      seen.add(key);\n      result.push(log);\n    }\n  }\n  return result;\n}\n\nexport function getLogsRefetchInterval(query: { state: { error: unknown } }) {\n  if (\n    isUnsupportedObservabilityOperationError(query.state.error, 'logs') ||\n    isObservabilityUnavailableError(query.state.error)\n  ) {\n    return false;\n  }\n  return LOGS_REFETCH_INTERVAL_MS;\n}\n\nexport const useLogs: (props?: LogsFilters) => UseLogsReturn = ({\n  filters,\n  orderBy = DEFAULT_LOGS_ORDER_BY,\n  queryOptions,\n}: LogsFilters = {}) => {\n  const client = useMastraClient();\n  const { inView: isEndOfListInView, setRef: setEndOfListElement } = useInView();\n\n  const query = useInfiniteQuery<ListLogsResponse, Error, ReturnType<typeof selectLogs>, readonly unknown[], number>({\n    queryKey: ['logs', filters, orderBy],\n    queryFn: ({ pageParam }) =>\n      client.listLogsVNext({\n        pagination: { page: pageParam, perPage: LOGS_PER_PAGE },\n        filters,\n        orderBy,\n      }),\n    initialPageParam: 0,\n    getNextPageParam,\n    select: selectLogs,\n    retry: false,\n    refetchInterval: getLogsRefetchInterval,\n    ...queryOptions,\n  });\n\n  const { hasNextPage, isFetchingNextPage, fetchNextPage, data, isLoading, isError, error } = query;\n\n  useEffect(() => {\n    if (isEndOfListInView && hasNextPage && !isFetchingNextPage) {\n      void fetchNextPage();\n    }\n  }, [isEndOfListInView, hasNextPage, isFetchingNextPage, fetchNextPage]);\n\n  return { data, hasNextPage, isFetchingNextPage, fetchNextPage, isLoading, isError, error, setEndOfListElement };\n};\n"],"mappings":";;;;;;;AAwBA,MAAM,gBAAgB;AACtB,MAAM,2BAA2B;AAIjC,MAAM,wBAAqC;CAAE,OAAO;CAAa,WAAW;AAAO;AAcnF,SAAS,iBAAiB,UAAwC,WAAoB,eAAuB;CAC3G,IAAI,UAAU,YAAY,SACxB,OAAO,gBAAgB;AAG3B;;;;;;AAOA,SAAS,WAAW,MAAqC;CACvD,MAAM,uBAAO,IAAI,IAAY;CAC7B,MAAM,SAAS,CAAC;CAChB,KAAK,MAAM,QAAQ,KAAK,OACtB,KAAK,MAAM,OAAO,KAAK,QAAQ,CAAC,GAAG;EACjC,MAAM,MAAM,IAAI,SAAS,KAAK,UAAU,GAAG;EAC3C,IAAI,KAAK,IAAI,GAAG,GAAG;EACnB,KAAK,IAAI,GAAG;EACZ,OAAO,KAAK,GAAG;CACjB;CAEF,OAAO;AACT;AAEA,SAAgB,uBAAuB,OAAsC;CAC3E,IACEA,oBAAAA,yCAAyC,MAAM,MAAM,OAAO,MAAM,KAClEC,oBAAAA,gCAAgC,MAAM,MAAM,KAAK,GAEjD,OAAO;CAET,OAAO;AACT;AAEA,MAAa,WAAmD,EAC9D,SACA,UAAU,uBACV,iBACe,CAAC,MAAM;CACtB,MAAM,SAASC,8BAAAA,gBAAgB;CAC/B,MAAM,EAAE,QAAQ,mBAAmB,QAAQ,wBAAwBC,oBAAAA,UAAU;CAkB7E,MAAM,EAAE,aAAa,oBAAoB,eAAe,MAAM,WAAW,SAAS,WAAA,GAAA,sBAAA,iBAAA,CAhBiC;EACjH,UAAU;GAAC;GAAQ;GAAS;EAAO;EACnC,UAAU,EAAE,gBACV,OAAO,cAAc;GACnB,YAAY;IAAE,MAAM;IAAW,SAAS;GAAc;GACtD;GACA;EACF,CAAC;EACH,kBAAkB;EAClB;EACA,QAAQ;EACR,OAAO;EACP,iBAAiB;EACjB,GAAG;CACL,CAEgG;CAEhG,CAAA,GAAA,MAAA,UAAA,OAAgB;EACd,IAAI,qBAAqB,eAAe,CAAC,oBACvC,cAAmB;CAEvB,GAAG;EAAC;EAAmB;EAAa;EAAoB;CAAa,CAAC;CAEtE,OAAO;EAAE;EAAM;EAAa;EAAoB;EAAe;EAAW;EAAS;EAAO;CAAoB;AAChH"}
+{"version":3,"file":"index.cjs","names":["isUnsupportedObservabilityOperationError","isObservabilityUnavailableError","useMastraClient","useInView","useInfiniteQuery"],"sources":["../../../src/hooks/logs/use-logs.ts"],"sourcesContent":["import type { MastraClient } from '@mastra/client-js';\nimport { useInfiniteQuery } from '@tanstack/react-query';\nimport { useEffect } from 'react';\nimport { useMastraClient } from '../../mastra-client-context';\nimport { isObservabilityUnavailableError, isUnsupportedObservabilityOperationError } from '../query/query-utils';\nimport type { MastraInfiniteQueryOptions } from '../shared/query-options';\n\nimport { useInView } from '../shared/use-in-view';\nimport type { LogRecord } from './types';\n\ntype ListLogsArgs = NonNullable<Parameters<MastraClient['listLogsVNext']>[0]>;\ntype ListLogsResponse = Awaited<ReturnType<MastraClient['listLogsVNext']>>;\n\ninterface UseLogsReturn {\n  data: LogRecord[] | undefined;\n  hasNextPage: boolean;\n  isFetchingNextPage: boolean;\n  fetchNextPage: () => void;\n  isLoading: boolean;\n  isError: boolean;\n  error: Error | null;\n  setEndOfListElement: (node: HTMLDivElement | null) => void;\n}\n\nconst LOGS_PER_PAGE = 20;\nconst LOGS_REFETCH_INTERVAL_MS = 10000;\n\nexport type LogsOrderBy = NonNullable<ListLogsArgs['orderBy']>;\n\nconst DEFAULT_LOGS_ORDER_BY: LogsOrderBy = { field: 'timestamp', direction: 'DESC' };\n\nexport interface LogsFilters {\n  filters?: ListLogsArgs['filters'];\n  orderBy?: LogsOrderBy;\n  /** TanStack overrides spread last. The hook returns the selected log list, so `select` must keep that shape. */\n  queryOptions?: MastraInfiniteQueryOptions<\n    ListLogsResponse,\n    ReturnType<typeof selectLogs>,\n    readonly unknown[],\n    number\n  >;\n}\n\nfunction getNextPageParam(lastPage: ListLogsResponse | undefined, _allPages: unknown, lastPageParam: number) {\n  if (lastPage?.pagination?.hasMore) {\n    return lastPageParam + 1;\n  }\n  return undefined;\n}\n\n/** Deduplicates logs across loaded pages. Offset-based pagination duplicates rows at the page\n *  boundary whenever new logs are inserted between sequential `fetchNextPage` calls — without\n *  this filter the duplicates produce duplicate React keys and chaotic virtualizer offsets.\n *  Falls back to the full serialized record when `logId` is absent so logs that happen to\n *  share `(timestamp, message, data)` but differ in other fields aren't collapsed. */\nfunction selectLogs(data: { pages: ListLogsResponse[] }) {\n  const seen = new Set<string>();\n  const result = [];\n  for (const page of data.pages) {\n    for (const log of page.logs ?? []) {\n      const key = log.logId ?? JSON.stringify(log);\n      if (seen.has(key)) continue;\n      seen.add(key);\n      result.push(log);\n    }\n  }\n  return result;\n}\n\nexport function getLogsRefetchInterval(query: { state: { error: unknown } }) {\n  if (\n    isUnsupportedObservabilityOperationError(query.state.error, 'logs') ||\n    isObservabilityUnavailableError(query.state.error)\n  ) {\n    return false;\n  }\n  return LOGS_REFETCH_INTERVAL_MS;\n}\n\nexport const useLogs: (props?: LogsFilters) => UseLogsReturn = ({\n  filters,\n  orderBy = DEFAULT_LOGS_ORDER_BY,\n  queryOptions,\n}: LogsFilters = {}) => {\n  const client = useMastraClient();\n  const { inView: isEndOfListInView, setRef: setEndOfListElement } = useInView();\n\n  const query = useInfiniteQuery<ListLogsResponse, Error, ReturnType<typeof selectLogs>, readonly unknown[], number>({\n    queryKey: ['logs', filters, orderBy],\n    queryFn: ({ pageParam }) =>\n      client.listLogsVNext({\n        pagination: { page: pageParam, perPage: LOGS_PER_PAGE },\n        filters,\n        orderBy,\n      }),\n    initialPageParam: 0,\n    getNextPageParam,\n    select: selectLogs,\n    retry: false,\n    refetchInterval: getLogsRefetchInterval,\n    ...queryOptions,\n  });\n\n  const { hasNextPage, isFetchingNextPage, fetchNextPage, data, isLoading, isError, error } = query;\n\n  useEffect(() => {\n    if (isEndOfListInView && hasNextPage && !isFetchingNextPage) {\n      void fetchNextPage();\n    }\n  }, [isEndOfListInView, hasNextPage, isFetchingNextPage, fetchNextPage]);\n\n  return { data, hasNextPage, isFetchingNextPage, fetchNextPage, isLoading, isError, error, setEndOfListElement };\n};\n"],"mappings":";;;;;;;AAwBA,MAAM,gBAAgB;AACtB,MAAM,2BAA2B;AAIjC,MAAM,wBAAqC;CAAE,OAAO;CAAa,WAAW;AAAO;AAcnF,SAAS,iBAAiB,UAAwC,WAAoB,eAAuB;CAC3G,IAAI,UAAU,YAAY,SACxB,OAAO,gBAAgB;AAG3B;;;;;;AAOA,SAAS,WAAW,MAAqC;CACvD,MAAM,uBAAO,IAAI,IAAY;CAC7B,MAAM,SAAS,CAAC;CAChB,KAAK,MAAM,QAAQ,KAAK,OACtB,KAAK,MAAM,OAAO,KAAK,QAAQ,CAAC,GAAG;EACjC,MAAM,MAAM,IAAI,SAAS,KAAK,UAAU,GAAG;EAC3C,IAAI,KAAK,IAAI,GAAG,GAAG;EACnB,KAAK,IAAI,GAAG;EACZ,OAAO,KAAK,GAAG;CACjB;CAEF,OAAO;AACT;AAEA,SAAgB,uBAAuB,OAAsC;CAC3E,IACEA,oBAAAA,yCAAyC,MAAM,MAAM,OAAO,MAAM,KAClEC,oBAAAA,gCAAgC,MAAM,MAAM,KAAK,GAEjD,OAAO;CAET,OAAO;AACT;AAEA,MAAa,WAAmD,EAC9D,SACA,UAAU,uBACV,iBACe,CAAC,MAAM;CACtB,MAAM,SAASC,8BAAAA,gBAAgB;CAC/B,MAAM,EAAE,QAAQ,mBAAmB,QAAQ,wBAAwBC,oBAAAA,UAAU;CAkB7E,MAAM,EAAE,aAAa,oBAAoB,eAAe,MAAM,WAAW,SAAS,WAAA,GAhBpEC,sBAAAA,iBAAAA,CAAqG;EACjH,UAAU;GAAC;GAAQ;GAAS;EAAO;EACnC,UAAU,EAAE,gBACV,OAAO,cAAc;GACnB,YAAY;IAAE,MAAM;IAAW,SAAS;GAAc;GACtD;GACA;EACF,CAAC;EACH,kBAAkB;EAClB;EACA,QAAQ;EACR,OAAO;EACP,iBAAiB;EACjB,GAAG;CACL,CAEgG;CAEhG,CAAA,GAAA,MAAA,UAAA,OAAgB;EACd,IAAI,qBAAqB,eAAe,CAAC,oBACvC,cAAmB;CAEvB,GAAG;EAAC;EAAmB;EAAa;EAAoB;CAAa,CAAC;CAEtE,OAAO;EAAE;EAAM;EAAa;EAAoB;EAAe;EAAW;EAAS;EAAO;CAAoB;AAChH"}

package/dist/hooks/logs/use-logs.d.ts

@@ -37,13 +37,13 @@ declare function selectLogs(data: {
     } | null) | undefined;
     traceId?: (string | null) | undefined;
     spanId?: (string | null) | undefined;
-    entityType?: (("agent" | "input_processor" | "input_step_processor" | "mcp_server" | "memory" | "output_processor" | "output_step_processor" | "rag_ingestion" | "scorer" | "tool" | "tool_result_processor" | "trajectory" | "workflow_run" | "workflow_step") | null) | undefined;
+    entityType?: (("agent" | "classifier" | "input_processor" | "input_step_processor" | "mcp_server" | "memory" | "output_processor" | "output_step_processor" | "rag_ingestion" | "scorer" | "tool" | "tool_result_processor" | "trajectory" | "workflow_run" | "workflow_step") | null) | undefined;
     entityId?: (string | null) | undefined;
     entityName?: (string | null) | undefined;
-    parentEntityType?: (("agent" | "input_processor" | "input_step_processor" | "mcp_server" | "memory" | "output_processor" | "output_step_processor" | "rag_ingestion" | "scorer" | "tool" | "tool_result_processor" | "trajectory" | "workflow_run" | "workflow_step") | null) | undefined;
+    parentEntityType?: (("agent" | "classifier" | "input_processor" | "input_step_processor" | "mcp_server" | "memory" | "output_processor" | "output_step_processor" | "rag_ingestion" | "scorer" | "tool" | "tool_result_processor" | "trajectory" | "workflow_run" | "workflow_step") | null) | undefined;
     parentEntityId?: (string | null) | undefined;
     parentEntityName?: (string | null) | undefined;
-    rootEntityType?: (("agent" | "input_processor" | "input_step_processor" | "mcp_server" | "memory" | "output_processor" | "output_step_processor" | "rag_ingestion" | "scorer" | "tool" | "tool_result_processor" | "trajectory" | "workflow_run" | "workflow_step") | null) | undefined;
+    rootEntityType?: (("agent" | "classifier" | "input_processor" | "input_step_processor" | "mcp_server" | "memory" | "output_processor" | "output_step_processor" | "rag_ingestion" | "scorer" | "tool" | "tool_result_processor" | "trajectory" | "workflow_run" | "workflow_step") | null) | undefined;
     rootEntityId?: (string | null) | undefined;
     rootEntityName?: (string | null) | undefined;
     userId?: (string | null) | undefined;

package/dist/hooks/mcps/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","names":["useMastraClient","useMastraClient","useMastraClient","useMastraClient","useMastraClient","useMastraClient","useMastraClient"],"sources":["../../../src/hooks/mcps/use-mcp-app-html.ts","../../../src/hooks/mcps/use-mcp-app-tools.ts","../../../src/hooks/mcps/use-mcp-server-tool.ts","../../../src/hooks/mcps/use-mcp-server-tools.ts","../../../src/hooks/mcps/use-mcp-server-tools-by-id.ts","../../../src/hooks/mcps/use-mcp-servers.ts","../../../src/hooks/mcps/use-try-connect-mcp.ts"],"sourcesContent":["import type { MastraClient } from '@mastra/client-js';\nimport { useQuery } from '@tanstack/react-query';\nimport type { UseQueryResult } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraQueryOptions } from '../shared/query-options';\n\n/** The app HTML, or `null` when the resource has no text. */\ntype McpAppHtml = string | null;\n\nexport type ReadMcpServerResourceResponse = Awaited<ReturnType<MastraClient['readMcpServerResource']>>;\n\n/**\n * The HTML of an MCP tool's `ui://` app resource (from the tool's `_meta`), for rendering the tool's own UI.\n * Skips the fetch when the tool has no app resource.\n */\nexport const useMcpAppHtml = <TData = McpAppHtml>({\n  serverId,\n  appResourceUri,\n  queryOptions,\n}: {\n  serverId: string;\n  appResourceUri?: string;\n  queryOptions?: MastraQueryOptions<McpAppHtml, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['mcp-app-resource', serverId, appResourceUri],\n    queryFn: async () => {\n      if (!appResourceUri) return null;\n      const response = await client.readMcpServerResource(serverId, appResourceUri);\n      return response.contents[0]?.text ?? null;\n    },\n    enabled: appResourceUri !== undefined,\n    ...queryOptions,\n  });\n};\n","import 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\nexport interface McpAppToolInfo {\n  serverId: string;\n  toolId: string;\n  toolName: string;\n  resourceUri: string;\n}\n\n/**\n * Builds a map of tool names → MCP App info for tools across all registered\n * MCP servers that have `_meta.ui.resourceUri` (i.e., tools with interactive\n * MCP App UIs).\n *\n * For each app tool, entries are created under both the original tool name\n * and the namespaced name (`${serverId}_${toolName}`) so that agent tools\n * sourced via `MCPClient.listTools()` (which namespaces) can be resolved.\n */\nexport function useMcpAppTools<TData = Record<string, McpAppToolInfo>>({\n  queryOptions,\n}: { queryOptions?: MastraQueryOptions<Record<string, McpAppToolInfo>, TData> } = {}): UseQueryResult<TData, Error> {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['mcp-app-tools'],\n    queryFn: async () => {\n      const map: Record<string, McpAppToolInfo> = {};\n\n      const { servers } = await client.getMcpServers();\n      if (!servers?.length) return map;\n\n      const results = await Promise.allSettled(\n        servers.map(async server => {\n          const { tools } = await client.getMcpServerTools(server.id);\n          return { serverId: server.id, tools };\n        }),\n      );\n\n      for (const result of results) {\n        if (result.status !== 'fulfilled') continue;\n        const { serverId, tools } = result.value;\n        for (const tool of tools) {\n          const meta = tool._meta as { ui?: { resourceUri?: string } } | undefined;\n          const resourceUri = meta?.ui?.resourceUri;\n          if (resourceUri) {\n            const toolId = tool.name;\n            const info: McpAppToolInfo = { serverId, toolId, toolName: tool.name, resourceUri };\n            map[toolId] = info;\n            if (tool.name !== toolId) {\n              map[tool.name] = info;\n            }\n            // Also map the namespaced name used by MCPClient.listTools()\n            map[`${serverId}_${tool.name}`] = info;\n          }\n        }\n      }\n\n      return map;\n    },\n    staleTime: 60_000,\n    refetchOnWindowFocus: false,\n    ...queryOptions,\n  });\n}\n","import type { MastraClient } from '@mastra/client-js';\nimport type { RequestContext } from '@mastra/core/request-context';\nimport { useMutation, useQuery } from '@tanstack/react-query';\nimport type { UseMutationResult, UseQueryResult } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraMutationOptions, MastraQueryOptions } from '../shared/query-options';\n\ntype MCPServerToolDetails = Awaited<ReturnType<ReturnType<MastraClient['getMcpServerTool']>['details']>>;\n\n/**\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useMCPServerTool = <TData = MCPServerToolDetails>({\n  serverId,\n  toolId,\n  requestContext,\n  queryOptions,\n}: {\n  serverId: string;\n  toolId: string;\n  requestContext?: Record<string, any>;\n  queryOptions?: MastraQueryOptions<MCPServerToolDetails, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['mcp-server-tool', serverId, toolId],\n    queryFn: () => {\n      const instance = client.getMcpServerTool(serverId, toolId);\n      return instance.details(requestContext);\n    },\n    ...queryOptions,\n  });\n};\n\nexport type ExecuteMCPToolResponse = Awaited<ReturnType<ReturnType<MastraClient['getMcpServerTool']>['execute']>>;\ntype ExecuteMCPToolVariables = { data: any; requestContext?: Record<string, any> };\n\nexport const useExecuteMCPTool = ({\n  serverId,\n  toolId,\n  queryOptions,\n}: {\n  serverId: string;\n  toolId: string;\n  queryOptions?: MastraMutationOptions<ExecuteMCPToolResponse, ExecuteMCPToolVariables>;\n}): UseMutationResult<ExecuteMCPToolResponse, Error, ExecuteMCPToolVariables> => {\n  const client = useMastraClient();\n\n  return useMutation({\n    mutationFn: ({ data, requestContext }: ExecuteMCPToolVariables) => {\n      const instance = client.getMcpServerTool(serverId, toolId);\n      return instance.execute({ data, requestContext: requestContext as RequestContext });\n    },\n    ...queryOptions,\n  });\n};\n","import type { McpToolInfo as SdkMcpToolInfo } from '@mastra/client-js';\nimport type { ServerInfo } from '@mastra/core/mcp';\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\nexport const useMCPServerTools = <TData = Record<string, SdkMcpToolInfo>>({\n  selectedServer,\n  queryOptions,\n}: {\n  selectedServer: ServerInfo;\n  queryOptions?: MastraQueryOptions<Record<string, SdkMcpToolInfo>, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['mcpserver-tools', selectedServer?.id],\n    queryFn: async () => {\n      const response = await client.getMcpServerTools(selectedServer?.id);\n      const fetchedToolsArray: SdkMcpToolInfo[] = response.tools;\n      const transformedTools: Record<string, SdkMcpToolInfo> = {};\n      fetchedToolsArray.forEach((sdkToolInfo: SdkMcpToolInfo) => {\n        transformedTools[sdkToolInfo.name] = sdkToolInfo;\n      });\n\n      return transformedTools;\n    },\n    retry: false,\n    refetchOnWindowFocus: false,\n    ...queryOptions,\n  });\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 MCPServerToolsById = Record<string, Awaited<ReturnType<MastraClient['getMcpServerTools']>>['tools'][number]>;\n\n/**\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useMCPServerToolsById = <TData = MCPServerToolsById>({\n  serverId,\n  queryOptions,\n}: {\n  serverId: string | null;\n  queryOptions?: MastraQueryOptions<MCPServerToolsById, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['mcpserver-tools', serverId],\n    queryFn: async (): Promise<MCPServerToolsById> => {\n      const response = await client.getMcpServerTools(serverId!);\n      return Object.fromEntries(response.tools.map(tool => [tool.name, tool]));\n    },\n    retry: false,\n    refetchOnWindowFocus: false,\n    ...queryOptions,\n  });\n};\n","import type { McpServerListResponse } 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\nexport const useMCPServers = <TData = McpServerListResponse['servers']>({\n  requestContext,\n  queryOptions,\n}: {\n  requestContext?: Record<string, any>;\n  queryOptions?: MastraQueryOptions<McpServerListResponse['servers'], TData>;\n} = {}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['mcp-servers'],\n    queryFn: async () => {\n      const mcpServers: McpServerListResponse['servers'] = (await client.getMcpServers(requestContext)).servers;\n      return mcpServers;\n    },\n    ...queryOptions,\n  });\n};\n","import { useMutation } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraMutationOptions } from '../shared/query-options';\n\ninterface McpTool {\n  name: string;\n  description?: string;\n  inputSchema?: Record<string, unknown>;\n}\n\nexport interface TryConnectResult {\n  tools: McpTool[];\n}\n\nasync function parseResponse(response: Response): Promise<unknown> {\n  const contentType = response.headers.get('content-type') ?? '';\n\n  if (contentType.includes('text/event-stream')) {\n    const text = await response.text();\n    const lines = text.split('\\n');\n\n    for (const line of lines) {\n      if (line.startsWith('data:')) {\n        const data = line.slice('data:'.length).trim();\n        if (data) {\n          return JSON.parse(data);\n        }\n      }\n    }\n\n    throw new Error('No data found in SSE response');\n  }\n\n  return response.json();\n}\n\nasync function connectAndListTools(url: string, clientHeaders?: Record<string, string>): Promise<TryConnectResult> {\n  const baseHeaders: Record<string, string> = {\n    ...clientHeaders,\n    'Content-Type': 'application/json',\n    Accept: 'application/json, text/event-stream',\n  };\n\n  // Step 1: Initialize\n  const initResponse = await fetch(url, {\n    method: 'POST',\n    headers: baseHeaders,\n    credentials: 'include',\n    body: JSON.stringify({\n      jsonrpc: '2.0',\n      id: 1,\n      method: 'initialize',\n      params: {\n        protocolVersion: '2025-03-26',\n        capabilities: {},\n        clientInfo: { name: 'mastra-playground', version: '1.0.0' },\n      },\n    }),\n  });\n\n  if (!initResponse.ok) {\n    throw new Error(`Initialize failed: ${initResponse.status} ${initResponse.statusText}`);\n  }\n\n  const sessionId = initResponse.headers.get('Mcp-Session-Id');\n  await parseResponse(initResponse);\n\n  const sessionHeaders: Record<string, string> = {\n    ...baseHeaders,\n  };\n  if (sessionId) {\n    sessionHeaders['Mcp-Session-Id'] = sessionId;\n  }\n\n  // Step 2: Send initialized notification\n  await fetch(url, {\n    method: 'POST',\n    headers: sessionHeaders,\n    credentials: 'include',\n    body: JSON.stringify({\n      jsonrpc: '2.0',\n      method: 'notifications/initialized',\n    }),\n  });\n\n  // Step 3: List tools\n  const toolsResponse = await fetch(url, {\n    method: 'POST',\n    headers: sessionHeaders,\n    credentials: 'include',\n    body: JSON.stringify({\n      jsonrpc: '2.0',\n      id: 2,\n      method: 'tools/list',\n    }),\n  });\n\n  if (!toolsResponse.ok) {\n    throw new Error(`tools/list failed: ${toolsResponse.status} ${toolsResponse.statusText}`);\n  }\n\n  const toolsResult = (await parseResponse(toolsResponse)) as { result?: { tools?: McpTool[] } };\n\n  return {\n    tools: toolsResult.result?.tools ?? [],\n  };\n}\n\nexport const useTryConnectMcp = ({\n  queryOptions,\n}: { queryOptions?: MastraMutationOptions<TryConnectResult, string> } = {}) => {\n  const client = useMastraClient();\n  const clientHeaders = (client.options?.headers as Record<string, string>) ?? {};\n\n  return useMutation({\n    mutationFn: (url: string) => connectAndListTools(url, clientHeaders),\n    ...queryOptions,\n  });\n};\n\nexport type TryConnectMcpMutation = ReturnType<typeof useTryConnectMcp>;\n"],"mappings":";;;;;;;;AAeA,MAAa,iBAAqC,EAChD,UACA,gBACA,mBAKkC;CAClC,MAAM,SAASA,8BAAAA,gBAAgB;CAE/B,QAAA,GAAA,sBAAA,SAAA,CAAgB;EACd,UAAU;GAAC;GAAoB;GAAU;EAAc;EACvD,SAAS,YAAY;GACnB,IAAI,CAAC,gBAAgB,OAAO;GAE5B,QAAO,MADgB,OAAO,sBAAsB,UAAU,cAAc,EAAA,CAC5D,SAAS,EAAE,EAAE,QAAQ;EACvC;EACA,SAAS,mBAAmB,KAAA;EAC5B,GAAG;CACL,CAAC;AACH;;;;;;;;;;;;ACfA,SAAgB,eAAuD,EACrE,iBACgF,CAAC,GAAiC;CAClH,MAAM,SAASC,8BAAAA,gBAAgB;CAE/B,QAAA,GAAA,sBAAA,SAAA,CAAgB;EACd,UAAU,CAAC,eAAe;EAC1B,SAAS,YAAY;GACnB,MAAM,MAAsC,CAAC;GAE7C,MAAM,EAAE,YAAY,MAAM,OAAO,cAAc;GAC/C,IAAI,CAAC,SAAS,QAAQ,OAAO;GAE7B,MAAM,UAAU,MAAM,QAAQ,WAC5B,QAAQ,IAAI,OAAM,WAAU;IAC1B,MAAM,EAAE,UAAU,MAAM,OAAO,kBAAkB,OAAO,EAAE;IAC1D,OAAO;KAAE,UAAU,OAAO;KAAI;IAAM;GACtC,CAAC,CACH;GAEA,KAAK,MAAM,UAAU,SAAS;IAC5B,IAAI,OAAO,WAAW,aAAa;IACnC,MAAM,EAAE,UAAU,UAAU,OAAO;IACnC,KAAK,MAAM,QAAQ,OAAO;KAExB,MAAM,cADO,KAAK,OACQ,IAAI;KAC9B,IAAI,aAAa;MACf,MAAM,SAAS,KAAK;MACpB,MAAM,OAAuB;OAAE;OAAU;OAAQ,UAAU,KAAK;OAAM;MAAY;MAClF,IAAI,UAAU;MACd,IAAI,KAAK,SAAS,QAChB,IAAI,KAAK,QAAQ;MAGnB,IAAI,GAAG,SAAS,GAAG,KAAK,UAAU;KACpC;IACF;GACF;GAEA,OAAO;EACT;EACA,WAAW;EACX,sBAAsB;EACtB,GAAG;CACL,CAAC;AACH;;;;;;ACtDA,MAAa,oBAAkD,EAC7D,UACA,QACA,gBACA,mBAMkC;CAClC,MAAM,SAASC,8BAAAA,gBAAgB;CAE/B,QAAA,GAAA,sBAAA,SAAA,CAAgB;EACd,UAAU;GAAC;GAAmB;GAAU;EAAM;EAC9C,eAAe;GAEb,OADiB,OAAO,iBAAiB,UAAU,MACrC,CAAC,CAAC,QAAQ,cAAc;EACxC;EACA,GAAG;CACL,CAAC;AACH;AAKA,MAAa,qBAAqB,EAChC,UACA,QACA,mBAK+E;CAC/E,MAAM,SAASA,8BAAAA,gBAAgB;CAE/B,QAAA,GAAA,sBAAA,YAAA,CAAmB;EACjB,aAAa,EAAE,MAAM,qBAA8C;GAEjE,OADiB,OAAO,iBAAiB,UAAU,MACrC,CAAC,CAAC,QAAQ;IAAE;IAAsB;GAAiC,CAAC;EACpF;EACA,GAAG;CACL,CAAC;AACH;;;ACjDA,MAAa,qBAA6D,EACxE,gBACA,mBAIkC;CAClC,MAAM,SAASC,8BAAAA,gBAAgB;CAE/B,QAAA,GAAA,sBAAA,SAAA,CAAgB;EACd,UAAU,CAAC,mBAAmB,gBAAgB,EAAE;EAChD,SAAS,YAAY;GAEnB,MAAM,qBAAsC,MADrB,OAAO,kBAAkB,gBAAgB,EAAE,EAAA,CACb;GACrD,MAAM,mBAAmD,CAAC;GAC1D,kBAAkB,SAAS,gBAAgC;IACzD,iBAAiB,YAAY,QAAQ;GACvC,CAAC;GAED,OAAO;EACT;EACA,OAAO;EACP,sBAAsB;EACtB,GAAG;CACL,CAAC;AACH;;;;;;ACrBA,MAAa,yBAAqD,EAChE,UACA,mBAIkC;CAClC,MAAM,SAASC,8BAAAA,gBAAgB;CAE/B,QAAA,GAAA,sBAAA,SAAA,CAAgB;EACd,UAAU,CAAC,mBAAmB,QAAQ;EACtC,SAAS,YAAyC;GAChD,MAAM,WAAW,MAAM,OAAO,kBAAkB,QAAS;GACzD,OAAO,OAAO,YAAY,SAAS,MAAM,KAAI,SAAQ,CAAC,KAAK,MAAM,IAAI,CAAC,CAAC;EACzE;EACA,OAAO;EACP,sBAAsB;EACtB,GAAG;CACL,CAAC;AACH;;;ACxBA,MAAa,iBAA2D,EACtE,gBACA,iBAIE,CAAC,MAAoC;CACvC,MAAM,SAASC,8BAAAA,gBAAgB;CAE/B,QAAA,GAAA,sBAAA,SAAA,CAAgB;EACd,UAAU,CAAC,aAAa;EACxB,SAAS,YAAY;GAEnB,QADsD,MAAM,OAAO,cAAc,cAAc,EAAA,CAAG;EAEpG;EACA,GAAG;CACL,CAAC;AACH;;;ACTA,eAAe,cAAc,UAAsC;CAGjE,KAFoB,SAAS,QAAQ,IAAI,cAAc,KAAK,GAAA,CAE5C,SAAS,mBAAmB,GAAG;EAE7C,MAAM,SAAQ,MADK,SAAS,KAAK,EAAA,CACd,MAAM,IAAI;EAE7B,KAAK,MAAM,QAAQ,OACjB,IAAI,KAAK,WAAW,OAAO,GAAG;GAC5B,MAAM,OAAO,KAAK,MAAM,CAAc,CAAC,CAAC,KAAK;GAC7C,IAAI,MACF,OAAO,KAAK,MAAM,IAAI;EAE1B;EAGF,MAAM,IAAI,MAAM,+BAA+B;CACjD;CAEA,OAAO,SAAS,KAAK;AACvB;AAEA,eAAe,oBAAoB,KAAa,eAAmE;CACjH,MAAM,cAAsC;EAC1C,GAAG;EACH,gBAAgB;EAChB,QAAQ;CACV;CAGA,MAAM,eAAe,MAAM,MAAM,KAAK;EACpC,QAAQ;EACR,SAAS;EACT,aAAa;EACb,MAAM,KAAK,UAAU;GACnB,SAAS;GACT,IAAI;GACJ,QAAQ;GACR,QAAQ;IACN,iBAAiB;IACjB,cAAc,CAAC;IACf,YAAY;KAAE,MAAM;KAAqB,SAAS;IAAQ;GAC5D;EACF,CAAC;CACH,CAAC;CAED,IAAI,CAAC,aAAa,IAChB,MAAM,IAAI,MAAM,sBAAsB,aAAa,OAAO,GAAG,aAAa,YAAY;CAGxF,MAAM,YAAY,aAAa,QAAQ,IAAI,gBAAgB;CAC3D,MAAM,cAAc,YAAY;CAEhC,MAAM,iBAAyC,EAC7C,GAAG,YACL;CACA,IAAI,WACF,eAAe,oBAAoB;CAIrC,MAAM,MAAM,KAAK;EACf,QAAQ;EACR,SAAS;EACT,aAAa;EACb,MAAM,KAAK,UAAU;GACnB,SAAS;GACT,QAAQ;EACV,CAAC;CACH,CAAC;CAGD,MAAM,gBAAgB,MAAM,MAAM,KAAK;EACrC,QAAQ;EACR,SAAS;EACT,aAAa;EACb,MAAM,KAAK,UAAU;GACnB,SAAS;GACT,IAAI;GACJ,QAAQ;EACV,CAAC;CACH,CAAC;CAED,IAAI,CAAC,cAAc,IACjB,MAAM,IAAI,MAAM,sBAAsB,cAAc,OAAO,GAAG,cAAc,YAAY;CAK1F,OAAO,EACL,QAAO,MAHkB,cAAc,aAAa,EAAA,CAGjC,QAAQ,SAAS,CAAC,EACvC;AACF;AAEA,MAAa,oBAAoB,EAC/B,iBACsE,CAAC,MAAM;CAE7E,MAAM,gBADSC,8BAAAA,gBACa,CAAC,CAAC,SAAS,WAAsC,CAAC;CAE9E,QAAA,GAAA,sBAAA,YAAA,CAAmB;EACjB,aAAa,QAAgB,oBAAoB,KAAK,aAAa;EACnE,GAAG;CACL,CAAC;AACH"}
+{"version":3,"file":"index.cjs","names":["useMastraClient","useQuery","useMastraClient","useQuery","useMastraClient","useQuery","useMutation","useMastraClient","useQuery","useMastraClient","useQuery","useMastraClient","useQuery","useMastraClient","useMutation"],"sources":["../../../src/hooks/mcps/use-mcp-app-html.ts","../../../src/hooks/mcps/use-mcp-app-tools.ts","../../../src/hooks/mcps/use-mcp-server-tool.ts","../../../src/hooks/mcps/use-mcp-server-tools.ts","../../../src/hooks/mcps/use-mcp-server-tools-by-id.ts","../../../src/hooks/mcps/use-mcp-servers.ts","../../../src/hooks/mcps/use-try-connect-mcp.ts"],"sourcesContent":["import type { MastraClient } from '@mastra/client-js';\nimport { useQuery } from '@tanstack/react-query';\nimport type { UseQueryResult } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraQueryOptions } from '../shared/query-options';\n\n/** The app HTML, or `null` when the resource has no text. */\ntype McpAppHtml = string | null;\n\nexport type ReadMcpServerResourceResponse = Awaited<ReturnType<MastraClient['readMcpServerResource']>>;\n\n/**\n * The HTML of an MCP tool's `ui://` app resource (from the tool's `_meta`), for rendering the tool's own UI.\n * Skips the fetch when the tool has no app resource.\n */\nexport const useMcpAppHtml = <TData = McpAppHtml>({\n  serverId,\n  appResourceUri,\n  queryOptions,\n}: {\n  serverId: string;\n  appResourceUri?: string;\n  queryOptions?: MastraQueryOptions<McpAppHtml, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['mcp-app-resource', serverId, appResourceUri],\n    queryFn: async () => {\n      if (!appResourceUri) return null;\n      const response = await client.readMcpServerResource(serverId, appResourceUri);\n      return response.contents[0]?.text ?? null;\n    },\n    enabled: appResourceUri !== undefined,\n    ...queryOptions,\n  });\n};\n","import 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\nexport interface McpAppToolInfo {\n  serverId: string;\n  toolId: string;\n  toolName: string;\n  resourceUri: string;\n}\n\n/**\n * Builds a map of tool names → MCP App info for tools across all registered\n * MCP servers that have `_meta.ui.resourceUri` (i.e., tools with interactive\n * MCP App UIs).\n *\n * For each app tool, entries are created under both the original tool name\n * and the namespaced name (`${serverId}_${toolName}`) so that agent tools\n * sourced via `MCPClient.listTools()` (which namespaces) can be resolved.\n */\nexport function useMcpAppTools<TData = Record<string, McpAppToolInfo>>({\n  queryOptions,\n}: { queryOptions?: MastraQueryOptions<Record<string, McpAppToolInfo>, TData> } = {}): UseQueryResult<TData, Error> {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['mcp-app-tools'],\n    queryFn: async () => {\n      const map: Record<string, McpAppToolInfo> = {};\n\n      const { servers } = await client.getMcpServers();\n      if (!servers?.length) return map;\n\n      const results = await Promise.allSettled(\n        servers.map(async server => {\n          const { tools } = await client.getMcpServerTools(server.id);\n          return { serverId: server.id, tools };\n        }),\n      );\n\n      for (const result of results) {\n        if (result.status !== 'fulfilled') continue;\n        const { serverId, tools } = result.value;\n        for (const tool of tools) {\n          const meta = tool._meta as { ui?: { resourceUri?: string } } | undefined;\n          const resourceUri = meta?.ui?.resourceUri;\n          if (resourceUri) {\n            const toolId = tool.name;\n            const info: McpAppToolInfo = { serverId, toolId, toolName: tool.name, resourceUri };\n            map[toolId] = info;\n            if (tool.name !== toolId) {\n              map[tool.name] = info;\n            }\n            // Also map the namespaced name used by MCPClient.listTools()\n            map[`${serverId}_${tool.name}`] = info;\n          }\n        }\n      }\n\n      return map;\n    },\n    staleTime: 60_000,\n    refetchOnWindowFocus: false,\n    ...queryOptions,\n  });\n}\n","import type { MastraClient } from '@mastra/client-js';\nimport type { RequestContext } from '@mastra/core/request-context';\nimport { useMutation, useQuery } from '@tanstack/react-query';\nimport type { UseMutationResult, UseQueryResult } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraMutationOptions, MastraQueryOptions } from '../shared/query-options';\n\ntype MCPServerToolDetails = Awaited<ReturnType<ReturnType<MastraClient['getMcpServerTool']>['details']>>;\n\n/**\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useMCPServerTool = <TData = MCPServerToolDetails>({\n  serverId,\n  toolId,\n  requestContext,\n  queryOptions,\n}: {\n  serverId: string;\n  toolId: string;\n  requestContext?: Record<string, any>;\n  queryOptions?: MastraQueryOptions<MCPServerToolDetails, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['mcp-server-tool', serverId, toolId],\n    queryFn: () => {\n      const instance = client.getMcpServerTool(serverId, toolId);\n      return instance.details(requestContext);\n    },\n    ...queryOptions,\n  });\n};\n\nexport type ExecuteMCPToolResponse = Awaited<ReturnType<ReturnType<MastraClient['getMcpServerTool']>['execute']>>;\ntype ExecuteMCPToolVariables = { data: any; requestContext?: Record<string, any> };\n\nexport const useExecuteMCPTool = ({\n  serverId,\n  toolId,\n  queryOptions,\n}: {\n  serverId: string;\n  toolId: string;\n  queryOptions?: MastraMutationOptions<ExecuteMCPToolResponse, ExecuteMCPToolVariables>;\n}): UseMutationResult<ExecuteMCPToolResponse, Error, ExecuteMCPToolVariables> => {\n  const client = useMastraClient();\n\n  return useMutation({\n    mutationFn: ({ data, requestContext }: ExecuteMCPToolVariables) => {\n      const instance = client.getMcpServerTool(serverId, toolId);\n      return instance.execute({ data, requestContext: requestContext as RequestContext });\n    },\n    ...queryOptions,\n  });\n};\n","import type { McpToolInfo as SdkMcpToolInfo } from '@mastra/client-js';\nimport type { ServerInfo } from '@mastra/core/mcp';\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\nexport const useMCPServerTools = <TData = Record<string, SdkMcpToolInfo>>({\n  selectedServer,\n  queryOptions,\n}: {\n  selectedServer: ServerInfo;\n  queryOptions?: MastraQueryOptions<Record<string, SdkMcpToolInfo>, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['mcpserver-tools', selectedServer?.id],\n    queryFn: async () => {\n      const response = await client.getMcpServerTools(selectedServer?.id);\n      const fetchedToolsArray: SdkMcpToolInfo[] = response.tools;\n      const transformedTools: Record<string, SdkMcpToolInfo> = {};\n      fetchedToolsArray.forEach((sdkToolInfo: SdkMcpToolInfo) => {\n        transformedTools[sdkToolInfo.name] = sdkToolInfo;\n      });\n\n      return transformedTools;\n    },\n    retry: false,\n    refetchOnWindowFocus: false,\n    ...queryOptions,\n  });\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 MCPServerToolsById = Record<string, Awaited<ReturnType<MastraClient['getMcpServerTools']>>['tools'][number]>;\n\n/**\n * Does not guard on empty ids; pass `queryOptions: { enabled }` to skip the fetch.\n */\nexport const useMCPServerToolsById = <TData = MCPServerToolsById>({\n  serverId,\n  queryOptions,\n}: {\n  serverId: string | null;\n  queryOptions?: MastraQueryOptions<MCPServerToolsById, TData>;\n}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['mcpserver-tools', serverId],\n    queryFn: async (): Promise<MCPServerToolsById> => {\n      const response = await client.getMcpServerTools(serverId!);\n      return Object.fromEntries(response.tools.map(tool => [tool.name, tool]));\n    },\n    retry: false,\n    refetchOnWindowFocus: false,\n    ...queryOptions,\n  });\n};\n","import type { McpServerListResponse } 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\nexport const useMCPServers = <TData = McpServerListResponse['servers']>({\n  requestContext,\n  queryOptions,\n}: {\n  requestContext?: Record<string, any>;\n  queryOptions?: MastraQueryOptions<McpServerListResponse['servers'], TData>;\n} = {}): UseQueryResult<TData, Error> => {\n  const client = useMastraClient();\n\n  return useQuery({\n    queryKey: ['mcp-servers'],\n    queryFn: async () => {\n      const mcpServers: McpServerListResponse['servers'] = (await client.getMcpServers(requestContext)).servers;\n      return mcpServers;\n    },\n    ...queryOptions,\n  });\n};\n","import { useMutation } from '@tanstack/react-query';\nimport { useMastraClient } from '../../mastra-client-context';\nimport type { MastraMutationOptions } from '../shared/query-options';\n\ninterface McpTool {\n  name: string;\n  description?: string;\n  inputSchema?: Record<string, unknown>;\n}\n\nexport interface TryConnectResult {\n  tools: McpTool[];\n}\n\nasync function parseResponse(response: Response): Promise<unknown> {\n  const contentType = response.headers.get('content-type') ?? '';\n\n  if (contentType.includes('text/event-stream')) {\n    const text = await response.text();\n    const lines = text.split('\\n');\n\n    for (const line of lines) {\n      if (line.startsWith('data:')) {\n        const data = line.slice('data:'.length).trim();\n        if (data) {\n          return JSON.parse(data);\n        }\n      }\n    }\n\n    throw new Error('No data found in SSE response');\n  }\n\n  return response.json();\n}\n\nasync function connectAndListTools(url: string, clientHeaders?: Record<string, string>): Promise<TryConnectResult> {\n  const baseHeaders: Record<string, string> = {\n    ...clientHeaders,\n    'Content-Type': 'application/json',\n    Accept: 'application/json, text/event-stream',\n  };\n\n  // Step 1: Initialize\n  const initResponse = await fetch(url, {\n    method: 'POST',\n    headers: baseHeaders,\n    credentials: 'include',\n    body: JSON.stringify({\n      jsonrpc: '2.0',\n      id: 1,\n      method: 'initialize',\n      params: {\n        protocolVersion: '2025-03-26',\n        capabilities: {},\n        clientInfo: { name: 'mastra-playground', version: '1.0.0' },\n      },\n    }),\n  });\n\n  if (!initResponse.ok) {\n    throw new Error(`Initialize failed: ${initResponse.status} ${initResponse.statusText}`);\n  }\n\n  const sessionId = initResponse.headers.get('Mcp-Session-Id');\n  await parseResponse(initResponse);\n\n  const sessionHeaders: Record<string, string> = {\n    ...baseHeaders,\n  };\n  if (sessionId) {\n    sessionHeaders['Mcp-Session-Id'] = sessionId;\n  }\n\n  // Step 2: Send initialized notification\n  await fetch(url, {\n    method: 'POST',\n    headers: sessionHeaders,\n    credentials: 'include',\n    body: JSON.stringify({\n      jsonrpc: '2.0',\n      method: 'notifications/initialized',\n    }),\n  });\n\n  // Step 3: List tools\n  const toolsResponse = await fetch(url, {\n    method: 'POST',\n    headers: sessionHeaders,\n    credentials: 'include',\n    body: JSON.stringify({\n      jsonrpc: '2.0',\n      id: 2,\n      method: 'tools/list',\n    }),\n  });\n\n  if (!toolsResponse.ok) {\n    throw new Error(`tools/list failed: ${toolsResponse.status} ${toolsResponse.statusText}`);\n  }\n\n  const toolsResult = (await parseResponse(toolsResponse)) as { result?: { tools?: McpTool[] } };\n\n  return {\n    tools: toolsResult.result?.tools ?? [],\n  };\n}\n\nexport const useTryConnectMcp = ({\n  queryOptions,\n}: { queryOptions?: MastraMutationOptions<TryConnectResult, string> } = {}) => {\n  const client = useMastraClient();\n  const clientHeaders = (client.options?.headers as Record<string, string>) ?? {};\n\n  return useMutation({\n    mutationFn: (url: string) => connectAndListTools(url, clientHeaders),\n    ...queryOptions,\n  });\n};\n\nexport type TryConnectMcpMutation = ReturnType<typeof useTryConnectMcp>;\n"],"mappings":";;;;;;;;AAeA,MAAa,iBAAqC,EAChD,UACA,gBACA,mBAKkC;CAClC,MAAM,SAASA,8BAAAA,gBAAgB;CAE/B,QAAA,GAAOC,sBAAAA,SAAAA,CAAS;EACd,UAAU;GAAC;GAAoB;GAAU;EAAc;EACvD,SAAS,YAAY;GACnB,IAAI,CAAC,gBAAgB,OAAO;GAE5B,QAAO,MADgB,OAAO,sBAAsB,UAAU,cAAc,EAAA,CAC5D,SAAS,EAAE,EAAE,QAAQ;EACvC;EACA,SAAS,mBAAmB,KAAA;EAC5B,GAAG;CACL,CAAC;AACH;;;;;;;;;;;;ACfA,SAAgB,eAAuD,EACrE,iBACgF,CAAC,GAAiC;CAClH,MAAM,SAASC,8BAAAA,gBAAgB;CAE/B,QAAA,GAAOC,sBAAAA,SAAAA,CAAS;EACd,UAAU,CAAC,eAAe;EAC1B,SAAS,YAAY;GACnB,MAAM,MAAsC,CAAC;GAE7C,MAAM,EAAE,YAAY,MAAM,OAAO,cAAc;GAC/C,IAAI,CAAC,SAAS,QAAQ,OAAO;GAE7B,MAAM,UAAU,MAAM,QAAQ,WAC5B,QAAQ,IAAI,OAAM,WAAU;IAC1B,MAAM,EAAE,UAAU,MAAM,OAAO,kBAAkB,OAAO,EAAE;IAC1D,OAAO;KAAE,UAAU,OAAO;KAAI;IAAM;GACtC,CAAC,CACH;GAEA,KAAK,MAAM,UAAU,SAAS;IAC5B,IAAI,OAAO,WAAW,aAAa;IACnC,MAAM,EAAE,UAAU,UAAU,OAAO;IACnC,KAAK,MAAM,QAAQ,OAAO;KAExB,MAAM,cADO,KAAK,OACQ,IAAI;KAC9B,IAAI,aAAa;MACf,MAAM,SAAS,KAAK;MACpB,MAAM,OAAuB;OAAE;OAAU;OAAQ,UAAU,KAAK;OAAM;MAAY;MAClF,IAAI,UAAU;MACd,IAAI,KAAK,SAAS,QAChB,IAAI,KAAK,QAAQ;MAGnB,IAAI,GAAG,SAAS,GAAG,KAAK,UAAU;KACpC;IACF;GACF;GAEA,OAAO;EACT;EACA,WAAW;EACX,sBAAsB;EACtB,GAAG;CACL,CAAC;AACH;;;;;;ACtDA,MAAa,oBAAkD,EAC7D,UACA,QACA,gBACA,mBAMkC;CAClC,MAAM,SAASC,8BAAAA,gBAAgB;CAE/B,QAAA,GAAOC,sBAAAA,SAAAA,CAAS;EACd,UAAU;GAAC;GAAmB;GAAU;EAAM;EAC9C,eAAe;GAEb,OADiB,OAAO,iBAAiB,UAAU,MACrC,CAAC,CAAC,QAAQ,cAAc;EACxC;EACA,GAAG;CACL,CAAC;AACH;AAKA,MAAa,qBAAqB,EAChC,UACA,QACA,mBAK+E;CAC/E,MAAM,SAASD,8BAAAA,gBAAgB;CAE/B,QAAA,GAAOE,sBAAAA,YAAAA,CAAY;EACjB,aAAa,EAAE,MAAM,qBAA8C;GAEjE,OADiB,OAAO,iBAAiB,UAAU,MACrC,CAAC,CAAC,QAAQ;IAAE;IAAsB;GAAiC,CAAC;EACpF;EACA,GAAG;CACL,CAAC;AACH;;;ACjDA,MAAa,qBAA6D,EACxE,gBACA,mBAIkC;CAClC,MAAM,SAASC,8BAAAA,gBAAgB;CAE/B,QAAA,GAAOC,sBAAAA,SAAAA,CAAS;EACd,UAAU,CAAC,mBAAmB,gBAAgB,EAAE;EAChD,SAAS,YAAY;GAEnB,MAAM,qBAAsC,MADrB,OAAO,kBAAkB,gBAAgB,EAAE,EAAA,CACb;GACrD,MAAM,mBAAmD,CAAC;GAC1D,kBAAkB,SAAS,gBAAgC;IACzD,iBAAiB,YAAY,QAAQ;GACvC,CAAC;GAED,OAAO;EACT;EACA,OAAO;EACP,sBAAsB;EACtB,GAAG;CACL,CAAC;AACH;;;;;;ACrBA,MAAa,yBAAqD,EAChE,UACA,mBAIkC;CAClC,MAAM,SAASC,8BAAAA,gBAAgB;CAE/B,QAAA,GAAOC,sBAAAA,SAAAA,CAAS;EACd,UAAU,CAAC,mBAAmB,QAAQ;EACtC,SAAS,YAAyC;GAChD,MAAM,WAAW,MAAM,OAAO,kBAAkB,QAAS;GACzD,OAAO,OAAO,YAAY,SAAS,MAAM,KAAI,SAAQ,CAAC,KAAK,MAAM,IAAI,CAAC,CAAC;EACzE;EACA,OAAO;EACP,sBAAsB;EACtB,GAAG;CACL,CAAC;AACH;;;ACxBA,MAAa,iBAA2D,EACtE,gBACA,iBAIE,CAAC,MAAoC;CACvC,MAAM,SAASC,8BAAAA,gBAAgB;CAE/B,QAAA,GAAOC,sBAAAA,SAAAA,CAAS;EACd,UAAU,CAAC,aAAa;EACxB,SAAS,YAAY;GAEnB,QADsD,MAAM,OAAO,cAAc,cAAc,EAAA,CAAG;EAEpG;EACA,GAAG;CACL,CAAC;AACH;;;ACTA,eAAe,cAAc,UAAsC;CAGjE,KAFoB,SAAS,QAAQ,IAAI,cAAc,KAAK,GAAA,CAE5C,SAAS,mBAAmB,GAAG;EAE7C,MAAM,SAAQ,MADK,SAAS,KAAK,EAAA,CACd,MAAM,IAAI;EAE7B,KAAK,MAAM,QAAQ,OACjB,IAAI,KAAK,WAAW,OAAO,GAAG;GAC5B,MAAM,OAAO,KAAK,MAAM,CAAc,CAAC,CAAC,KAAK;GAC7C,IAAI,MACF,OAAO,KAAK,MAAM,IAAI;EAE1B;EAGF,MAAM,IAAI,MAAM,+BAA+B;CACjD;CAEA,OAAO,SAAS,KAAK;AACvB;AAEA,eAAe,oBAAoB,KAAa,eAAmE;CACjH,MAAM,cAAsC;EAC1C,GAAG;EACH,gBAAgB;EAChB,QAAQ;CACV;CAGA,MAAM,eAAe,MAAM,MAAM,KAAK;EACpC,QAAQ;EACR,SAAS;EACT,aAAa;EACb,MAAM,KAAK,UAAU;GACnB,SAAS;GACT,IAAI;GACJ,QAAQ;GACR,QAAQ;IACN,iBAAiB;IACjB,cAAc,CAAC;IACf,YAAY;KAAE,MAAM;KAAqB,SAAS;IAAQ;GAC5D;EACF,CAAC;CACH,CAAC;CAED,IAAI,CAAC,aAAa,IAChB,MAAM,IAAI,MAAM,sBAAsB,aAAa,OAAO,GAAG,aAAa,YAAY;CAGxF,MAAM,YAAY,aAAa,QAAQ,IAAI,gBAAgB;CAC3D,MAAM,cAAc,YAAY;CAEhC,MAAM,iBAAyC,EAC7C,GAAG,YACL;CACA,IAAI,WACF,eAAe,oBAAoB;CAIrC,MAAM,MAAM,KAAK;EACf,QAAQ;EACR,SAAS;EACT,aAAa;EACb,MAAM,KAAK,UAAU;GACnB,SAAS;GACT,QAAQ;EACV,CAAC;CACH,CAAC;CAGD,MAAM,gBAAgB,MAAM,MAAM,KAAK;EACrC,QAAQ;EACR,SAAS;EACT,aAAa;EACb,MAAM,KAAK,UAAU;GACnB,SAAS;GACT,IAAI;GACJ,QAAQ;EACV,CAAC;CACH,CAAC;CAED,IAAI,CAAC,cAAc,IACjB,MAAM,IAAI,MAAM,sBAAsB,cAAc,OAAO,GAAG,cAAc,YAAY;CAK1F,OAAO,EACL,QAAO,MAHkB,cAAc,aAAa,EAAA,CAGjC,QAAQ,SAAS,CAAC,EACvC;AACF;AAEA,MAAa,oBAAoB,EAC/B,iBACsE,CAAC,MAAM;CAE7E,MAAM,gBADSC,8BAAAA,gBACa,CAAC,CAAC,SAAS,WAAsC,CAAC;CAE9E,QAAA,GAAOC,sBAAAA,YAAAA,CAAY;EACjB,aAAa,QAAgB,oBAAoB,KAAK,aAAa;EACnE,GAAG;CACL,CAAC;AACH"}