How tayda64/google-k8--ax works
AX is a declarative orchestrator for sandboxed agent tasks. The main workflow is to apply Task, Workspace, Gateway, and Model manifests with the CLI; the API stores them and emits task events; controller workers reconcile tasks into Substrate actors; and the task runner prepares workspaces and serves task metadata and optional guest access. The graph also shows model configuration and SSH access. README documents additional workspace inputs and lifecycle commands; source excerpts do not establish every integration’s wiring.
Main entry points: main.go AX CLI · store Resource store · worker.go Task worker · runner.go Runner lifecycle
Components 20
Control plane
- AX CLI
cmd/ax/main.go - AX gRPC API
internal/server/server.go - Resource API types
pkg/apis/v1alpha1 - Resource store
internal/store - Redis backend
internal/store/redis/store.go - Task event stream
internal/store
Task provisioning
- Task worker
internal/controller/worker.go - Task reconciler
internal/controller/reconciler.go - Agent Substrate
- Workspace planner
internal/workspace - Gateway policy
internal/controller/reconciler.go
Sandbox runtime
- Task runner entry
cmd/ax-task-runner/main.go - Runner lifecycle
runner/runner.go - Task metadata
internal/metadata/server.go - Guest services
internal/guest/client.go - Model client
internal/model/client.go
Agent access
- Cluster tunnel
internal/tunnel
Other components
- Operator
- Google model API
- Task process
Connections 22
- Operator → AX CLI: invokes
- AX CLI → Resource API types: encodes manifests
- AX CLI → AX gRPC API: calls gRPC
- AX gRPC API → Resource store: reads and writes
- Resource store → Redis backend: persists data
- Resource store → Task event stream: publishes task events
- Task worker → Task event stream: consumes events
- Task worker → Resource store: loads resources
- Task worker → Task reconciler: reconciles task
- Task reconciler → Agent Substrate: creates actors
- Task worker → Workspace planner: resolves workspaces
- Task reconciler → Gateway policy: applies egress policy
- Agent Substrate → Task runner entry: starts task container
- Task runner entry → Runner lifecycle: runs task
- Runner lifecycle → Workspace planner: plans workspace
- Runner lifecycle → Task metadata: serves task metadata
- Runner lifecycle → Guest services: enables debug services
- Task process → Task metadata: reads metadata
- AX CLI → Cluster tunnel: connects to cluster
- AX CLI → Guest services: runs SSH commands
- Resource store → Model client: loads model config
- Model client → Google model API: generates responses