refactor: remove ate build tag

Makefile

@@ -83,20 +83,20 @@ clean-logs:
 
 ax-image:
 	@echo "Building container image with ko..."
-	GOFLAGS="-tags=ate" ko build --base-import-paths ./cmd/ax
+	ko build --base-import-paths ./cmd/ax
 
 ax-server-image:
 	@echo "Building ax-server (harness path) container image with ko..."
 	GOFLAGS="-tags=harness" ko build --base-import-paths ./cmd/ax
 
 axepp-image:
 	@echo "Building axepp container image with ko..."
-	GOFLAGS="-tags=ate" ko build --base-import-paths ./cmd/axepp
+	ko build --base-import-paths ./cmd/axepp
 
 ax-shell-image:
 	# Used to debug ax servers within a cluster.
 	@echo "Building ax shell container image with ko using busybox..."
-	KO_DOCKER_REPO=$(KO_DOCKER_REPO)/ax-shell KO_DEFAULTBASEIMAGE=busybox:1.36 GOFLAGS="-tags=ate" ko build --base-import-paths ./cmd/ax
+	KO_DOCKER_REPO=$(KO_DOCKER_REPO)/ax-shell KO_DEFAULTBASEIMAGE=busybox:1.36 ko build --base-import-paths ./cmd/ax
 
 # Build all container images
 images: ax-image axepp-image ax-shell-image

cmd/axepp/main.go

@@ -1,4 +1,3 @@
-//go:build ate
 
 // Copyright 2026 Google LLC
 //

hack/install-ax.sh

@@ -45,7 +45,7 @@ run_kubectl() {
 }
 
 run_ko() {
-  GOFLAGS="-tags=ate" ko apply \
+  ko apply \
     ${KUBECTL_CONTEXT:+--context=${KUBECTL_CONTEXT}} \
     "$@"
 }

internal/server/server_ate.go

@@ -1,4 +1,4 @@
-//go:build !harness && ate
+//go:build !harness
 
 // Copyright 2026 Google LLC
 //
@@ -29,7 +29,7 @@ import (
 	"google.golang.org/grpc/credentials"
 )
 
-func createATEClient() any {
+func createATEClient() *ate.Client {
 	if os.Getenv("AX_SUBSTRATE") != "1" {
 		return nil
 	}
@@ -76,16 +76,15 @@ func suspendActor(actorID string) {
 	if client == nil {
 		return
 	}
-	c := client.(*ate.Client)
-	defer c.Close()
+	defer client.Close()
 
 	log.Printf("Automatically suspending actor %s in 50 milliseconds...", actorID)
 	time.Sleep(50 * time.Millisecond)
 
 	suspendCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
 	defer cancel()
 
-	if _, err := c.SuspendActor(suspendCtx, actorID); err != nil {
+	if _, err := client.SuspendActor(suspendCtx, actorID); err != nil {
 		log.Printf("Failed to automatically suspend actor %s: %v", actorID, err)
 	} else {
 		log.Printf("Successfully suspended actor %s", actorID)