diff --git a/.github/scripts/triage_issue.py b/.github/scripts/triage_issue.py index 2e6f1975f..167678911 100644 --- a/.github/scripts/triage_issue.py +++ b/.github/scripts/triage_issue.py @@ -18,8 +18,8 @@ import sys def get_gemini_response(api_key, prompt): - # Using the stable Gemini 3.5 Flash - url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-3.5-flash:generateContent?key={api_key}" + model = os.getenv("GEMINI_MODEL", "gemini-3.8-flash") + url = f"https://generativelanguage.googleapis.com/v1beta/models/{model}:generateContent?key={api_key}" headers = {'Content-Type': 'application/json'} data = { "contents": [{ diff --git a/.github/workflows/triage-issue.yml b/.github/workflows/triage-issue.yml index 0a2a4ad9b..cf14e1a8c 100644 --- a/.github/workflows/triage-issue.yml +++ b/.github/workflows/triage-issue.yml @@ -45,6 +45,7 @@ jobs: id: run_script env: GEMINI_API_KEY: ${{ secrets.GEMINI_API_KEY }} + GEMINI_MODEL: ${{ vars.GEMINI_MODEL }} ISSUE_TITLE: ${{ github.event.issue.title || github.event.inputs.title }} ISSUE_BODY: ${{ github.event.issue.body || github.event.inputs.body }} run: | diff --git a/visual-testing/src/main/java/com/google/maps/android/visualtesting/GeminiVisualTestHelper.kt b/visual-testing/src/main/java/com/google/maps/android/visualtesting/GeminiVisualTestHelper.kt index b4c84454f..7228cf23b 100644 --- a/visual-testing/src/main/java/com/google/maps/android/visualtesting/GeminiVisualTestHelper.kt +++ b/visual-testing/src/main/java/com/google/maps/android/visualtesting/GeminiVisualTestHelper.kt @@ -93,9 +93,7 @@ class GeminiVisualTestHelper { val fullPrompt = "$systemPrompt\n\nCommand: \"$prompt\"\n\nUI Hierarchy:\n$hierarchyXml" - // Use a simpler text-only model for this task as no image is involved. - // The user mentioned "gemini-flash 2.5" works, but we should use a standard name. - val modelName = "gemini-2.5-flash-lite" + val modelName = System.getenv("GEMINI_MODEL") ?: "gemini-3.8-flash" val request = GeminiRequest(contents = listOf(Content(parts = listOf(Part(text = fullPrompt))))) var attempts = 0 @@ -229,7 +227,8 @@ class GeminiVisualTestHelper { var response: HttpResponse? = null while (attempts < 3) { attempts++ - response = client.post("https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-lite:generateContent?key=$apiKey") { + val model = System.getenv("GEMINI_MODEL") ?: "gemini-3.8-flash" + response = client.post("https://generativelanguage.googleapis.com/v1beta/models/$model:generateContent?key=$apiKey") { contentType(ContentType.Application.Json) setBody(request) }