From 95b3d1b5baf0ebd129017748a90d7851ab664dc4 Mon Sep 17 00:00:00 2001
From: Guillaume Aquilina <guillaume@workflowai.com>
Date: Tue, 15 Apr 2025 17:56:51 +0200
Subject: [PATCH] feat: add mistral and openai models

---
 pyproject.toml                  |  2 +-
 workflowai/core/domain/model.py | 11 +++++++++++
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 64d2952..fa162a3 100644
--- a/pyproject.toml
+++ b/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 <guillaume@workflowai.com>"]
 readme = "README.md"
diff --git a/workflowai/core/domain/model.py b/workflowai/core/domain/model.py
index 2c521b1..efa7576 100644
--- a/workflowai/core/domain/model.py
+++ b/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"
 
     # --------------------------------------------------------------------------