Skip to content

Feat/model library worker reconnect compose#9

Merged
rwilliamspbg-ops merged 3 commits into
mainfrom
feat/model-library-worker-reconnect-compose
Jun 23, 2026
Merged

Feat/model library worker reconnect compose#9
rwilliamspbg-ops merged 3 commits into
mainfrom
feat/model-library-worker-reconnect-compose

Conversation

@rwilliamspbg-ops

@rwilliamspbg-ops rwilliamspbg-ops commented Jun 23, 2026

Copy link
Copy Markdown
Owner

Summary

This branch hardens the model library, worker lifecycle, GUI backend wiring, and Docker runtime setup for the Mohawk Inference Engine.

What Changed

Model library

  • Added persistent local model library tracking in mohawk/models/loader.py.
  • Enabled HuggingFace downloads to be registered automatically in the local model index.
  • Added validation for blank model IDs and safer HuggingFace download argument handling.
  • Added tests for local model registration, HuggingFace download registration, and empty-ID rejection.

Worker lifecycle and secure execution

  • Added worker join, leave, and reconnect APIs.
  • Added failover/reshare handling for worker churn.
  • Fixed deterministic shared-key derivation for secure worker/controller communication.
  • Improved slice serialization to preserve shapes and support backward compatibility.
  • Added regression tests covering worker lifecycle and reconnect flows.

GUI backend wiring

  • Rewired the desktop GUI to use real controller API endpoints for models, workers, queue, sessions, metrics, and security actions.
  • Improved API error propagation so backend failures surface meaningful details in the UI.
  • Added a controller service backend that exposes the GUI-facing API contract.

Docker and CI

  • Corrected compose files and Dockerfiles so the runtime commands match the service ports.
  • Added CI compose smoke validation.
  • Fixed the controller Dockerfile healthcheck/exposed port mismatch so the image matches the actual uvicorn service port.

Validation

  • Full test suite: 87 passed, 4 skipped
  • Focused controller/model tests: passed
  • Compose config validation: passed

Notes

  • Docker compose runtime still requires a machine with a working Docker daemon for full up --build execution.

- replace remaining GUI fallback paths with real API-driven wiring\n- implement controller service endpoints for models, workers, sessions, queue, chat, and security actions\n- wire main compose backend to real controller_service by default\n- update Dockerfile to include runtime deps and prototype backend code for controller service\n- add controller service endpoint tests for GUI contract coverage\n- add CI compose smoke job to boot stack and verify health + GUI-facing endpoints\n- align pytest warning filtering and startup docs with updated runtime behavior\n\nValidation:\n- pytest -q -> 87 passed, 4 skipped\n- docker compose config (main + dev) -> OK
@rwilliamspbg-ops rwilliamspbg-ops merged commit ebb97b2 into main Jun 23, 2026
4 of 6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant