Dataset 类

添加于： @mastra/core@1.4.0

Dataset 类提供用于管理单个数据集条目 CRUD、版本和实验的方法。可通过 mastra.datasets.get() 或 mastra.datasets.create() 获取。

使用示例

添加条目并运行实验

src/mastra/datasets/usage.ts

import { Mastra } from '@mastra/core'

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

const dataset = await mastra.datasets.create({
  name: 'QA pairs',
})

// Add items
await dataset.addItems({
  items: [
    { input: { question: 'What is AI?' }, groundTruth: { answer: 'Artificial Intelligence' } },
    { input: { question: 'What is ML?' }, groundTruth: { answer: 'Machine Learning' } },
  ],
})

// Run an experiment
const summary = await dataset.startExperiment({
  targetType: 'agent',
  targetId: 'my-agent',
  scorers: ['accuracy'],
})

console.log(`${summary.succeededCount}/${summary.totalItems} succeeded`)

管理版本和历史记录

src/mastra/datasets/versioning.ts

import { Mastra } from '@mastra/core'

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

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

// List versions
const { versions } = await dataset.listVersions()

// List items at a specific version
const items = await dataset.listItems({ version: 2 })

// Get item change history
const history = await dataset.getItemHistory({ itemId: 'item-id' })

获取方式

不要直接实例化 Dataset。请通过 DatasetsManager 获取：

const dataset = await mastra.datasets.get({ id: 'dataset-id' })
// or
const dataset = await mastra.datasets.create({ name: 'My dataset' })

属性

要获取完整的数据集记录（名称、描述、schema、版本和时间戳），请调用 dataset.getDetails()。

id:

string

数据集的唯一标识符。只读。

相关内容

• DatasetsManager 类
• dataset.startExperiment()
• dataset.addItems()
• dataset.listVersions()