/*
 * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
 * @generated-id: 2f1e89a21914
 */

import { betaObservabilityDatasetsCreate } from "../funcs/betaObservabilityDatasetsCreate.js";
import { betaObservabilityDatasetsCreateRecord } from "../funcs/betaObservabilityDatasetsCreateRecord.js";
import { betaObservabilityDatasetsDelete } from "../funcs/betaObservabilityDatasetsDelete.js";
import { betaObservabilityDatasetsExportToJsonl } from "../funcs/betaObservabilityDatasetsExportToJsonl.js";
import { betaObservabilityDatasetsFetch } from "../funcs/betaObservabilityDatasetsFetch.js";
import { betaObservabilityDatasetsFetchTask } from "../funcs/betaObservabilityDatasetsFetchTask.js";
import { betaObservabilityDatasetsImportFromCampaign } from "../funcs/betaObservabilityDatasetsImportFromCampaign.js";
import { betaObservabilityDatasetsImportFromDatasetRecords } from "../funcs/betaObservabilityDatasetsImportFromDatasetRecords.js";
import { betaObservabilityDatasetsImportFromExplorer } from "../funcs/betaObservabilityDatasetsImportFromExplorer.js";
import { betaObservabilityDatasetsImportFromFile } from "../funcs/betaObservabilityDatasetsImportFromFile.js";
import { betaObservabilityDatasetsImportFromPlayground } from "../funcs/betaObservabilityDatasetsImportFromPlayground.js";
import { betaObservabilityDatasetsList } from "../funcs/betaObservabilityDatasetsList.js";
import { betaObservabilityDatasetsListRecords } from "../funcs/betaObservabilityDatasetsListRecords.js";
import { betaObservabilityDatasetsListTasks } from "../funcs/betaObservabilityDatasetsListTasks.js";
import { betaObservabilityDatasetsUpdate } from "../funcs/betaObservabilityDatasetsUpdate.js";
import { ClientSDK, RequestOptions } from "../lib/sdks.js";
import * as components from "../models/components/index.js";
import * as operations from "../models/operations/index.js";
import { unwrapAsync } from "../types/fp.js";
import { Records } from "./records.js";

export class Datasets extends ClientSDK {
  private _records?: Records;
  get records(): Records {
    return (this._records ??= new Records(this._options));
  }

  /**
   * Create a new empty dataset
   */
  async create(
    request: components.CreateDatasetRequest,
    options?: RequestOptions,
  ): Promise<components.Dataset> {
    return unwrapAsync(betaObservabilityDatasetsCreate(
      this,
      request,
      options,
    ));
  }

  /**
   * List existing datasets
   */
  async list(
    request?:
      | operations.GetDatasetsV1ObservabilityDatasetsGetRequest
      | undefined,
    options?: RequestOptions,
  ): Promise<components.ListDatasetsResponse> {
    return unwrapAsync(betaObservabilityDatasetsList(
      this,
      request,
      options,
    ));
  }

  /**
   * Get dataset by id
   */
  async fetch(
    request:
      operations.GetDatasetByIdV1ObservabilityDatasetsDatasetIdGetRequest,
    options?: RequestOptions,
  ): Promise<components.DatasetPreview> {
    return unwrapAsync(betaObservabilityDatasetsFetch(
      this,
      request,
      options,
    ));
  }

  /**
   * Delete a dataset
   */
  async delete(
    request:
      operations.DeleteDatasetV1ObservabilityDatasetsDatasetIdDeleteRequest,
    options?: RequestOptions,
  ): Promise<void> {
    return unwrapAsync(betaObservabilityDatasetsDelete(
      this,
      request,
      options,
    ));
  }

  /**
   * Patch dataset
   */
  async update(
    request:
      operations.UpdateDatasetV1ObservabilityDatasetsDatasetIdPatchRequest,
    options?: RequestOptions,
  ): Promise<components.DatasetPreview> {
    return unwrapAsync(betaObservabilityDatasetsUpdate(
      this,
      request,
      options,
    ));
  }

