Skip to content

Commit 4859b96

Browse files
authored
fix: use valid model name for Vertex AI connectivity check (#919)
The `_test_vertex_connection` function used `gemini-3-flash-preview` which does not exist, causing the countTokens API to return 404 and the connectivity check to always fail. Replace with `gemini-2.5-flash`.
1 parent 12a447f commit 4859b96

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • agent_starter_pack/cli/utils

agent_starter_pack/cli/utils/gcp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def _test_vertex_connection(
120120

121121
try:
122122
response = requests.post(
123-
f"https://us-central1-aiplatform.googleapis.com/v1beta1/projects/{project}/locations/global/publishers/google/models/gemini-3-flash-preview:countTokens",
123+
f"https://us-central1-aiplatform.googleapis.com/v1beta1/projects/{project}/locations/global/publishers/google/models/gemini-2.5-flash:countTokens",
124124
headers={
125125
"Authorization": f"Bearer {token}",
126126
"Content-Type": "application/json",

0 commit comments

Comments
 (0)