Package io.debezium.ai.embeddings

Class HuggingFaceModelFactory<R extends org.apache.kafka.connect.connector.ConnectRecord<R>>

java.lang.Object
io.debezium.ai.embeddings.HuggingFaceModelFactory<R>
All Implemented Interfaces:
EmbeddingsModelFactory
public class HuggingFaceModelFactory<R extends org.apache.kafka.connect.connector.ConnectRecord<R>> extends Object implements EmbeddingsModelFactory
EmbeddingsModelFactory for FieldToEmbedding SMT, which uses Hugging Face serverless inference for serving embedding models.
Author:
vjuranek

  • Field Details

    • DEFAULT_OPERATION_TIMEOUT

      private static final int DEFAULT_OPERATION_TIMEOUT
      See Also:

    • HUGGING_FACE_PREFIX

      public static final String HUGGING_FACE_PREFIX
      See Also:

    • HUGGING_FACE_BASE_URL

      private static final Field HUGGING_FACE_BASE_URL

    • HUGGING_FACE_ACCESS_TOKEN

      private static final Field HUGGING_FACE_ACCESS_TOKEN

    • MODEL_NAME

      private static final Field MODEL_NAME

    • OPERATION_TIMEOUT

      private static final Field OPERATION_TIMEOUT

    • ALL_FIELDS

      public static final Field.Set ALL_FIELDS

    • baseUrl

      private String baseUrl

    • accessToken

      private String accessToken

    • modelName

      private String modelName

    • operationTimeout

      private int operationTimeout

  • Constructor Details

    • HuggingFaceModelFactory

      public HuggingFaceModelFactory()

  • Method Details