@mastra/playground-ui 59.0.0 → 60.0.0-alpha.1

package/dist/domains/datasets/hooks/use-dataset-mutations.cjs.js

@@ -0,0 +1,157 @@
+'use strict';
+
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+
+const react = require('@mastra/react');
+const reactQuery = require('@tanstack/react-query');
+
+const useDatasetMutations = () => {
+  const client = react.useMastraClient();
+  const queryClient = reactQuery.useQueryClient();
+  const createDataset = reactQuery.useMutation({
+    mutationFn: (params) => client.createDataset(params),
+    onSuccess: () => {
+      void queryClient.invalidateQueries({ queryKey: ["datasets"] });
+    }
+  });
+  const updateDataset = reactQuery.useMutation({
+    mutationFn: (params) => client.updateDataset(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["datasets"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset", variables.datasetId] });
+    }
+  });
+  const deleteDataset = reactQuery.useMutation({
+    mutationFn: (datasetId) => client.deleteDataset(datasetId),
+    onSuccess: () => {
+      void queryClient.invalidateQueries({ queryKey: ["datasets"] });
+    }
+  });
+  const addItem = reactQuery.useMutation({
+    mutationFn: (params) => client.addDatasetItem(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-versions", variables.datasetId] });
+    }
+  });
+  const updateItem = reactQuery.useMutation({
+    mutationFn: (params) => client.updateDatasetItem(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-item", variables.datasetId, variables.itemId] });
+      void queryClient.invalidateQueries({
+        queryKey: ["dataset-item-versions", variables.datasetId, variables.itemId]
+      });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-versions", variables.datasetId] });
+    }
+  });
+  const deleteItem = reactQuery.useMutation({
+    mutationFn: ({ datasetId, itemId }) => client.deleteDatasetItem(datasetId, itemId),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-versions", variables.datasetId] });
+    }
+  });
+  const purgeItem = reactQuery.useMutation({
+    mutationFn: ({ datasetId, itemId }) => client.purgeDatasetItem(datasetId, itemId),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-item", variables.datasetId, variables.itemId] });
+      void queryClient.invalidateQueries({
+        queryKey: ["dataset-item-versions", variables.datasetId, variables.itemId]
+      });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiment-results"] });
+      void queryClient.invalidateQueries({ queryKey: ["experiment-results"] });
+      void queryClient.invalidateQueries({ queryKey: ["review-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-review-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-completed-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["experiment-review-summary"] });
+    }
+  });
+  const batchInsertItems = reactQuery.useMutation({
+    mutationFn: (params) => client.batchInsertDatasetItems(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-versions", variables.datasetId] });
+    }
+  });
+  const batchDeleteItems = reactQuery.useMutation({
+    mutationFn: (params) => client.batchDeleteDatasetItems(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-versions", variables.datasetId] });
+    }
+  });
+  const deleteItems = reactQuery.useMutation({
+    mutationFn: async ({ datasetId, itemIds }) => {
+      return client.batchDeleteDatasetItems({ datasetId, itemIds });
+    },
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-versions", variables.datasetId] });
+    }
+  });
+  const triggerExperiment = reactQuery.useMutation({
+    mutationFn: (params) => client.triggerDatasetExperiment(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiments", variables.datasetId] });
+    }
+  });
+  const deleteExperiment = reactQuery.useMutation({
+    mutationFn: (experimentId) => client.deleteExperiment(experimentId),
+    onSuccess: () => {
+      void queryClient.invalidateQueries({ queryKey: ["experiments"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiments"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiment"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiment-results"] });
+      void queryClient.invalidateQueries({ queryKey: ["review-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["completed-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["experiment-review-summary"] });
+    }
+  });
+  const updateExperiment = reactQuery.useMutation({
+    mutationFn: (params) => client.updateDatasetExperiment(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["experiments"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiments", variables.datasetId] });
+      void queryClient.invalidateQueries({
+        queryKey: ["dataset-experiment", variables.datasetId, variables.experimentId]
+      });
+    }
+  });
+  const updateExperimentResult = reactQuery.useMutation({
+    mutationFn: (params) => client.updateDatasetExperimentResult(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["experiment-results", variables.experimentId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiment-results"] });
+      void queryClient.invalidateQueries({ queryKey: ["review-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-review-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-completed-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["experiment-review-summary"] });
+    }
+  });
+  return {
+    createDataset,
+    updateDataset,
+    deleteDataset,
+    addItem,
+    updateItem,
+    deleteItem,
+    purgeItem,
+    deleteItems,
+    batchInsertItems,
+    batchDeleteItems,
+    triggerExperiment,
+    deleteExperiment,
+    updateExperiment,
+    updateExperimentResult
+  };
+};
+
+exports.useDatasetMutations = useDatasetMutations;
+//# sourceMappingURL=use-dataset-mutations.cjs.js.map

package/dist/domains/datasets/hooks/use-dataset-mutations.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-dataset-mutations.cjs.js","sources":["../../../../src/domains/datasets/hooks/use-dataset-mutations.ts"],"sourcesContent":["import type {\n  CreateDatasetParams,\n  UpdateDatasetParams,\n  AddDatasetItemParams,\n  UpdateDatasetItemParams,\n  TriggerDatasetExperimentParams,\n  UpdateDatasetExperimentParams,\n  UpdateExperimentResultParams,\n  BatchInsertDatasetItemsParams,\n  BatchDeleteDatasetItemsParams,\n} from '@mastra/client-js';\nimport { useMastraClient } from '@mastra/react';\nimport { useMutation, useQueryClient } from '@tanstack/react-query';\n\ntype DatasetItemMutationVariables = {\n  datasetId: string;\n  itemId: string;\n};\n\n/**\n * Hook providing mutation functions for datasets, items, and runs\n * All mutations invalidate relevant query caches on success\n */\nexport const useDatasetMutations = () => {\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  });\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  });\n\n  const deleteDataset = useMutation({\n    mutationFn: (datasetId: string) => client.deleteDataset(datasetId),\n    onSuccess: () => {\n      void queryClient.invalidateQueries({ queryKey: ['datasets'] });\n    },\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  });\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  });\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  });\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  });\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  });\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  });\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  });\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  });\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: ['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  });\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: ['dataset-experiments', variables.datasetId] });\n      void queryClient.invalidateQueries({\n        queryKey: ['dataset-experiment', variables.datasetId, variables.experimentId],\n      });\n    },\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  });\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"],"names":["useMastraClient","useQueryClient","useMutation"],"mappings":";;;;;;;AAuBO,MAAM,sBAAsB,MAAM;AACvC,EAAA,MAAM,SAASA,qBAAA,EAAgB;AAC/B,EAAA,MAAM,cAAcC,yBAAA,EAAe;AAEnC,EAAA,MAAM,gBAAgBC,sBAAA,CAAY;AAAA,IAChC,UAAA,EAAY,CAAC,MAAA,KAAgC,MAAA,CAAO,cAAc,MAAM,CAAA;AAAA,IACxE,WAAW,MAAM;AACf,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,UAAU,GAAG,CAAA;AAAA,IAC/D;AAAA,GACD,CAAA;AAED,EAAA,MAAM,gBAAgBA,sBAAA,CAAY;AAAA,IAChC,UAAA,EAAY,CAAC,MAAA,KAAgC,MAAA,CAAO,cAAc,MAAM,CAAA;AAAA,IACxE,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,UAAU,GAAG,CAAA;AAC7D,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,SAAA,EAAW,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IACnF;AAAA,GACD,CAAA;AAED,EAAA,MAAM,gBAAgBA,sBAAA,CAAY;AAAA,IAChC,UAAA,EAAY,CAAC,SAAA,KAAsB,MAAA,CAAO,cAAc,SAAS,CAAA;AAAA,IACjE,WAAW,MAAM;AACf,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,UAAU,GAAG,CAAA;AAAA,IAC/D;AAAA,GACD,CAAA;AAED,EAAA,MAAM,UAAUA,sBAAA,CAAY;AAAA,IAC1B,UAAA,EAAY,CAAC,MAAA,KAAiC,MAAA,CAAO,eAAe,MAAM,CAAA;AAAA,IAC1E,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,SAAA,EAAW,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACjF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,kBAAA,EAAoB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC5F;AAAA,GACD,CAAA;AAED,EAAA,MAAM,aAAaA,sBAAA,CAAY;AAAA,IAC7B,UAAA,EAAY,CAAC,MAAA,KAAoC,MAAA,CAAO,kBAAkB,MAAM,CAAA;AAAA,IAChF,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,iBAAA,CAAkB,EAAE,QAAA,EAAU,CAAC,cAAA,EAAgB,SAAA,CAAU,SAAA,EAAW,SAAA,CAAU,MAAM,CAAA,EAAG,CAAA;AACxG,MAAA,KAAK,YAAY,iBAAA,CAAkB;AAAA,QACjC,UAAU,CAAC,uBAAA,EAAyB,SAAA,CAAU,SAAA,EAAW,UAAU,MAAM;AAAA,OAC1E,CAAA;AACD,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,kBAAA,EAAoB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC5F;AAAA,GACD,CAAA;AAED,EAAA,MAAM,aAAaA,sBAAA,CAAY;AAAA,IAC7B,UAAA,EAAY,CAAC,EAAE,SAAA,EAAW,QAAO,KAAoC,MAAA,CAAO,iBAAA,CAAkB,SAAA,EAAW,MAAM,CAAA;AAAA,IAC/G,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,SAAA,EAAW,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACjF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,kBAAA,EAAoB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC5F;AAAA,GACD,CAAA;AAED,EAAA,MAAM,YAAYA,sBAAA,CAAY;AAAA,IAC5B,UAAA,EAAY,CAAC,EAAE,SAAA,EAAW,QAAO,KAAoC,MAAA,CAAO,gBAAA,CAAiB,SAAA,EAAW,MAAM,CAAA;AAAA,IAC9G,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,iBAAA,CAAkB,EAAE,QAAA,EAAU,CAAC,cAAA,EAAgB,SAAA,CAAU,SAAA,EAAW,SAAA,CAAU,MAAM,CAAA,EAAG,CAAA;AACxG,MAAA,KAAK,YAAY,iBAAA,CAAkB;AAAA,QACjC,UAAU,CAAC,uBAAA,EAAyB,SAAA,CAAU,SAAA,EAAW,UAAU,MAAM;AAAA,OAC1E,CAAA;AACD,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,4BAA4B,GAAG,CAAA;AAC/E,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,oBAAoB,GAAG,CAAA;AACvE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,cAAc,GAAG,CAAA;AACjE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,sBAAsB,GAAG,CAAA;AACzE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,yBAAyB,GAAG,CAAA;AAC5E,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,2BAA2B,GAAG,CAAA;AAAA,IAChF;AAAA,GACD,CAAA;AAGD,EAAA,MAAM,mBAAmBA,sBAAA,CAAY;AAAA,IACnC,UAAA,EAAY,CAAC,MAAA,KAA0C,MAAA,CAAO,wBAAwB,MAAM,CAAA;AAAA,IAC5F,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,SAAA,EAAW,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACjF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,kBAAA,EAAoB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC5F;AAAA,GACD,CAAA;AAGD,EAAA,MAAM,mBAAmBA,sBAAA,CAAY;AAAA,IACnC,UAAA,EAAY,CAAC,MAAA,KAA0C,MAAA,CAAO,wBAAwB,MAAM,CAAA;AAAA,IAC5F,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,SAAA,EAAW,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACjF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,kBAAA,EAAoB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC5F;AAAA,GACD,CAAA;AAGD,EAAA,MAAM,cAAcA,sBAAA,CAAY;AAAA,IAC9B,UAAA,EAAY,OAAO,EAAE,SAAA,EAAW,SAAQ,KAAgD;AACtF,MAAA,OAAO,MAAA,CAAO,uBAAA,CAAwB,EAAE,SAAA,EAAW,SAAS,CAAA;AAAA,IAC9D,CAAA;AAAA,IACA,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,SAAA,EAAW,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACjF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,kBAAA,EAAoB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC5F;AAAA,GACD,CAAA;AAED,EAAA,MAAM,oBAAoBA,sBAAA,CAAY;AAAA,IACpC,UAAA,EAAY,CAAC,MAAA,KAA2C,MAAA,CAAO,yBAAyB,MAAM,CAAA;AAAA,IAC9F,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,qBAAA,EAAuB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC/F;AAAA,GACD,CAAA;AAED,EAAA,MAAM,mBAAmBA,sBAAA,CAAY;AAAA,IACnC,UAAA,EAAY,CAAC,YAAA,KAAyB,MAAA,CAAO,iBAAiB,YAAY,CAAA;AAAA,IAC1E,WAAW,MAAM;AACf,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,aAAa,GAAG,CAAA;AAChE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,qBAAqB,GAAG,CAAA;AACxE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,oBAAoB,GAAG,CAAA;AACvE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,4BAA4B,GAAG,CAAA;AAC/E,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,cAAc,GAAG,CAAA;AACjE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,iBAAiB,GAAG,CAAA;AACpE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,2BAA2B,GAAG,CAAA;AAAA,IAChF;AAAA,GACD,CAAA;AAED,EAAA,MAAM,mBAAmBA,sBAAA,CAAY;AAAA,IACnC,UAAA,EAAY,CAAC,MAAA,KAA0C,MAAA,CAAO,wBAAwB,MAAM,CAAA;AAAA,IAC5F,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,aAAa,GAAG,CAAA;AAChE,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,qBAAA,EAAuB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAC7F,MAAA,KAAK,YAAY,iBAAA,CAAkB;AAAA,QACjC,UAAU,CAAC,oBAAA,EAAsB,SAAA,CAAU,SAAA,EAAW,UAAU,YAAY;AAAA,OAC7E,CAAA;AAAA,IACH;AAAA,GACD,CAAA;AAED,EAAA,MAAM,yBAAyBA,sBAAA,CAAY;AAAA,IACzC,UAAA,EAAY,CAAC,MAAA,KAAyC,MAAA,CAAO,8BAA8B,MAAM,CAAA;AAAA,IACjG,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,oBAAA,EAAsB,SAAA,CAAU,YAAY,CAAA,EAAG,CAAA;AAC/F,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,4BAA4B,GAAG,CAAA;AAC/E,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,cAAc,GAAG,CAAA;AACjE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,sBAAsB,GAAG,CAAA;AACzE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,yBAAyB,GAAG,CAAA;AAC5E,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,2BAA2B,GAAG,CAAA;AAAA,IAChF;AAAA,GACD,CAAA;AAED,EAAA,OAAO;AAAA,IACL,aAAA;AAAA,IACA,aAAA;AAAA,IACA,aAAA;AAAA,IACA,OAAA;AAAA,IACA,UAAA;AAAA,IACA,UAAA;AAAA,IACA,SAAA;AAAA,IACA,WAAA;AAAA,IACA,gBAAA;AAAA,IACA,gBAAA;AAAA,IACA,iBAAA;AAAA,IACA,gBAAA;AAAA,IACA,gBAAA;AAAA,IACA;AAAA,GACF;AACF;;;;"}

package/dist/domains/datasets/hooks/use-dataset-mutations.d.ts

@@ -0,0 +1,2 @@
+export * from '../../../src/domains/datasets/hooks/use-dataset-mutations'
+export {}

package/dist/domains/datasets/hooks/use-dataset-mutations.es.js

@@ -0,0 +1,153 @@
+import { useMastraClient } from '@mastra/react';
+import { useQueryClient, useMutation } from '@tanstack/react-query';
+
+const useDatasetMutations = () => {
+  const client = useMastraClient();
+  const queryClient = useQueryClient();
+  const createDataset = useMutation({
+    mutationFn: (params) => client.createDataset(params),
+    onSuccess: () => {
+      void queryClient.invalidateQueries({ queryKey: ["datasets"] });
+    }
+  });
+  const updateDataset = useMutation({
+    mutationFn: (params) => client.updateDataset(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["datasets"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset", variables.datasetId] });
+    }
+  });
+  const deleteDataset = useMutation({
+    mutationFn: (datasetId) => client.deleteDataset(datasetId),
+    onSuccess: () => {
+      void queryClient.invalidateQueries({ queryKey: ["datasets"] });
+    }
+  });
+  const addItem = useMutation({
+    mutationFn: (params) => client.addDatasetItem(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-versions", variables.datasetId] });
+    }
+  });
+  const updateItem = useMutation({
+    mutationFn: (params) => client.updateDatasetItem(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-item", variables.datasetId, variables.itemId] });
+      void queryClient.invalidateQueries({
+        queryKey: ["dataset-item-versions", variables.datasetId, variables.itemId]
+      });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-versions", variables.datasetId] });
+    }
+  });
+  const deleteItem = useMutation({
+    mutationFn: ({ datasetId, itemId }) => client.deleteDatasetItem(datasetId, itemId),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-versions", variables.datasetId] });
+    }
+  });
+  const purgeItem = useMutation({
+    mutationFn: ({ datasetId, itemId }) => client.purgeDatasetItem(datasetId, itemId),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-item", variables.datasetId, variables.itemId] });
+      void queryClient.invalidateQueries({
+        queryKey: ["dataset-item-versions", variables.datasetId, variables.itemId]
+      });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiment-results"] });
+      void queryClient.invalidateQueries({ queryKey: ["experiment-results"] });
+      void queryClient.invalidateQueries({ queryKey: ["review-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-review-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-completed-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["experiment-review-summary"] });
+    }
+  });
+  const batchInsertItems = useMutation({
+    mutationFn: (params) => client.batchInsertDatasetItems(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-versions", variables.datasetId] });
+    }
+  });
+  const batchDeleteItems = useMutation({
+    mutationFn: (params) => client.batchDeleteDatasetItems(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-versions", variables.datasetId] });
+    }
+  });
+  const deleteItems = useMutation({
+    mutationFn: async ({ datasetId, itemIds }) => {
+      return client.batchDeleteDatasetItems({ datasetId, itemIds });
+    },
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-items", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset", variables.datasetId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-versions", variables.datasetId] });
+    }
+  });
+  const triggerExperiment = useMutation({
+    mutationFn: (params) => client.triggerDatasetExperiment(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiments", variables.datasetId] });
+    }
+  });
+  const deleteExperiment = useMutation({
+    mutationFn: (experimentId) => client.deleteExperiment(experimentId),
+    onSuccess: () => {
+      void queryClient.invalidateQueries({ queryKey: ["experiments"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiments"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiment"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiment-results"] });
+      void queryClient.invalidateQueries({ queryKey: ["review-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["completed-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["experiment-review-summary"] });
+    }
+  });
+  const updateExperiment = useMutation({
+    mutationFn: (params) => client.updateDatasetExperiment(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["experiments"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiments", variables.datasetId] });
+      void queryClient.invalidateQueries({
+        queryKey: ["dataset-experiment", variables.datasetId, variables.experimentId]
+      });
+    }
+  });
+  const updateExperimentResult = useMutation({
+    mutationFn: (params) => client.updateDatasetExperimentResult(params),
+    onSuccess: (_, variables) => {
+      void queryClient.invalidateQueries({ queryKey: ["experiment-results", variables.experimentId] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-experiment-results"] });
+      void queryClient.invalidateQueries({ queryKey: ["review-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-review-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["dataset-completed-items"] });
+      void queryClient.invalidateQueries({ queryKey: ["experiment-review-summary"] });
+    }
+  });
+  return {
+    createDataset,
+    updateDataset,
+    deleteDataset,
+    addItem,
+    updateItem,
+    deleteItem,
+    purgeItem,
+    deleteItems,
+    batchInsertItems,
+    batchDeleteItems,
+    triggerExperiment,
+    deleteExperiment,
+    updateExperiment,
+    updateExperimentResult
+  };
+};
+
+export { useDatasetMutations };
+//# sourceMappingURL=use-dataset-mutations.es.js.map

package/dist/domains/datasets/hooks/use-dataset-mutations.es.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-dataset-mutations.es.js","sources":["../../../../src/domains/datasets/hooks/use-dataset-mutations.ts"],"sourcesContent":["import type {\n  CreateDatasetParams,\n  UpdateDatasetParams,\n  AddDatasetItemParams,\n  UpdateDatasetItemParams,\n  TriggerDatasetExperimentParams,\n  UpdateDatasetExperimentParams,\n  UpdateExperimentResultParams,\n  BatchInsertDatasetItemsParams,\n  BatchDeleteDatasetItemsParams,\n} from '@mastra/client-js';\nimport { useMastraClient } from '@mastra/react';\nimport { useMutation, useQueryClient } from '@tanstack/react-query';\n\ntype DatasetItemMutationVariables = {\n  datasetId: string;\n  itemId: string;\n};\n\n/**\n * Hook providing mutation functions for datasets, items, and runs\n * All mutations invalidate relevant query caches on success\n */\nexport const useDatasetMutations = () => {\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  });\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  });\n\n  const deleteDataset = useMutation({\n    mutationFn: (datasetId: string) => client.deleteDataset(datasetId),\n    onSuccess: () => {\n      void queryClient.invalidateQueries({ queryKey: ['datasets'] });\n    },\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  });\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  });\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  });\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  });\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  });\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  });\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  });\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  });\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: ['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  });\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: ['dataset-experiments', variables.datasetId] });\n      void queryClient.invalidateQueries({\n        queryKey: ['dataset-experiment', variables.datasetId, variables.experimentId],\n      });\n    },\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  });\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"],"names":[],"mappings":";;;AAuBO,MAAM,sBAAsB,MAAM;AACvC,EAAA,MAAM,SAAS,eAAA,EAAgB;AAC/B,EAAA,MAAM,cAAc,cAAA,EAAe;AAEnC,EAAA,MAAM,gBAAgB,WAAA,CAAY;AAAA,IAChC,UAAA,EAAY,CAAC,MAAA,KAAgC,MAAA,CAAO,cAAc,MAAM,CAAA;AAAA,IACxE,WAAW,MAAM;AACf,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,UAAU,GAAG,CAAA;AAAA,IAC/D;AAAA,GACD,CAAA;AAED,EAAA,MAAM,gBAAgB,WAAA,CAAY;AAAA,IAChC,UAAA,EAAY,CAAC,MAAA,KAAgC,MAAA,CAAO,cAAc,MAAM,CAAA;AAAA,IACxE,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,UAAU,GAAG,CAAA;AAC7D,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,SAAA,EAAW,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IACnF;AAAA,GACD,CAAA;AAED,EAAA,MAAM,gBAAgB,WAAA,CAAY;AAAA,IAChC,UAAA,EAAY,CAAC,SAAA,KAAsB,MAAA,CAAO,cAAc,SAAS,CAAA;AAAA,IACjE,WAAW,MAAM;AACf,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,UAAU,GAAG,CAAA;AAAA,IAC/D;AAAA,GACD,CAAA;AAED,EAAA,MAAM,UAAU,WAAA,CAAY;AAAA,IAC1B,UAAA,EAAY,CAAC,MAAA,KAAiC,MAAA,CAAO,eAAe,MAAM,CAAA;AAAA,IAC1E,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,SAAA,EAAW,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACjF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,kBAAA,EAAoB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC5F;AAAA,GACD,CAAA;AAED,EAAA,MAAM,aAAa,WAAA,CAAY;AAAA,IAC7B,UAAA,EAAY,CAAC,MAAA,KAAoC,MAAA,CAAO,kBAAkB,MAAM,CAAA;AAAA,IAChF,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,iBAAA,CAAkB,EAAE,QAAA,EAAU,CAAC,cAAA,EAAgB,SAAA,CAAU,SAAA,EAAW,SAAA,CAAU,MAAM,CAAA,EAAG,CAAA;AACxG,MAAA,KAAK,YAAY,iBAAA,CAAkB;AAAA,QACjC,UAAU,CAAC,uBAAA,EAAyB,SAAA,CAAU,SAAA,EAAW,UAAU,MAAM;AAAA,OAC1E,CAAA;AACD,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,kBAAA,EAAoB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC5F;AAAA,GACD,CAAA;AAED,EAAA,MAAM,aAAa,WAAA,CAAY;AAAA,IAC7B,UAAA,EAAY,CAAC,EAAE,SAAA,EAAW,QAAO,KAAoC,MAAA,CAAO,iBAAA,CAAkB,SAAA,EAAW,MAAM,CAAA;AAAA,IAC/G,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,SAAA,EAAW,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACjF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,kBAAA,EAAoB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC5F;AAAA,GACD,CAAA;AAED,EAAA,MAAM,YAAY,WAAA,CAAY;AAAA,IAC5B,UAAA,EAAY,CAAC,EAAE,SAAA,EAAW,QAAO,KAAoC,MAAA,CAAO,gBAAA,CAAiB,SAAA,EAAW,MAAM,CAAA;AAAA,IAC9G,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,iBAAA,CAAkB,EAAE,QAAA,EAAU,CAAC,cAAA,EAAgB,SAAA,CAAU,SAAA,EAAW,SAAA,CAAU,MAAM,CAAA,EAAG,CAAA;AACxG,MAAA,KAAK,YAAY,iBAAA,CAAkB;AAAA,QACjC,UAAU,CAAC,uBAAA,EAAyB,SAAA,CAAU,SAAA,EAAW,UAAU,MAAM;AAAA,OAC1E,CAAA;AACD,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,4BAA4B,GAAG,CAAA;AAC/E,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,oBAAoB,GAAG,CAAA;AACvE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,cAAc,GAAG,CAAA;AACjE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,sBAAsB,GAAG,CAAA;AACzE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,yBAAyB,GAAG,CAAA;AAC5E,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,2BAA2B,GAAG,CAAA;AAAA,IAChF;AAAA,GACD,CAAA;AAGD,EAAA,MAAM,mBAAmB,WAAA,CAAY;AAAA,IACnC,UAAA,EAAY,CAAC,MAAA,KAA0C,MAAA,CAAO,wBAAwB,MAAM,CAAA;AAAA,IAC5F,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,SAAA,EAAW,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACjF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,kBAAA,EAAoB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC5F;AAAA,GACD,CAAA;AAGD,EAAA,MAAM,mBAAmB,WAAA,CAAY;AAAA,IACnC,UAAA,EAAY,CAAC,MAAA,KAA0C,MAAA,CAAO,wBAAwB,MAAM,CAAA;AAAA,IAC5F,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,SAAA,EAAW,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACjF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,kBAAA,EAAoB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC5F;AAAA,GACD,CAAA;AAGD,EAAA,MAAM,cAAc,WAAA,CAAY;AAAA,IAC9B,UAAA,EAAY,OAAO,EAAE,SAAA,EAAW,SAAQ,KAAgD;AACtF,MAAA,OAAO,MAAA,CAAO,uBAAA,CAAwB,EAAE,SAAA,EAAW,SAAS,CAAA;AAAA,IAC9D,CAAA;AAAA,IACA,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,eAAA,EAAiB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACvF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,SAAA,EAAW,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AACjF,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,kBAAA,EAAoB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC5F;AAAA,GACD,CAAA;AAED,EAAA,MAAM,oBAAoB,WAAA,CAAY;AAAA,IACpC,UAAA,EAAY,CAAC,MAAA,KAA2C,MAAA,CAAO,yBAAyB,MAAM,CAAA;AAAA,IAC9F,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,qBAAA,EAAuB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAAA,IAC/F;AAAA,GACD,CAAA;AAED,EAAA,MAAM,mBAAmB,WAAA,CAAY;AAAA,IACnC,UAAA,EAAY,CAAC,YAAA,KAAyB,MAAA,CAAO,iBAAiB,YAAY,CAAA;AAAA,IAC1E,WAAW,MAAM;AACf,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,aAAa,GAAG,CAAA;AAChE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,qBAAqB,GAAG,CAAA;AACxE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,oBAAoB,GAAG,CAAA;AACvE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,4BAA4B,GAAG,CAAA;AAC/E,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,cAAc,GAAG,CAAA;AACjE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,iBAAiB,GAAG,CAAA;AACpE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,2BAA2B,GAAG,CAAA;AAAA,IAChF;AAAA,GACD,CAAA;AAED,EAAA,MAAM,mBAAmB,WAAA,CAAY;AAAA,IACnC,UAAA,EAAY,CAAC,MAAA,KAA0C,MAAA,CAAO,wBAAwB,MAAM,CAAA;AAAA,IAC5F,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,aAAa,GAAG,CAAA;AAChE,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,qBAAA,EAAuB,SAAA,CAAU,SAAS,CAAA,EAAG,CAAA;AAC7F,MAAA,KAAK,YAAY,iBAAA,CAAkB;AAAA,QACjC,UAAU,CAAC,oBAAA,EAAsB,SAAA,CAAU,SAAA,EAAW,UAAU,YAAY;AAAA,OAC7E,CAAA;AAAA,IACH;AAAA,GACD,CAAA;AAED,EAAA,MAAM,yBAAyB,WAAA,CAAY;AAAA,IACzC,UAAA,EAAY,CAAC,MAAA,KAAyC,MAAA,CAAO,8BAA8B,MAAM,CAAA;AAAA,IACjG,SAAA,EAAW,CAAC,CAAA,EAAG,SAAA,KAAc;AAC3B,MAAA,KAAK,WAAA,CAAY,kBAAkB,EAAE,QAAA,EAAU,CAAC,oBAAA,EAAsB,SAAA,CAAU,YAAY,CAAA,EAAG,CAAA;AAC/F,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,4BAA4B,GAAG,CAAA;AAC/E,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,cAAc,GAAG,CAAA;AACjE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,sBAAsB,GAAG,CAAA;AACzE,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,yBAAyB,GAAG,CAAA;AAC5E,MAAA,KAAK,YAAY,iBAAA,CAAkB,EAAE,UAAU,CAAC,2BAA2B,GAAG,CAAA;AAAA,IAChF;AAAA,GACD,CAAA;AAED,EAAA,OAAO;AAAA,IACL,aAAA;AAAA,IACA,aAAA;AAAA,IACA,aAAA;AAAA,IACA,OAAA;AAAA,IACA,UAAA;AAAA,IACA,UAAA;AAAA,IACA,SAAA;AAAA,IACA,WAAA;AAAA,IACA,gBAAA;AAAA,IACA,gBAAA;AAAA,IACA,iBAAA;AAAA,IACA,gBAAA;AAAA,IACA,gBAAA;AAAA,IACA;AAAA,GACF;AACF;;;;"}

package/dist/domains/datasets/hooks/use-datasets.cjs.js

@@ -0,0 +1,55 @@
+'use strict';
+
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+
+const react = require('@mastra/react');
+const reactQuery = require('@tanstack/react-query');
+const React = require('react');
+const hooks_useInView = require('../../../hooks/use-in-view.cjs.js');
+
+const useDatasets = (pagination) => {
+  const client = react.useMastraClient();
+  return reactQuery.useQuery({
+    queryKey: ["datasets", pagination],
+    queryFn: () => client.listDatasets(pagination),
+    placeholderData: (previousData) => previousData
+  });
+};
+const DATASETS_PER_PAGE = 20;
+const useInfiniteDatasets = (filter, orderBy) => {
+  const client = react.useMastraClient();
+  const { inView: isEndOfListInView, setRef: setEndOfListElement } = hooks_useInView.useInView();
+  const targetType = filter?.targetType || void 0;
+  const targetIds = filter?.targetId ? [filter.targetId] : void 0;
+  const query = reactQuery.useInfiniteQuery({
+    queryKey: ["datasets", "infinite", { targetType, targetIds, orderBy }],
+    queryFn: ({ pageParam }) => client.listDatasets({ page: pageParam, perPage: DATASETS_PER_PAGE, targetType, targetIds, orderBy }),
+    initialPageParam: 0,
+    getNextPageParam: (lastPage, _, lastPageParam) => {
+      if (!lastPage?.datasets?.length || !lastPage.pagination?.hasMore) {
+        return void 0;
+      }
+      return lastPageParam + 1;
+    },
+    select: (data) => data.pages.flatMap((page) => page?.datasets ?? [])
+  });
+  React.useEffect(() => {
+    if (isEndOfListInView && query.hasNextPage && !query.isFetchingNextPage) {
+      void query.fetchNextPage();
+    }
+  }, [isEndOfListInView, query]);
+  return { ...query, setEndOfListElement };
+};
+const useDataset = (datasetId) => {
+  const client = react.useMastraClient();
+  return reactQuery.useQuery({
+    queryKey: ["dataset", datasetId],
+    queryFn: () => client.getDataset(datasetId),
+    enabled: Boolean(datasetId)
+  });
+};
+
+exports.useDataset = useDataset;
+exports.useDatasets = useDatasets;
+exports.useInfiniteDatasets = useInfiniteDatasets;
+//# sourceMappingURL=use-datasets.cjs.js.map

package/dist/domains/datasets/hooks/use-datasets.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-datasets.cjs.js","sources":["../../../../src/domains/datasets/hooks/use-datasets.ts"],"sourcesContent":["import type { ExperimentTargetType, ListDatasetsParams } from '@mastra/client-js';\nimport { useMastraClient } from '@mastra/react';\nimport { useInfiniteQuery, useQuery } from '@tanstack/react-query';\nimport { useEffect } from 'react';\nimport { useInView } from '@/hooks/use-in-view';\n\n/**\n * Hook to list all datasets with optional pagination\n */\nexport const useDatasets = (pagination?: { page?: number; perPage?: number }) => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['datasets', pagination],\n    queryFn: () => client.listDatasets(pagination),\n    placeholderData: previousData => previousData,\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\n/**\n * Hook to list datasets with infinite scroll pagination, optionally scoped server-side to a target.\n */\nexport const useInfiniteDatasets = (filter?: DatasetTargetFilter, orderBy?: DatasetsOrderBy) => {\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 => data.pages.flatMap(page => page?.datasets ?? []),\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 */\nexport const useDataset = (datasetId: string) => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['dataset', datasetId],\n    queryFn: () => client.getDataset(datasetId),\n    enabled: Boolean(datasetId),\n  });\n};\n"],"names":["useMastraClient","useQuery","useInView","useInfiniteQuery","useEffect"],"mappings":";;;;;;;;;AASO,MAAM,WAAA,GAAc,CAAC,UAAA,KAAqD;AAC/E,EAAA,MAAM,SAASA,qBAAA,EAAgB;AAC/B,EAAA,OAAOC,mBAAA,CAAS;AAAA,IACd,QAAA,EAAU,CAAC,UAAA,EAAY,UAAU,CAAA;AAAA,IACjC,OAAA,EAAS,MAAM,MAAA,CAAO,YAAA,CAAa,UAAU,CAAA;AAAA,IAC7C,iBAAiB,CAAA,YAAA,KAAgB;AAAA,GAClC,CAAA;AACH;AAEA,MAAM,iBAAA,GAAoB,EAAA;AAYnB,MAAM,mBAAA,GAAsB,CAAC,MAAA,EAA8B,OAAA,KAA8B;AAC9F,EAAA,MAAM,SAASD,qBAAA,EAAgB;AAC/B,EAAA,MAAM,EAAE,MAAA,EAAQ,iBAAA,EAAmB,MAAA,EAAQ,mBAAA,KAAwBE,yBAAA,EAAU;AAC7E,EAAA,MAAM,UAAA,GAAa,QAAQ,UAAA,IAAc,MAAA;AACzC,EAAA,MAAM,YAAY,MAAA,EAAQ,QAAA,GAAW,CAAC,MAAA,CAAO,QAAQ,CAAA,GAAI,MAAA;AAEzD,EAAA,MAAM,QAAQC,2BAAA,CAAiB;AAAA,IAC7B,QAAA,EAAU,CAAC,UAAA,EAAY,UAAA,EAAY,EAAE,UAAA,EAAY,SAAA,EAAW,SAAS,CAAA;AAAA,IACrE,OAAA,EAAS,CAAC,EAAE,SAAA,OACV,MAAA,CAAO,YAAA,CAAa,EAAE,IAAA,EAAM,WAAW,OAAA,EAAS,iBAAA,EAAmB,UAAA,EAAY,SAAA,EAAW,SAAS,CAAA;AAAA,IACrG,gBAAA,EAAkB,CAAA;AAAA,IAClB,gBAAA,EAAkB,CAAC,QAAA,EAAU,CAAA,EAAG,aAAA,KAAkB;AAChD,MAAA,IAAI,CAAC,QAAA,EAAU,QAAA,EAAU,UAAU,CAAC,QAAA,CAAS,YAAY,OAAA,EAAS;AAChE,QAAA,OAAO,MAAA;AAAA,MACT;AACA,MAAA,OAAO,aAAA,GAAgB,CAAA;AAAA,IACzB,CAAA;AAAA,IACA,MAAA,EAAQ,UAAQ,IAAA,CAAK,KAAA,CAAM,QAAQ,CAAA,IAAA,KAAQ,IAAA,EAAM,QAAA,IAAY,EAAE;AAAA,GAChE,CAAA;AAED,EAAAC,eAAA,CAAU,MAAM;AACd,IAAA,IAAI,iBAAA,IAAqB,KAAA,CAAM,WAAA,IAAe,CAAC,MAAM,kBAAA,EAAoB;AACvE,MAAA,KAAK,MAAM,aAAA,EAAc;AAAA,IAC3B;AAAA,EACF,CAAA,EAAG,CAAC,iBAAA,EAAmB,KAAK,CAAC,CAAA;AAE7B,EAAA,OAAO,EAAE,GAAG,KAAA,EAAO,mBAAA,EAAoB;AACzC;AAKO,MAAM,UAAA,GAAa,CAAC,SAAA,KAAsB;AAC/C,EAAA,MAAM,SAASJ,qBAAA,EAAgB;AAC/B,EAAA,OAAOC,mBAAA,CAAS;AAAA,IACd,QAAA,EAAU,CAAC,SAAA,EAAW,SAAS,CAAA;AAAA,IAC/B,OAAA,EAAS,MAAM,MAAA,CAAO,UAAA,CAAW,SAAS,CAAA;AAAA,IAC1C,OAAA,EAAS,QAAQ,SAAS;AAAA,GAC3B,CAAA;AACH;;;;;;"}

