From 6be2ef3fc68e411a5c62a8cf99fe529762cee92b Mon Sep 17 00:00:00 2001 From: pedrohsdb Date: Fri, 9 Jan 2026 12:06:19 -0800 Subject: [PATCH] preserve preview suffix in vertex cache model name (#4426) --- skyvern/forge/agent.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/skyvern/forge/agent.py b/skyvern/forge/agent.py index 76c34126..550cf341 100644 --- a/skyvern/forge/agent.py +++ b/skyvern/forge/agent.py @@ -2699,8 +2699,11 @@ class ForgeAgent: except Exception as e: LOG.debug("Failed to extract model name from config, using default", error=str(e)) - # Normalize model name to the canonical Vertex identifier (e.g., gemini-2.5-pro) - match = re.search(r"(gemini-\d+(?:\.\d+)?-(?:flash-lite|flash|pro))", model_name, re.IGNORECASE) + # Normalize model name to the canonical Vertex identifier (e.g., gemini-2.5-pro). + # Preserve preview suffixes so we don't strip required identifiers (e.g., gemini-3-flash-preview). + match = re.search( + r"(gemini-\d+(?:\.\d+)?-(?:flash-lite|flash|pro)(?:-preview)?)", model_name, re.IGNORECASE + ) if match: model_name = match.group(1).lower()