Skip to content
This repository was archived by the owner on Dec 11, 2025. It is now read-only.

Commit bc32d0d

Browse files
committed
chore: fix missing gemini model
1 parent 391b25e commit bc32d0d

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

notdiamond/llms/providers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -123,8 +123,8 @@ class NDLLMProviders(Enum):
123123
GEMINI_15_PRO_LATEST = ("google", "gemini-1.5-pro-latest")
124124
GEMINI_15_PRO_EXP_0801 = ("google", "gemini-1.5-pro-exp-0801")
125125
GEMINI_15_FLASH_LATEST = ("google", "gemini-1.5-flash-latest")
126-
# GEMINI_20_FLASH = ("google", "gemini-20-flash")
127-
# GEMINI_20_FLASH_001 = ("google", "gemini-20-flash-001")
126+
GEMINI_20_FLASH = ("google", "gemini-2.0-flash")
127+
GEMINI_20_FLASH_001 = ("google", "gemini-2.0-flash-001")
128128

129129
COMMAND_R = ("cohere", "command-r")
130130
COMMAND_R_PLUS = ("cohere", "command-r-plus")

0 commit comments

Comments
 (0)