feat: Add GEMINI_MODEL support (#214)

vivekkairi · Vivek Kairi · web-flow · commit 796d21938533 · 2025-08-21T13:33:05.000Z
Co-authored-by: Vivek Kairi &lt;vivek@zerodhafundhouse.com&gt;
diff --git a/.github/workflows/gemini-cli.yml b/.github/workflows/gemini-cli.yml
@@ -128,6 +128,7 @@ jobs:
           google_api_key: '${{ secrets.GOOGLE_API_KEY }}'
           use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'
           gemini_debug: '${{ fromJSON(vars.DEBUG || vars.ACTIONS_STEP_DEBUG || false) }}'
+          gemini_model: '${{ vars.GEMINI_MODEL }}'
           settings: |-
             {
               "maxSessionTurns": 50,
diff --git a/.github/workflows/gemini-issue-automated-triage.yml b/.github/workflows/gemini-issue-automated-triage.yml
@@ -90,6 +90,7 @@ jobs:
           google_api_key: '${{ secrets.GOOGLE_API_KEY }}'
           use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'
           gemini_debug: '${{ fromJSON(vars.DEBUG || vars.ACTIONS_STEP_DEBUG || false) }}'
+          gemini_model: '${{ vars.GEMINI_MODEL }}'
           settings: |-
             {
               "maxSessionTurns": 25,
diff --git a/.github/workflows/gemini-issue-scheduled-triage.yml b/.github/workflows/gemini-issue-scheduled-triage.yml
@@ -98,6 +98,7 @@ jobs:
           google_api_key: '${{ secrets.GOOGLE_API_KEY }}'
           use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'
           gemini_debug: '${{ fromJSON(vars.DEBUG || vars.ACTIONS_STEP_DEBUG || false) }}'
+          gemini_model: '${{ vars.GEMINI_MODEL }}'
           settings: |-
             {
               "maxSessionTurns": 25,
diff --git a/.github/workflows/gemini-pr-review.yml b/.github/workflows/gemini-pr-review.yml
@@ -172,6 +172,7 @@ jobs:
           google_api_key: '${{ secrets.GOOGLE_API_KEY }}'
           use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'
           gemini_debug: '${{ fromJSON(vars.DEBUG || vars.ACTIONS_STEP_DEBUG || false) }}'
+          gemini_model: '${{ vars.GEMINI_MODEL }}'
           settings: |-
             {
               "maxSessionTurns": 20,
diff --git a/README.md b/README.md
@@ -165,6 +165,10 @@ go to the [Gemini CLI workflow documentation](./examples/workflows/gemini-cli).
 
 -   <a name="google_api_key"></a><a href="#user-content-google_api_key"><code>google_api_key</code></a>: _(Optional)_ The Vertex AI API key to use with Gemini.
 
+-   <a name="gemini_debug"></a><a href="#user-content-gemini_debug"><code>gemini_debug</code></a>: _(Optional)_ Enable debug logging and output streaming.
+
+-   <a name="gemini_model"></a><a href="#user-content-gemini_model"><code>gemini_model</code></a>: _(Optional)_ The model to use with Gemini.
+
 
 <!-- END_AUTOGEN_INPUTS -->
 
diff --git a/action.yml b/action.yml
@@ -61,6 +61,9 @@ inputs:
   gemini_debug:
     description: 'Enable debug logging and output streaming.'
     required: false
+  gemini_model:
+    description: 'The model to use with Gemini.'
+    required: false
 
 outputs:
   summary:
@@ -211,6 +214,7 @@ runs:
         GOOGLE_GENAI_USE_GCA: '${{ inputs.use_gemini_code_assist }}'
         GOOGLE_CLOUD_ACCESS_TOKEN: '${{steps.auth.outputs.access_token}}'
         PROMPT: '${{ inputs.prompt }}'
+        GEMINI_MODEL: '${{ inputs.gemini_model }}'
 
 branding:
   icon: 'terminal'
diff --git a/examples/workflows/gemini-cli/gemini-cli.yml b/examples/workflows/gemini-cli/gemini-cli.yml
@@ -214,6 +214,7 @@ jobs:
           use_vertex_ai: '${{ vars.GOOGLE_GENAI_USE_VERTEXAI }}'
           use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'
           gemini_debug: '${{ fromJSON(vars.DEBUG || vars.ACTIONS_STEP_DEBUG || false) }}'
+          gemini_model: '${{ vars.GEMINI_MODEL }}'
           settings: |-
             {
               "maxSessionTurns": 50,
diff --git a/examples/workflows/issue-triage/gemini-issue-automated-triage.yml b/examples/workflows/issue-triage/gemini-issue-automated-triage.yml
@@ -89,6 +89,7 @@ jobs:
           use_vertex_ai: '${{ vars.GOOGLE_GENAI_USE_VERTEXAI }}'
           use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'
           gemini_debug: '${{ fromJSON(vars.DEBUG || vars.ACTIONS_STEP_DEBUG || false) }}'
+          gemini_model: '${{ vars.GEMINI_MODEL }}'
           settings: |-
             {
               "maxSessionTurns": 25,
diff --git a/examples/workflows/issue-triage/gemini-issue-scheduled-triage.yml b/examples/workflows/issue-triage/gemini-issue-scheduled-triage.yml
@@ -97,6 +97,7 @@ jobs:
           use_vertex_ai: '${{ vars.GOOGLE_GENAI_USE_VERTEXAI }}'
           use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'
           gemini_debug: '${{ fromJSON(vars.DEBUG || vars.ACTIONS_STEP_DEBUG || false) }}'
+          gemini_model: '${{ vars.GEMINI_MODEL }}'
           settings: |-
             {
               "maxSessionTurns": 25,
diff --git a/examples/workflows/pr-review/gemini-pr-review.yml b/examples/workflows/pr-review/gemini-pr-review.yml
@@ -171,6 +171,7 @@ jobs:
           use_vertex_ai: '${{ vars.GOOGLE_GENAI_USE_VERTEXAI }}'
           use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'
           gemini_debug: '${{ fromJSON(vars.DEBUG || vars.ACTIONS_STEP_DEBUG || false) }}'
+          gemini_model: '${{ vars.GEMINI_MODEL }}'
           settings: |-
             {
               "maxSessionTurns": 20,

Original file line number	Diff line number	Diff line change
`@@ -128,6 +128,7 @@ jobs:`
`128`	`128`	`google_api_key: '${{ secrets.GOOGLE_API_KEY }}'`
`129`	`129`	`use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'`
`130`	`130`	`gemini_debug: '${{ fromJSON(vars.DEBUG \|\| vars.ACTIONS_STEP_DEBUG \|\| false) }}'`
	`131`	`+ gemini_model: '${{ vars.GEMINI_MODEL }}'`
`131`	`132`	`settings: \|-`
`132`	`133`	`{`
`133`	`134`	`"maxSessionTurns": 50,`
Original file line number	Diff line number	Diff line change
`@@ -90,6 +90,7 @@ jobs:`
`90`	`90`	`google_api_key: '${{ secrets.GOOGLE_API_KEY }}'`
`91`	`91`	`use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'`
`92`	`92`	`gemini_debug: '${{ fromJSON(vars.DEBUG \|\| vars.ACTIONS_STEP_DEBUG \|\| false) }}'`
	`93`	`+ gemini_model: '${{ vars.GEMINI_MODEL }}'`
`93`	`94`	`settings: \|-`
`94`	`95`	`{`
`95`	`96`	`"maxSessionTurns": 25,`
Original file line number	Diff line number	Diff line change
`@@ -98,6 +98,7 @@ jobs:`
`98`	`98`	`google_api_key: '${{ secrets.GOOGLE_API_KEY }}'`
`99`	`99`	`use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'`
`100`	`100`	`gemini_debug: '${{ fromJSON(vars.DEBUG \|\| vars.ACTIONS_STEP_DEBUG \|\| false) }}'`
	`101`	`+ gemini_model: '${{ vars.GEMINI_MODEL }}'`
`101`	`102`	`settings: \|-`
`102`	`103`	`{`
`103`	`104`	`"maxSessionTurns": 25,`
Original file line number	Diff line number	Diff line change
`@@ -172,6 +172,7 @@ jobs:`
`172`	`172`	`google_api_key: '${{ secrets.GOOGLE_API_KEY }}'`
`173`	`173`	`use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'`
`174`	`174`	`gemini_debug: '${{ fromJSON(vars.DEBUG \|\| vars.ACTIONS_STEP_DEBUG \|\| false) }}'`
	`175`	`+ gemini_model: '${{ vars.GEMINI_MODEL }}'`
`175`	`176`	`settings: \|-`
`176`	`177`	`{`
`177`	`178`	`"maxSessionTurns": 20,`
Original file line number	Diff line number	Diff line change
`@@ -214,6 +214,7 @@ jobs:`
`214`	`214`	`use_vertex_ai: '${{ vars.GOOGLE_GENAI_USE_VERTEXAI }}'`
`215`	`215`	`use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'`
`216`	`216`	`gemini_debug: '${{ fromJSON(vars.DEBUG \|\| vars.ACTIONS_STEP_DEBUG \|\| false) }}'`
	`217`	`+ gemini_model: '${{ vars.GEMINI_MODEL }}'`
`217`	`218`	`settings: \|-`
`218`	`219`	`{`
`219`	`220`	`"maxSessionTurns": 50,`
Original file line number	Diff line number	Diff line change
`@@ -89,6 +89,7 @@ jobs:`
`89`	`89`	`use_vertex_ai: '${{ vars.GOOGLE_GENAI_USE_VERTEXAI }}'`
`90`	`90`	`use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'`
`91`	`91`	`gemini_debug: '${{ fromJSON(vars.DEBUG \|\| vars.ACTIONS_STEP_DEBUG \|\| false) }}'`
	`92`	`+ gemini_model: '${{ vars.GEMINI_MODEL }}'`
`92`	`93`	`settings: \|-`
`93`	`94`	`{`
`94`	`95`	`"maxSessionTurns": 25,`
Original file line number	Diff line number	Diff line change
`@@ -97,6 +97,7 @@ jobs:`
`97`	`97`	`use_vertex_ai: '${{ vars.GOOGLE_GENAI_USE_VERTEXAI }}'`
`98`	`98`	`use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'`
`99`	`99`	`gemini_debug: '${{ fromJSON(vars.DEBUG \|\| vars.ACTIONS_STEP_DEBUG \|\| false) }}'`
	`100`	`+ gemini_model: '${{ vars.GEMINI_MODEL }}'`
`100`	`101`	`settings: \|-`
`101`	`102`	`{`
`102`	`103`	`"maxSessionTurns": 25,`
Original file line number	Diff line number	Diff line change
`@@ -171,6 +171,7 @@ jobs:`
`171`	`171`	`use_vertex_ai: '${{ vars.GOOGLE_GENAI_USE_VERTEXAI }}'`
`172`	`172`	`use_gemini_code_assist: '${{ vars.GOOGLE_GENAI_USE_GCA }}'`
`173`	`173`	`gemini_debug: '${{ fromJSON(vars.DEBUG \|\| vars.ACTIONS_STEP_DEBUG \|\| false) }}'`
	`174`	`+ gemini_model: '${{ vars.GEMINI_MODEL }}'`
`174`	`175`	`settings: \|-`
`175`	`176`	`{`
`176`	`177`	`"maxSessionTurns": 20,`