added modelName enums

This commit is contained in:
stuart committed 2025-12-12 17:48:25 +01:00
1 parent 7126a66331
commit d09008827b
1 file changed
+10
+10
View File
@@ -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"