mAi
b282325663
Implements the Replicate API backend (FLUX schnell / FLUX dev) per ImaGen issue #3: - internal/backend/replicate.go — Backend adapter. Supports model refs as "owner/name" (uses /v1/models/{owner}/{name}/predictions) and "owner/name:hash" (uses /v1/predictions with explicit version). Polls /v1/predictions/{id} every 500ms with model-aware timeout (60s schnell, 120s dev). Resilience: 401 names api_token_env, 429 with exp backoff up to 3 retries (honours Retry-After), 5xx retries once, image download retries once on transient failure. - internal/backend/replicate_pricing.go — hardcoded per-image USD rates for known FLUX models, snapshotted from replicate.com/pricing with a refresh TODO. - internal/backend/replicate_test.go — mocked-HTTP unit tests covering happy path (model + version-pinned), 401, 429 retry policy, failed prediction, poll timeout, image-download retry, ctx cancel, BackendOpts passthrough, default_steps, aspect-ratio reduction, sha256 prompt hash. - internal/usage/usage.go — Supabase REST sink + read-side query for mai.imagen_usage. Adapter writes are best-effort: failures warn but the image still lands. - cmd/imagen/usage.go — `imagen usage [--since DATE] [--raw]` reads the table and prints a tab-aligned grouped or raw table with totals. - cmd/imagen/backends.go — instances of type=replicate now report "ok" or "not configured (set REPLICATE_API_TOKEN)" depending on env. - internal/config/config.go — sample adds flux-schnell-replicate + flux-dev-replicate; default_backend stays flux-schnell-local. - Supabase migration mai.imagen_usage (id, created_at, backend, model, seed, prompt_hash, latency_ms, cost_usd_estimate, caller) + indexes on (created_at DESC) and (caller). The raw prompt is never stored. Caller identity resolves from MAI_FROM_ID, then the tmux pane's @mai-name option, mirroring the maimcp identity logic. Prompt hash is sha256 of the user-facing prompt; raw prompt never reaches the table.
ImaGen
Model-agnostic image-generation framework: pluggable backends (local FLUX on mRock, Replicate, DALL-E, …) behind a single CLI / skill / API.
imagen generate "a cat in a fishbowl" --backend flux-schnell-local --size 1024x1024
See CLAUDE.md for the design — backend contract, registry,
config layout, how to add a new adapter.
Install
go install mgit.msbls.de/m/ImaGen/cmd/imagen@latest
Or from a checkout:
make build # writes ./bin/imagen
make install # installs into $GOBIN (defaults to ~/go/bin)
First run
mkdir -p ~/.config
imagen config init > ~/.config/imagen.yaml
imagen config validate
imagen backends
imagen generate "test prompt" --backend mock --output /tmp/x.png
The mock backend ships in this repo and produces a deterministic gradient PNG — useful for smoke-testing the pipeline without reaching any model.
Status
| Component | Status |
|---|---|
| Backend interface | done (#1) |
| Mock backend | done (#1) |
| ComfyUI / FLUX on mRock | open (#2) |
| Replicate adapter | open (#3) |
/imagine skill |
open (#4) |
| HTTP server | stubbed (#1) |
Issues live at https://mgit.msbls.de/m/ImaGen/issues.