Datasets

import { init, datasets } from 'gentrace';

init({
  apiKey: process.env.GENTRACE_API_KEY,
});

// List datasets
const datasetList = await datasets.list();

// Access the datasets
for (const dataset of datasetList.data) {
  console.log(dataset.name);
  console.log(dataset.description);
}

import os
from gentrace import init, datasets

init(api_key=os.environ["GENTRACE_API_KEY"])

# List datasets
dataset_list = await datasets.list()

# Access the datasets
for dataset in dataset_list.data:
    print(dataset.name)
    print(dataset.description)

const dataset = await datasets.create({
  name: 'Production feedback - Q1 2025',
  description: 'Negative user feedback from production',
  pipelineId: 'pipeline-uuid', // or use pipelineSlug
});

dataset = await datasets.create(
    name="Production feedback - Q1 2025",
    description="Negative user feedback from production",
    pipeline_id="pipeline-uuid",  # or use pipeline_slug
)

const dataset = await datasets.retrieve('dataset-id');
console.log(dataset.name);

dataset = await datasets.retrieve("dataset-id")
print(dataset.name)

const dataset = await datasets.update('dataset-id', {
  name: 'Updated dataset name',
  isArchived: false,
  isGolden: true, // Mark as golden dataset
});

dataset = await datasets.update(
    "dataset-id",
    name="Updated dataset name",
    is_archived=False,
    is_golden=True,  # Mark as golden dataset
)

// Filter by pipeline
const datasetList = await datasets.list({
  pipelineId: 'pipeline-id',
  archived: false, // Exclude archived datasets
});

# Filter by pipeline
dataset_list = await datasets.list(
    pipeline_id="pipeline-id",
    archived=False,  # Exclude archived datasets
)

// Get test cases for a dataset
const testCasesList = await testCases.list({
  datasetId: dataset.id,
});

from gentrace import test_cases_async

# Get test cases for a dataset
test_case_list = await test_cases_async.list(
    dataset_id=dataset.id
)