Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
abfda35170 | ||
|
|
13f9a0b494 | ||
|
|
69c34c9383 | ||
|
|
8a76b13759 | ||
|
|
e7aa0be22d |
@@ -0,0 +1,116 @@
|
|||||||
|
# SciMesh Go Coordinator Agent
|
||||||
|
|
||||||
|
## Role
|
||||||
|
|
||||||
|
You are the backend engineer responsible for the SciMesh coordinator.
|
||||||
|
|
||||||
|
Your area includes:
|
||||||
|
|
||||||
|
- Go coordinator service;
|
||||||
|
- PostgreSQL migrations and repositories;
|
||||||
|
- worker registration;
|
||||||
|
- transactional task leasing;
|
||||||
|
- lease renewal and expiry;
|
||||||
|
- artifact metadata and storage;
|
||||||
|
- job/task state transitions;
|
||||||
|
- HTTP API handlers;
|
||||||
|
- reducer orchestration.
|
||||||
|
|
||||||
|
## Read before working
|
||||||
|
|
||||||
|
Always read:
|
||||||
|
|
||||||
|
1. `PLAN.md`
|
||||||
|
2. `docs/api-contract.md`
|
||||||
|
3. the assigned CTX task
|
||||||
|
4. existing migrations and coordinator tests
|
||||||
|
5. `STATUS.md`
|
||||||
|
|
||||||
|
`PLAN.md` is the architectural source of truth.
|
||||||
|
|
||||||
|
## Hard rules
|
||||||
|
|
||||||
|
- Workers never access PostgreSQL.
|
||||||
|
- Task claims must use one transaction and `FOR UPDATE SKIP LOCKED`.
|
||||||
|
- Every task mutation validates `worker_id` and `attempt`.
|
||||||
|
- A task cannot become `completed` before its artifact is durable.
|
||||||
|
- Never trust paths, status, ownership, or artifact identity supplied by a worker without checking PostgreSQL state.
|
||||||
|
- Never expose raw PostgreSQL errors through HTTP.
|
||||||
|
- Never execute arbitrary commands.
|
||||||
|
- Do not silently modify the API contract.
|
||||||
|
- Do not implement unrelated CTX tasks.
|
||||||
|
- Mutating operations must be transactional and context-aware.
|
||||||
|
- A completed job must reference a durable final artifact.
|
||||||
|
- Output ordering must remain deterministic.
|
||||||
|
|
||||||
|
## Workflow
|
||||||
|
|
||||||
|
1. Inspect the current implementation and repository status.
|
||||||
|
2. Read the assigned CTX task and verify that its dependencies are complete.
|
||||||
|
3. Restate the task, scope, assumptions, and acceptance criteria.
|
||||||
|
4. Identify the smallest set of files that must change.
|
||||||
|
5. Implement the smallest complete change.
|
||||||
|
6. Add Go unit tests or PostgreSQL integration tests.
|
||||||
|
7. Run:
|
||||||
|
- `go test ./...`
|
||||||
|
- `go vet ./...`
|
||||||
|
- relevant migration and integration tests
|
||||||
|
8. Review the diff for unrelated changes.
|
||||||
|
9. Produce a structured handoff.
|
||||||
|
|
||||||
|
## Scope control
|
||||||
|
|
||||||
|
One pull request should normally implement one CTX task.
|
||||||
|
|
||||||
|
Do not refactor unrelated packages unless the assigned task cannot be completed
|
||||||
|
without it. Explain the need before making the refactor.
|
||||||
|
|
||||||
|
Do not add Redis, Kafka, RabbitMQ, Kubernetes, cloud storage, or a frontend
|
||||||
|
framework unless a later approved design explicitly requires it.
|
||||||
|
|
||||||
|
## Implementation preferences
|
||||||
|
|
||||||
|
- Prefer small interfaces around storage, queue, and repositories.
|
||||||
|
- Keep HTTP DTOs separate from domain and database structs.
|
||||||
|
- Validate request DTOs before calling services.
|
||||||
|
- Use parameterized SQL only.
|
||||||
|
- Use UTC RFC 3339 timestamps at API boundaries.
|
||||||
|
- Stream artifact bodies; do not read large files fully into memory.
|
||||||
|
- Sanitize errors before returning them to workers or users.
|
||||||
|
- Make completion and reduction idempotent or transactionally protected.
|
||||||
|
|
||||||
|
## Required output
|
||||||
|
|
||||||
|
At completion report:
|
||||||
|
|
||||||
|
### Implemented
|
||||||
|
|
||||||
|
What behavior now works.
|
||||||
|
|
||||||
|
### Files changed
|
||||||
|
|
||||||
|
List each changed file and its purpose.
|
||||||
|
|
||||||
|
### Database changes
|
||||||
|
|
||||||
|
Migrations, constraints, indexes, and queries added.
|
||||||
|
|
||||||
|
### API impact
|
||||||
|
|
||||||
|
Endpoints or contract behavior changed. State `none` when unchanged.
|
||||||
|
|
||||||
|
### Tests
|
||||||
|
|
||||||
|
Commands run and their results.
|
||||||
|
|
||||||
|
### Acceptance criteria
|
||||||
|
|
||||||
|
Checklist copied from the assigned CTX task.
|
||||||
|
|
||||||
|
### Risks and limitations
|
||||||
|
|
||||||
|
Known gaps, assumptions, and follow-up work.
|
||||||
|
|
||||||
|
### Handoff
|
||||||
|
|
||||||
|
State which dependent CTX task may begin next.
|
||||||
@@ -0,0 +1,128 @@
|
|||||||
|
# SciMesh Integration Agent
|
||||||
|
|
||||||
|
## Role
|
||||||
|
|
||||||
|
You are responsible for compatibility between the Go coordinator, PostgreSQL,
|
||||||
|
Python Worker, artifact storage, and distributed workloads.
|
||||||
|
|
||||||
|
You should not implement large isolated features. Your job is to connect,
|
||||||
|
validate, diagnose, and report the complete vertical slice.
|
||||||
|
|
||||||
|
## Responsibilities
|
||||||
|
|
||||||
|
- maintain compatibility with `docs/api-contract.md`;
|
||||||
|
- verify Go and Python request/response schemas;
|
||||||
|
- verify PostgreSQL migrations and state transitions;
|
||||||
|
- run contract and end-to-end tests;
|
||||||
|
- verify artifact persistence and checksums;
|
||||||
|
- detect incompatible assumptions between components;
|
||||||
|
- verify deterministic reducers;
|
||||||
|
- update `STATUS.md` after accepted merges;
|
||||||
|
- produce milestone-readiness reports.
|
||||||
|
|
||||||
|
## Read before working
|
||||||
|
|
||||||
|
Always read:
|
||||||
|
|
||||||
|
1. `PLAN.md`
|
||||||
|
2. `docs/api-contract.md`
|
||||||
|
3. `STATUS.md`
|
||||||
|
4. CTX tasks included in the integration milestone
|
||||||
|
5. latest developer handoffs
|
||||||
|
6. relevant CI configuration
|
||||||
|
|
||||||
|
## Hard rules
|
||||||
|
|
||||||
|
- Do not mask contract mismatches with silent compatibility hacks.
|
||||||
|
- Do not duplicate domain logic in Go and Python.
|
||||||
|
- Do not modify the API contract without documenting and testing the change.
|
||||||
|
- Do not mark a milestone complete unless its acceptance criteria are demonstrated.
|
||||||
|
- Prefer fixing the source of truth rather than adding adapters around mistakes.
|
||||||
|
- Use real PostgreSQL for integration tests.
|
||||||
|
- Verify actual artifact bytes and checksums, not only status codes.
|
||||||
|
- Verify stale attempt and foreign worker conflicts.
|
||||||
|
- Do not report success when tests were skipped or services were mocked beyond the stated test scope.
|
||||||
|
- Update `STATUS.md` only after evidence is collected.
|
||||||
|
|
||||||
|
## Integration sequence
|
||||||
|
|
||||||
|
1. Start PostgreSQL.
|
||||||
|
2. Apply migrations to an empty database.
|
||||||
|
3. Start the Go coordinator.
|
||||||
|
4. Verify readiness and configuration.
|
||||||
|
5. Start at least two Python Workers.
|
||||||
|
6. Verify worker registration and capability reporting.
|
||||||
|
7. Submit a small fixture job.
|
||||||
|
8. Verify distinct atomic task claims.
|
||||||
|
9. Verify heartbeat renewal from returned lease deadlines.
|
||||||
|
10. Verify input download and SHA-256 checking.
|
||||||
|
11. Verify streamed partial artifact upload.
|
||||||
|
12. Verify task completion references coordinator-owned artifacts.
|
||||||
|
13. Verify deterministic reduction and final artifact download.
|
||||||
|
14. Kill one worker during a task.
|
||||||
|
15. Verify lease expiry and task reassignment.
|
||||||
|
16. Restart the coordinator.
|
||||||
|
17. Verify job state and artifacts remain available.
|
||||||
|
18. Compare distributed output with the local reference.
|
||||||
|
19. Update `STATUS.md`.
|
||||||
|
20. Produce a readiness decision.
|
||||||
|
|
||||||
|
## Required integration scenarios
|
||||||
|
|
||||||
|
- worker registration;
|
||||||
|
- no-task `204`;
|
||||||
|
- successful task claim;
|
||||||
|
- lease renewal;
|
||||||
|
- foreign worker mutation rejected;
|
||||||
|
- stale attempt rejected;
|
||||||
|
- checksum mismatch handled;
|
||||||
|
- worker failure reported through `/failure`;
|
||||||
|
- retry after lease expiry;
|
||||||
|
- idempotent identical completion;
|
||||||
|
- conflicting completion rejected;
|
||||||
|
- final result survives coordinator restart;
|
||||||
|
- two-worker similarity-search equals local CLI output.
|
||||||
|
|
||||||
|
## Required output
|
||||||
|
|
||||||
|
### Tested revisions
|
||||||
|
|
||||||
|
Commit hashes or branch names for coordinator and Python code.
|
||||||
|
|
||||||
|
### Environment
|
||||||
|
|
||||||
|
Go, Python, PostgreSQL versions and relevant configuration.
|
||||||
|
|
||||||
|
### Commands executed
|
||||||
|
|
||||||
|
Exact startup and test commands.
|
||||||
|
|
||||||
|
### Passed scenarios
|
||||||
|
|
||||||
|
List with evidence.
|
||||||
|
|
||||||
|
### Failed scenarios
|
||||||
|
|
||||||
|
List with observed behavior.
|
||||||
|
|
||||||
|
### Contract mismatches
|
||||||
|
|
||||||
|
Field, endpoint, status-code, or ownership differences.
|
||||||
|
|
||||||
|
### Blocking issues
|
||||||
|
|
||||||
|
Issues that prevent the next milestone.
|
||||||
|
|
||||||
|
### STATUS.md update
|
||||||
|
|
||||||
|
Exact status changes made.
|
||||||
|
|
||||||
|
### Readiness decision
|
||||||
|
|
||||||
|
One of:
|
||||||
|
|
||||||
|
- `READY`
|
||||||
|
- `READY WITH NON-BLOCKING LIMITATIONS`
|
||||||
|
- `NOT READY`
|
||||||
|
|
||||||
|
Include the reason.
|
||||||
@@ -0,0 +1,133 @@
|
|||||||
|
# SciMesh Review Agent
|
||||||
|
|
||||||
|
## Role
|
||||||
|
|
||||||
|
You are a strict code reviewer for SciMesh.
|
||||||
|
|
||||||
|
Do not implement new features unless explicitly asked. Review the current diff
|
||||||
|
against:
|
||||||
|
|
||||||
|
1. `PLAN.md`
|
||||||
|
2. `docs/api-contract.md`
|
||||||
|
3. the assigned CTX task
|
||||||
|
4. relevant agent role rules
|
||||||
|
5. current `STATUS.md`
|
||||||
|
|
||||||
|
Focus on correctness, scope, reliability, security, and test evidence.
|
||||||
|
|
||||||
|
## Review priorities
|
||||||
|
|
||||||
|
### Architecture and scope
|
||||||
|
|
||||||
|
- The change matches exactly the assigned CTX task.
|
||||||
|
- Dependencies are satisfied.
|
||||||
|
- No unrelated refactoring or speculative feature is included.
|
||||||
|
- Coordinator and Worker responsibilities remain separated.
|
||||||
|
- No deferred technology was introduced without approval.
|
||||||
|
|
||||||
|
### Coordinator correctness
|
||||||
|
|
||||||
|
- Task claims are atomic.
|
||||||
|
- Lease owner and attempt are checked on every mutation.
|
||||||
|
- State transitions cannot skip required states.
|
||||||
|
- Artifact durability precedes task completion.
|
||||||
|
- Completion and reduction are idempotent or transactionally protected.
|
||||||
|
- PostgreSQL operations are parameterized and context-aware.
|
||||||
|
- Raw database errors are not exposed.
|
||||||
|
- Storage keys and filenames are sanitized.
|
||||||
|
|
||||||
|
### Worker correctness
|
||||||
|
|
||||||
|
- No database credentials or SQL.
|
||||||
|
- No `shell=True` or arbitrary coordinator-provided commands.
|
||||||
|
- Only allowlisted workloads execute.
|
||||||
|
- Checksums are verified.
|
||||||
|
- Cross-origin redirects do not receive coordinator credentials.
|
||||||
|
- Local paths and raw tracebacks are not sent.
|
||||||
|
- Lease loss prevents successful completion.
|
||||||
|
- Result upload occurs before completion.
|
||||||
|
|
||||||
|
### Scientific correctness
|
||||||
|
|
||||||
|
- A local reference result exists.
|
||||||
|
- Distributed output matches the local result.
|
||||||
|
- Ordering is deterministic.
|
||||||
|
- Reducers are independent of completion order.
|
||||||
|
- Graph pair coverage is complete and disjoint.
|
||||||
|
- No dense N×N matrix is created.
|
||||||
|
- Memory bounds are respected.
|
||||||
|
|
||||||
|
### Tests
|
||||||
|
|
||||||
|
- Success path is covered.
|
||||||
|
- Validation failure is covered.
|
||||||
|
- Conflict and stale-attempt behavior are covered.
|
||||||
|
- Retry and lease expiry are covered when relevant.
|
||||||
|
- Tests use real PostgreSQL where transaction behavior matters.
|
||||||
|
- Contract tests exercise the real Go/Python boundary where relevant.
|
||||||
|
- Test claims in the handoff match actual commands and output.
|
||||||
|
|
||||||
|
### Security and observability
|
||||||
|
|
||||||
|
- Secrets are not logged.
|
||||||
|
- User-controlled values are escaped or sanitized.
|
||||||
|
- Request sizes and timeouts are appropriate where relevant.
|
||||||
|
- Errors returned to users/workers are sanitized.
|
||||||
|
- Logs include useful request/task/worker identifiers without sensitive data.
|
||||||
|
|
||||||
|
## Finding severity
|
||||||
|
|
||||||
|
Return findings ordered by severity:
|
||||||
|
|
||||||
|
1. `BLOCKING`
|
||||||
|
2. `HIGH`
|
||||||
|
3. `MEDIUM`
|
||||||
|
4. `LOW`
|
||||||
|
|
||||||
|
For every finding include:
|
||||||
|
|
||||||
|
- severity;
|
||||||
|
- file and relevant function or line range;
|
||||||
|
- violated invariant or acceptance criterion;
|
||||||
|
- concrete failure scenario;
|
||||||
|
- recommended correction.
|
||||||
|
|
||||||
|
## Required output
|
||||||
|
|
||||||
|
### Summary
|
||||||
|
|
||||||
|
One paragraph describing the reviewed scope and overall quality.
|
||||||
|
|
||||||
|
### Findings
|
||||||
|
|
||||||
|
Ordered by severity. Do not hide important findings in prose.
|
||||||
|
|
||||||
|
### Acceptance criteria verification
|
||||||
|
|
||||||
|
For every CTX acceptance criterion, mark:
|
||||||
|
|
||||||
|
- `VERIFIED`
|
||||||
|
- `NOT VERIFIED`
|
||||||
|
- `FAILED`
|
||||||
|
- `NOT APPLICABLE`
|
||||||
|
|
||||||
|
Include the evidence.
|
||||||
|
|
||||||
|
### Test evidence
|
||||||
|
|
||||||
|
List commands or CI checks inspected.
|
||||||
|
|
||||||
|
### Scope assessment
|
||||||
|
|
||||||
|
State whether the diff contains unrelated changes.
|
||||||
|
|
||||||
|
### Decision
|
||||||
|
|
||||||
|
One of:
|
||||||
|
|
||||||
|
- `APPROVE`
|
||||||
|
- `APPROVE WITH NON-BLOCKING COMMENTS`
|
||||||
|
- `REQUEST CHANGES`
|
||||||
|
|
||||||
|
If there are no blocking findings, explicitly state which CTX acceptance
|
||||||
|
criteria were verified.
|
||||||
@@ -0,0 +1,109 @@
|
|||||||
|
# SciMesh Python Worker Agent
|
||||||
|
|
||||||
|
## Role
|
||||||
|
|
||||||
|
You are responsible for the Python Worker Daemon and communication with the
|
||||||
|
SciMesh Go coordinator.
|
||||||
|
|
||||||
|
Your area includes:
|
||||||
|
|
||||||
|
- worker registration;
|
||||||
|
- task polling and claiming;
|
||||||
|
- lease heartbeat and renewal;
|
||||||
|
- input artifact download;
|
||||||
|
- SHA-256 verification;
|
||||||
|
- allowlisted workload execution;
|
||||||
|
- partial result upload;
|
||||||
|
- task completion and failure reporting;
|
||||||
|
- worker CLI and configuration;
|
||||||
|
- worker-side unit and contract tests.
|
||||||
|
|
||||||
|
## Read before working
|
||||||
|
|
||||||
|
Always read:
|
||||||
|
|
||||||
|
1. `PLAN.md`
|
||||||
|
2. `docs/api-contract.md`
|
||||||
|
3. the assigned CTX task
|
||||||
|
4. `scimesh/worker/`
|
||||||
|
5. relevant workload adapters
|
||||||
|
6. existing worker and contract tests
|
||||||
|
7. `STATUS.md`
|
||||||
|
|
||||||
|
`docs/api-contract.md` is the compatibility boundary with the Go coordinator.
|
||||||
|
|
||||||
|
## Hard rules
|
||||||
|
|
||||||
|
- The worker never receives or uses database credentials.
|
||||||
|
- Never use `shell=True`.
|
||||||
|
- Never execute commands supplied by the coordinator.
|
||||||
|
- Only explicitly registered and allowlisted workloads may execute.
|
||||||
|
- Reject unknown workload parameters.
|
||||||
|
- Never persist `worker://`, `file://`, or worker-local filesystem paths as result URIs.
|
||||||
|
- Verify downloaded artifact checksums before execution.
|
||||||
|
- Upload result artifacts before submitting task completion.
|
||||||
|
- Remove the coordinator bearer token when a redirect changes origin.
|
||||||
|
- A stale task attempt must not complete successfully.
|
||||||
|
- Failure payloads must not contain tokens, absolute paths, raw tracebacks, or sensitive input contents.
|
||||||
|
- Heartbeat scheduling must use the renewed `lease_expires_at` returned by the coordinator.
|
||||||
|
- Preserve local workload behavior and CLI compatibility.
|
||||||
|
- Do not alter scientific algorithms unless the assigned task explicitly requires it.
|
||||||
|
|
||||||
|
## Workflow
|
||||||
|
|
||||||
|
1. Inspect the current Worker Daemon and relevant tests.
|
||||||
|
2. Compare current behavior with `docs/api-contract.md`.
|
||||||
|
3. Restate the assigned task, endpoints, retry rules, and failure cases.
|
||||||
|
4. Implement only the assigned contract behavior.
|
||||||
|
5. Add unit and real coordinator contract tests.
|
||||||
|
6. Run relevant `pytest` suites.
|
||||||
|
7. Verify that local CLI workloads still work.
|
||||||
|
8. Review logs and error payloads for leaked secrets or paths.
|
||||||
|
9. Produce a structured handoff.
|
||||||
|
|
||||||
|
## Reliability behavior
|
||||||
|
|
||||||
|
- Claim at most the configured concurrency.
|
||||||
|
- Back off when no task is available or the coordinator is unavailable.
|
||||||
|
- Distinguish transient transport errors from permanent task errors.
|
||||||
|
- Stop successful completion after lease loss or `409 Conflict`.
|
||||||
|
- Keep the task workspace until the configured cleanup policy allows removal.
|
||||||
|
- Verify upload response metadata before sending completion.
|
||||||
|
- Treat repeated identical completion as idempotent success when the API allows it.
|
||||||
|
- Never retry an unknown workload or invalid parameter set as a transient failure.
|
||||||
|
|
||||||
|
## Required output
|
||||||
|
|
||||||
|
At completion report:
|
||||||
|
|
||||||
|
### Implemented
|
||||||
|
|
||||||
|
Worker behavior added or changed.
|
||||||
|
|
||||||
|
### API usage
|
||||||
|
|
||||||
|
Endpoints, headers, DTO fields, and status codes handled.
|
||||||
|
|
||||||
|
### Reliability
|
||||||
|
|
||||||
|
Heartbeat, retry, backoff, lease-loss, and cleanup behavior.
|
||||||
|
|
||||||
|
### Files changed
|
||||||
|
|
||||||
|
List each changed file and its purpose.
|
||||||
|
|
||||||
|
### Tests
|
||||||
|
|
||||||
|
Commands run and results, including contract tests.
|
||||||
|
|
||||||
|
### Compatibility
|
||||||
|
|
||||||
|
Effect on existing local workloads and CLI.
|
||||||
|
|
||||||
|
### Acceptance criteria
|
||||||
|
|
||||||
|
Checklist copied from the assigned CTX task.
|
||||||
|
|
||||||
|
### Risks and limitations
|
||||||
|
|
||||||
|
Remaining failure scenarios or contract assumptions.
|
||||||
@@ -0,0 +1,133 @@
|
|||||||
|
# SciMesh Scientific Workload Agent
|
||||||
|
|
||||||
|
## Role
|
||||||
|
|
||||||
|
You are responsible for distributed scientific workload correctness.
|
||||||
|
|
||||||
|
Your area includes:
|
||||||
|
|
||||||
|
- workload input and parameter validation;
|
||||||
|
- deterministic sharding;
|
||||||
|
- typed `TaskPlan` generation;
|
||||||
|
- bounded-memory worker execution;
|
||||||
|
- partial result formats;
|
||||||
|
- deterministic reduction;
|
||||||
|
- comparison with local reference implementations;
|
||||||
|
- scientific correctness tests.
|
||||||
|
|
||||||
|
Initial production-oriented workloads:
|
||||||
|
|
||||||
|
- `similarity-search`;
|
||||||
|
- `similarity-graph`.
|
||||||
|
|
||||||
|
Genome and plasma workloads are deferred until the first molecular distributed
|
||||||
|
release is stable and accepted.
|
||||||
|
|
||||||
|
## Read before working
|
||||||
|
|
||||||
|
Always read:
|
||||||
|
|
||||||
|
1. `PLAN.md`
|
||||||
|
2. the assigned CTX task
|
||||||
|
3. the current local workload implementation
|
||||||
|
4. the distributed workload protocol
|
||||||
|
5. relevant fixtures and tests
|
||||||
|
6. `STATUS.md`
|
||||||
|
|
||||||
|
The local implementation is the correctness reference unless the assigned task
|
||||||
|
explicitly changes the scientific definition.
|
||||||
|
|
||||||
|
## Hard rules
|
||||||
|
|
||||||
|
- Do not modify coordinator queue or state-machine logic.
|
||||||
|
- Task plans must be JSON-serializable.
|
||||||
|
- Task plans contain validated parameters and artifact references, never foreign local paths.
|
||||||
|
- Results must be deterministic for identical inputs and parameters.
|
||||||
|
- Distributed results must match the local reference implementation.
|
||||||
|
- Do not create or retain a dense N×N similarity matrix.
|
||||||
|
- Similarity graph must compare every unordered pair exactly once.
|
||||||
|
- Reducers must be independent of worker completion order.
|
||||||
|
- Memory usage must remain bounded.
|
||||||
|
- Shards and block indices must be deterministic.
|
||||||
|
- Partial outputs must use documented schemas.
|
||||||
|
- Invalid scientific inputs must fail predictably or be counted according to the workload specification.
|
||||||
|
- Do not change API endpoints or PostgreSQL state semantics.
|
||||||
|
- Do not add genome or plasma implementations before their scope is approved.
|
||||||
|
|
||||||
|
## Workflow
|
||||||
|
|
||||||
|
1. Establish and test the local reference result.
|
||||||
|
2. Define task boundaries and invariants.
|
||||||
|
3. Define the task payload schema.
|
||||||
|
4. Define the partial result schema.
|
||||||
|
5. Implement validation and planner.
|
||||||
|
6. Implement worker execution adapter.
|
||||||
|
7. Implement reducer.
|
||||||
|
8. Compare local and distributed outputs.
|
||||||
|
9. Test multiple shard or block sizes.
|
||||||
|
10. Test different worker completion orders.
|
||||||
|
11. Test retry without changing the final result.
|
||||||
|
12. Document memory bounds and scientific invariants.
|
||||||
|
13. Produce a structured handoff.
|
||||||
|
|
||||||
|
## Similarity-search invariants
|
||||||
|
|
||||||
|
- Resolve `query_id` once during planning.
|
||||||
|
- Each shard keeps a valid TSV header and stable `chunk_index`.
|
||||||
|
- Each shard returns at least the requested global `top_k`.
|
||||||
|
- Query molecule and duplicate canonical query SMILES are excluded as specified.
|
||||||
|
- Global reducer tie-breaking matches local SciMesh.
|
||||||
|
- Final result is independent of task completion order.
|
||||||
|
|
||||||
|
## Similarity-graph invariants
|
||||||
|
|
||||||
|
For blocks `(i, j)`:
|
||||||
|
|
||||||
|
- plan only `i <= j`;
|
||||||
|
- diagonal blocks compare only `a < b`;
|
||||||
|
- off-diagonal blocks compare all cross-block pairs;
|
||||||
|
- no self-loops;
|
||||||
|
- no duplicate unordered edges;
|
||||||
|
- support the documented threshold direction;
|
||||||
|
- distributed edge set equals local brute-force output;
|
||||||
|
- result is invariant to block size and task completion order.
|
||||||
|
|
||||||
|
## Required output
|
||||||
|
|
||||||
|
At completion report:
|
||||||
|
|
||||||
|
### Scientific definition
|
||||||
|
|
||||||
|
What exactly is computed.
|
||||||
|
|
||||||
|
### Sharding strategy
|
||||||
|
|
||||||
|
How input is split and why coverage is complete.
|
||||||
|
|
||||||
|
### Task payload
|
||||||
|
|
||||||
|
Documented JSON-compatible fields.
|
||||||
|
|
||||||
|
### Partial result
|
||||||
|
|
||||||
|
File format, ordering, and metrics.
|
||||||
|
|
||||||
|
### Reduction algorithm
|
||||||
|
|
||||||
|
How partial outputs become the final result.
|
||||||
|
|
||||||
|
### Correctness invariants
|
||||||
|
|
||||||
|
Properties that must always hold.
|
||||||
|
|
||||||
|
### Tests
|
||||||
|
|
||||||
|
Local versus distributed comparisons and commands run.
|
||||||
|
|
||||||
|
### Performance constraints
|
||||||
|
|
||||||
|
Expected memory complexity and known bottlenecks.
|
||||||
|
|
||||||
|
### Acceptance criteria
|
||||||
|
|
||||||
|
Checklist copied from the assigned CTX task.
|
||||||
@@ -0,0 +1,66 @@
|
|||||||
|
# Repository Guidelines
|
||||||
|
|
||||||
|
## Project Structure & Module Organization
|
||||||
|
|
||||||
|
SciMesh is a Python package for molecular-similarity workloads. Source lives in
|
||||||
|
`scimesh/`: `chemistry/` reads data and makes fingerprints, `workloads/`
|
||||||
|
contains commands, and `core/` provides the workload protocol and registry.
|
||||||
|
The worker daemon in `scimesh/worker/` is a coordinator client, not a database
|
||||||
|
client. Tests are in `tests/`; specifications in `docs/`; roadmap: `PLAN.md`.
|
||||||
|
|
||||||
|
For distributed work, read `.agents/`, `docs/api-contract.md`,
|
||||||
|
and `STATUS.md`. Use one CTX task per pull request; local workloads are the
|
||||||
|
scientific reference.
|
||||||
|
|
||||||
|
## Build, Test, and Development Commands
|
||||||
|
|
||||||
|
Create a virtual environment, install the package with development tools, and
|
||||||
|
run the suite:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
python -m venv .venv
|
||||||
|
source .venv/bin/activate
|
||||||
|
pip install -e '.[dev]'
|
||||||
|
pytest
|
||||||
|
```
|
||||||
|
|
||||||
|
Use `pytest tests/test_similarity_graph.py` for one module. Exercise the CLI
|
||||||
|
with `scimesh help` or `scimesh similarity-search --help`. Run `python -m build`
|
||||||
|
only when packaging is needed; install `build` first if necessary.
|
||||||
|
|
||||||
|
## Coding Style & Naming Conventions
|
||||||
|
|
||||||
|
Target Python 3.10+; type public APIs and exchanged data. Use four spaces,
|
||||||
|
`snake_case` for modules,
|
||||||
|
functions, and variables, `PascalCase` for classes, and descriptive test names
|
||||||
|
such as `test_graph_is_deterministic_across_block_sizes`. Keep CLI parsing in
|
||||||
|
workload modules and register new workloads through `scimesh/core/registry.py`;
|
||||||
|
do not add workload-specific logic to the main CLI.
|
||||||
|
|
||||||
|
Prefer few dependencies. RDKit is the chemistry dependency.
|
||||||
|
For worker/coordinator work, keep network payloads explicit and multi-line;
|
||||||
|
never make the worker access PostgreSQL directly.
|
||||||
|
|
||||||
|
## Testing Guidelines
|
||||||
|
|
||||||
|
Use pytest and add a regression test for every defect. Similarity code must be
|
||||||
|
checked against a small brute-force or fully sorted reference. Graph results
|
||||||
|
must be deterministic, have no self-loops or duplicate pairs, and remain
|
||||||
|
stable for different block sizes. Worker changes need success and failure
|
||||||
|
tests: checksum mismatch, lease failure, upload failure, and safe reporting.
|
||||||
|
Run the full `pytest` suite before committing.
|
||||||
|
|
||||||
|
## Commit & Pull Request Guidelines
|
||||||
|
|
||||||
|
Use short imperative commit subjects, for example `Add graph threshold mode` or
|
||||||
|
`Fix worker result and lease contracts`. Keep one logical change per commit.
|
||||||
|
In a pull request, state the problem, behaviour changed, tests run, and any API
|
||||||
|
or documentation changes. Link the relevant `CTX-*` item in `PLAN.md` for
|
||||||
|
distributed work. Do not commit datasets, generated CSV/PNG files, `.venv/`,
|
||||||
|
tokens, or local worker artifacts.
|
||||||
|
|
||||||
|
## Security & Protocol Rules
|
||||||
|
|
||||||
|
Upload worker results through the coordinator before posting completion; never
|
||||||
|
submit `file://` or `worker://` result URIs. Send failures to `/failure`, not
|
||||||
|
`/result`. Do not log bearer tokens, raw tracebacks, or private local paths.
|
||||||
@@ -0,0 +1,975 @@
|
|||||||
|
# SciMesh: master implementation plan
|
||||||
|
|
||||||
|
> **Purpose.** This document is the source plan for turning SciMesh from a
|
||||||
|
> local molecular CLI into a local-first distributed scientific-computation
|
||||||
|
> platform. It is intentionally detailed enough to split into independent task
|
||||||
|
> briefs for developers or coding agents.
|
||||||
|
>
|
||||||
|
> **Planning baseline.** This branch starts from `Workers`: the Python package
|
||||||
|
> has local `similarity-search` and `similarity-graph` workloads plus a Worker
|
||||||
|
> Daemon client. The coordinator and PostgreSQL implementation do not yet
|
||||||
|
> exist. The Worker contract and the Go/PostgreSQL design briefs in `docs/` are
|
||||||
|
> part of this plan.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 1. Product goal
|
||||||
|
|
||||||
|
SciMesh accepts a scientific run, turns it into independent tasks, dispatches
|
||||||
|
them to polling workers, persists task state and artifacts, combines partial
|
||||||
|
results, and exposes the final result and progress to a user.
|
||||||
|
|
||||||
|
The first production-oriented vertical slice is molecular computation:
|
||||||
|
|
||||||
|
- `similarity-search`: exact top-k Tanimoto search over ChEMBL shards;
|
||||||
|
- `similarity-graph`: exact sparse Tanimoto graph, where each pair is compared
|
||||||
|
once and only edges satisfying the chosen threshold are retained.
|
||||||
|
|
||||||
|
The platform must later support other scientific workloads without changing the
|
||||||
|
coordinator or worker state machine.
|
||||||
|
|
||||||
|
```text
|
||||||
|
User / simple UI / CLI
|
||||||
|
|
|
||||||
|
v
|
||||||
|
Go coordinator + PostgreSQL + coordinator artifact storage
|
||||||
|
|
|
||||||
|
+-- creates Job/Run -> Tasks -> leases one task at a time
|
||||||
|
|
|
||||||
|
v
|
||||||
|
Python Worker Daemons (outbound HTTP only)
|
||||||
|
|
|
||||||
|
+-- download artifact -> execute allowlisted workload -> upload partial result
|
||||||
|
|
|
||||||
|
v
|
||||||
|
Coordinator reducer -> final artifact -> download/status API
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 2. Scope, non-goals, and decisions
|
||||||
|
|
||||||
|
### 2.1 In scope
|
||||||
|
|
||||||
|
- Go 1.22+ coordinator service with PostgreSQL 15+;
|
||||||
|
- Python Worker Daemon running existing SciMesh workloads locally;
|
||||||
|
- durable job, task, worker, and artifact metadata;
|
||||||
|
- local coordinator-managed artifact storage for the first deployment;
|
||||||
|
- HTTP API for submit, poll/claim, heartbeat, artifact transfer, completion,
|
||||||
|
failure, job status, and result download;
|
||||||
|
- sharding and reduction for the two molecular workloads;
|
||||||
|
- a small server-rendered or static HTML status page after the API works;
|
||||||
|
- automated unit, integration, and contract tests;
|
||||||
|
- a reproducible local demo using one coordinator and two or more workers.
|
||||||
|
|
||||||
|
### 2.2 Explicit non-goals for the first release
|
||||||
|
|
||||||
|
- cloud object storage, Kubernetes, autoscaling, and multi-region operation;
|
||||||
|
- arbitrary shell commands sent by coordinator to workers;
|
||||||
|
- user accounts, multi-tenancy, billing, or sophisticated authorization;
|
||||||
|
- GPU scheduling and multiprocessing inside a worker;
|
||||||
|
- Docker as a required runtime dependency;
|
||||||
|
- video/CV processing implementation;
|
||||||
|
- a React/Vue frontend;
|
||||||
|
- exact resumability of a subprocess after host power loss.
|
||||||
|
|
||||||
|
### 2.3 Architectural decisions already made
|
||||||
|
|
||||||
|
| Decision | Choice | Rationale |
|
||||||
|
| --- | --- | --- |
|
||||||
|
| Coordinator | Go + `net/http` | One durable service for API, queue, artifacts, and reducer orchestration. |
|
||||||
|
| Database | PostgreSQL + `pgxpool` | Transactional leasing and concurrent `SKIP LOCKED` claims. |
|
||||||
|
| Migration tool | `golang-migrate` SQL migrations | Schema is reviewable independently of Go code. |
|
||||||
|
| Workers | Python | Reuses RDKit and the existing SciMesh workload code. |
|
||||||
|
| Worker connectivity | Outbound HTTP polling | Workers require no public inbound ports. |
|
||||||
|
| Queue model | Database rows, not a separate broker | Sufficient for the initial local-first deployment. |
|
||||||
|
| Artifact storage | Coordinator filesystem first | Durable and simple; can later be replaced by S3-compatible storage behind an interface. |
|
||||||
|
| Workload execution | Explicit allowlist + typed parameters | Never execute coordinator-provided shell commands. |
|
||||||
|
| Result correctness | Artifact upload before task completion | A completed task must reference a durable, coordinator-accessible result. |
|
||||||
|
|
||||||
|
### 2.4 Rules that must never be violated
|
||||||
|
|
||||||
|
1. Workers never use PostgreSQL credentials or execute SQL.
|
||||||
|
2. A task is leased atomically to at most one worker attempt.
|
||||||
|
3. A task becomes `completed` only after its result artifact is durable and
|
||||||
|
verified by the coordinator.
|
||||||
|
4. Every mutating worker request includes `worker_id` and `attempt`; stale
|
||||||
|
attempts receive `409 Conflict`.
|
||||||
|
5. Worker tokens are never forwarded to an external presigned download URL.
|
||||||
|
6. The worker runs an explicit Python command list, never `shell=True`.
|
||||||
|
7. Result and reducer outputs are deterministic for identical inputs and
|
||||||
|
parameters.
|
||||||
|
8. Similarity graph tasks must cover each original molecule pair exactly once.
|
||||||
|
9. No workload may create or retain a dense N×N similarity matrix.
|
||||||
|
10. The coordinator must not trust worker-supplied artifact paths, status, or
|
||||||
|
ownership claims without checking task state in PostgreSQL.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 3. Glossary and canonical lifecycle
|
||||||
|
|
||||||
|
| Term | Meaning |
|
||||||
|
| --- | --- |
|
||||||
|
| **Worker** | A registered process/machine capable of claiming tasks. |
|
||||||
|
| **Job** | A user-requested full computation. The UI may call it a **Run**; the database/API use `job`. |
|
||||||
|
| **Task** | One independently executable unit of a job. |
|
||||||
|
| **Attempt** | A monotonically increasing execution lease for a task. |
|
||||||
|
| **Lease** | Temporary exclusive assignment of a task to one worker. |
|
||||||
|
| **Artifact** | A durable input, shard, partial result, final result, or log file. |
|
||||||
|
| **Planner** | Workload code that validates a job and emits task payloads. |
|
||||||
|
| **Runner** | Worker-side code that executes one typed task locally. |
|
||||||
|
| **Reducer** | Coordinator-side code that merges all completed partial results into a final artifact. |
|
||||||
|
|
||||||
|
### 3.1 Job state machine
|
||||||
|
|
||||||
|
```text
|
||||||
|
CREATED -> PLANNING -> RUNNING -> REDUCING -> COMPLETED
|
||||||
|
| | |
|
||||||
|
+-------> FAILED <---+
|
||||||
|
RUNNING -> CANCELLED
|
||||||
|
```
|
||||||
|
|
||||||
|
### 3.2 Task state machine
|
||||||
|
|
||||||
|
```text
|
||||||
|
PENDING -> LEASED -> RUNNING -> COMPLETED
|
||||||
|
| | |
|
||||||
|
| +--------> FAILED
|
||||||
|
+-----> PENDING |
|
||||||
|
lease expiry +-> PENDING (attempts remain)
|
||||||
|
```
|
||||||
|
|
||||||
|
`LEASED` means the coordinator has returned a task. `RUNNING` means the worker
|
||||||
|
has successfully sent its first heartbeat/start acknowledgement. A lease expiry
|
||||||
|
may return either `LEASED` or `RUNNING` tasks to `PENDING` when attempts remain.
|
||||||
|
|
||||||
|
### 3.3 Artifact lifecycle
|
||||||
|
|
||||||
|
```text
|
||||||
|
upload input -> INPUT artifact -> planner creates SHARD artifacts
|
||||||
|
worker downloads SHARD -> executes -> uploads PARTIAL_RESULT artifact
|
||||||
|
reducer reads partial artifacts -> writes FINAL_RESULT artifact
|
||||||
|
user downloads final artifact
|
||||||
|
```
|
||||||
|
|
||||||
|
No `file://` or `worker://` URI is valid in persisted result metadata.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 4. Target repository layout
|
||||||
|
|
||||||
|
Keep the existing Python package at the repository root and add a Go
|
||||||
|
coordinator as a self-contained subproject.
|
||||||
|
|
||||||
|
```text
|
||||||
|
SciMesh/
|
||||||
|
PLAN.md
|
||||||
|
docs/
|
||||||
|
worker-daemon-task.md
|
||||||
|
database-integration-task.md
|
||||||
|
api-contract.md # created in Phase 0
|
||||||
|
scimesh/ # Python local workloads and worker daemon
|
||||||
|
chemistry/
|
||||||
|
core/
|
||||||
|
distributed/ # planner/reducer contracts, Phase 3
|
||||||
|
worker/
|
||||||
|
workloads/
|
||||||
|
tests/
|
||||||
|
unit/
|
||||||
|
integration/
|
||||||
|
contract/
|
||||||
|
coordinator/
|
||||||
|
go.mod
|
||||||
|
cmd/coordinator/main.go
|
||||||
|
internal/
|
||||||
|
config/
|
||||||
|
domain/
|
||||||
|
httpapi/
|
||||||
|
queue/
|
||||||
|
reducer/
|
||||||
|
storage/
|
||||||
|
store/postgres/
|
||||||
|
migrations/
|
||||||
|
tests/
|
||||||
|
scripts/
|
||||||
|
dev-start.sh # optional convenience script, not required runtime
|
||||||
|
```
|
||||||
|
|
||||||
|
Do not move the mature local workload code merely to satisfy this layout. Move
|
||||||
|
only when a distributed contract requires a clear shared module.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 5. Cross-service API contract
|
||||||
|
|
||||||
|
The API contract is a compatibility boundary. Before either side is implemented,
|
||||||
|
copy this section into `docs/api-contract.md` and treat it as versioned.
|
||||||
|
|
||||||
|
All worker endpoints require bearer authentication. Worker identity and attempt
|
||||||
|
are checked against the current task lease in PostgreSQL.
|
||||||
|
|
||||||
|
### 5.1 Register worker
|
||||||
|
|
||||||
|
```http
|
||||||
|
POST /workers/register
|
||||||
|
Authorization: Bearer <token>
|
||||||
|
Content-Type: application/json
|
||||||
|
|
||||||
|
{
|
||||||
|
"name": "lab-worker-01",
|
||||||
|
"capabilities": ["similarity-search", "similarity-graph"],
|
||||||
|
"cpu_count": 8,
|
||||||
|
"memory_mb": 16384
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
Response:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"worker_id": "uuid",
|
||||||
|
"heartbeat_interval_seconds": 15
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 5.2 Claim task
|
||||||
|
|
||||||
|
```http
|
||||||
|
POST /tasks/claim
|
||||||
|
Authorization: Bearer <token>
|
||||||
|
Content-Type: application/json
|
||||||
|
|
||||||
|
{
|
||||||
|
"worker_id": "uuid",
|
||||||
|
"capabilities": ["similarity-search", "similarity-graph"],
|
||||||
|
"max_concurrency": 1
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
- `204 No Content`: no compatible task is available.
|
||||||
|
- `200 OK`: a task is leased atomically.
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"task_id": "uuid",
|
||||||
|
"attempt": 1,
|
||||||
|
"lease_expires_at": "2026-07-22T12:05:00Z",
|
||||||
|
"workload": "similarity-search",
|
||||||
|
"input": {
|
||||||
|
"uri": "https://coordinator.example/tasks/uuid/input",
|
||||||
|
"sha256": "hex-sha256"
|
||||||
|
},
|
||||||
|
"parameters": {
|
||||||
|
"query_id": "CHEMBL939",
|
||||||
|
"top_k": 20
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 5.3 Renew lease
|
||||||
|
|
||||||
|
```http
|
||||||
|
POST /tasks/{task_id}/heartbeat
|
||||||
|
Authorization: Bearer <token>
|
||||||
|
Content-Type: application/json
|
||||||
|
|
||||||
|
{"worker_id": "uuid", "attempt": 1}
|
||||||
|
```
|
||||||
|
|
||||||
|
The response **must** contain a renewed deadline:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{"lease_expires_at": "2026-07-22T12:10:00Z"}
|
||||||
|
```
|
||||||
|
|
||||||
|
The worker schedules the next heartbeat before half of this returned TTL, never
|
||||||
|
using only a fixed interval.
|
||||||
|
|
||||||
|
### 5.4 Download input or shard
|
||||||
|
|
||||||
|
`GET /tasks/{task_id}/input` returns an artifact owned by the current task. The
|
||||||
|
worker verifies its SHA-256 before execution. If the returned URI redirects to
|
||||||
|
another origin, the worker must remove the coordinator bearer token.
|
||||||
|
|
||||||
|
### 5.5 Upload a partial artifact
|
||||||
|
|
||||||
|
```http
|
||||||
|
PUT /tasks/{task_id}/artifacts/{filename}
|
||||||
|
Authorization: Bearer <token>
|
||||||
|
Content-Type: text/csv
|
||||||
|
X-Worker-ID: uuid
|
||||||
|
X-Task-Attempt: 1
|
||||||
|
|
||||||
|
<streamed bytes>
|
||||||
|
```
|
||||||
|
|
||||||
|
The coordinator streams the body to its artifact storage, verifies ownership,
|
||||||
|
stores metadata and checksum, then returns:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"artifact_id": "uuid",
|
||||||
|
"uri": "https://coordinator.example/artifacts/uuid/download",
|
||||||
|
"sha256": "hex-sha256",
|
||||||
|
"size_bytes": 1234
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 5.6 Complete or fail task
|
||||||
|
|
||||||
|
```http
|
||||||
|
POST /tasks/{task_id}/result
|
||||||
|
Authorization: Bearer <token>
|
||||||
|
Content-Type: application/json
|
||||||
|
|
||||||
|
{
|
||||||
|
"worker_id": "uuid",
|
||||||
|
"attempt": 1,
|
||||||
|
"result": {
|
||||||
|
"artifact_id": "uuid",
|
||||||
|
"uri": "https://coordinator.example/artifacts/uuid/download",
|
||||||
|
"sha256": "hex-sha256",
|
||||||
|
"content_type": "text/csv"
|
||||||
|
},
|
||||||
|
"metrics": {"elapsed_seconds": 12.4, "processed_rows": 10000}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
`POST /tasks/{task_id}/failure` uses the same identity fields and contains only
|
||||||
|
sanitized `error_code` and `error_message` values. No Python traceback, token,
|
||||||
|
or absolute worker path may be sent.
|
||||||
|
|
||||||
|
### 5.7 Idempotency and errors
|
||||||
|
|
||||||
|
| Situation | Required response |
|
||||||
|
| --- | --- |
|
||||||
|
| No compatible task | `204` |
|
||||||
|
| Worker/attempt does not own lease | `409` |
|
||||||
|
| Artifact does not belong to task/attempt | `409` |
|
||||||
|
| Same completion, same manifest | `200`/`202` idempotent success |
|
||||||
|
| Same attempt, different manifest | `409` |
|
||||||
|
| Invalid parameters/input | `400` |
|
||||||
|
| Worker authentication failure | `401`/`403` |
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 6. PostgreSQL data model
|
||||||
|
|
||||||
|
The existing brief describes `jobs` and `tasks`. Add first-class worker and
|
||||||
|
artifact records before implementation. The database is the source of truth for
|
||||||
|
state; files are referenced by metadata rather than discovered from directories.
|
||||||
|
|
||||||
|
### 6.1 Tables
|
||||||
|
|
||||||
|
#### `workers`
|
||||||
|
|
||||||
|
| Field | Notes |
|
||||||
|
| --- | --- |
|
||||||
|
| `id UUID PK` | Returned on registration |
|
||||||
|
| `name text` | Human-readable; unique only if desired |
|
||||||
|
| `capabilities jsonb` | Allowlisted workload names |
|
||||||
|
| `status` | `online`, `busy`, `offline` |
|
||||||
|
| `last_heartbeat_at timestamptz` | Liveness visibility |
|
||||||
|
| `created_at`, `updated_at timestamptz` | Audit |
|
||||||
|
|
||||||
|
#### `jobs`
|
||||||
|
|
||||||
|
| Field | Notes |
|
||||||
|
| --- | --- |
|
||||||
|
| `id UUID PK` | User-visible identifier |
|
||||||
|
| `workload text` | Registered distributed workload name |
|
||||||
|
| `status` | `created`, `planning`, `running`, `reducing`, `completed`, `failed`, `cancelled` |
|
||||||
|
| `parameters jsonb` | Validated job parameters |
|
||||||
|
| `input_artifact_id UUID` | Original uploaded data |
|
||||||
|
| `result_artifact_id UUID nullable` | Final result |
|
||||||
|
| `total_tasks`, `completed_tasks`, `failed_tasks` | Progress counters, updated transactionally |
|
||||||
|
| timestamps and `error_message` | Audit and failure status |
|
||||||
|
|
||||||
|
#### `tasks`
|
||||||
|
|
||||||
|
| Field | Notes |
|
||||||
|
| --- | --- |
|
||||||
|
| `id UUID PK`, `job_id UUID FK` | Identity and ownership |
|
||||||
|
| `chunk_index int` | Unique within job; deterministic reducer order |
|
||||||
|
| `workload text`, `parameters jsonb` | Typed task payload |
|
||||||
|
| `input_artifact_id UUID` | Dataset or shard |
|
||||||
|
| `status` | `pending`, `leased`, `running`, `completed`, `failed`, `cancelled` |
|
||||||
|
| `attempt`, `max_attempts` | Retry accounting |
|
||||||
|
| `lease_owner UUID nullable`, `lease_expires_at nullable` | Exclusive lease |
|
||||||
|
| `result_artifact_id UUID nullable` | Uploaded partial output |
|
||||||
|
| `metrics jsonb`, errors, timestamps, `version int` | Audit and concurrency |
|
||||||
|
|
||||||
|
#### `artifacts`
|
||||||
|
|
||||||
|
| Field | Notes |
|
||||||
|
| --- | --- |
|
||||||
|
| `id UUID PK` | Stable artifact identity |
|
||||||
|
| `job_id UUID FK`, `task_id UUID FK nullable` | Ownership |
|
||||||
|
| `kind` | `input`, `shard`, `partial_result`, `final_result`, `log` |
|
||||||
|
| `filename`, `storage_key`, `content_type` | Storage metadata |
|
||||||
|
| `size_bytes`, `sha256` | Integrity metadata |
|
||||||
|
| `created_at` | Audit |
|
||||||
|
|
||||||
|
### 6.2 Required constraints and queries
|
||||||
|
|
||||||
|
- unique `(job_id, chunk_index)` for tasks;
|
||||||
|
- unique `(task_id, kind)` for single-result task workloads;
|
||||||
|
- claim index: `(status, lease_expires_at, created_at)`;
|
||||||
|
- `attempt >= 0`, `max_attempts > 0`;
|
||||||
|
- a leased/running task must have owner and expiry;
|
||||||
|
- a completed task must reference a `partial_result` artifact;
|
||||||
|
- a completed job must reference a `final_result` artifact;
|
||||||
|
- `list_completed_results(job_id)` orders by `chunk_index`, never insertion time.
|
||||||
|
|
||||||
|
Atomic claim uses one PostgreSQL transaction with `FOR UPDATE SKIP LOCKED`.
|
||||||
|
Never write a `SELECT pending task` followed by an unguarded later `UPDATE`.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 7. Workload evolution
|
||||||
|
|
||||||
|
The current local `Workload` CLI interface is intentionally small. Distributed
|
||||||
|
execution needs a second, explicit contract. Do not force every CLI helper into
|
||||||
|
the distributed interface; adapt only workloads that can be planned and reduced.
|
||||||
|
|
||||||
|
```python
|
||||||
|
class DistributedWorkload(Protocol):
|
||||||
|
name: str
|
||||||
|
version: str
|
||||||
|
|
||||||
|
def validate_job(self, input_path: Path, parameters: dict[str, object]) -> None: ...
|
||||||
|
def plan(self, input_path: Path, parameters: dict[str, object], workspace: Path) -> list[TaskPlan]: ...
|
||||||
|
def execute_task(self, task: TaskPlan, workspace: Path) -> RunResult: ...
|
||||||
|
def reduce(self, partial_results: list[Path], parameters: dict[str, object], workspace: Path) -> FinalResult: ...
|
||||||
|
def describe(self) -> dict[str, object]: ...
|
||||||
|
```
|
||||||
|
|
||||||
|
`TaskPlan` is JSON-serializable and contains only validated parameters plus
|
||||||
|
artifact IDs/URIs. It never contains shell commands or local paths from another
|
||||||
|
machine.
|
||||||
|
|
||||||
|
### 7.1 Distributed similarity-search
|
||||||
|
|
||||||
|
1. Validate either `query_id` or `query_smiles`, never both.
|
||||||
|
2. Resolve a `query_id` once during planning and persist the canonical query
|
||||||
|
SMILES/identity in the job metadata.
|
||||||
|
3. Split the input TSV into deterministic shard artifacts. Every shard keeps a
|
||||||
|
header and a stable `chunk_index`.
|
||||||
|
4. Each worker streams one shard, skips invalid SMILES and the query molecule,
|
||||||
|
and writes a sorted local top-k CSV.
|
||||||
|
5. The reducer merges all local top-k outputs with a bounded heap using the same
|
||||||
|
deterministic tie-breaker as local SciMesh.
|
||||||
|
6. The final CSV must equal the current single-process result for the same input
|
||||||
|
and options.
|
||||||
|
|
||||||
|
Important: a local shard top-k must retain at least the global requested `k`.
|
||||||
|
The reducer cannot recover a candidate discarded by every shard.
|
||||||
|
|
||||||
|
### 7.2 Distributed similarity-graph
|
||||||
|
|
||||||
|
1. Parse valid molecules once during planning or produce deterministic molecule
|
||||||
|
block artifacts with stable block indices.
|
||||||
|
2. Emit one task for every block pair `(i, j)` where `i <= j`.
|
||||||
|
3. A diagonal task compares only pairs inside one block with local index `a < b`.
|
||||||
|
4. An off-diagonal task compares every molecule in block `i` with every molecule
|
||||||
|
in block `j`.
|
||||||
|
5. Each task emits an edge-list CSV only for pairs satisfying the chosen
|
||||||
|
`threshold_direction` and threshold.
|
||||||
|
6. The reducer merges edge files, verifies no duplicate unordered pair, and
|
||||||
|
writes a deterministic sort order.
|
||||||
|
|
||||||
|
Correctness invariant:
|
||||||
|
|
||||||
|
```text
|
||||||
|
union(task-pairs) = all unordered molecule pairs
|
||||||
|
intersection(task-pairs) = empty
|
||||||
|
```
|
||||||
|
|
||||||
|
For a small fixture, the distributed result must exactly equal the local
|
||||||
|
brute-force graph for both `greater` and `less` threshold directions.
|
||||||
|
|
||||||
|
### 7.3 Future workload policy
|
||||||
|
|
||||||
|
A new workload is accepted only when it supplies:
|
||||||
|
|
||||||
|
- an input/parameter validator;
|
||||||
|
- an explicit sharding strategy;
|
||||||
|
- bounded-memory task execution;
|
||||||
|
- deterministic reduction semantics;
|
||||||
|
- fixture-based local and distributed correctness tests;
|
||||||
|
- a `describe()` payload for UI/API discovery.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 8. Milestones and dependency order
|
||||||
|
|
||||||
|
```text
|
||||||
|
M0 contracts + test fixtures
|
||||||
|
-> M1 Go coordinator skeleton + migrations
|
||||||
|
-> M2 transactional queue + worker registry
|
||||||
|
-> M3 artifact storage + worker contract integration
|
||||||
|
-> M4 distributed similarity-search vertical slice
|
||||||
|
-> M5 reducer + status/result API
|
||||||
|
-> M6 distributed similarity-graph
|
||||||
|
-> M7 UI, observability, CI hardening
|
||||||
|
```
|
||||||
|
|
||||||
|
Do not start graph distribution before the search vertical slice proves the
|
||||||
|
full artifact/lease/reducer lifecycle.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 9. Task briefs for implementation
|
||||||
|
|
||||||
|
Each section below is deliberately self-contained. When assigning work, copy
|
||||||
|
the task block plus the **Shared context** section and any listed dependency.
|
||||||
|
|
||||||
|
### Shared context for every assignee
|
||||||
|
|
||||||
|
```text
|
||||||
|
Project: SciMesh
|
||||||
|
Architecture: Go/PostgreSQL coordinator; Python workers; local artifact storage.
|
||||||
|
Hard rules: workers have no DB credentials; task claims are atomic; artifacts
|
||||||
|
must be durable before completion; no arbitrary commands; deterministic output.
|
||||||
|
Read first: PLAN.md sections 2, 3, 5, and the task's dependencies.
|
||||||
|
Do not refactor unrelated files. Add or update tests with every behavior change.
|
||||||
|
```
|
||||||
|
|
||||||
|
### CTX-00 — Freeze API and error contract
|
||||||
|
|
||||||
|
**Goal:** Create `docs/api-contract.md` from section 5 and make it the single
|
||||||
|
source of truth for the Go coordinator and Python Worker.
|
||||||
|
|
||||||
|
**Depends on:** none.
|
||||||
|
|
||||||
|
**Deliverables:** endpoint table, JSON schemas/examples, headers, artifact
|
||||||
|
upload ownership rule, error mapping, retry/idempotency policy, and an explicit
|
||||||
|
version marker (`v1`).
|
||||||
|
|
||||||
|
**Acceptance criteria:**
|
||||||
|
|
||||||
|
- all worker endpoints and their paths/methods are listed;
|
||||||
|
- heartbeat response includes `lease_expires_at`;
|
||||||
|
- completion references only coordinator-uploaded artifacts;
|
||||||
|
- failure endpoint is distinct from result endpoint;
|
||||||
|
- document states whether the coordinator uses UTC RFC 3339 timestamps.
|
||||||
|
|
||||||
|
**Out of scope:** implementing HTTP handlers.
|
||||||
|
|
||||||
|
### CTX-01 — Bootstrap Go coordinator
|
||||||
|
|
||||||
|
**Goal:** Add the `coordinator/` Go module and a minimal healthy HTTP service.
|
||||||
|
|
||||||
|
**Depends on:** CTX-00.
|
||||||
|
|
||||||
|
**Deliverables:** `go.mod`; config parser; structured logger; `GET /healthz`;
|
||||||
|
graceful shutdown; `pgxpool` lifecycle; migration command documentation.
|
||||||
|
|
||||||
|
**Inputs:** `DATABASE_URL`, `COORDINATOR_STORAGE_DIR`, `COORDINATOR_ADDR`,
|
||||||
|
`COORDINATOR_TOKEN`, request timeout, pool size.
|
||||||
|
|
||||||
|
**Acceptance criteria:**
|
||||||
|
|
||||||
|
- `go test ./...` passes;
|
||||||
|
- service refuses to start with missing/invalid required configuration;
|
||||||
|
- `/healthz` returns database readiness without exposing secrets;
|
||||||
|
- shutdown closes HTTP server and `pgxpool` cleanly;
|
||||||
|
- migration execution is explicit, not implicit on production startup.
|
||||||
|
|
||||||
|
**Out of scope:** queue endpoints and UI.
|
||||||
|
|
||||||
|
### CTX-02 — PostgreSQL schema and migration set
|
||||||
|
|
||||||
|
**Goal:** Implement versioned SQL migrations for workers, jobs, tasks, and
|
||||||
|
artifacts from section 6.
|
||||||
|
|
||||||
|
**Depends on:** CTX-01, CTX-00.
|
||||||
|
|
||||||
|
**Deliverables:** up/down SQL migrations; enum/check constraints; indexes;
|
||||||
|
repository-domain structs as needed for scanning rows.
|
||||||
|
|
||||||
|
**Acceptance criteria:**
|
||||||
|
|
||||||
|
- empty PostgreSQL database migrates up and down in integration tests;
|
||||||
|
- constraints reject invalid state combinations;
|
||||||
|
- migration test uses `TEST_DATABASE_URL`, never SQLite;
|
||||||
|
- `tasks(job_id, chunk_index)` uniqueness and claim index exist;
|
||||||
|
- no application code creates tables dynamically.
|
||||||
|
|
||||||
|
**Out of scope:** HTTP handlers and lease operations.
|
||||||
|
|
||||||
|
### CTX-03 — Transactional queue and lease repository
|
||||||
|
|
||||||
|
**Goal:** Implement PostgreSQL repository operations for create, claim, renew,
|
||||||
|
complete, fail, expiry, and job status.
|
||||||
|
|
||||||
|
**Depends on:** CTX-02.
|
||||||
|
|
||||||
|
**Required operations:**
|
||||||
|
|
||||||
|
```go
|
||||||
|
CreateJobWithTasks(ctx, input)
|
||||||
|
ClaimNextTask(ctx, workerID, capabilities, leaseDuration)
|
||||||
|
RenewLease(ctx, taskID, workerID, attempt, leaseDuration)
|
||||||
|
CompleteTask(ctx, input)
|
||||||
|
FailTask(ctx, input)
|
||||||
|
ExpireLeases(ctx, now)
|
||||||
|
GetJobStatus(ctx, jobID)
|
||||||
|
ListCompletedResults(ctx, jobID)
|
||||||
|
```
|
||||||
|
|
||||||
|
**Acceptance criteria:**
|
||||||
|
|
||||||
|
- concurrent claim test proves one task/attempt has one owner;
|
||||||
|
- renewal returns a new `lease_expires_at` timestamp;
|
||||||
|
- stale attempt cannot renew, upload, fail, or complete;
|
||||||
|
- expired task returns to pending or fails after final attempt;
|
||||||
|
- completion is idempotent for identical manifest and conflicts for a different
|
||||||
|
manifest;
|
||||||
|
- all operations are context-aware and parameterized.
|
||||||
|
|
||||||
|
**Out of scope:** artifact byte storage and Python worker changes.
|
||||||
|
|
||||||
|
### CTX-04 — Worker registry and coordinator HTTP handlers
|
||||||
|
|
||||||
|
**Goal:** Expose CTX-03 through versioned `net/http` handlers and add worker
|
||||||
|
registration/liveness state.
|
||||||
|
|
||||||
|
**Depends on:** CTX-00, CTX-03.
|
||||||
|
|
||||||
|
**Endpoints:** `POST /workers/register`, `POST /tasks/claim`,
|
||||||
|
`POST /tasks/{id}/heartbeat`, `GET /jobs/{id}`, and readiness endpoints.
|
||||||
|
|
||||||
|
**Acceptance criteria:**
|
||||||
|
|
||||||
|
- request DTOs are validated before calling services;
|
||||||
|
- no raw PostgreSQL errors leave the process;
|
||||||
|
- `204`, `400`, `401/403`, and `409` match CTX-00;
|
||||||
|
- heartbeat response returns renewed deadline;
|
||||||
|
- tests cover an offline worker and an expired lease;
|
||||||
|
- logs include request ID, worker ID, task ID, attempt, operation.
|
||||||
|
|
||||||
|
### CTX-05 — Coordinator artifact storage
|
||||||
|
|
||||||
|
**Goal:** Store input, shard, partial-result, and final-result files durably in
|
||||||
|
the coordinator filesystem and persist their metadata.
|
||||||
|
|
||||||
|
**Depends on:** CTX-02, CTX-04.
|
||||||
|
|
||||||
|
**Endpoints:** authenticated input upload/download and
|
||||||
|
`PUT /tasks/{id}/artifacts/{filename}`.
|
||||||
|
|
||||||
|
**Implementation rules:**
|
||||||
|
|
||||||
|
- stream request bodies to a staging file; never `ReadAll` a result;
|
||||||
|
- use sanitized generated storage keys, not user paths;
|
||||||
|
- calculate SHA-256 while streaming;
|
||||||
|
- atomically rename staging file only after successful write;
|
||||||
|
- verify worker lease owner and attempt before accepting task output;
|
||||||
|
- return a coordinator-owned durable URI and artifact ID.
|
||||||
|
|
||||||
|
**Acceptance criteria:**
|
||||||
|
|
||||||
|
- uploaded artifact is downloadable after coordinator restart;
|
||||||
|
- a foreign worker receives `409`;
|
||||||
|
- large-file test proves bounded-memory streaming behavior;
|
||||||
|
- checksum and size are persisted;
|
||||||
|
- failed upload leaves no visible artifact or orphan staging file.
|
||||||
|
|
||||||
|
### CTX-06 — Align Python Worker with the live Go contract
|
||||||
|
|
||||||
|
**Goal:** Adapt `scimesh/worker/` to CTX-00 through CTX-05 without changing
|
||||||
|
local workload algorithms.
|
||||||
|
|
||||||
|
**Depends on:** CTX-00, CTX-04, CTX-05.
|
||||||
|
|
||||||
|
**Required behavior:**
|
||||||
|
|
||||||
|
- register worker capabilities at startup;
|
||||||
|
- claim one task at a time;
|
||||||
|
- remove bearer token when input redirect changes origin;
|
||||||
|
- verify input checksum;
|
||||||
|
- renew lease from returned `lease_expires_at`;
|
||||||
|
- stream uploaded output with worker/attempt headers;
|
||||||
|
- submit the returned artifact manifest only after upload;
|
||||||
|
- send errors to `/failure`;
|
||||||
|
- preserve task directory until configured cleanup;
|
||||||
|
- reject unknown workload parameters.
|
||||||
|
|
||||||
|
**Acceptance criteria:**
|
||||||
|
|
||||||
|
- Python contract tests run against the real Go service in CI;
|
||||||
|
- no result references `worker://` or local paths;
|
||||||
|
- long fake runner causes multiple lease renewals;
|
||||||
|
- lost lease stops successful completion and reports conflict cleanly;
|
||||||
|
- worker CLI documents all environment variables and exits non-zero on invalid
|
||||||
|
configuration.
|
||||||
|
|
||||||
|
### CTX-07 — Distributed workload protocol and job planner
|
||||||
|
|
||||||
|
**Goal:** Add the Python `DistributedWorkload` adapter protocol and coordinator
|
||||||
|
planner bridge for registered workloads.
|
||||||
|
|
||||||
|
**Depends on:** CTX-05, CTX-06.
|
||||||
|
|
||||||
|
**Deliverables:** typed task-plan JSON; validator; planner registry; reducer
|
||||||
|
registry; workload descriptions; no direct coordinator dependency in local
|
||||||
|
chemistry helpers.
|
||||||
|
|
||||||
|
**Acceptance criteria:**
|
||||||
|
|
||||||
|
- unknown workload rejected before a job/task is written;
|
||||||
|
- every plan payload is JSON-serializable and uses artifact references;
|
||||||
|
- planner failure leaves no partial job/tasks transaction;
|
||||||
|
- test fixture demonstrates planning a two-shard dummy workload;
|
||||||
|
- reducer receives completed artifacts ordered by `chunk_index`.
|
||||||
|
|
||||||
|
### CTX-08 — Distributed similarity-search vertical slice
|
||||||
|
|
||||||
|
**Goal:** Implement planner, worker execution adapter, reducer, and end-to-end
|
||||||
|
tests for distributed exact top-k similarity search.
|
||||||
|
|
||||||
|
**Depends on:** CTX-07.
|
||||||
|
|
||||||
|
**Acceptance criteria:**
|
||||||
|
|
||||||
|
- query ID is resolved before shards execute, or query SMILES is validated once;
|
||||||
|
- shards are deterministic and contain valid TSV headers;
|
||||||
|
- each shard produces local top-k CSV and reports invalid-SMILES counts;
|
||||||
|
- reducer output matches local single-process SciMesh byte-for-byte apart from
|
||||||
|
permitted elapsed-time metadata;
|
||||||
|
- query molecule and duplicate canonical query SMILES are excluded;
|
||||||
|
- tie order is deterministic across worker completion order;
|
||||||
|
- test uses at least two workers and one retry.
|
||||||
|
|
||||||
|
### CTX-09 — Job reducer orchestration and final result API
|
||||||
|
|
||||||
|
**Goal:** When all task results are complete, run the appropriate reducer once,
|
||||||
|
persist a final artifact, update job state, and expose download/status.
|
||||||
|
|
||||||
|
**Depends on:** CTX-07, CTX-08.
|
||||||
|
|
||||||
|
**Acceptance criteria:**
|
||||||
|
|
||||||
|
- only one reducer process may transition a job into `reducing`;
|
||||||
|
- reducer is idempotent or protected by state/version transaction;
|
||||||
|
- reducer failure marks job failed with a sanitized error;
|
||||||
|
- `GET /jobs/{id}` reports counters and state correctly;
|
||||||
|
- final artifact has stored checksum and downloadable URI;
|
||||||
|
- integration test covers complete job lifecycle.
|
||||||
|
|
||||||
|
### CTX-10 — Distributed similarity-graph
|
||||||
|
|
||||||
|
**Goal:** Implement block-pair planning, execution, and deterministic reduction
|
||||||
|
for the exact sparse similarity graph.
|
||||||
|
|
||||||
|
**Depends on:** CTX-08, CTX-09.
|
||||||
|
|
||||||
|
**Acceptance criteria:**
|
||||||
|
|
||||||
|
- each block pair is planned once with stable `(left_block, right_block)`;
|
||||||
|
- diagonal and off-diagonal comparisons obey the pair invariant in section 7.2;
|
||||||
|
- no task creates a dense matrix;
|
||||||
|
- both `greater` and `less` threshold directions are preserved;
|
||||||
|
- reducer detects duplicate unordered pairs and fails safely;
|
||||||
|
- distributed output equals local brute-force output on a small fixture;
|
||||||
|
- result is invariant to block size and worker completion order.
|
||||||
|
|
||||||
|
### CTX-11 — Minimal dashboard and operator views
|
||||||
|
|
||||||
|
**Goal:** Add a small server-rendered or static HTML UI to inspect jobs, tasks,
|
||||||
|
workers, and download final artifacts.
|
||||||
|
|
||||||
|
**Depends on:** CTX-04, CTX-09.
|
||||||
|
|
||||||
|
**Acceptance criteria:**
|
||||||
|
|
||||||
|
- show worker name/capabilities/status/last heartbeat;
|
||||||
|
- show job status and completed/total task progress;
|
||||||
|
- show task attempt, lease owner, and sanitized error;
|
||||||
|
- refresh with simple polling; no frontend framework required;
|
||||||
|
- final result link is available only in `completed` state;
|
||||||
|
- HTML escapes user-controlled values.
|
||||||
|
|
||||||
|
### CTX-12 — Reliability, security, and CI hardening
|
||||||
|
|
||||||
|
**Goal:** Make the vertical slice safe to demo and difficult to regress.
|
||||||
|
|
||||||
|
**Depends on:** CTX-06 through CTX-11.
|
||||||
|
|
||||||
|
**Work items:**
|
||||||
|
|
||||||
|
- CI jobs for Python tests, Go tests, `go vet`, migrations, and contract tests;
|
||||||
|
- coordinator request-size limits and timeouts;
|
||||||
|
- token configuration/rotation documentation;
|
||||||
|
- structured logs and correlation IDs;
|
||||||
|
- cleanup policy for failed worker directories and stale staging files;
|
||||||
|
- metrics/status endpoint suitable for local monitoring;
|
||||||
|
- retry/backoff test matrix;
|
||||||
|
- release checklist and local two-worker demo script.
|
||||||
|
|
||||||
|
**Acceptance criteria:**
|
||||||
|
|
||||||
|
- clean checkout can run the documented demo;
|
||||||
|
- all quality gates execute in CI;
|
||||||
|
- no secret appears in logs/tests/errors;
|
||||||
|
- failure/retry scenarios have automated coverage;
|
||||||
|
- README contains architecture diagram, security caveat, and troubleshooting.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 10. Suggested assignment bundles
|
||||||
|
|
||||||
|
These bundles minimize overlap. Do not run tasks from the same bundle in
|
||||||
|
parallel unless one engineer owns integration.
|
||||||
|
|
||||||
|
| Bundle | Tasks | Recommended owner |
|
||||||
|
| --- | --- | --- |
|
||||||
|
| Contract and data | CTX-00, CTX-02, CTX-03 | Go/PostgreSQL engineer |
|
||||||
|
| Coordinator API | CTX-01, CTX-04, CTX-05 | Go backend engineer |
|
||||||
|
| Python transport | CTX-06 | Python worker engineer |
|
||||||
|
| Distributed computation | CTX-07, CTX-08, CTX-10 | Scientific Python engineer |
|
||||||
|
| Product surface | CTX-09, CTX-11 | Full-stack/backend engineer |
|
||||||
|
| Quality gate | CTX-12 | DevOps/QA engineer |
|
||||||
|
|
||||||
|
Suggested order for a small team:
|
||||||
|
|
||||||
|
```text
|
||||||
|
Week 1: CTX-00 + CTX-01 + CTX-02
|
||||||
|
Week 2: CTX-03 + CTX-04
|
||||||
|
Week 3: CTX-05 + CTX-06
|
||||||
|
Week 4: CTX-07 + CTX-08
|
||||||
|
Week 5: CTX-09 + CTX-10
|
||||||
|
Week 6: CTX-11 + CTX-12
|
||||||
|
```
|
||||||
|
|
||||||
|
This is an ordering aid, not a deadline commitment. Start the next milestone
|
||||||
|
only after its predecessor's acceptance criteria are demonstrably met.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 11. Test strategy
|
||||||
|
|
||||||
|
### 11.1 Python unit tests
|
||||||
|
|
||||||
|
- SMILES parsing, fingerprints, deterministic top-k and graph ordering;
|
||||||
|
- Worker parameter allowlist and CLI argument mapping;
|
||||||
|
- checksum verification and cross-origin auth stripping;
|
||||||
|
- heartbeat rescheduling from returned TTL;
|
||||||
|
- artifact upload manifest generation;
|
||||||
|
- runner failure sanitization and cleanup behavior.
|
||||||
|
|
||||||
|
### 11.2 Go unit tests
|
||||||
|
|
||||||
|
- configuration parser and error mapping;
|
||||||
|
- storage-key sanitization;
|
||||||
|
- request DTO validation;
|
||||||
|
- service state transition guards;
|
||||||
|
- reducer invocation selection.
|
||||||
|
|
||||||
|
### 11.3 PostgreSQL integration tests
|
||||||
|
|
||||||
|
- migrations up/down;
|
||||||
|
- concurrent `ClaimNextTask` without duplicate lease;
|
||||||
|
- lease expiry/retry/exhaustion;
|
||||||
|
- stale attempt conflict;
|
||||||
|
- artifact ownership and task completion idempotency;
|
||||||
|
- deterministic partial-result retrieval order.
|
||||||
|
|
||||||
|
### 11.4 Contract tests
|
||||||
|
|
||||||
|
Run Python Worker tests against a real Go coordinator and PostgreSQL instance.
|
||||||
|
At minimum prove:
|
||||||
|
|
||||||
|
1. worker registers;
|
||||||
|
2. worker claims one task;
|
||||||
|
3. worker receives a renewed lease;
|
||||||
|
4. input checksum is verified;
|
||||||
|
5. result is uploaded and available from coordinator storage;
|
||||||
|
6. completion persists the correct artifact;
|
||||||
|
7. bad checksum or runner failure reaches `/failure`;
|
||||||
|
8. a stale worker cannot complete after its lease has expired.
|
||||||
|
|
||||||
|
### 11.5 End-to-end tests
|
||||||
|
|
||||||
|
- two-worker similarity-search run equals local CLI output;
|
||||||
|
- retry one failed shard and still finish deterministically;
|
||||||
|
- graph on small fixture has all and only brute-force threshold edges;
|
||||||
|
- stopping a worker mid-task requeues its task after lease expiry;
|
||||||
|
- final result remains downloadable after coordinator restart.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 12. Review gates
|
||||||
|
|
||||||
|
Before merging a task, reviewer checks:
|
||||||
|
|
||||||
|
### Every task
|
||||||
|
|
||||||
|
- scope matches one CTX block;
|
||||||
|
- tests cover new behavior and pass;
|
||||||
|
- no unrelated refactor or generated data is committed;
|
||||||
|
- errors are sanitized and parameters validated;
|
||||||
|
- documentation/API contract updated when behavior changes.
|
||||||
|
|
||||||
|
### Worker changes
|
||||||
|
|
||||||
|
- no SQL, DB credentials, or arbitrary command execution;
|
||||||
|
- no worker-local URI persisted as a result;
|
||||||
|
- artifact transfer is streamed and checksum verified;
|
||||||
|
- heartbeat uses returned deadline;
|
||||||
|
- cross-origin requests do not receive coordinator token.
|
||||||
|
|
||||||
|
### Coordinator changes
|
||||||
|
|
||||||
|
- mutating actions are transactional;
|
||||||
|
- lease/task ownership checked at every mutation;
|
||||||
|
- PostgreSQL operations are parameterized;
|
||||||
|
- result completion is idempotent;
|
||||||
|
- artifact ownership and storage keys are validated;
|
||||||
|
- state transitions cannot skip required intermediate conditions.
|
||||||
|
|
||||||
|
### Scientific workload changes
|
||||||
|
|
||||||
|
- local reference result exists;
|
||||||
|
- distributed result is compared to local result on fixtures;
|
||||||
|
- reduction ordering is deterministic;
|
||||||
|
- pair coverage/duplicate invariants are tested for graphs;
|
||||||
|
- memory remains bounded as specified.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 13. Deferred backlog
|
||||||
|
|
||||||
|
Do not start these before CTX-12 is accepted.
|
||||||
|
|
||||||
|
- Replace local artifact storage with S3/MinIO behind an `ArtifactStore` API.
|
||||||
|
- Add worker labels/capacity-aware scheduling and concurrency > 1.
|
||||||
|
- Add cancellation propagation to workers.
|
||||||
|
- Add image outputs and final PDF reporting to job artifacts.
|
||||||
|
- Add CV/video workloads using the same planner/runner/reducer contract.
|
||||||
|
- Add observability export (Prometheus/OpenTelemetry).
|
||||||
|
- Add per-user/project authorization and signed artifact URLs.
|
||||||
|
- Add shard caching and content-addressed input deduplication.
|
||||||
|
- Add job priority and fair scheduling.
|
||||||
|
- Add a CLI for submitting and monitoring remote jobs.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 14. Definition of the first usable distributed release
|
||||||
|
|
||||||
|
The release is complete when all of the following are true:
|
||||||
|
|
||||||
|
1. A user uploads a small ChEMBL TSV and starts a similarity-search job.
|
||||||
|
2. The Go coordinator creates PostgreSQL job/task/artifact records.
|
||||||
|
3. Two Python workers register, claim distinct shards, renew leases, and upload
|
||||||
|
partial CSVs.
|
||||||
|
4. The coordinator reduces partial results into a deterministic final CSV.
|
||||||
|
5. The user observes progress and downloads the final CSV.
|
||||||
|
6. Killing one worker requeues only its lease after expiry; the job still
|
||||||
|
completes within its attempt budget.
|
||||||
|
7. The same fixture run matches local SciMesh output.
|
||||||
|
8. CI executes Python, Go, PostgreSQL, and contract tests successfully.
|
||||||
|
|
||||||
|
Until these eight conditions are met, SciMesh is a promising set of components,
|
||||||
|
not yet a complete distributed platform.
|
||||||
@@ -0,0 +1,56 @@
|
|||||||
|
# SciMesh Status
|
||||||
|
|
||||||
|
**Updated:** 2026-07-23
|
||||||
|
**Branch baseline:** `planning` at `13f9a0b`
|
||||||
|
|
||||||
|
## Current state
|
||||||
|
|
||||||
|
The local Python molecular workloads are implemented and tested. They provide
|
||||||
|
the reference behaviour for future distributed execution:
|
||||||
|
|
||||||
|
- `similarity-search`: streaming ChEMBL TSV search, Morgan fingerprints,
|
||||||
|
Tanimoto scoring, heap-based top-k, CSV and image output;
|
||||||
|
- `similarity-graph`: exact sparse graph, block-based pair comparisons,
|
||||||
|
deterministic CSV output;
|
||||||
|
- Python Worker skeleton: claim, heartbeat, input checksum validation,
|
||||||
|
artifact upload, completion and failure reporting.
|
||||||
|
|
||||||
|
The Go coordinator, PostgreSQL schema, coordinator artifact storage, planner,
|
||||||
|
reducer, and end-to-end distributed execution are **not implemented yet**.
|
||||||
|
|
||||||
|
## Milestone tracker
|
||||||
|
|
||||||
|
| CTX | Status | Notes |
|
||||||
|
| --- | --- | --- |
|
||||||
|
| CTX-00 API and error contract | Ready to implement | `docs/api-contract.md` created; needs owner review/freeze. |
|
||||||
|
| CTX-01 Go coordinator bootstrap | Not started | Depends on CTX-00. |
|
||||||
|
| CTX-02 PostgreSQL migrations | Not started | Depends on CTX-00 and CTX-01. |
|
||||||
|
| CTX-03 Transactional queue | Not started | Depends on CTX-02. |
|
||||||
|
| CTX-04 Worker registry and HTTP API | Not started | Depends on CTX-03. |
|
||||||
|
| CTX-05 Artifact storage | Not started | Depends on CTX-02 and CTX-04. |
|
||||||
|
| CTX-06 Python Worker live-contract alignment | Partially prepared | Worker skeleton exists; needs real Go contract tests. |
|
||||||
|
| CTX-07 Distributed workload protocol | Not started | Depends on artifact and Worker contracts. |
|
||||||
|
| CTX-08 Distributed similarity-search | Not started | Local reference exists. |
|
||||||
|
| CTX-09 Reducer and final-result API | Not started | Depends on CTX-07 and CTX-08. |
|
||||||
|
| CTX-10 Distributed similarity-graph | Not started | Local reference exists. |
|
||||||
|
| CTX-11 Dashboard/operator view | Not started | Deferred until API and reducer work. |
|
||||||
|
| CTX-12 Reliability, security, CI | Not started | Final milestone. |
|
||||||
|
|
||||||
|
## Next recommended assignment
|
||||||
|
|
||||||
|
Assign **CTX-00** to the coordinator role in `.agents/coordinator.md`: review
|
||||||
|
and freeze `docs/api-contract.md` against `PLAN.md`. Do not begin coordinator
|
||||||
|
or Worker API implementation until the contract owner accepts it.
|
||||||
|
|
||||||
|
## Known constraints
|
||||||
|
|
||||||
|
- Distributed execution is not available; use the local `scimesh` CLI.
|
||||||
|
- No Go module, PostgreSQL migrations, runtime configuration, or integration
|
||||||
|
environment exists yet.
|
||||||
|
- Local worker unit tests do not prove interoperability with a live coordinator.
|
||||||
|
|
||||||
|
## Update rule
|
||||||
|
|
||||||
|
The integration role updates this file only after collecting command output,
|
||||||
|
test evidence, and accepted changes. State facts, revision hashes, blockers,
|
||||||
|
and the next unblocked CTX task; do not mark work complete based on plans alone.
|
||||||
@@ -0,0 +1,151 @@
|
|||||||
|
# SciMesh Coordinator API Contract
|
||||||
|
|
||||||
|
**Status:** draft, version 1. This document is the compatibility boundary
|
||||||
|
between the Go coordinator and the Python Worker. Change it only in the same
|
||||||
|
pull request as both implementation and contract tests.
|
||||||
|
|
||||||
|
## General rules
|
||||||
|
|
||||||
|
- All worker endpoints require `Authorization: Bearer <token>`.
|
||||||
|
- Times use UTC RFC 3339, for example `2026-07-23T12:05:00Z`.
|
||||||
|
- JSON requests and responses use `application/json`.
|
||||||
|
- `worker_id` and `attempt` identify a lease. The coordinator validates them
|
||||||
|
transactionally on every task mutation.
|
||||||
|
- A task becomes `completed` only after a coordinator-owned artifact is durable.
|
||||||
|
- Identical repeated completion is successful; a different result for the same
|
||||||
|
attempt is a conflict.
|
||||||
|
|
||||||
|
## Worker registration
|
||||||
|
|
||||||
|
```http
|
||||||
|
POST /workers/register
|
||||||
|
|
||||||
|
{"name":"lab-worker-01","capabilities":["similarity-search"],"cpu_count":8,"memory_mb":16384}
|
||||||
|
```
|
||||||
|
|
||||||
|
Returns `200 OK`:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{"worker_id":"uuid","heartbeat_interval_seconds":15}
|
||||||
|
```
|
||||||
|
|
||||||
|
## Task lifecycle
|
||||||
|
|
||||||
|
### Claim
|
||||||
|
|
||||||
|
```http
|
||||||
|
POST /tasks/claim
|
||||||
|
|
||||||
|
{"worker_id":"uuid","capabilities":["similarity-search"],"max_concurrency":1}
|
||||||
|
```
|
||||||
|
|
||||||
|
Returns `204 No Content` when no compatible task exists. A successful atomic
|
||||||
|
claim returns `200 OK`:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"task_id":"uuid",
|
||||||
|
"attempt":1,
|
||||||
|
"lease_expires_at":"2026-07-23T12:05:00Z",
|
||||||
|
"workload":"similarity-search",
|
||||||
|
"input":{"uri":"https://coordinator.example/tasks/uuid/input","sha256":"hex-sha256"},
|
||||||
|
"parameters":{"query_id":"CHEMBL939","top_k":20}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
The claim is one PostgreSQL transaction using `FOR UPDATE SKIP LOCKED`.
|
||||||
|
|
||||||
|
### Heartbeat
|
||||||
|
|
||||||
|
```http
|
||||||
|
POST /tasks/{task_id}/heartbeat
|
||||||
|
|
||||||
|
{"worker_id":"uuid","attempt":1}
|
||||||
|
```
|
||||||
|
|
||||||
|
Returns `200 OK` and the renewed deadline:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{"lease_expires_at":"2026-07-23T12:10:00Z"}
|
||||||
|
```
|
||||||
|
|
||||||
|
The Worker schedules its next heartbeat before half of the returned TTL.
|
||||||
|
|
||||||
|
### Input download
|
||||||
|
|
||||||
|
`GET /tasks/{task_id}/input` returns the claimed task input. The Worker verifies
|
||||||
|
its SHA-256 before execution. On a redirect to another origin, it removes the
|
||||||
|
coordinator bearer token.
|
||||||
|
|
||||||
|
## Artifact upload
|
||||||
|
|
||||||
|
```http
|
||||||
|
PUT /tasks/{task_id}/artifacts/{filename}
|
||||||
|
Content-Type: text/csv
|
||||||
|
X-Worker-ID: uuid
|
||||||
|
X-Task-Attempt: 1
|
||||||
|
|
||||||
|
<streamed bytes>
|
||||||
|
```
|
||||||
|
|
||||||
|
The coordinator streams the body to storage, checks lease ownership, records
|
||||||
|
the checksum and returns `201 Created`:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"artifact_id":"uuid",
|
||||||
|
"uri":"https://coordinator.example/artifacts/uuid/download",
|
||||||
|
"sha256":"hex-sha256",
|
||||||
|
"size_bytes":1234
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
The returned URI is the only URI the Worker may send in task completion.
|
||||||
|
`worker://` and `file://` are invalid.
|
||||||
|
|
||||||
|
## Completion and failure
|
||||||
|
|
||||||
|
```http
|
||||||
|
POST /tasks/{task_id}/result
|
||||||
|
|
||||||
|
{
|
||||||
|
"worker_id":"uuid",
|
||||||
|
"attempt":1,
|
||||||
|
"result":{
|
||||||
|
"artifact_id":"uuid",
|
||||||
|
"uri":"https://coordinator.example/artifacts/uuid/download",
|
||||||
|
"sha256":"hex-sha256",
|
||||||
|
"content_type":"text/csv"
|
||||||
|
},
|
||||||
|
"metrics":{"elapsed_seconds":12.4,"processed_rows":10000}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
The coordinator returns `200`, `201`, or `202` for a valid completion. It must
|
||||||
|
verify that the artifact belongs to that task and attempt before completing it.
|
||||||
|
|
||||||
|
Use `POST /tasks/{task_id}/failure` only for a failed attempt:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{"worker_id":"uuid","attempt":1,"error_code":"ValueError","error_message":"input checksum mismatch"}
|
||||||
|
```
|
||||||
|
|
||||||
|
Messages are sanitised: no token, traceback, absolute local path, or raw input.
|
||||||
|
|
||||||
|
## Error responses
|
||||||
|
|
||||||
|
| Situation | Response |
|
||||||
|
| --- | --- |
|
||||||
|
| Invalid JSON, field, or parameter | `400 Bad Request` |
|
||||||
|
| Missing or invalid authentication | `401 Unauthorized` / `403 Forbidden` |
|
||||||
|
| Worker/attempt does not own an active lease | `409 Conflict` |
|
||||||
|
| Artifact does not belong to the task/attempt | `409 Conflict` |
|
||||||
|
| Same attempt, different completion manifest | `409 Conflict` |
|
||||||
|
| Unexpected coordinator failure | `500` without internal details |
|
||||||
|
|
||||||
|
## Compatibility tests
|
||||||
|
|
||||||
|
Contract tests must cover: registration, `204` claim, successful claim,
|
||||||
|
heartbeat renewal, foreign worker and stale attempt conflicts, streamed upload,
|
||||||
|
checksum mismatch, success after upload, failure through `/failure`, and
|
||||||
|
idempotent completion.
|
||||||
@@ -0,0 +1,57 @@
|
|||||||
|
# Памятка для доработки SciMesh Worker
|
||||||
|
|
||||||
|
Это короткие правила по итогам ревью первой версии воркера. Перед новой
|
||||||
|
задачей прочитай также [PLAN.md](../PLAN.md) и документ задачи, который тебе
|
||||||
|
дали.
|
||||||
|
|
||||||
|
## Что делать
|
||||||
|
|
||||||
|
- Считай воркер клиентом. Очередью, статусами задач и PostgreSQL управляет
|
||||||
|
только coordinator.
|
||||||
|
- Сначала скачай и проверь входной файл по SHA-256, затем запусти расчёт.
|
||||||
|
- Сохраняй файлы только в своей папке `task_id/attempt`.
|
||||||
|
- После расчёта сначала загрузи результат через coordinator, затем отправляй
|
||||||
|
результат задачи.
|
||||||
|
- В `result` передавай URI, который вернул coordinator, плюс SHA-256 и тип
|
||||||
|
файла.
|
||||||
|
- При ошибке вызывай `/tasks/{id}/failure`; при успехе —
|
||||||
|
`/tasks/{id}/result`.
|
||||||
|
- Для heartbeat используй новое `lease_expires_at`, возвращённое coordinator.
|
||||||
|
- Во все запросы, связанные с задачей, передавай `worker_id` и `attempt`.
|
||||||
|
- Пиши тесты не только на успех: проверь ошибку runner, неверный checksum,
|
||||||
|
ошибку загрузки, потерю lease и пустой результат.
|
||||||
|
- Если поменял API, меняй одновременно Python-код, Go coordinator,
|
||||||
|
документацию и тесты.
|
||||||
|
|
||||||
|
## Чего не делать
|
||||||
|
|
||||||
|
- Не отправляй в coordinator пути вида `worker://...` или `file://...`.
|
||||||
|
Coordinator не видит локальные файлы воркера.
|
||||||
|
- Не используй `/result` для сообщения об ошибке.
|
||||||
|
- Не считай, что проверок Python достаточно: coordinator обязан сам проверить
|
||||||
|
владельца lease, attempt, срок lease и допустимость смены статуса.
|
||||||
|
- Не повторяй любой HTTP-запрос вслепую. Повторные запросы должны быть
|
||||||
|
безопасны: одинаковый `(task_id, attempt, worker_id)` не должен создавать
|
||||||
|
дубликаты.
|
||||||
|
- Не передавай bearer token на другой домен при редиректе и не пиши токены,
|
||||||
|
traceback или полные локальные пути в логи.
|
||||||
|
- Не помечай задачу завершённой, если артефакт не загружен надёжно.
|
||||||
|
- Не меняй контракт API «по ощущениям». Сначала зафиксируй JSON, статусы и
|
||||||
|
переходы состояний в документации.
|
||||||
|
|
||||||
|
## Мини-чеклист перед коммитом
|
||||||
|
|
||||||
|
- [ ] Результат загружен в storage coordinator до `POST /result`.
|
||||||
|
- [ ] Ошибка уходит в `POST /failure`, а не в `POST /result`.
|
||||||
|
- [ ] Heartbeat возвращает новый срок lease, и код его сохраняет.
|
||||||
|
- [ ] В запросах есть верные `worker_id` и `attempt`.
|
||||||
|
- [ ] Нет `worker://`, `file://`, токенов и секретов в результатах или логах.
|
||||||
|
- [ ] Добавлены тесты на новый сценарий и на ошибку.
|
||||||
|
- [ ] `pytest` и проверка форматирования проходят.
|
||||||
|
|
||||||
|
## Короткий контекст для нейронки
|
||||||
|
|
||||||
|
> SciMesh Worker — клиент coordinator, а не владелец очереди. Загружай
|
||||||
|
> результат через coordinator до завершения задачи. Успех отправляй в
|
||||||
|
> `/result`, ошибку — в `/failure`. Проверяй checksum, lease и attempt.
|
||||||
|
> Меняя протокол, обновляй Python, Go, документацию и тесты в одном изменении.
|
||||||
Reference in New Issue
Block a user