跳至主要內容

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