Files
wehub-resource-sync 5a558eb09e
TypeScript SDK Compatibility V1.x E2E Tests / Select Node version matrix (push) Has been cancelled
TypeScript SDK Compatibility V1.x E2E Tests / TypeScript SDK Compatibility V1.x E2E Tests Node ${{matrix.node_version}} (push) Has been cancelled
TypeScript SDK E2E Tests / TypeScript SDK E2E Tests Node ${{matrix.node_version}} (push) Has been cancelled
Opik Optimizer - E2E Tests / build-opik (push) Has been cancelled
TypeScript SDK Compatibility V1.x E2E Tests / build-opik (push) Has been cancelled
Python SDK E2E Tests / Select Python version matrix (push) Has been cancelled
Python SDK E2E Tests / Python SDK E2E Tests ${{matrix.python_version}} (push) Has been cancelled
Python SDK E2E Tests / build-opik (push) Has been cancelled
Python SDK Compatibility V1.x E2E Tests / Select Python version matrix (push) Has been cancelled
Python SDK Compatibility V1.x E2E Tests / Python SDK Compatibility V1.x E2E Tests ${{matrix.python_version}} (push) Has been cancelled
Python SDK Compatibility V1.x E2E Tests / build-opik (push) Has been cancelled
TypeScript SDK E2E Tests / Select Node version matrix (push) Has been cancelled
TypeScript SDK E2E Tests / build-opik (push) Has been cancelled
Opik Optimizer - E2E Tests / Opik Optimizer E2E Tests Python ${{matrix.python_version}} (push) Has been cancelled
Opik Optimizer - E2E Tests / Opik Optimizer Integration Smoke Tests (push) Has been cancelled
🐙 Code Quality / detect (push) Has been cancelled
🐙 Code Quality / lint (${{ matrix.leg.name }}) (push) Has been cancelled
🐙 Code Quality / summary (push) Has been cancelled
TypeScript SDK Library Integration Tests / Check Secrets (push) Has been cancelled
TypeScript SDK Library Integration Tests / opik-vercel (Vercel AI SDK / eve) (push) Has been cancelled
SDK Library Integration Tests Runner / Check Secrets (push) Has been cancelled
SDK Library Integration Tests Runner / Missed OpenAI API Key Warning (push) Has been cancelled
SDK Library Integration Tests Runner / Build (push) Has been cancelled
SDK Library Integration Tests Runner / openai_tests (push) Has been cancelled
SDK Library Integration Tests Runner / langchain_tests (push) Has been cancelled
SDK Library Integration Tests Runner / langchain_legacy_tests (push) Has been cancelled
SDK Library Integration Tests Runner / llama_index_tests (push) Has been cancelled
SDK Library Integration Tests Runner / anthropic_tests (push) Has been cancelled
SDK Library Integration Tests Runner / mistral_tests (push) Has been cancelled
SDK Library Integration Tests Runner / groq_tests (push) Has been cancelled
SDK Library Integration Tests Runner / aisuite_tests (push) Has been cancelled
SDK Library Integration Tests Runner / haystack_tests (push) Has been cancelled
SDK Library Integration Tests Runner / dspy_tests (push) Has been cancelled
SDK Library Integration Tests Runner / crewai_v0_tests (push) Has been cancelled
SDK Library Integration Tests Runner / crewai_v1_tests (push) Has been cancelled
SDK Library Integration Tests Runner / genai_tests (push) Has been cancelled
SDK Library Integration Tests Runner / adk_tests (push) Has been cancelled
SDK Library Integration Tests Runner / adk_legacy_1_3_0_tests (push) Has been cancelled
SDK Library Integration Tests Runner / evaluation_metrics_tests (push) Has been cancelled
SDK Library Integration Tests Runner / bedrock_tests (push) Has been cancelled
SDK Library Integration Tests Runner / litellm_tests (push) Has been cancelled
SDK Library Integration Tests Runner / harbor_tests (push) Has been cancelled
SDK Library Integration Tests Runner / Slack Notification (push) Has been cancelled
Lint Opik Helm Chart / render-equality (push) Has been cancelled
Opik Optimizer - Unit Tests / Opik Optimizer Unit Tests Python ${{matrix.python_version}} (push) Has been cancelled
Python BE E2E Tests / Python BE E2E (push) Has been cancelled
Python Backend Tests / run-python-backend-tests (push) Has been cancelled
Python SDK Unit Tests / Python SDK Unit Tests ${{matrix.python_version}} (push) Has been cancelled
Release Drafter / update_release_draft (push) Has been cancelled
SDK E2E Libraries Integration Tests / Check Secrets (push) Has been cancelled
SDK E2E Libraries Integration Tests / Missed OpenAI API Key Warning (push) Has been cancelled
SDK E2E Libraries Integration Tests / build-opik (push) Has been cancelled
SDK E2E Libraries Integration Tests / E2E Lib Integration Python ${{matrix.python_version}} (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-gemini) (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-langchain) (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-openai) (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-otel) (push) Has been cancelled
TypeScript SDK Integration Build & Publish / build-and-publish (opik-vercel) (push) Has been cancelled
TypeScript SDK Build & Publish / build-and-publish (push) Has been cancelled
TypeScript SDK Unit Tests / Test on Node ${{ matrix.node-version }} (push) Has been cancelled
Backend Tests / discover-tests (push) Has been cancelled
Backend Tests / ${{ matrix.name }} (push) Has been cancelled
Build and Publish SDK / build-and-publish (push) Has been cancelled
Build Opik Docker Images / set-version (push) Has been cancelled
Build Opik Docker Images / build-backend (push) Has been cancelled
Build Opik Docker Images / build-sandbox-executor-python (push) Has been cancelled
Build Opik Docker Images / build-python-backend (push) Has been cancelled
Build Opik Docker Images / build-frontend (push) Has been cancelled
Build Opik Docker Images / create-git-tag (push) Has been cancelled
ClickHouse Migration Cluster Check / validate-clickhouse-migrations (push) Has been cancelled
Docs - Publish / run (push) Has been cancelled
E2E Tests - Post Merge (v2) / 🧪 E2E v2 Tests (${{ github.event.inputs.tier || 't1' }}) (push) Has been cancelled
E2E Tests - Post Merge (v2) / 📢 Slack Notification (push) Has been cancelled
Frontend Unit Tests / Test on Node 20 (push) Has been cancelled
Guardrails E2E Tests / Select Python version matrix (push) Has been cancelled
Guardrails E2E Tests / Guardrails E2E Tests ${{matrix.python_version}} (push) Has been cancelled
Guardrails E2E Tests / 📢 Slack Notification (push) Has been cancelled
Guardrails Backend Unit Tests / Guardrails Backend Unit Tests (push) Has been cancelled
Guardrails Backend Unit Tests / 📢 Slack Notification (push) Has been cancelled
Lint Opik Helm Chart / lint-helm-chart (Helm v3.21.0) (push) Has been cancelled
Lint Opik Helm Chart / lint-helm-chart (Helm v4.2.0) (push) Has been cancelled
Lint Opik Helm Chart / unittest-helm-chart (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 13:25:44 +08:00

790 lines
29 KiB
Plaintext
Raw Permalink Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
---
headline: Manage datasets | Opik Documentation
og:description: Evaluate your LLM using datasets. Learn to create and manage them
via Python SDK, TypeScript SDK, or the Traces table.
og:site_name: Opik Documentation
og:title: Manage datasets effectively with Opik
subtitle: Guides you through the process of creating and managing datasets
title: Manage datasets
canonical-url: https://www.comet.com/docs/opik/evaluation/advanced/manage_datasets
---
<Note>
In Opik 2.0, datasets are project-scoped. Make sure to specify a `project_name` when creating datasets so they are associated with the correct project.
</Note>
Datasets can be used to track test cases you would like to evaluate your LLM on. Each dataset is made up of a dictionary
with any key value pairs. When getting started, we recommend having an `input` and optional `expected_output` fields for
example. These datasets can be created from:
- Python SDK: You can use the Python SDK to create a dataset and add items to it.
- TypeScript SDK: You can use the TypeScript SDK to create a dataset and add items to it.
- Traces table: You can add existing logged traces (from a production application for example) to a dataset.
- The Opik UI: You can manually create a dataset and add items to it.
Once a dataset has been created, you can run Experiments on it. Each Experiment will evaluate an LLM application based
on the test cases in the dataset using an evaluation metric and report the results back to the dataset.
## Create a dataset via the UI
The simplest and fastest way to create a dataset is directly in the Opik UI.
This is ideal for quickly bootstrapping datasets from CSV files without needing to write any code.
Steps:
1. Navigate to **Evaluation > Datasets** in the Opik UI.
2. Click **Create new dataset**.
3. In the pop-up modal:
- Provide a name and an optional description
- Optionally, upload a CSV file with your data
4. Click **Create dataset**.
<Frame>
<img src="/img/evaluation/create_dataset.png" />
</Frame>
If you need to create a dataset with more than 1,000 rows, you [can use the SDK](/v1/evaluation/manage_datasets#creating-a-dataset-using-the-sdk).
<Tip>
The UI dataset creation has some limitations:
* File size is limited to 1,000 rows via the UI.
* No support for nested JSON structures in the CSV itself.
For datasets requiring rich metadata, complex schemas, or programmatic control, use the SDK instead (see the next section).
</Tip>
<Note>
When you create a dataset with a CSV file, this creates the first version (v1)
of your dataset. All subsequent modifications will create new versions automatically.
</Note>
## Understanding dataset versioning
Dataset versioning in Opik creates **immutable snapshots** of your data. Every time you modify a dataset—whether adding, editing, or deleting items—a new version is automatically created. This ensures complete reproducibility, provides an audit trail of all changes, and allows easy rollback to any previous state.
Each dataset version contains:
- **Version name**: Auto-generated sequential name (v1, v2, v3, etc.)
- **Change description**: Optional note describing what changed
- **Tags**: Labels for categorizing versions (e.g., `production`, `baseline`)
- **Item statistics**: Count of items added, modified, and deleted
- **Timestamp and author**: When the version was created and by whom
Once a version is created, its data cannot be changed—any modification creates a new version instead. Restoring a previous version also creates a *new* version with the same data, preserving your complete version timeline.
<Note>
The special `latest` tag always points to the most recent version.
When running experiments without specifying a version, `latest` is used by default.
</Note>
## Working with draft mode (UI)
When making changes to a dataset in the Opik UI, all modifications go into a **draft state** first. This gives you a staging area to review changes before committing them as a new version. The draft is visible only to you, and AI-generated samples from "Expand with AI" also go to draft for review.
When a dataset has unsaved draft changes, an orange **"Draft"** tag appears next to the dataset name, and **Save changes** / **Discard changes** buttons appear in the toolbar. Items show colored borders: green for newly added items, amber for modified items.
<Frame>
<img src="/img/evaluation/dataset_draft_mode.png" />
</Frame>
### Saving or discarding changes
To commit your draft as a new version:
1. Click **Save changes** in the toolbar
2. Enter a **version note** describing what changed
3. Optionally add **tags** to categorize this version
4. Click **Save**
<Frame>
<img src="/img/evaluation/save_version_dialog.png" />
</Frame>
To abandon your draft, click **Discard changes** and confirm. If you try to navigate away with unsaved changes, Opik displays a warning to prevent accidental loss of work.
<Tip>
Use draft mode to batch related changes into a single, well-documented version.
</Tip>
## Version history
To view the complete timeline of dataset changes, navigate to your dataset and click the **Version history** tab. The table shows each version's name, change summary (items added/modified/deleted), version note, tags, item count, and creation timestamp.
<Frame>
<img src="/img/evaluation/version_history_tab.png" />
</Frame>
From this view you can:
- **View items**: Click a version row and select **View items** to see the exact data at that point in time
- **Restore**: Click the **⋮** menu and select **Restore this version** to create a new version with that data
- **Edit metadata**: Click the **⋮** menu and select **Edit** to update the version note or tags (the data itself remains immutable)
<Note>
Restoring a version creates a **new** version with the same data.
No history is lost or overwritten.
</Note>
### Managing dataset and version tags from the SDK
The `Dataset` object exposes `get_tags()` to read the current tags, but does not yet provide a dedicated setter. To write tags programmatically — for example to drive an `env:prod` / `env:stage` promotion workflow — use the REST client exposed on the Opik client.
There are two tag surfaces, depending on what you want to scope the tag to:
- **Dataset-level tags** apply to the dataset as a whole and persist across versions. Use `update_dataset` — this **replaces** the existing tag list.
- **Version-level tags** apply to a specific dataset version. Use `update_dataset_version` — this is **additive** (it adds to the version's existing tags).
```python {pytest_codeblocks_skip=true}
import opik
client = opik.Opik()
dataset = client.get_or_create_dataset(name="my-eval", project_name="my-project")
# Read current dataset-level tags
print(dataset.get_tags())
# Set dataset-level tags (replaces the existing list)
client.rest_client.datasets.update_dataset(
id=dataset.id,
name=dataset.name,
tags=["env:prod"],
)
# Add tags to a specific version (additive)
client.rest_client.datasets.update_dataset_version(
id=dataset.id,
version_hash=dataset.version_hash,
tags_to_add=["env:prod"],
)
```
<Note>
`client.rest_client` is a thin wrapper around the public REST API. The underlying endpoints are stable, but the Python wrapper itself is not guaranteed to be backward-compatible across SDK versions. A first-class `Dataset.set_tags()` / `Dataset.add_version_tags()` helper is on the roadmap — this snippet is the supported interim path.
</Note>
You can then filter dataset items by these tags via [`get_items(filter_string=...)`](#querying-dataset-items) using the `tags contains` operator.
## Adding traces to a dataset
One of the most powerful ways to build evaluation datasets is by converting production traces into dataset items. This allows you to leverage real-world interactions from your LLM application to create test cases for evaluation.
### Adding traces via the UI
To add traces to a dataset from the Opik UI:
1. Navigate to the traces page
2. Select one or more traces you want to add to a dataset
3. Click the **Add to dataset** button in the toolbar
4. In the dialog that appears:
- Select an existing dataset or create a new one
- Choose which trace metadata to include:
- **Nested spans**: Include all child spans within the trace
- **Tags**: Include trace tags
- **Feedback scores**: Include any feedback scores attached to the trace
- **Comments**: Include comments added to the trace
- **Usage metrics**: Include token usage and cost information
- **Metadata**: Include custom metadata fields
5. Click on the dataset name to add the selected traces
<Frame>
<img src="/img/evaluation/add_traces_to_dataset.png" alt="Add traces to dataset modal" />
</Frame>
<Tip>
By default, all metadata options are enabled. You can uncheck any options you don't need. The trace's input and output are always included.
</Tip>
### What gets added to the dataset
When you add a trace to a dataset, the following structure is created:
- **input**: The trace's input data
- **expected_output**: The trace's output data (stored as `expected_output` for evaluation purposes)
- **spans** (optional): Array of nested spans with their inputs, outputs, and metadata
- **tags** (optional): Array of tags associated with the trace
- **feedback_scores** (optional): Array of feedback scores with name, value, and source
- **comments** (optional): Array of comments with text and ID
- **usage** (optional): Token usage and cost information
- **metadata** (optional): Custom metadata fields
This rich structure allows you to:
- Evaluate complex multi-step workflows by including nested spans
- Filter and analyze based on tags and metadata
- Use existing feedback scores as ground truth for evaluation
- Preserve context through comments and annotations
## Creating a dataset using the SDK
<Tip>
In Opik 2.0, datasets are project-scoped. Specify a `project_name` to associate your dataset with the correct project.
</Tip>
You can create a dataset and log items to it using the `get_or_create_dataset` method:
<CodeBlocks>
```typescript title="TypeScript SDK" language="typescript"
import { Opik } from "opik";
// Create a dataset
const client = new Opik();
const dataset = await client.getOrCreateDataset("My dataset", "Evaluation dataset", "my-project");
```
```python title="Python SDK" language="python"
from opik import Opik
# Create a dataset
client = Opik()
dataset = client.get_or_create_dataset(name="My dataset", project_name="my-project")
```
</CodeBlocks>
If a dataset with the given name already exists, the existing dataset will be returned.
### Insert items
#### Inserting dictionary items
You can insert items to a dataset using the `insert` method:
<CodeBlocks>
```typescript title="TypeScript" language="typescript"
import { Opik } from "opik";
const client = new Opik();
const dataset = await client.getOrCreateDataset("My dataset", "Evaluation dataset", "my-project");
dataset.insert([
{ user_question: "Hello, world!", expected_output: { assistant_answer: "Hello, world!" } },
{ user_question: "What is the capital of France?", expected_output: { assistant_answer: "Paris" } },
]);
```
```python title="Python" language="python"
import opik
# Get or create a dataset
client = opik.Opik()
dataset = client.get_or_create_dataset(name="My dataset", project_name="my-project")
# Add dataset items to it
dataset.insert([
{"user_question": "Hello, world!", "expected_output": {"assistant_answer": "Hello, world!"}},
{"user_question": "What is the capital of France?", "expected_output": {"assistant_answer": "Paris"}},
])
```
</CodeBlocks>
<Tip>
Opik automatically deduplicates items that are inserted into a dataset when using the Python SDK. This means that you
can insert the same item multiple times without duplicating it in the dataset. This combined with the `get or create
dataset` methods means that you can use the SDK to manage your datasets in a "fire and forget" manner.
</Tip>
<Note>
When using the SDK to insert items, a new dataset version is automatically created.
If you insert items in multiple batches within a single `insert()` call, they are grouped into one version.
</Note>
Once the items have been inserted, you can view them in the Opik UI:
<Frame>
<img src="/img/evaluation/dataset_items_page.png" />
</Frame>
#### Inserting items from a JSONL file
You can also insert items from a JSONL file:
<CodeBlocks>
```python title="Python" language="python"
import opik
client = opik.Opik()
dataset = client.get_or_create_dataset(name="My dataset", project_name="my-project")
dataset.read_jsonl_from_file("path/to/file.jsonl")
```
</CodeBlocks>
#### Inserting items from a Pandas DataFrame
You can also insert items from a Pandas DataFrame:
<CodeBlocks>
```python title="Python" language="python"
import opik
client = opik.Opik()
dataset = client.get_or_create_dataset(name="My dataset", project_name="my-project")
dataset.insert_from_pandas(dataframe=df)
# You can also specify an optional keys_mapping parameter
dataset.insert_from_pandas(dataframe=df, keys_mapping={"Expected output": "expected_output"})
```
</CodeBlocks>
### Deleting items
You can delete items in a dataset by using the `delete` method:
<CodeBlocks>
```typescript title="TypeScript" language="typescript"
import { Opik } from "opik";
// Get or create a dataset
client = new Opik();
dataset = await client.getDataset("My dataset")
await dataset.delete(["123", "456"])
// Or to delete all items
await dataset.clear()
```
```python title="Python" language="python"
from opik import Opik
# Get or create a dataset
client = Opik()
dataset = client.get_dataset(name="My dataset")
dataset.delete(items_ids=["123", "456"])
# Or to delete all items
dataset.clear()
```
</CodeBlocks>
<Note>
Deleting items creates a new version of the dataset. The deleted items remain accessible
in previous versions through the version history, ensuring you never permanently lose data.
</Note>
## Downloading a dataset from Opik
You can download a dataset from Opik using the `get_dataset` method:
<CodeBlocks>
```typescript title="TypeScript" language="typescript"
import { Opik } from "opik";
const client = new Opik();
const dataset = await client.getDataset("My dataset");
const items = await dataset.getItems();
console.log(items);
```
```python title="Python" language="python"
from opik import Opik
client = Opik()
dataset = client.get_dataset(name="My dataset")
# Get items as list of DatasetItem objects
items = dataset.get_items()
# Convert to a Pandas DataFrame
dataset.to_pandas()
# Convert to a JSON array
dataset.to_json()
```
</CodeBlocks>
## Filtering datasets programmatically
You can filter dataset items using the `filter_string` parameter on the `get_items()` method or when
running evaluations with `evaluate_prompt()`. This allows you to work with specific subsets of your data.
### Basic filtering
<CodeBlocks>
```python title="Python" language="python"
from opik import Opik
client = Opik()
dataset = client.get_dataset(name="my_dataset")
# Get filtered items
failed_items = dataset.get_items(filter_string='tags contains "failed"')
```
</CodeBlocks>
### Filter syntax
The filter string uses Opik Query Language (OQL) syntax. Supported columns include:
| Column | Type | Description |
|--------|------|-------------|
| `id` | String | Unique identifier for the dataset item |
| `source` | String | Source of the dataset item |
| `trace_id` | String | Associated trace ID |
| `span_id` | String | Associated span ID |
| `data` | Dictionary | Use dot notation for nested fields (e.g., `data.category`) |
| `tags` | List | Use "contains" operator (e.g., `tags contains "test"`) |
| `created_at` | DateTime | ISO 8601 format (e.g., `created_at >= "2024-01-01T00:00:00Z"`) |
| `last_updated_at` | DateTime | ISO 8601 format |
| `created_by` | String | User who created the item |
| `last_updated_by` | String | User who last updated the item |
### Filter examples
<CodeBlocks>
```python title="Python" language="python"
from opik import Opik
client = Opik()
dataset = client.get_dataset(name="my_dataset")
# Filter by tag
failed_items = dataset.get_items(filter_string='tags contains "failed"')
# Filter by data field
finance_items = dataset.get_items(filter_string='data.category = "finance"')
# Filter by date
recent_items = dataset.get_items(
filter_string='created_at >= "2024-06-01T00:00:00Z"'
)
# Multiple conditions
filtered_items = dataset.get_items(
filter_string='tags contains "production" AND data.difficulty = "hard"'
)
```
</CodeBlocks>
## Running experiments with dataset versions
When you run an experiment, Opik automatically links it to the specific dataset version that was used. This ensures complete reproducibility—you can always know exactly which data was used for any experiment.
### Automatic version association
Every experiment records which dataset version it used:
- When running from the UI or SDK without specifying a version, the `latest` version is used
- The experiment results page shows the associated dataset version
- You can click the version to see the exact data that was evaluated
This association is permanent. Even if you later modify the dataset, your experiment results remain linked to the original version used.
### Selecting a specific version in Playground
When running experiments from the Playground:
1. Open the Playground and configure your prompt
2. In the dataset selector, choose your dataset
3. A nested dropdown appears showing available versions
4. Select the specific version you want to use, or choose `latest` for the most recent
<Frame>
<img src="/img/evaluation/dataset_version_select.png" />
</Frame>
<Tip>
When comparing experiments or running A/B tests, use the same dataset version
to isolate the effect of your changes. This ensures differences in results
are due to your prompt or model changes, not data variations.
</Tip>
### Selecting a specific version in the SDK
When running experiments programmatically, you can specify which dataset version to use by passing a `DatasetVersion` object to `evaluate()`:
<CodeBlocks>
```python title="Python" language="python"
from opik import Opik
from opik.evaluation import evaluate
client = Opik()
dataset = client.get_dataset(name="My dataset")
# Run experiment on the latest version (default behavior)
result = evaluate(
experiment_name="baseline-experiment",
dataset=dataset,
task=my_task_function,
scoring_metrics=[my_metric],
project_name="my-project",
)
# Run experiment on a specific version
v1_view = dataset.get_version_view("v1")
result = evaluate(
experiment_name="v1-experiment",
dataset=v1_view, # Pass the DatasetVersion object
task=my_task_function,
scoring_metrics=[my_metric],
project_name="my-project",
)
```
```typescript title="TypeScript" language="typescript"
import { Opik, evaluate } from "opik";
const client = new Opik();
const dataset = await client.getDataset("My dataset");
// Run experiment on the latest version (default)
const result = await evaluate({
experimentName: "baseline-experiment",
dataset: dataset,
task: myTaskFunction,
scoringMetrics: [myMetric],
projectName: "my-project",
});
// Run experiment on a specific version
const v2 = await dataset.getVersionView("v2");
const pinnedResult = await evaluate({
experimentName: "pinned-experiment",
dataset: v2,
task: myTaskFunction,
scoringMetrics: [myMetric],
projectName: "my-project",
});
```
</CodeBlocks>
### Working with dataset versions programmatically
The SDK provides methods for inspecting and working with dataset versions:
<CodeBlocks>
```python title="Python" language="python"
from opik import Opik
client = Opik()
dataset = client.get_dataset(name="My dataset")
# Get the current (latest) version name
current_version = dataset.get_current_version_name()
print(f"Current version: {current_version}") # e.g., "v3"
# Get detailed version info (returns DatasetVersionPublic)
version_info = dataset.get_version_info()
print(f"Version ID: {version_info.id}")
print(f"Version name: {version_info.version_name}")
print(f"Items total: {version_info.items_total}")
print(f"Created at: {version_info.created_at}")
# Get a read-only view of a specific version
v1_view = dataset.get_version_view("v1")
# Access version metadata
print(f"Version: {v1_view.version_name}")
print(f"Items in v1: {v1_view.items_total}")
print(f"Items added: {v1_view.items_added}")
print(f"Items modified: {v1_view.items_modified}")
print(f"Items deleted: {v1_view.items_deleted}")
# Get items from a specific version
v1_items = v1_view.get_items()
# Export version data
v1_df = v1_view.to_pandas()
v1_json = v1_view.to_json()
```
```typescript title="TypeScript" language="typescript"
import { Opik } from "opik";
const client = new Opik();
const dataset = await client.getDataset("My dataset");
// Get the current (latest) version name
const currentVersion = await dataset.getCurrentVersionName();
console.log(`Current version: ${currentVersion}`); // e.g., "v3"
// Get detailed version info (returns DatasetVersionPublic)
const versionInfo = await dataset.getVersionInfo();
console.log(`Version ID: ${versionInfo?.id}`);
console.log(`Version name: ${versionInfo?.versionName}`);
console.log(`Items total: ${versionInfo?.itemsTotal}`);
console.log(`Created at: ${versionInfo?.createdAt}`);
// Get a read-only view of a specific version
const v1View = await dataset.getVersionView("v1");
// Access version metadata
console.log(`Version: ${v1View.versionName}`);
console.log(`Items in v1: ${v1View.itemsTotal}`);
console.log(`Items added: ${v1View.itemsAdded}`);
console.log(`Items modified: ${v1View.itemsModified}`);
console.log(`Items deleted: ${v1View.itemsDeleted}`);
// Get items from a specific version
const v1Items = await v1View.getItems();
// Export version data as JSON
const v1Json = await v1View.toJson();
```
</CodeBlocks>
<Note>
`DatasetVersion` is a read-only view. You cannot insert, update, or delete items
through a `DatasetVersion` object. All mutations must be done through the `Dataset` object.
</Note>
## Expanding a dataset with AI
Dataset expansion allows you to use AI to generate additional synthetic samples based on your existing dataset. This is particularly useful when you have a small dataset and want to create more diverse test cases to improve your evaluation coverage.
The AI analyzes the patterns in your existing data and generates new samples that follow similar structures while introducing variations. This helps you:
- **Increase dataset size** for more comprehensive evaluation
- **Create edge cases** and variations you might not have considered
- **Improve model robustness** by testing against diverse inputs
- **Scale your evaluation** without manual data creation
### How to expand a dataset
To expand a dataset with AI:
1. **Navigate to your dataset** in the Opik UI (Evaluation > Datasets > [Your Dataset])
2. **Click the "Expand with AI" button** in the dataset view
3. **Configure the expansion settings**:
- **Model**: Choose the LLM model to use for generation (supports GPT-4, GPT-5, Claude, and other models)
- **Sample Count**: Specify how many new samples to generate (1-100)
- **Preserve Fields**: Select which fields from your original data to keep unchanged
- **Variation Instructions**: Provide specific guidance on how to vary the data (e.g., "Create variations that test edge cases" or "Generate examples with different complexity levels")
- **Custom Prompt**: Optionally provide a custom prompt template instead of the auto-generated one
4. **Start the expansion** - The AI will analyze your data and generate new samples
5. **Review the results** - Generated samples are added to your **draft**. You can review, edit, or remove them before saving to create a new version
<Frame>
<img src="/img/evaluation/dataset_expansion_modal.png" />
</Frame>
### Configuration options
**Sample Count**: Start with a smaller number (10-20) to review the quality before generating larger batches.
**Preserve Fields**: Use this to maintain consistency in certain fields while allowing variation in others. For example, preserve the `category` field while varying the `input` and `expected_output`.
**Variation Instructions**: Provide specific guidance such as:
- "Create variations with different difficulty levels"
- "Generate edge cases and error scenarios"
- "Add examples with different input formats"
- "Include multilingual variations"
### Best practices
- **Start small**: Generate 10-20 samples first to evaluate quality before scaling up
- **Review generated content**: Always review AI-generated samples for accuracy and relevance
- **Use variation instructions**: Provide clear guidance on the type of variations you want
- **Preserve key fields**: Use field preservation to maintain important categorizations or metadata
- **Iterate and refine**: Use the custom prompt option to fine-tune generation for your specific needs
<Tip>
Dataset expansion works best when you have at least 5-10 high-quality examples in your original dataset. The AI uses
these examples to understand the patterns and generate similar but varied content.
</Tip>
## Managing dataset item tags
Tags are a powerful way to organize, categorize, and filter your dataset items. You can use tags to:
- **Categorize test cases** by type, difficulty, or domain (e.g., `edge-case`, `production`, `multilingual`)
- **Track data sources** where items originated from (e.g., `user-feedback`, `synthetic`, `real-world`)
- **Mark review status** during dataset curation (e.g., `needs-review`, `validated`, `archived`)
- **Filter for evaluation** to run experiments on specific subsets of your data
- **Organize workflows** by marking items for different stages or teams
Each dataset item can have multiple tags.
### Adding tags to dataset items
#### Adding tags to individual items
To add tags to a single dataset item:
1. **Navigate to your dataset** in the Opik UI (Evaluation > Datasets > [Your Dataset])
2. **Click on any dataset item** to open the details panel
3. **In the Tags section**, click the **"+" button**
4. **Type the tag name** and press Enter
5. The tag will be immediately added and saved
You can remove tags by clicking the **"×" icon** next to any tag in the details panel.
#### Adding tags to multiple items (batch operation)
To add the same tag to multiple dataset items at once:
1. **Navigate to your dataset** in the Opik UI
2. **Select multiple items** by clicking the checkboxes next to each item
3. **Click the "Add tags" button** in the toolbar (visible when items are selected)
4. **Enter the tag name** in the dialog that appears
5. **Click "Add tag"** to apply the tag to all selected items
This is particularly useful when you want to categorize a group of related test cases or mark items from the same data source.
<Tip>
Tags are case-sensitive and support alphanumeric characters, hyphens, and underscores. Choose consistent naming conventions for your tags to make filtering easier.
</Tip>
### Filtering dataset items by tags
Once you've tagged your dataset items, you can filter them to work with specific subsets:
1. **Navigate to your dataset** in the Opik UI
2. **Click the "Filters" button** next to the search bar
3. **Select "Tags" from the Column dropdown**
4. **Choose "contains" as the operator**
5. **Enter the tag name** you want to filter by
6. **Close the dialog** to apply the filter
The dataset items table will update to show only items matching your filter criteria. You can:
- **View filtered items** to focus on specific categories
- **Run experiments** on filtered subsets by using the filtered view
- **Export filtered data** for specific test case groups
- **Combine with other filters** to create complex queries
The filter is saved in the URL, so you can bookmark or share specific filtered views of your dataset.
## Bulk operations
Opik supports bulk operations for efficiently managing large datasets. These operations help you work with many items at once without tedious individual selections.
### Select all functionality
When working with datasets that span multiple pages:
1. **Select items on the current page** using the checkbox in the table header
2. A banner appears offering to **"Select all items"** across all pages
3. Click to select all items matching your current filter criteria
<Frame>
<img src="/img/evaluation/dataset_bulk_select.png" />
</Frame>
This works with filtered views too—if you have a filter applied, "Select all" only selects items matching that filter.
### Available bulk operations
Once you have items selected, the toolbar shows available operations:
- **Add tags**: Apply one or more tags to all selected items
- **Delete**: Remove selected items (creates a new version with items removed)
- **Export**: Download selected items as CSV or JSON
### Processing indicators
For large bulk operations:
- A loading indicator shows "Your dataset is still processing..."
- The operation runs in the background—you can continue browsing
- A success message appears when processing completes
<Tip>
For very large datasets, bulk operations are processed in batches. The UI remains
responsive during processing, and you'll see progress indicators for long-running operations.
</Tip>