models + refactor

2026-02-27 20:37:24 +03:00
parent d9caececd7
commit e011805186
31 changed files with 234 additions and 223 deletions
--- a/models/enums.py
+++ b/models/enums.py
@@ -52,3 +52,20 @@ class GenType(str, Enum):
            GenType.TEXT: 'Text',
            GenType.IMAGE: 'Image',
        }[self]
+
+
+class TextModel(str, Enum):
+    GEMINI_3_1_PRO_PREVIEW = "gemini-3.1-pro-preview"
+
+    @property
+    def value_model(self) -> str:
+        return self.value
+
+
+class ImageModel(str, Enum):
+    GEMINI_3_PRO_IMAGE_PREVIEW = "gemini-3-pro-image-preview"
+    GEMINI_3_1_FLASH_IMAGE_PREVIEW = "gemini-3.1-flash-image-preview"
+
+    @property
+    def value_model(self) -> str:
+        return self.value