vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

preserve preview suffix in vertex cache model name (#4426)

Browse Source
This commit is contained in:
pedrohsdb
2026-01-09 12:06:19 -08:00
committed by GitHub
parent 37f5d6b8b2
commit 6be2ef3fc6
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
skyvern/forge/agent.py
View File

@@ -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()