package/dist/domains/datasets/hooks/use-datasets.d.ts

@@ -0,0 +1,2 @@
+export * from '../../../src/domains/datasets/hooks/use-datasets'
+export {}

package/dist/domains/datasets/hooks/use-datasets.es.js

@@ -0,0 +1,49 @@
+import { useMastraClient } from '@mastra/react';
+import { useQuery, useInfiniteQuery } from '@tanstack/react-query';
+import { useEffect } from 'react';
+import { useInView } from '../../../hooks/use-in-view.es.js';
+
+const useDatasets = (pagination) => {
+  const client = useMastraClient();
+  return useQuery({
+    queryKey: ["datasets", pagination],
+    queryFn: () => client.listDatasets(pagination),
+    placeholderData: (previousData) => previousData
+  });
+};
+const DATASETS_PER_PAGE = 20;
+const useInfiniteDatasets = (filter, orderBy) => {
+  const client = useMastraClient();
+  const { inView: isEndOfListInView, setRef: setEndOfListElement } = useInView();
+  const targetType = filter?.targetType || void 0;
+  const targetIds = filter?.targetId ? [filter.targetId] : void 0;
+  const query = useInfiniteQuery({
+    queryKey: ["datasets", "infinite", { targetType, targetIds, orderBy }],
+    queryFn: ({ pageParam }) => client.listDatasets({ page: pageParam, perPage: DATASETS_PER_PAGE, targetType, targetIds, orderBy }),
+    initialPageParam: 0,
+    getNextPageParam: (lastPage, _, lastPageParam) => {
+      if (!lastPage?.datasets?.length || !lastPage.pagination?.hasMore) {
+        return void 0;
+      }
+      return lastPageParam + 1;
+    },
+    select: (data) => data.pages.flatMap((page) => page?.datasets ?? [])
+  });
+  useEffect(() => {
+    if (isEndOfListInView && query.hasNextPage && !query.isFetchingNextPage) {
+      void query.fetchNextPage();
+    }
+  }, [isEndOfListInView, query]);
+  return { ...query, setEndOfListElement };
+};
+const useDataset = (datasetId) => {
+  const client = useMastraClient();
+  return useQuery({
+    queryKey: ["dataset", datasetId],
+    queryFn: () => client.getDataset(datasetId),
+    enabled: Boolean(datasetId)
+  });
+};
+
+export { useDataset, useDatasets, useInfiniteDatasets };
+//# sourceMappingURL=use-datasets.es.js.map

