Skip to content

Commit d2b2bc7

Browse files
kylexqianclaude
andauthored
feat: add latest Claude, OpenAI, Gemini, and Grok models to TEE_LLM enum (#286)
Adds 13 new models across all four providers: - OpenAI: gpt-4.1-mini, gpt-4.1-nano, o3, gpt-5.4, gpt-5.4-mini, gpt-5.4-nano, gpt-5.5 - Anthropic: claude-opus-4-7 (latest flagship) - Google: gemini-3.1-pro-preview, gemini-3.1-flash-lite-preview (Gemini 2.5 replacements) - xAI: grok-4.20-reasoning, grok-4.20-non-reasoning, grok-code-fast-1 Updates Gemini deprecation comment to reflect replacements are now available. Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 52f778f commit d2b2bc7

1 file changed

Lines changed: 15 additions & 2 deletions

File tree

src/opengradient/types.py

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -516,32 +516,45 @@ class TEE_LLM(str, Enum):
516516

517517
# OpenAI models via TEE
518518
GPT_4_1_2025_04_14 = "openai/gpt-4.1-2025-04-14"
519+
GPT_4_1_MINI = "openai/gpt-4.1-mini"
520+
GPT_4_1_NANO = "openai/gpt-4.1-nano"
521+
O3 = "openai/o3"
519522
O4_MINI = "openai/o4-mini"
520523
GPT_5 = "openai/gpt-5"
521524
GPT_5_MINI = "openai/gpt-5-mini"
522525
GPT_5_2 = "openai/gpt-5.2"
526+
GPT_5_4 = "openai/gpt-5.4"
527+
GPT_5_4_MINI = "openai/gpt-5.4-mini"
528+
GPT_5_4_NANO = "openai/gpt-5.4-nano"
529+
GPT_5_5 = "openai/gpt-5.5"
523530

524531
# Anthropic models via TEE
525532
CLAUDE_SONNET_4_5 = "anthropic/claude-sonnet-4-5"
526533
CLAUDE_SONNET_4_6 = "anthropic/claude-sonnet-4-6"
527534
CLAUDE_HAIKU_4_5 = "anthropic/claude-haiku-4-5"
528535
CLAUDE_OPUS_4_5 = "anthropic/claude-opus-4-5"
529536
CLAUDE_OPUS_4_6 = "anthropic/claude-opus-4-6"
537+
CLAUDE_OPUS_4_7 = "anthropic/claude-opus-4-7"
530538

531539
# Google models via TEE
532540
# Note: gemini-2.5-flash, gemini-2.5-pro, and gemini-2.5-flash-lite are scheduled
533-
# for deprecation on June 17, 2026 (flash-lite: July 22, 2026). Replacements will be
534-
# gemini-3-flash-preview, gemini-3.1-pro-preview, and gemini-3.1-flash-lite-preview.
541+
# for deprecation on June 17, 2026 (flash-lite: July 22, 2026). Use the Gemini 3
542+
# replacements below for new integrations.
535543
GEMINI_2_5_FLASH = "google/gemini-2.5-flash"
536544
GEMINI_2_5_PRO = "google/gemini-2.5-pro"
537545
GEMINI_2_5_FLASH_LITE = "google/gemini-2.5-flash-lite"
538546
GEMINI_3_FLASH = "google/gemini-3-flash-preview"
547+
GEMINI_3_1_PRO_PREVIEW = "google/gemini-3.1-pro-preview"
548+
GEMINI_3_1_FLASH_LITE_PREVIEW = "google/gemini-3.1-flash-lite-preview"
539549

540550
# xAI Grok models via TEE
541551
GROK_4 = "x-ai/grok-4"
542552
GROK_4_FAST = "x-ai/grok-4-fast"
543553
GROK_4_1_FAST = "x-ai/grok-4-1-fast"
544554
GROK_4_1_FAST_NON_REASONING = "x-ai/grok-4-1-fast-non-reasoning"
555+
GROK_4_20_REASONING = "x-ai/grok-4.20-reasoning"
556+
GROK_4_20_NON_REASONING = "x-ai/grok-4.20-non-reasoning"
557+
GROK_CODE_FAST_1 = "x-ai/grok-code-fast-1"
545558

546559

547560
@dataclass

0 commit comments

Comments
 (0)