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

import { llmEmbeddingsEmbed } from "../funcs/llmEmbeddingsEmbed.js";
import { llmEmbeddingsListModels } from "../funcs/llmEmbeddingsListModels.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 Embeddings extends ClientSDK {
  /**
   * List available embedding models
   *
   * @remarks
   * Returns a list of available embedding models with their limits. Use this endpoint to discover which models are available and their constraints (batch size, input length) before making embedding requests.
   */
  async listModels(
    request: operations.ListEmbeddingModelsRequest,
    options?: RequestOptions,
  ): Promise<components.EmbeddingModelListResponse> {
    return unwrapAsync(llmEmbeddingsListModels(
      this,
      request,
      options,
    ));
  }

  /**
   * Create text embeddings
   *
   * @remarks
   * Generates vector embeddings for single or multiple text inputs. Returns floating-point vectors along with token usage statistics.
   */
  async embed(
    request: operations.EmbedRequest,
    options?: RequestOptions,
  ): Promise<components.EmbeddingsResponse> {
    return unwrapAsync(llmEmbeddingsEmbed(
      this,
      request,
      options,
    ));
  }
}