package/dist/domains/datasets/hooks/use-datasets.es.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-datasets.es.js","sources":["../../../../src/domains/datasets/hooks/use-datasets.ts"],"sourcesContent":["import type { ExperimentTargetType, ListDatasetsParams } from '@mastra/client-js';\nimport { useMastraClient } from '@mastra/react';\nimport { useInfiniteQuery, useQuery } from '@tanstack/react-query';\nimport { useEffect } from 'react';\nimport { useInView } from '@/hooks/use-in-view';\n\n/**\n * Hook to list all datasets with optional pagination\n */\nexport const useDatasets = (pagination?: { page?: number; perPage?: number }) => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['datasets', pagination],\n    queryFn: () => client.listDatasets(pagination),\n    placeholderData: previousData => previousData,\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\n/**\n * Hook to list datasets with infinite scroll pagination, optionally scoped server-side to a target.\n */\nexport const useInfiniteDatasets = (filter?: DatasetTargetFilter, orderBy?: DatasetsOrderBy) => {\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 => data.pages.flatMap(page => page?.datasets ?? []),\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 */\nexport const useDataset = (datasetId: string) => {\n  const client = useMastraClient();\n  return useQuery({\n    queryKey: ['dataset', datasetId],\n    queryFn: () => client.getDataset(datasetId),\n    enabled: Boolean(datasetId),\n  });\n};\n"],"names":[],"mappings":";;;;;AASO,MAAM,WAAA,GAAc,CAAC,UAAA,KAAqD;AAC/E,EAAA,MAAM,SAAS,eAAA,EAAgB;AAC/B,EAAA,OAAO,QAAA,CAAS;AAAA,IACd,QAAA,EAAU,CAAC,UAAA,EAAY,UAAU,CAAA;AAAA,IACjC,OAAA,EAAS,MAAM,MAAA,CAAO,YAAA,CAAa,UAAU,CAAA;AAAA,IAC7C,iBAAiB,CAAA,YAAA,KAAgB;AAAA,GAClC,CAAA;AACH;AAEA,MAAM,iBAAA,GAAoB,EAAA;AAYnB,MAAM,mBAAA,GAAsB,CAAC,MAAA,EAA8B,OAAA,KAA8B;AAC9F,EAAA,MAAM,SAAS,eAAA,EAAgB;AAC/B,EAAA,MAAM,EAAE,MAAA,EAAQ,iBAAA,EAAmB,MAAA,EAAQ,mBAAA,KAAwB,SAAA,EAAU;AAC7E,EAAA,MAAM,UAAA,GAAa,QAAQ,UAAA,IAAc,MAAA;AACzC,EAAA,MAAM,YAAY,MAAA,EAAQ,QAAA,GAAW,CAAC,MAAA,CAAO,QAAQ,CAAA,GAAI,MAAA;AAEzD,EAAA,MAAM,QAAQ,gBAAA,CAAiB;AAAA,IAC7B,QAAA,EAAU,CAAC,UAAA,EAAY,UAAA,EAAY,EAAE,UAAA,EAAY,SAAA,EAAW,SAAS,CAAA;AAAA,IACrE,OAAA,EAAS,CAAC,EAAE,SAAA,OACV,MAAA,CAAO,YAAA,CAAa,EAAE,IAAA,EAAM,WAAW,OAAA,EAAS,iBAAA,EAAmB,UAAA,EAAY,SAAA,EAAW,SAAS,CAAA;AAAA,IACrG,gBAAA,EAAkB,CAAA;AAAA,IAClB,gBAAA,EAAkB,CAAC,QAAA,EAAU,CAAA,EAAG,aAAA,KAAkB;AAChD,MAAA,IAAI,CAAC,QAAA,EAAU,QAAA,EAAU,UAAU,CAAC,QAAA,CAAS,YAAY,OAAA,EAAS;AAChE,QAAA,OAAO,MAAA;AAAA,MACT;AACA,MAAA,OAAO,aAAA,GAAgB,CAAA;AAAA,IACzB,CAAA;AAAA,IACA,MAAA,EAAQ,UAAQ,IAAA,CAAK,KAAA,CAAM,QAAQ,CAAA,IAAA,KAAQ,IAAA,EAAM,QAAA,IAAY,EAAE;AAAA,GAChE,CAAA;AAED,EAAA,SAAA,CAAU,MAAM;AACd,IAAA,IAAI,iBAAA,IAAqB,KAAA,CAAM,WAAA,IAAe,CAAC,MAAM,kBAAA,EAAoB;AACvE,MAAA,KAAK,MAAM,aAAA,EAAc;AAAA,IAC3B;AAAA,EACF,CAAA,EAAG,CAAC,iBAAA,EAAmB,KAAK,CAAC,CAAA;AAE7B,EAAA,OAAO,EAAE,GAAG,KAAA,EAAO,mBAAA,EAAoB;AACzC;AAKO,MAAM,UAAA,GAAa,CAAC,SAAA,KAAsB;AAC/C,EAAA,MAAM,SAAS,eAAA,EAAgB;AAC/B,EAAA,OAAO,QAAA,CAAS;AAAA,IACd,QAAA,EAAU,CAAC,SAAA,EAAW,SAAS,CAAA;AAAA,IAC/B,OAAA,EAAS,MAAM,MAAA,CAAO,UAAA,CAAW,SAAS,CAAA;AAAA,IAC1C,OAAA,EAAS,QAAQ,SAAS;AAAA,GAC3B,CAAA;AACH;;;;"}

package/dist/domains/datasets/hooks.cjs.js

@@ -0,0 +1,14 @@
+'use strict';
+
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+
+const domains_datasets_hooks_useDatasets = require('./hooks/use-datasets.cjs.js');
+const domains_datasets_hooks_useDatasetMutations = require('./hooks/use-dataset-mutations.cjs.js');
+
+
+
+exports.useDataset = domains_datasets_hooks_useDatasets.useDataset;
+exports.useDatasets = domains_datasets_hooks_useDatasets.useDatasets;
+exports.useInfiniteDatasets = domains_datasets_hooks_useDatasets.useInfiniteDatasets;
+exports.useDatasetMutations = domains_datasets_hooks_useDatasetMutations.useDatasetMutations;
+//# sourceMappingURL=hooks.cjs.js.map

package/dist/domains/datasets/hooks.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hooks.cjs.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;;;"}

package/dist/domains/datasets/hooks.d.ts

@@ -0,0 +1,2 @@
+export * from '../../src/domains/datasets/hooks/index'
+export {}

package/dist/domains/datasets/hooks.es.js

@@ -0,0 +1,3 @@
+export { useDataset, useDatasets, useInfiniteDatasets } from './hooks/use-datasets.es.js';
+export { useDatasetMutations } from './hooks/use-dataset-mutations.es.js';
+//# sourceMappingURL=hooks.es.js.map

package/dist/domains/datasets/hooks.es.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hooks.es.js","sources":[],"sourcesContent":[],"names":[],"mappings":";"}

package/dist/domains/datasets.cjs.js

@@ -0,0 +1,16 @@
+'use strict';
+
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+
+const domains_datasets_components_saveAsDatasetItemDialog = require('./datasets/components/save-as-dataset-item-dialog.cjs.js');
+const domains_datasets_hooks_useDatasets = require('./datasets/hooks/use-datasets.cjs.js');
+const domains_datasets_hooks_useDatasetMutations = require('./datasets/hooks/use-dataset-mutations.cjs.js');
+
+
+
+exports.SaveAsDatasetItemDialog = domains_datasets_components_saveAsDatasetItemDialog.SaveAsDatasetItemDialog;
+exports.useDataset = domains_datasets_hooks_useDatasets.useDataset;
+exports.useDatasets = domains_datasets_hooks_useDatasets.useDatasets;
+exports.useInfiniteDatasets = domains_datasets_hooks_useDatasets.useInfiniteDatasets;
+exports.useDatasetMutations = domains_datasets_hooks_useDatasetMutations.useDatasetMutations;
+//# sourceMappingURL=datasets.cjs.js.map

package/dist/domains/datasets.cjs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"datasets.cjs.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;;;;;;;;;;;;;"}

