Consolidate Gemini 3 Pro into version-agnostic VERTEX_GEMINI_3_PRO (#4926)

This commit is contained in:
pedrohsdb 2026-02-27 19:13:23 -08:00 committed by GitHub
parent 2e663beec0
commit 5bf555077f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 18 additions and 20 deletions

View file

@ -18,7 +18,7 @@ block_module = sys.modules["skyvern.forge.sdk.workflow.models.block"]
("model_name", "expected_llm_key"),
[
("gemini-2.5-flash", "VERTEX_GEMINI_2.5_FLASH"),
("gemini-3-pro-preview", "VERTEX_GEMINI_3.0_PRO"),
("gemini-3-pro-preview", "VERTEX_GEMINI_3_PRO"),
],
)
async def test_text_prompt_block_uses_selected_model(monkeypatch, model_name, expected_llm_key):