Tasks

import { listTasks } from "@arizeai/ax-client";

const { data: tasks, pagination } = await listTasks({
  space: "my-space",    // space name or ID (optional)
  name: "quality",      // substring filter on task name (optional)
  project: "my-project",  // project name or ID (optional)
  dataset: "my-dataset",  // dataset name or ID (optional)
  type: "TEMPLATE_EVALUATION",  // optional
  limit: 10,
});

import { createTask } from "@arizeai/ax-client";

const task = await createTask({
  name: "Production Quality Monitor",
  type: "TEMPLATE_EVALUATION",
  space: "my-space",    // required when project is a name
  project: "my-project",
  isContinuous: true,
  samplingRate: 0.1,    // sample 10% of spans
  evaluators: [
    {
      evaluatorId: "your_evaluator_id",
      columnMappings: { input: "question", output: "answer" },
    },
  ],
});

import { createTask, triggerTaskRun, waitForTaskRun } from "@arizeai/ax-client";

const task = await createTask({
  name: "Weekly Quality Check",
  type: "TEMPLATE_EVALUATION",
  space: "my-space",
  dataset: "my-dataset",
  evaluators: [
    {
      evaluatorId: "your_evaluator_id",
      columnMappings: { input: "question", output: "answer" },
    },
  ],
});

// Immediately trigger a run and wait for it to finish
const run = await triggerTaskRun({ task: task.id });
const finalRun = await waitForTaskRun({ runId: run.id });
console.log(finalRun.status); // "COMPLETED" | "FAILED" | "CANCELLED"

import { createEvaluationTask, triggerTaskRun, waitForTaskRun } from "@arizeai/ax-client";

const task = await createEvaluationTask({
  name: "Weekly Quality Check",
  type: "TEMPLATE_EVALUATION",
  space: "my-space",
  project: "my-project",
  evaluators: [
    {
      evaluatorId: "your_evaluator_id",
      columnMappings: { input: "question", output: "answer" },
    },
  ],
});

const run = await triggerTaskRun({ task: task.id });
const finalRun = await waitForTaskRun({ runId: run.id });

import { createRunExperimentTask, triggerTaskRun, waitForTaskRun } from "@arizeai/ax-client";

const task = await createRunExperimentTask({
  name: "GPT-4o Baseline Task",
  dataset: "my-dataset",
  space: "my-space",
  runConfiguration: {
    experiment_type: "LLM_GENERATION",
    aiIntegration: "my-openai-integration",
    model_name: "gpt-4o",
    input_variable_format: "F_STRING",
    messages: [
      { role: "SYSTEM", content: "You are a helpful assistant." },
      { role: "USER", content: "Answer: {question}" },
    ],
  },
});

const run = await triggerTaskRun({ task: task.id });
const finalRun = await waitForTaskRun({ runId: run.id });

import { getTask } from "@arizeai/ax-client";

// By task ID
const task = await getTask({ task: "your_task_id" });

// By task name (requires space)
const task = await getTask({ task: "My Task", space: "my-space" });

import { updateTask } from "@arizeai/ax-client";

const updated = await updateTask({
  task: "My Task",     // task name or ID
  space: "my-space",   // required when task is a name
  name: "Renamed Task",
  samplingRate: 0.25,  // project-scoped tasks only
});

import { deleteTask } from "@arizeai/ax-client";

await deleteTask({
  task: "My Task",     // task name or ID
  space: "my-space",   // required when task is a name
});

import { triggerTaskRun } from "@arizeai/ax-client";

const run = await triggerTaskRun({
  task: "My Task",     // task name or ID
  space: "my-space",  // required when task is a name
  dataStartTime: new Date("2026-03-01T00:00:00Z"),
  dataEndTime: new Date("2026-03-08T00:00:00Z"),
  maxSpans: 5000,
  overrideEvaluations: false,  // skip already-evaluated spans
});

import { listTaskRuns } from "@arizeai/ax-client";

const { data: runs, pagination } = await listTaskRuns({
  task: "My Task",     // task name or ID
  space: "my-space",  // required when task is a name
  status: "COMPLETED",  // optional status filter
  limit: 10,
});

for (const run of runs) {
  console.log(run.id, run.numSuccesses, run.numErrors);
}

import { getTaskRun } from "@arizeai/ax-client";

const run = await getTaskRun({ runId: "your_run_id" });
console.log(run.status);       // e.g. "RUNNING"
console.log(run.numSuccesses); // spans evaluated so far

import { triggerTaskRun, waitForTaskRun } from "@arizeai/ax-client";

const run = await triggerTaskRun({ task: "your_task_id" });
const finalRun = await waitForTaskRun({
  runId: run.id,
  pollInterval: 3_000,      // poll every 3 seconds (default: 5000)
  timeout: 5 * 60_000,      // give up after 5 minutes (default: 10 minutes)
});

if (finalRun.status === "COMPLETED") {
  console.log(`${finalRun.numSuccesses} spans evaluated successfully`);
} else {
  console.error(`Run ended with status: ${finalRun.status}`);
}

import { cancelTaskRun } from "@arizeai/ax-client";

const run = await cancelTaskRun({ runId: "your_run_id" });
console.log(run.status); // "CANCELLED"