cog-review

Cog architecture review guidelines

Cog packages ML models into production-ready containers. Use this skill for changes that cross language boundaries or touch core architecture.

Component overview

• CLI (Go): cmd/cog/ and pkg/ -- builds, runs, and deploys models
• Python SDK: python/cog/ -- predictor interface, types, HTTP/queue server
• Coglet (Rust): crates/ -- prediction server inside containers (HTTP, worker management, IPC)

Key design patterns

Wheel resolution: The CLI discovers SDK and coglet wheels from dist/ at Docker build time. Wheels are NOT embedded in the binary. Changes to build artifacts need to account for this.

Dockerfile generation: pkg/dockerfile/ generates Dockerfiles from cog.yaml config. Template injection and escaping matter here.