Skip to content

Commit ae2da5d

Browse files
committed
feat: add gemini 2.0 models
1 parent 4d915b6 commit ae2da5d

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

workflowai/core/domain/model.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,10 @@ class Model(str, Enum):
4343
# --------------------------------------------------------------------------
4444
# Gemini Models
4545
# --------------------------------------------------------------------------
46+
GEMINI_2_0_FLASH_LATEST = "gemini-2.0-flash-latest"
47+
GEMINI_2_0_FLASH_001 = "gemini-2.0-flash-001"
48+
GEMINI_2_0_FLASH_LITE_PREVIEW_2502 = "gemini-2.0-flash-lite-preview-02-05"
49+
GEMINI_2_0_PRO_EXP = "gemini-2.0-pro-exp-02-05"
4650
GEMINI_2_0_FLASH_EXP = "gemini-2.0-flash-exp"
4751
GEMINI_2_0_FLASH_THINKING_EXP_1219 = "gemini-2.0-flash-thinking-exp-1219"
4852
GEMINI_2_0_FLASH_THINKING_EXP_0121 = "gemini-2.0-flash-thinking-exp-01-21"

0 commit comments

Comments
 (0)