  /**
   * List existing records in the dataset
   */
  async listRecords(
    request:
      operations.GetDatasetRecordsV1ObservabilityDatasetsDatasetIdRecordsGetRequest,
    options?: RequestOptions,
  ): Promise<components.ListDatasetRecordsResponse> {
    return unwrapAsync(betaObservabilityDatasetsListRecords(
      this,
      request,
      options,
    ));
  }

  /**
   * Add a conversation to the dataset
   */
  async createRecord(
    request:
      operations.CreateDatasetRecordV1ObservabilityDatasetsDatasetIdRecordsPostRequest,
    options?: RequestOptions,
  ): Promise<components.DatasetRecord> {
    return unwrapAsync(betaObservabilityDatasetsCreateRecord(
      this,
      request,
      options,
    ));
  }

  /**
   * Populate the dataset with a campaign
   */
  async importFromCampaign(
    request:
      operations.PostDatasetRecordsFromCampaignV1ObservabilityDatasetsDatasetIdImportsFromCampaignPostRequest,
    options?: RequestOptions,
  ): Promise<components.DatasetImportTask> {
    return unwrapAsync(betaObservabilityDatasetsImportFromCampaign(
      this,
      request,
      options,
    ));
  }

  /**
   * Populate the dataset with samples from the explorer
   */
  async importFromExplorer(
    request:
      operations.PostDatasetRecordsFromExplorerV1ObservabilityDatasetsDatasetIdImportsFromExplorerPostRequest,
    options?: RequestOptions,
  ): Promise<components.DatasetImportTask> {
    return unwrapAsync(betaObservabilityDatasetsImportFromExplorer(
      this,
      request,
      options,
    ));
  }

  /**
   * Populate the dataset with samples from an uploaded file
   */
  async importFromFile(
    request:
      operations.PostDatasetRecordsFromFileV1ObservabilityDatasetsDatasetIdImportsFromFilePostRequest,
    options?: RequestOptions,
  ): Promise<components.DatasetImportTask> {
    return unwrapAsync(betaObservabilityDatasetsImportFromFile(
      this,
      request,
      options,
    ));
  }

  /**
   * Populate the dataset with samples from the playground
   */
  async importFromPlayground(
    request:
      operations.PostDatasetRecordsFromPlaygroundV1ObservabilityDatasetsDatasetIdImportsFromPlaygroundPostRequest,
    options?: RequestOptions,
  ): Promise<components.DatasetImportTask> {
    return unwrapAsync(betaObservabilityDatasetsImportFromPlayground(
      this,
      request,
      options,
    ));
  }

  /**
   * Populate the dataset with samples from another dataset
   */
  async importFromDatasetRecords(
    request:
      operations.PostDatasetRecordsFromDatasetV1ObservabilityDatasetsDatasetIdImportsFromDatasetPostRequest,
    options?: RequestOptions,
  ): Promise<components.DatasetImportTask> {
    return unwrapAsync(betaObservabilityDatasetsImportFromDatasetRecords(
      this,
      request,
      options,
    ));
  }

  /**
   * Export to the Files API and retrieve presigned URL to download the resulting JSONL file
   */
  async exportToJsonl(
    request:
      operations.ExportDatasetToJsonlV1ObservabilityDatasetsDatasetIdExportsToJsonlGetRequest,
    options?: RequestOptions,
  ): Promise<components.ExportDatasetResponse> {
    return unwrapAsync(betaObservabilityDatasetsExportToJsonl(
      this,
      request,
      options,
    ));
  }

  /**
   * Get status of a dataset import task
   */
  async fetchTask(
    request:
      operations.GetDatasetImportTaskV1ObservabilityDatasetsDatasetIdTasksTaskIdGetRequest,
    options?: RequestOptions,
  ): Promise<components.DatasetImportTask> {
    return unwrapAsync(betaObservabilityDatasetsFetchTask(
      this,
      request,
      options,
    ));
  }

  /**
   * List import tasks for the given dataset
   */
  async listTasks(
    request:
      operations.GetDatasetImportTasksV1ObservabilityDatasetsDatasetIdTasksGetRequest,
    options?: RequestOptions,
  ): Promise<components.ListDatasetImportTasksResponse> {
    return unwrapAsync(betaObservabilityDatasetsListTasks(
      this,
      request,
      options,
    ));
  }
}