MediaPipeTasksText Framework Reference

TextEmbeddingType

enum TextEmbeddingType : UInt, @unchecked Sendable

The embedding task type, used to format input text for Gecko and task-aware embedding models.

• retrievalQuery

  Embedding for a retrieval query.

  Declaration

  Swift

  case retrievalQuery = 1

• retrievalDocument

  Embedding for a document to be indexed.

  Declaration

  Swift

  case retrievalDocument = 2

• semanticSimilarity

  Embedding for semantic similarity comparison.

  Declaration

  Swift

  case semanticSimilarity = 3

• classification

  Embedding for classification.

  Declaration

  Swift

  case classification = 4

• clustering

  Embedding for clustering.

  Declaration

  Swift

  case clustering = 5

• questionAnswering

  Embedding for question answering.

  Declaration

  Swift

  case questionAnswering = 6

• factChecking

  Embedding for fact checking.

  Declaration

  Swift

  case factChecking = 7

• codeRetrieval

  Embedding for code retrieval.

  Declaration

  Swift

  case codeRetrieval = 8