diff --git a/anansyTypes/enums.py b/anansyTypes/enums.py index 0121ef0..d5693b1 100644 --- a/anansyTypes/enums.py +++ b/anansyTypes/enums.py @@ -3,6 +3,16 @@ class ModelType(Enum): EMBEDDING = "embedding" TRANSCRIPTION = "transcription" RESPONSE = "response" + +class EmbeddingModelName(Enum): + NB_BERT = "nb_bert" + +class TranscriptionModelName(Enum): + WHISPER_SMALL = "whisper_small" + +class ResponseModelName(Enum): + GENERIC = "generic" + class PreprocessorType(Enum): NORWEGIAN = "norwegian" DEFAULT = "norwegian"