From 8f0291d438cf4a45cd8d678d1da1344d639b6764 Mon Sep 17 00:00:00 2001
From: Claude <noreply@anthropic.com>
Date: Sat, 30 May 2026 01:24:57 +0000
Subject: [PATCH] Add Gemini 3.5 Flash to TEE_LLM enum

Register google/gemini-3.5-flash in the TEE_LLM enum and refresh the
Google model list in the README.
---
 README.md                 | 4 +++-
 src/opengradient/types.py | 1 +
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 4937b64..9d9b9cc 100644
--- a/README.md
+++ b/README.md
@@ -217,8 +217,10 @@ The SDK provides access to models from multiple providers via the `og.TEE_LLM` e
 - Gemini 2.5 Flash
 - Gemini 2.5 Pro
 - Gemini 2.5 Flash Lite
-- Gemini 3 Pro
 - Gemini 3 Flash
+- Gemini 3.1 Pro
+- Gemini 3.1 Flash Lite
+- Gemini 3.5 Flash
 
 #### xAI
 - Grok 4
diff --git a/src/opengradient/types.py b/src/opengradient/types.py
index 38e54e1..d9e72c1 100644
--- a/src/opengradient/types.py
+++ b/src/opengradient/types.py
@@ -549,6 +549,7 @@ class TEE_LLM(str, Enum):
     GEMINI_3_FLASH = "google/gemini-3-flash-preview"
     GEMINI_3_1_PRO_PREVIEW = "google/gemini-3.1-pro-preview"
     GEMINI_3_1_FLASH_LITE_PREVIEW = "google/gemini-3.1-flash-lite-preview"
+    GEMINI_3_5_FLASH = "google/gemini-3.5-flash"
 
     # xAI Grok models via TEE
     GROK_4 = "x-ai/grok-4"