import { Adaline } from '@adaline/client';

const adaline = new Adaline();
const datasets = adaline.datasets; // DatasetsClient

import { DatasetsClient } from '@adaline/client';

import type {
  Dataset,
  DatasetSummary,
  CreateDatasetRequest,
  UpdateDatasetRequest,
  ListDatasetsResponse,
  SortOrder,
} from '@adaline/api';

list(options: {
  projectId: string;
  sort?: SortOrder;
  createdAfter?: number;
  createdBefore?: number;
  limit?: number;
  cursor?: string;
}): Promise<ListDatasetsResponse>

const { data, pagination } = await adaline.datasets.list({
  projectId: 'project_abc123',
  sort: 'createdAt:desc',
  limit: 50,
});

for (const dataset of data) {
  console.log(dataset.id, dataset.title);
}

if (pagination.hasMore) {
  const next = await adaline.datasets.list({
    projectId: 'project_abc123',
    cursor: pagination.nextCursor,
  });
}

create(options: { dataset: CreateDatasetRequest }): Promise<Dataset>

interface CreateDatasetRequest {
  projectId: string;
  title: string;
  icon?: BaseEntityIcon;
  columns?: Array<{
    name: string;
    type: 'input' | 'output' | 'metadata' | /* dynamic types */;
    settings?: Record<string, unknown>;
  }>;
}

const dataset = await adaline.datasets.create({
  dataset: {
    projectId: 'project_abc123',
    title: 'Support triage eval set',
    icon: { type: 'emoji', value: '📚' },
    columns: [
      { name: 'question', type: 'input' },
      { name: 'expected', type: 'input' },
    ],
  },
});

console.log(`Created ${dataset.id} with ${dataset.columns.length} columns`);

get(options: { datasetId: string }): Promise<Dataset>

const dataset = await adaline.datasets.get({ datasetId: 'dataset_abc123' });

for (const column of dataset.columns) {
  console.log(column.name, column.type);
}

update(options: {
  datasetId: string;
  dataset: UpdateDatasetRequest;
}): Promise<DatasetSummary>

await adaline.datasets.update({
  datasetId: 'dataset_abc123',
  dataset: { title: 'Renamed dataset' },
});

delete(options: { datasetId: string }): Promise<void>

await adaline.datasets.delete({ datasetId: 'dataset_abc123' });