Retry remote server health checks

Author: lawrencecchen

cmd/subrouter/cx_server.go

@@ -302,7 +302,7 @@ func (r cxRunner) serverInstall(ctx context.Context, store cxServerStore, args [
 		"curl -fsSL " + shellQuote(publicInstallScriptURL) + " | sudo env SUBROUTER_VERSION=" + shellQuote(*version) + " sh",
 		"sudo /usr/local/bin/sr install-systemd --addr " + shellQuote(*addr) + " --cx-switch-interval " + shellQuote(*cxSwitchInterval) + " --extra-args " + shellQuote(*extraArgs),
 		"if [ -n \"$tailscale_auth_key\" ]; then sudo tailscale up --auth-key \"$tailscale_auth_key\" --hostname " + shellQuote(hostname) + " --ssh --accept-routes=false --accept-dns=false; fi",
-		"curl -fsS http://127.0.0.1:31415/_subrouter/health >/dev/null",
+		"i=0; until curl -fsS http://127.0.0.1:31415/_subrouter/health >/dev/null; do i=$((i+1)); if [ \"$i\" -ge 30 ]; then exit 1; fi; sleep 1; done",
 		"/usr/local/bin/sr --help >/dev/null",
 	}, "\n")
 	sshArgs := []string{"compute", "ssh", server.GCPInstance, "--zone", server.GCPZone, "--command", remoteCommand}

cmd/subrouter/cx_server_test.go

@@ -142,6 +142,7 @@ func TestCXServerInstallUsesPublicInstallerAndSystemdCommand(t *testing.T) {
 		publicInstallScriptURL,
 		"SUBROUTER_VERSION='0.1.2'",
 		"/usr/local/bin/sr install-systemd",
+		"until curl -fsS http://127.0.0.1:31415/_subrouter/health",
 		"tailscale up",
 	} {
 		if !strings.Contains(installCommand, want) {

package.json

@@ -1,6 +1,6 @@
 {
   "name": "subrouter",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "Routes AI coding-agent traffic across subscription accounts and API keys.",
   "license": "MIT",
   "homepage": "https://github.com/manaflow-ai/subrouter#readme",

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "subrouter"
-version = "0.1.3"
+version = "0.1.4"
 description = "Routes AI coding-agent traffic across subscription accounts and API keys."
 readme = "README.md"
 requires-python = ">=3.9"

subrouter_cli/__init__.py

@@ -9,7 +9,7 @@
 import urllib.request
 from pathlib import Path
 
-VERSION = "0.1.3"
+VERSION = "0.1.4"
 
 
 def _go_platform() -> str: