From e2d689a84c72fc5cc9063ae3b4720b48423b2c23 Mon Sep 17 00:00:00 2001
From: Christoph Eder <ce1@live.de>
Date: Sat, 13 Dec 2025 11:53:30 +0100
Subject: [PATCH] Changes in enums.py     Added generalised ModelName

---
 anansyTypes/enums.py | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

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