diff --git a/anansyTypes/enums.py b/anansyTypes/enums.py index d5693b1..d288713 100644 --- a/anansyTypes/enums.py +++ b/anansyTypes/enums.py @@ -1,21 +1,33 @@ from enum import Enum +from typing import Union + + 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" - + GENERIC = "generic" + + +ModelName= Union[EmbeddingModelName, TranscriptionModelName, ResponseModelName] + + class PreprocessorType(Enum): NORWEGIAN = "norwegian" DEFAULT = "norwegian" + + class SymbolProvider(Enum): TOBII_DYNAVOX = "tobii_dynavox" @@ -24,4 +36,3 @@ class Language(Enum): NORWEGIAN = "norwegian" ENGLISH = "english" DEFAULT = "norwegian" -