@arizeai/phoenix-client 5.4.1 → 5.5.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arizeai/phoenix-client",
-  "version": "5.4.1",
+  "version": "5.5.1",
   "description": "A client for the Phoenix API",
   "main": "dist/src/index.js",
   "module": "dist/esm/index.js",
@@ -77,7 +77,7 @@
     "tiny-invariant": "^1.3.3",
     "zod": "^3.24.3",
     "zod-to-json-schema": "^3.24.3",
-    "@arizeai/phoenix-otel": "0.3.0"
+    "@arizeai/phoenix-otel": "0.3.1"
   },
   "engines": {
     "node": ">=18"

package/src/__generated__/api/v1.ts

@@ -3527,6 +3527,8 @@ export interface operations {
                     inputs: Record<string, unknown>[];
                     outputs?: Record<string, unknown>[];
                     metadata?: Record<string, unknown>[];
+                    /** @description Split per example: string, string array, or null */
+                    splits?: (string | string[] | null)[];
                 };
                 "multipart/form-data": {
                     /** @enum {string} */
@@ -3536,6 +3538,8 @@ export interface operations {
                     "input_keys[]": string[];
                     "output_keys[]": string[];
                     "metadata_keys[]"?: string[];
+                    /** @description Column names for auto-assigning examples to splits */
+                    "split_keys[]"?: string[];
                     /** Format: binary */
                     file: string;
                 };

package/src/datasets/appendDatasetExamples.ts

@@ -36,6 +36,9 @@ export async function appendDatasetExamples({
   const inputs = examples.map((example) => example.input);
   const outputs = examples.map((example) => example.output ?? {}); // Treat null as an empty object
   const metadata = examples.map((example) => example.metadata ?? {});
+  const splits = examples.map((example) =>
+    example.splits !== undefined ? example.splits : null
+  );
   let datasetName: string;
   if ("datasetName" in dataset) {
     datasetName = dataset.datasetName;
@@ -58,6 +61,7 @@ export async function appendDatasetExamples({
       inputs,
       outputs,
       metadata,
+      splits,
     },
   });
   invariant(appendResponse.data?.data, "Failed to append dataset examples");

package/src/datasets/createDataset.ts

@@ -37,6 +37,9 @@ export async function createDataset({
   const inputs = examples.map((example) => example.input);
   const outputs = examples.map((example) => example?.output ?? {}); // Treat null as an empty object
   const metadata = examples.map((example) => example?.metadata ?? {});
+  const splits = examples.map((example) =>
+    example?.splits !== undefined ? example.splits : null
+  );
   const createDatasetResponse = await client.POST("/v1/datasets/upload", {
     params: {
       query: {
@@ -51,6 +54,7 @@ export async function createDataset({
       inputs,
       outputs,
       metadata,
+      splits,
     },
   });
   invariant(createDatasetResponse.data?.data, "Failed to create dataset");

package/src/types/datasets.ts

@@ -45,6 +45,13 @@ 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;
 }
 
 /**