agentapi and ai task support

Author: hugodutka

registry/coder/modules/claude-code/main.tf

@@ -96,9 +96,57 @@ variable "experiment_tmux_session_save_interval" {
   default     = "15"
 }
 
+variable "install_agentapi" {
+  type        = bool
+  description = "Whether to install AgentAPI."
+  default     = true
+}
+
+variable "agentapi_version" {
+  type        = string
+  description = "The version of AgentAPI to install."
+  default     = "v0.2.2"
+}
+
 locals {
-  encoded_pre_install_script  = var.experiment_pre_install_script != null ? base64encode(var.experiment_pre_install_script) : ""
-  encoded_post_install_script = var.experiment_post_install_script != null ? base64encode(var.experiment_post_install_script) : ""
+  encoded_pre_install_script             = var.experiment_pre_install_script != null ? base64encode(var.experiment_pre_install_script) : ""
+  encoded_post_install_script            = var.experiment_post_install_script != null ? base64encode(var.experiment_post_install_script) : ""
+  agentapi_start_command                 = <<-EOT
+    #!/bin/bash
+    set -e
+    
+    # if the first argument is not empty, start claude with the prompt
+    if [ -n "$1" ]; then
+      prompt="$(cat ~/.claude-code-prompt)"
+    else
+      prompt=""
+    fi
+
+    # use low width to fit in the tasks UI sidebar. height is adjusted to ~match the default 80k (80x1000) characters
+    # visible in the terminal screen.
+    agentapi server --term-width 67 --term-height 1190 -- bash -c "claude --continue --dangerously-skip-permissions \"$prompt\""
+  EOT
+  agentapi_wait_for_start_command        = <<-EOT
+    #!/bin/bash
+    set -o errexit
+    set -o pipefail
+
+    echo "Waiting for agentapi server to start on port 3284..."
+    for i in $(seq 1 15); do
+      if lsof -i :3284 | grep -q 'LISTEN'; then
+        echo "agentapi server started on port 3284."
+        break
+      fi
+      echo "Waiting... ($i/15)"
+      sleep 1
+    done
+    if ! lsof -i :3284 | grep -q 'LISTEN'; then
+      echo "Error: agentapi server did not start on port 3284 after 15 seconds."
+      exit 1
+    fi
+  EOT
+  agentapi_start_command_base64          = base64encode(local.agentapi_start_command)
+  agentapi_wait_for_start_command_base64 = base64encode(local.agentapi_wait_for_start_command)
 }
 
 # Install and Initialize Claude Code
@@ -176,9 +224,38 @@ resource "coder_script" "claude_code" {
       npm install -g @anthropic-ai/claude-code@${var.claude_code_version}
     fi
 
+    # Install AgentAPI if enabled
+    if [ "${var.install_agentapi}" = "true" ]; then
+      echo "Installing AgentAPI..."
+      arch=$(uname -m)
+      if [ "$arch" = "x86_64" ]; then
+        binary_name="agentapi-linux-amd64"
+      elif [ "$arch" = "aarch64" ]; then
+        binary_name="agentapi-linux-arm64"
+      else
+        echo "Error: Unsupported architecture: $arch"
+        exit 1
+      fi
+      wget "https://github.com/coder/agentapi/releases/download/${var.agentapi_version}/$binary_name"
+      chmod +x "$binary_name"
+      sudo mv "$binary_name" /usr/local/bin/agentapi
+    fi
+    if ! command_exists agentapi; then
+      echo "Error: AgentAPI is not installed. Please enable install_agentapi or install it manually."
+      exit 1
+    fi
+
+    # save the prompt for the agentapi start command
+    echo -n "$CODER_MCP_CLAUDE_TASK_PROMPT" > ~/.claude-code-prompt
+
+    echo -n "${local.agentapi_start_command_base64}" | base64 -d > ~/.agentapi-start-command
+    chmod +x ~/.agentapi-start-command
+    echo -n "${local.agentapi_wait_for_start_command_base64}" | base64 -d > ~/.agentapi-wait-for-start-command
+    chmod +x ~/.agentapi-wait-for-start-command
+
     if [ "${var.experiment_report_tasks}" = "true" ]; then
       echo "Configuring Claude Code to report tasks via Coder MCP..."
-      coder exp mcp configure claude-code ${var.folder}
+      coder exp mcp configure claude-code ${var.folder} --ai-agentapi-url http://localhost:3284
     fi
 
     if [ -n "${local.encoded_post_install_script}" ]; then
@@ -257,17 +334,16 @@ EOF
       export LANG=en_US.UTF-8
       export LC_ALL=en_US.UTF-8
 
+      tmux new-session -d -s claude-code-agentapi -c ${var.folder} '~/.agentapi-start-command true; exec bash'
+      ~/.agentapi-wait-for-start-command
+
       if [ "${var.experiment_tmux_session_persistence}" = "true" ]; then
         sleep 3
+      fi
         
-        if ! tmux has-session -t claude-code 2>/dev/null; then
-          # Only create a new session if one doesn't exist
-          tmux new-session -d -s claude-code -c ${var.folder} "claude --dangerously-skip-permissions \"$CODER_MCP_CLAUDE_TASK_PROMPT\""
-        fi
-      else
-        if ! tmux has-session -t claude-code 2>/dev/null; then
-          tmux new-session -d -s claude-code -c ${var.folder} "claude --dangerously-skip-permissions \"$CODER_MCP_CLAUDE_TASK_PROMPT\""
-        fi
+      if ! tmux has-session -t claude-code 2>/dev/null; then
+        # Only create a new session if one doesn't exist
+        tmux new-session -d -s claude-code -c ${var.folder} "agentapi attach; exec bash"
       fi
     fi
 
@@ -297,9 +373,17 @@ EOF
       export LANG=en_US.UTF-8
       export LC_ALL=en_US.UTF-8
 
+      screen -U -dmS claude-code-agentapi bash -c '
+        cd ${var.folder}
+        # setting the first argument will make claude use the prompt
+        ~/.agentapi-start-command true
+        exec bash
+      '
+      ~/.agentapi-wait-for-start-command
+
       screen -U -dmS claude-code bash -c '
         cd ${var.folder}
-        claude --dangerously-skip-permissions "$CODER_MCP_CLAUDE_TASK_PROMPT" | tee -a "$HOME/.claude-code.log"
+        agentapi attach
         exec bash
       '
     else
@@ -312,6 +396,20 @@ EOF
   run_on_start = true
 }
 
