Changes in enums.py

Added generalised ModelName

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"
-