From d13d3bd1a0d26d6c5494f09cd6dec5a071006dd7 Mon Sep 17 00:00:00 2001 From: Guillaume Aquilina Date: Mon, 7 Apr 2025 09:09:14 +0200 Subject: [PATCH] feat: add missing models --- workflowai/core/domain/model.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/workflowai/core/domain/model.py b/workflowai/core/domain/model.py index eab529b..703ecce 100644 --- a/workflowai/core/domain/model.py +++ b/workflowai/core/domain/model.py @@ -47,6 +47,8 @@ class Model(str, Enum): # -------------------------------------------------------------------------- # Gemini Models # -------------------------------------------------------------------------- + GEMINI_2_5_PRO_PREVIEW_0325 = "gemini-2.5-pro-preview-03-25" + GEMINI_2_5_PRO_EXP_0325 = "gemini-2.5-pro-exp-03-25" GEMINI_2_0_FLASH_LATEST = "gemini-2.0-flash-latest" GEMINI_2_0_FLASH_001 = "gemini-2.0-flash-001" GEMINI_2_0_FLASH_LITE_001 = "gemini-2.0-flash-lite-001" @@ -88,6 +90,8 @@ class Model(str, Enum): # -------------------------------------------------------------------------- # Llama Models # -------------------------------------------------------------------------- + LLAMA_4_SCOUT_BASIC = "llama4-scout-instruct-basic" + LLAMA_4_MAVERICK_BASIC = "llama4-maverick-instruct-basic" LLAMA_3_3_70B = "llama-3.3-70b" LLAMA_3_2_90B = "llama-3.2-90b" LLAMA_3_2_11B = "llama-3.2-11b"