+resource "coder_app" "claude_code_web" {
+  slug         = "ccw"
+  display_name = "Claude Code Web"
+  agent_id     = var.agent_id
+  url          = "http://localhost:3284/"
+  icon         = var.icon
+  subdomain    = true
+  healthcheck {
+    url       = "http://localhost:3284/status"
+    interval  = 5
+    threshold = 3
+  }
+}
+
 resource "coder_app" "claude_code" {
   slug         = "claude-code"
   display_name = "Claude Code"
@@ -324,31 +422,51 @@ resource "coder_app" "claude_code" {
     export LC_ALL=en_US.UTF-8
 
     if [ "${var.experiment_use_tmux}" = "true" ]; then
+      if ! tmux has-session -t claude-code-agentapi 2>/dev/null; then
+        # start agentapi without claude using the prompt (no argument)
+        tmux new-session -d -s claude-code-agentapi -c ${var.folder} '~/.agentapi-start-command; exec bash'
+        ~/.agentapi-wait-for-start-command
+      fi
+
       if tmux has-session -t claude-code 2>/dev/null; then
         echo "Attaching to existing Claude Code tmux session." | tee -a "$HOME/.claude-code.log"
-        # If Claude isn't running in the session, start it without the prompt
-        if ! tmux list-panes -t claude-code -F '#{pane_current_command}' | grep -q "claude"; then
-          tmux send-keys -t claude-code "cd ${var.folder} && claude -c --dangerously-skip-permissions" C-m
+        # If agentapi attach isn't running in the session, start it
+        if ! tmux list-panes -t claude-code -F '#{pane_current_command}' | grep -q "agentapi"; then
+          tmux send-keys -t claude-code "cd ${var.folder} && agentapi attach" C-m
         fi
         tmux attach-session -t claude-code
       else
         echo "Starting a new Claude Code tmux session." | tee -a "$HOME/.claude-code.log"
-        tmux new-session -s claude-code -c ${var.folder} "claude --dangerously-skip-permissions | tee -a \"$HOME/.claude-code.log\"; exec bash"
+        tmux new-session -s claude-code -c ${var.folder} "agentapi attach; exec bash"
       fi
     elif [ "${var.experiment_use_screen}" = "true" ]; then
+      if ! screen -list | grep -q "claude-code-agentapi"; then
+        screen -S claude-code-agentapi bash -c '
+          cd ${var.folder}
+          # start agentapi without claude using the prompt (no argument)
+          ~/.agentapi-start-command
+          exec bash
+        '
+      fi
       if screen -list | grep -q "claude-code"; then
         echo "Attaching to existing Claude Code screen session." | tee -a "$HOME/.claude-code.log"
         screen -xRR claude-code
       else
         echo "Starting a new Claude Code screen session." | tee -a "$HOME/.claude-code.log"
-        screen -S claude-code bash -c 'claude --dangerously-skip-permissions | tee -a "$HOME/.claude-code.log"; exec bash'
+        screen -S claude-code bash -c 'agentapi attach; exec bash'
       fi
     else
       cd ${var.folder}
-      claude
+      agentapi attach
     fi
     EOT
   icon         = var.icon
   order        = var.order
   group        = var.group
 }
+
+resource "coder_ai_task" "claude_code" {
+  sidebar_app {
+    id = coder_app.claude_code.id
+  }
+}