Merge pull request #98 from WorkflowAI/guillaume/add-mistral-models

Add mistral and openai models

Author: guillaq

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "workflowai"
-version = "0.6.7"
+version = "0.6.8"
 description = "Python SDK for WorkflowAI"
 authors = ["Guillaume Aquilina <[email protected]>"]
 readme = "README.md"

workflowai/core/domain/model.py

@@ -16,6 +16,12 @@ class Model(str, Enum):
     # OpenAI Models
     # --------------------------------------------------------------------------
 
+    GPT_41_LATEST = "gpt-4.1-latest"
+    GPT_41_2025_04_14 = "gpt-4.1-2025-04-14"
+    GPT_41_MINI_LATEST = "gpt-4.1-mini-latest"
+    GPT_41_MINI_2025_04_14 = "gpt-4.1-mini-2025-04-14"
+    GPT_41_NANO_LATEST = "gpt-4.1-nano-latest"
+    GPT_41_NANO_2025_04_14 = "gpt-4.1-nano-2025-04-14"
     GPT_4O_LATEST = "gpt-4o-latest"
     GPT_4O_2024_11_20 = "gpt-4o-2024-11-20"
     GPT_4O_2024_08_06 = "gpt-4o-2024-08-06"
@@ -122,7 +128,12 @@ class Model(str, Enum):
     PIXTRAL_12B_2409 = "pixtral-12b-2409"
     MINISTRAL_3B_2410 = "ministral-3b-2410"
     MINISTRAL_8B_2410 = "ministral-8b-2410"
+    MISTRAL_SMALL_LATEST = "mistral-small-latest"
+    MISTRAL_SMALL_2503 = "mistral-small-2503"
+    MISTRAL_SMALL_2501 = "mistral-small-2501"
     MISTRAL_SMALL_2409 = "mistral-small-2409"
+    MISTRAL_SABA_2502 = "mistral-saba-2502"
+    CODESTRAL_2501 = "codestral-2501"
     CODESTRAL_MAMBA_2407 = "codestral-mamba-2407"
 
     # --------------------------------------------------------------------------