refactor(gemini): enhance task prompt handling and streamline start script execution

Author: DevelopmentCats

registry/coder-labs/modules/gemini/main.test.ts

@@ -204,7 +204,9 @@ describe("gemini", async () => {
         task_prompt: taskPrompt,
       },
     });
-    await execModuleScript(id);
+    await execModuleScript(id, {
+      GEMINI_TASK_PROMPT: taskPrompt,
+    });
     const resp = await readFileContainer(id, "/home/coder/.gemini-module/agentapi-start.log");
     expect(resp).toContain("Running automated task:");
   });

registry/coder-labs/modules/gemini/main.tf

@@ -185,12 +185,8 @@ module "agentapi" {
   web_app_group        = var.group
   web_app_icon         = var.icon
   web_app_display_name = "Gemini"
-  cli_app              = true
   cli_app_slug         = "${local.app_slug}-cli"
   cli_app_display_name = "Gemini CLI"
-  cli_app_icon         = var.icon
-  cli_app_order        = var.order
-  cli_app_group        = var.group
   module_dir_name      = local.module_dir_name
   install_agentapi     = var.install_agentapi
   agentapi_version     = var.agentapi_version
@@ -225,7 +221,7 @@ module "agentapi" {
       GEMINI_YOLO_MODE='${var.enable_yolo_mode}' \
      GEMINI_MODEL='${var.gemini_model}' \
      GEMINI_START_DIRECTORY='${var.folder}' \
-     GEMINI_TASK_PROMPT='${base64encode(var.task_prompt)}' \
+     GEMINI_TASK_PROMPT='${var.task_prompt}' \
      /tmp/start.sh
    EOT
 }

registry/coder-labs/modules/gemini/scripts/install.sh

@@ -150,9 +150,15 @@ function add_system_prompt_if_exists() {
     fi
 }
 
+function configure_mcp() {
+    export CODER_MCP_APP_STATUS_SLUG="gemini"
+    export CODER_MCP_AI_AGENTAPI_URL="http://localhost:3284"
+    coder exp mcp configure gemini "${GEMINI_START_DIRECTORY}"
+}
 
 install_gemini
 gemini --version
 populate_settings_json
 add_system_prompt_if_exists
+configure_mcp
 

registry/coder-labs/modules/gemini/scripts/start.sh

@@ -2,8 +2,7 @@
 set -o errexit
 set -o pipefail
 
-# Handle parameters from agentapi module
-USE_PROMPT="$1"
+
 AGENTAPI_PORT="$2"
 
 source "$HOME"/.bashrc
@@ -20,7 +19,8 @@ fi
 
 printf "Version: %s\n" "$(gemini --version)"
 
-GEMINI_TASK_PROMPT=$(echo -n "$GEMINI_TASK_PROMPT" | base64 -d)
+MODULE_DIR="$HOME/.gemini-module"
+mkdir -p "$MODULE_DIR"
 
 if command_exists gemini; then
     printf "Gemini is installed\n"
@@ -47,12 +47,12 @@ else
     }
 fi
 
-# Handle prompt logic based on parameters
-if [ -n "$USE_PROMPT" ] && [ "$USE_PROMPT" = "true" ] && [ -n "$GEMINI_TASK_PROMPT" ]; then
+if [ -n "$GEMINI_TASK_PROMPT" ]; then
     printf "Running automated task: %s\n" "$GEMINI_TASK_PROMPT"
     PROMPT="Every step of the way, report tasks to Coder with proper descriptions and statuses. Your task at hand: $GEMINI_TASK_PROMPT"
-    GEMINI_ARGS=(--prompt)
-    GEMINI_ARGS+=("$PROMPT")
+    PROMPT_FILE="$MODULE_DIR/prompt.txt"
+    echo -n "$PROMPT" >"$PROMPT_FILE"
+    GEMINI_ARGS=(--prompt "$PROMPT")
 else
     printf "Starting Gemini CLI in interactive mode.\n"
     GEMINI_ARGS=()
@@ -73,6 +73,6 @@ else
     printf "No API key provided (neither GEMINI_API_KEY nor GOOGLE_API_KEY)\n"
 fi
 
-# Use the port parameter if provided, otherwise default to 3284
 PORT=${AGENTAPI_PORT:-3284}
-agentapi server --port "$PORT" --term-width 67 --term-height 1190 -- gemini "${GEMINI_ARGS[@]}"
+agentapi server --port "$PORT" --term-width 67 --term-height 1190 -- \
+    bash -c "$(printf '%q ' gemini "${GEMINI_ARGS[@]}")"