diff --git a/.github/triage-issue.yml b/.github/triage-issue.yml index 103a2f9..887ad96 100644 --- a/.github/triage-issue.yml +++ b/.github/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/scripts/triage_issue.py b/scripts/triage_issue.py index d9f1c91..b52d306 100644 --- a/scripts/triage_issue.py +++ b/scripts/triage_issue.py @@ -4,8 +4,8 @@ import sys def get_gemini_response(api_key, prompt): -# Using the stable Gemini 2.5 Flash - url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.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": [{