| |
| """Sentinel script: check if a user message should re-engage the ml-intern harness. |
| |
| This is a Codex-plugin workaround for the lack of persistent event hooks. |
| The harness SKILL.md instructs Codex to run this before each response when |
| harness mode is active. If the output is RE-ENGAGE, Codex stays in harness mode. |
| If FREE, it may drop to generic behavior (unless the conversation history shows |
| prior harness activity). |
| """ |
|
|
| from __future__ import annotations |
|
|
| import sys |
|
|
| |
| HARNESS_KEYWORDS = [ |
| "train", "training", "fine-tune", "finetune", "fine tuning", "evaluation", |
| "evaluate", "benchmark", "dataset", "datasets", "model", "models", |
| "hugging face", "huggingface", "hf", "paper", "papers", "citation", |
| "rag", "embedding", "embeddings", "diffusion", "lora", "dpo", "grpo", |
| "sft", "trl", "transformers", "trackio", "peft", "accelerate", |
| "sentence-transformers", "sentence transformer", "inference", "deploy", |
| "space", "spaces", "job", "jobs", "gpu", "cuda", "oom", "batch size", |
| "learning rate", "epoch", "epochs", "loss", "metric", "metrics", |
| "accuracy", "f1", "bleu", "rouge", "perplexity", "kl divergence", |
| "tokenizer", "tokenization", "config", "yaml", "jsonl", "push_to_hub", |
| "hub_model_id", "smoke test", "preflight", "plan", "architecture", |
| "design", "research", "recipe", "hyperparameter", "ablation", "sweep", |
| "inspect", "validate", "schema", "split", "config", "parquet", |
| "messages", "prompt", "completion", "chosen", "rejected", |
| "go ahead", "do it", "now what", "continue", "next step", "proceed", |
| "ship", "artifact", "artifact", "publish", "upload", "repo", |
| ] |
|
|
|
|
| def should_re_engage(message: str) -> bool: |
| """Return True if the message should keep/re-engage harness mode.""" |
| msg_lower = message.lower() |
| for kw in HARNESS_KEYWORDS: |
| if kw in msg_lower: |
| return True |
| return False |
|
|
|
|
| def main() -> int: |
| message = sys.argv[1] if len(sys.argv) > 1 else "" |
| if should_re_engage(message): |
| print("RE-ENGAGE") |
| else: |
| print("FREE") |
| return 0 |
|
|
|
|
| if __name__ == "__main__": |
| sys.exit(main()) |
|
|
