@arizeai/phoenix-client/src/types/datasets.ts

A client for the Phoenix API

import { Node } from "./core";

type DatasetSelectorBase = { versionId?: string; splits?: string[] };

/**
 * A dataset can be identified by its datasetId, datasetName, or datasetVersionId
 */
export type DatasetSelector =
  | (DatasetSelectorBase & { datasetId: string })
  | (DatasetSelectorBase & { datasetName: string });

/**
 * Overview information about a dataset
 */
export interface DatasetInfo extends Node {
  name: string;
  description?: string | null;
  metadata?: Record<string, unknown>;
}

/**
 * Information about a dataset version
 */
export interface DatasetVersionInfo extends Node {
  description?: string | null;
  metadata?: Record<string, unknown>;
  createdAt: Date;
}

/**
 * A dataset's examples
 */
export interface DatasetExamples {
  examples: ExampleWithId[];
  /**
   * The version ID of the dataset examples
   */
  versionId: string;
}

/**
 * An example is a record to feed into an AI task
 */
export interface Example {
  input: Record<string, unknown>;
  output?: Record<string, unknown> | null;
  metadata?: Record<string, unknown> | null;
  /**
   * Split assignment for this example. Can be:
   * - A single string for one split: "train"
   * - An array of strings for multiple splits: ["train", "easy"]
   * - null for no split assignment
   */
  splits?: string | string[] | null;
}

/**
 * An example that has been synced to the server
 */
export interface ExampleWithId extends Example, Node {
  updatedAt: Date;
}

/**
 * A dataset is a collection of examples for an AI task
 */
export interface Dataset extends DatasetInfo, DatasetExamples, Node {}

/**
 * A dataset with its version information
 */
export interface DatasetWithVersion extends Dataset {
  versionInfo: DatasetVersionInfo;
}