DatasetsManager

添加于： @mastra/core@1.4.0

DatasetsManager 类提供管理数据集的公共 API，包括 CRUD 操作和跨数据集实验比较。可通过 mastra.datasets 访问。

使用示例

创建并列出数据集

src/mastra/datasets/manage.ts

import { Mastra } from '@mastra/core'

const mastra = new Mastra({/* storage config */})

// Create a dataset
const dataset = await mastra.datasets.create({
  name: 'QA pairs',
  description: 'Question-answer evaluation set',
})

// List all datasets
const { datasets } = await mastra.datasets.list()

获取和删除数据集

src/mastra/datasets/get-delete.ts

import { Mastra } from '@mastra/core'

const mastra = new Mastra({/* storage config */})

// Get by ID
const dataset = await mastra.datasets.get({ id: 'dataset-id' })

// Delete by ID
await mastra.datasets.delete({ id: 'dataset-id' })

获取实验

按 ID 获取特定实验（运行）。与 dataset.getExperiment() 不同，此方法适用于所有数据集，无需先获取数据集引用。

import { Mastra } from '@mastra/core'

const mastra = new Mastra({/* storage config */})

// Get experiment directly without knowing the dataset
const experiment = await mastra.datasets.getExperiment({
  experimentId: 'exp-id',
})

console.log(`Dataset: ${experiment.datasetId}`)
console.log(`Status: ${experiment.status}`)

比较实验

src/mastra/datasets/compare.ts

import { Mastra } from '@mastra/core'

const mastra = new Mastra({/* storage config */})

const comparison = await mastra.datasets.compareExperiments({
  experimentIds: ['exp-1', 'exp-2'],
  baselineId: 'exp-1',
})

获取方式

不要直接实例化 DatasetsManager。请从 Mastra 实例访问它：

const mastra = new Mastra({/* storage config */})
const datasetsManager = mastra.datasets

相关内容

• Dataset 类
• DatasetsManager.create()
• DatasetsManager.compareExperiments()