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

import { llmExtractCreateSchema } from "../funcs/llmExtractCreateSchema.js";
import { llmExtractDeleteSchema } from "../funcs/llmExtractDeleteSchema.js";
import { llmExtractExtract } from "../funcs/llmExtractExtract.js";
import { llmExtractExtractWithSchema } from "../funcs/llmExtractExtractWithSchema.js";
import { llmExtractGetSchema } from "../funcs/llmExtractGetSchema.js";
import { llmExtractUpdateSchema } from "../funcs/llmExtractUpdateSchema.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 ExtractT extends ClientSDK {
  /**
   * Extract structured data with inline JSON Schema
   *
   * @remarks
   * Extracts structured data from files using a custom inline JSON schema.
   */
  async extract(
    request: operations.ExtractRequest,
    options?: RequestOptions,
  ): Promise<components.ExtractionResponse> {
    return unwrapAsync(llmExtractExtract(
      this,
      request,
      options,
    ));
  }

  /**
   * Create reusable extraction schema template
   *
   * @remarks
   * Creates and stores a reusable JSON Schema template for structured data extraction.
   */
  async createSchema(
    request: operations.CreateExtractSchemaRequest,
    options?: RequestOptions,
  ): Promise<components.SchemaCreateResponse> {
    return unwrapAsync(llmExtractCreateSchema(
      this,
      request,
      options,
    ));
  }

  /**
   * Get extraction schema by ID
   *
   * @remarks
   * Retrieve a stored extraction schema and its metadata.
   */
  async getSchema(
    request: operations.GetExtractSchemaRequest,
    options?: RequestOptions,
  ): Promise<components.SchemaGetResponse> {
    return unwrapAsync(llmExtractGetSchema(
      this,
      request,
      options,
    ));
  }

  /**
   * Update extraction schema by ID
   *
   * @remarks
   * Update an existing extraction schema and its metadata.
   */
  async updateSchema(
    request: operations.UpdateExtractSchemaRequest,
    options?: RequestOptions,
  ): Promise<components.SchemaUpdateResponse> {
    return unwrapAsync(llmExtractUpdateSchema(
      this,
      request,
      options,
    ));
  }

  /**
   * Delete extraction schema by ID
   *
   * @remarks
   * Remove a stored extraction schema.
   */
  async deleteSchema(
    request: operations.DeleteExtractSchemaRequest,
    options?: RequestOptions,
  ): Promise<components.SchemaDeleteResponse> {
    return unwrapAsync(llmExtractDeleteSchema(
      this,
      request,
      options,
    ));
  }

  /**
   * Extract data using saved schema template
   *
   * @remarks
   * Extracts structured data from files using a previously saved extraction schema template.
   */
  async extractWithSchema(
    request: operations.ExtractWithSchemaRequest,
    options?: RequestOptions,
  ): Promise<components.ExtractionResponse> {
    return unwrapAsync(llmExtractExtractWithSchema(
      this,
      request,
      options,
    ));
  }
}