mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
ci: add workflow for agentic issue triage (#19839)
Adds a GH workflow to start a workspace with a pre-determined template, perform a first pass over a given GitHub issue, and persist the changes in a GCS bucket for later refining. Tested locally with `nektos/act`. Co-authored-by: Mathias Fredriksson <mafredri@gmail.com>
This commit is contained in:
Cian Johnston
co-authored by
Mathias Fredriksson
Mathias Fredriksson
parent
5c2b9a5b82
commit
12496830d6
Executable
+251
@@ -0,0 +1,251 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
SCRIPT_DIR=$(dirname "${BASH_SOURCE[0]}")
|
||||
# shellcheck source=scripts/lib.sh
|
||||
source "${SCRIPT_DIR}/lib.sh"
|
||||
|
||||
CODER_BIN=${CODER_BIN:-"$(which coder)"}
|
||||
AGENTAPI_SLUG=${AGENTAPI_SLUG:-""}
|
||||
|
||||
TEMPDIR=$(mktemp -d)
|
||||
trap 'rm -rf "${TEMPDIR}"' EXIT
|
||||
|
||||
[[ -n ${VERBOSE:-} ]] && set -x
|
||||
set -euo pipefail
|
||||
|
||||
usage() {
|
||||
echo "Usage: $0 <options>"
|
||||
exit 1
|
||||
}
|
||||
|
||||
create() {
|
||||
requiredenvs CODER_URL CODER_SESSION_TOKEN WORKSPACE_NAME TEMPLATE_NAME TEMPLATE_PARAMETERS
|
||||
"${CODER_BIN}" \
|
||||
--url "${CODER_URL}" \
|
||||
--token "${CODER_SESSION_TOKEN}" \
|
||||
create \
|
||||
--template "${TEMPLATE_NAME}" \
|
||||
--stop-after 30m \
|
||||
--parameter "${TEMPLATE_PARAMETERS}" \
|
||||
--yes \
|
||||
"${WORKSPACE_NAME}"
|
||||
exit 0
|
||||
}
|
||||
|
||||
prompt() {
|
||||
if [[ -z "${AGENTAPI_SLUG}" ]]; then
|
||||
prompt_ssh
|
||||
else
|
||||
prompt_agentapi
|
||||
fi
|
||||
}
|
||||
|
||||
ssh_config() {
|
||||
requiredenvs CODER_URL CODER_SESSION_TOKEN WORKSPACE_NAME
|
||||
|
||||
if [[ -n "${OPENSSH_CONFIG_FILE:-}" ]]; then
|
||||
echo "Using existing SSH config file: ${OPENSSH_CONFIG_FILE}"
|
||||
return
|
||||
fi
|
||||
|
||||
OPENSSH_CONFIG_FILE="${TEMPDIR}/coder-ssh.config"
|
||||
"${CODER_BIN}" \
|
||||
config-ssh \
|
||||
--url "${CODER_URL}" \
|
||||
--token "${CODER_SESSION_TOKEN}" \
|
||||
--ssh-config-file="${OPENSSH_CONFIG_FILE}" \
|
||||
--yes
|
||||
export OPENSSH_CONFIG_FILE
|
||||
}
|
||||
|
||||
prompt_ssh() {
|
||||
requiredenvs CODER_URL CODER_SESSION_TOKEN WORKSPACE_NAME PROMPT
|
||||
|
||||
ssh_config
|
||||
|
||||
# Execute claude over SSH and provide prompt via stdin
|
||||
# Note: use of cat to work around claude-code#7357
|
||||
ssh \
|
||||
-F "${OPENSSH_CONFIG_FILE}" \
|
||||
"${WORKSPACE_NAME}.coder" \
|
||||
-- \
|
||||
"cat | \"\${HOME}\"/.local/bin/claude --dangerously-skip-permissions --print --verbose --output-format=stream-json" \
|
||||
<<<"${PROMPT}"
|
||||
exit 0
|
||||
}
|
||||
|
||||
prompt_agentapi() {
|
||||
requiredenvs CODER_URL CODER_SESSION_TOKEN WORKSPACE_NAME AGENTAPI_SLUG PROMPT
|
||||
|
||||
wait_agentapi_stable
|
||||
|
||||
username=$(curl \
|
||||
--fail \
|
||||
--header "Coder-Session-Token: ${CODER_SESSION_TOKEN}" \
|
||||
--location \
|
||||
--show-error \
|
||||
--silent \
|
||||
"${CODER_URL}/api/v2/users/me" | jq -r '.username')
|
||||
|
||||
payload="{
|
||||
\"content\": \"${PROMPT}\",
|
||||
\"type\": \"user\"
|
||||
}"
|
||||
|
||||
response=$(curl \
|
||||
--data-raw "${payload}" \
|
||||
--fail \
|
||||
--header "Content-Type: application/json" \
|
||||
--header "Coder-Session-Token: ${CODER_SESSION_TOKEN}" \
|
||||
--location \
|
||||
--request POST \
|
||||
--show-error \
|
||||
--silent \
|
||||
"${CODER_URL}/@${username}/${WORKSPACE_NAME}/apps/${AGENTAPI_SLUG}/message" | jq -r '.ok')
|
||||
if [[ "${response}" != "true" ]]; then
|
||||
echo "Failed to send prompt"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
wait_agentapi_stable
|
||||
}
|
||||
|
||||
wait_agentapi_stable() {
|
||||
requiredenvs CODER_URL CODER_SESSION_TOKEN WORKSPACE_NAME PROMPT
|
||||
username=$(curl \
|
||||
--fail \
|
||||
--header "Coder-Session-Token: ${CODER_SESSION_TOKEN}" \
|
||||
--location \
|
||||
--show-error \
|
||||
--silent \
|
||||
"${CODER_URL}/api/v2/users/me" | jq -r '.username')
|
||||
|
||||
for attempt in {1..120}; do
|
||||
response=$(curl \
|
||||
--fail \
|
||||
--header "Content-Type: application/json" \
|
||||
--header "Coder-Session-Token: ${CODER_SESSION_TOKEN}" \
|
||||
--location \
|
||||
--request GET \
|
||||
--show-error \
|
||||
--silent \
|
||||
"${CODER_URL}/@${username}/${WORKSPACE_NAME}/apps/agentapi/status" | jq -r '.status')
|
||||
if [[ "${response}" == "stable" ]]; then
|
||||
echo "AgentAPI stable"
|
||||
break
|
||||
fi
|
||||
echo "Waiting for AgentAPI to report stable status (attempt ${attempt}/120)"
|
||||
sleep 5
|
||||
done
|
||||
}
|
||||
|
||||
archive() {
|
||||
requiredenvs CODER_URL CODER_SESSION_TOKEN WORKSPACE_NAME DESTINATION_PREFIX
|
||||
ssh_config
|
||||
|
||||
# We want the heredoc to be expanded locally and not remotely.
|
||||
# shellcheck disable=SC2087
|
||||
ARCHIVE_DEST=$(
|
||||
ssh -F "${OPENSSH_CONFIG_FILE}" \
|
||||
"${WORKSPACE_NAME}.coder" \
|
||||
bash <<-EOF
|
||||
#!/usr/bin/env bash
|
||||
set -euo pipefail
|
||||
ARCHIVE_PATH=\$(coder-archive-create)
|
||||
ARCHIVE_NAME=\$(basename "\${ARCHIVE_PATH}")
|
||||
ARCHIVE_DEST="${DESTINATION_PREFIX%%/}/\${ARCHIVE_NAME}"
|
||||
if [[ ! -f "\${ARCHIVE_PATH}" ]]; then
|
||||
echo "FATAL: Archive not found at expected path: \${ARCHIVE_PATH}"
|
||||
exit 1
|
||||
fi
|
||||
gcloud storage cp "\${ARCHIVE_PATH}" "\${ARCHIVE_DEST}"
|
||||
echo "\${ARCHIVE_DEST}"
|
||||
exit 0
|
||||
EOF
|
||||
)
|
||||
|
||||
echo "${ARCHIVE_DEST}"
|
||||
|
||||
exit 0
|
||||
}
|
||||
|
||||
commit_push() {
|
||||
requiredenvs CODER_URL CODER_SESSION_TOKEN WORKSPACE_NAME
|
||||
ssh_config
|
||||
|
||||
# We want the heredoc to be expanded locally and not remotely.
|
||||
# shellcheck disable=SC2087
|
||||
ssh \
|
||||
-F "${OPENSSH_CONFIG_FILE}" \
|
||||
"${WORKSPACE_NAME}.coder" \
|
||||
-- \
|
||||
bash <<-EOF
|
||||
#!/usr/bin/env bash
|
||||
set -euo pipefail
|
||||
BRANCH="traiage/${WORKSPACE_NAME}"
|
||||
if [[ \$(git branch --show-current) != "\${BRANCH}" ]]; then
|
||||
git checkout -b "\${BRANCH}"
|
||||
fi
|
||||
|
||||
if [[ -z \$(git status --porcelain) ]]; then
|
||||
echo "FATAL: No changes to commit"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
git add -A
|
||||
commit_msg=\$(echo -n 'You are a CLI utility that generates a commit message. Generate a concise git commit message for the currently staged changes. Print ONLY the commit message and nothing else.' | \${HOME}/.local/bin/claude --print)
|
||||
if [[ -z "\${commit_msg}" ]]; then
|
||||
commit_msg="Default commit message"
|
||||
fi
|
||||
git commit -am "\${commit_msg}"
|
||||
exit 0
|
||||
EOF
|
||||
|
||||
exit $?
|
||||
}
|
||||
|
||||
delete() {
|
||||
requiredenvs CODER_URL CODER_SESSION_TOKEN WORKSPACE_NAME
|
||||
"${CODER_BIN}" \
|
||||
--url "${CODER_URL}" \
|
||||
--token "${CODER_SESSION_TOKEN}" \
|
||||
delete \
|
||||
"${WORKSPACE_NAME}" \
|
||||
--yes
|
||||
exit 0
|
||||
}
|
||||
|
||||
main() {
|
||||
dependencies coder
|
||||
|
||||
if [[ $# -eq 0 ]]; then
|
||||
usage
|
||||
fi
|
||||
|
||||
case "$1" in
|
||||
create)
|
||||
create
|
||||
;;
|
||||
prompt)
|
||||
prompt
|
||||
;;
|
||||
archive)
|
||||
archive
|
||||
;;
|
||||
commit-push)
|
||||
commit_push
|
||||
;;
|
||||
delete)
|
||||
delete
|
||||
;;
|
||||
wait-agentapi-stable)
|
||||
wait_agentapi_stable
|
||||
;;
|
||||
*)
|
||||
echo "Unknown option: $1"
|
||||
usage
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
main "$@"
|
||||
Reference in New Issue
Block a user