package/dist/domains/datasets.d.ts

@@ -0,0 +1,2 @@
+export * from '../src/domains/datasets/index'
+export {}

package/dist/domains/datasets.es.js

@@ -0,0 +1,4 @@
+export { SaveAsDatasetItemDialog } from './datasets/components/save-as-dataset-item-dialog.es.js';
+export { useDataset, useDatasets, useInfiniteDatasets } from './datasets/hooks/use-datasets.es.js';
+export { useDatasetMutations } from './datasets/hooks/use-dataset-mutations.es.js';
+//# sourceMappingURL=datasets.es.js.map

package/dist/domains/datasets.es.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"datasets.es.js","sources":[],"sourcesContent":[],"names":[],"mappings":";;"}

package/dist/domains/logs/components/log-data-panel.cjs.js

@@ -9,7 +9,7 @@ const Button = require('../../../Button-BAMnPqos.cjs');
 const buttonsGroup = require('../../../buttons-group-CoZP3lJt.cjs');
 const copyButton = require('../../../copy-button-D2MTGxq_.cjs');
 const dataKeysAndValues = require('../../../data-keys-and-values-CMJ9zHKm.cjs');
-const dataPanel = require('../../../data-panel-C_BqhzeL.cjs');
+const dataPanel = require('../../../data-panel-Bq23Kt2W.cjs');
 const utils_dateFormat = require('../../../utils/date-format.cjs.js');
 
 function LogDataPanel({

package/dist/domains/logs/components/log-data-panel.es.js

@@ -5,7 +5,7 @@ import { B as Button } from '../../../Button-B8nF87ai.js';
 import { B as ButtonsGroup } from '../../../buttons-group-DsZgrZBZ.js';
 import { C as CopyButton } from '../../../copy-button-DcZ1mdw4.js';
 import { D as DataKeysAndValues } from '../../../data-keys-and-values-BTYDB-K2.js';
-import { D as DataPanel } from '../../../data-panel-BWgExbHs.js';
+import { D as DataPanel } from '../../../data-panel-D5ruDXyN.js';
 import { formatTimestampPrecise } from '../../../utils/date-format.es.js';
 
 function LogDataPanel({

package/dist/domains/memory/components/observation-detail-view.cjs.js

@@ -5,7 +5,7 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
 const jsxRuntime = require('react/jsx-runtime');
 const React = require('react');
 const checkbox = require('../../../checkbox-CI8JtZKS.cjs');
-const codeDiff = require('../../../code-diff-CG8veSa-.cjs');
+const codeDiff = require('../../../code-diff-yR3Gan2m.cjs');
 const EmptyState = require('../../../EmptyState-2XQgeju6.cjs');
 const skeleton = require('../../../skeleton-cXUjPrq9.cjs');
 const Txt = require('../../../Txt-DPmEpNbp.cjs');

package/dist/domains/memory/components/observation-detail-view.es.js

@@ -1,7 +1,7 @@
 import { jsxs, jsx } from 'react/jsx-runtime';
 import { useState, useMemo, useEffect } from 'react';
 import { C as Checkbox } from '../../../checkbox-CQedzgsi.js';
-import { C as CodeDiff } from '../../../code-diff-Hao35skN.js';
+import { C as CodeDiff } from '../../../code-diff-BgmXIX2v.js';
 import { E as EmptyState } from '../../../EmptyState-Bc4nJ9QH.js';
 import { S as Skeleton } from '../../../skeleton-BsRAXyh5.js';
 import { T as Txt } from '../../../Txt-BP1SRHmL.js';