/*
 * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
 */

import { llmFineTuningCancel } from "../funcs/llmFineTuningCancel.js";
import { llmFineTuningCreate } from "../funcs/llmFineTuningCreate.js";
import { llmFineTuningList } from "../funcs/llmFineTuningList.js";
import { llmFineTuningListCheckpoints } from "../funcs/llmFineTuningListCheckpoints.js";
import { llmFineTuningListEvents } from "../funcs/llmFineTuningListEvents.js";
import { llmFineTuningPause } from "../funcs/llmFineTuningPause.js";
import { llmFineTuningResume } from "../funcs/llmFineTuningResume.js";
import { llmFineTuningRetrieve } from "../funcs/llmFineTuningRetrieve.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";

export class FineTuning extends ClientSDK {
  /**
   * Create a fine-tuning job
   *
   * @remarks
   * Creates a fine-tuning job which begins the process of training a new model from a given dataset.
   */
  async create(
    request: components.CreateFineTuningJobRequest,
    options?: RequestOptions,
  ): Promise<components.FineTuningJob> {
    return unwrapAsync(llmFineTuningCreate(
      this,
      request,
      options,
    ));
  }

  /**
   * List fine-tuning jobs
   *
   * @remarks
   * List your organization's fine-tuning jobs.
   */
  async list(
    request: operations.ListFineTuningJobsRequest,
    options?: RequestOptions,
  ): Promise<components.ListFineTuningJobsResponse> {
    return unwrapAsync(llmFineTuningList(
      this,
      request,
      options,
    ));
  }

  /**
   * Retrieve a fine-tuning job
   *
   * @remarks
   * Get info about a fine-tuning job.
   */
  async retrieve(
    request: operations.RetrieveFineTuningJobRequest,
    options?: RequestOptions,
  ): Promise<components.FineTuningJob> {
    return unwrapAsync(llmFineTuningRetrieve(
      this,
      request,
      options,
    ));
  }

  /**
   * Cancel a fine-tuning job
   *
   * @remarks
   * Immediately cancel a fine-tuning job.
   */
  async cancel(
    request: operations.CancelFineTuningJobRequest,
    options?: RequestOptions,
  ): Promise<components.FineTuningJob> {
    return unwrapAsync(llmFineTuningCancel(
      this,
      request,
      options,
    ));
  }

  /**
   * Pause a fine-tuning job
   *
   * @remarks
   * Pause a running fine-tuning job.
   */
  async pause(
    request: operations.PauseFineTuningJobRequest,
    options?: RequestOptions,
  ): Promise<components.FineTuningJob> {
    return unwrapAsync(llmFineTuningPause(
      this,
      request,
      options,
    ));
  }

  /**
   * Resume a fine-tuning job
   *
   * @remarks
   * Resume a paused fine-tuning job.
   */
  async resume(
    request: operations.ResumeFineTuningJobRequest,
    options?: RequestOptions,
  ): Promise<components.FineTuningJob> {
    return unwrapAsync(llmFineTuningResume(
      this,
      request,
      options,
    ));
  }

  /**
   * List fine-tuning events
   *
   * @remarks
   * Get status updates for a fine-tuning job.
   */
  async listEvents(
    request: operations.ListFineTuningEventsRequest,
    options?: RequestOptions,
  ): Promise<components.ListEventsResponse> {
    return unwrapAsync(llmFineTuningListEvents(
      this,
      request,
      options,
    ));
  }

  /**
   * List fine-tuning checkpoints
   *
   * @remarks
   * List checkpoints for a fine-tuning job.
   */
  async listCheckpoints(
    request: operations.ListFineTuningCheckpointsRequest,
    options?: RequestOptions,
  ): Promise<components.ListCheckpointsResponse> {
    return unwrapAsync(llmFineTuningListCheckpoints(
      this,
      request,
      options,
    ));
  }
}