From d09008827bd5e887128da419337fe5d8655541b1 Mon Sep 17 00:00:00 2001
From: stuartgo <ottersenstuart@gmail.com>
Date: Fri, 12 Dec 2025 17:48:25 +0100
Subject: [PATCH] added modelName enums

---
 anansyTypes/enums.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

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"