forgeenv source snapshot for training job

scripts/jobs/train_repair_agent.py

@@ -78,11 +78,22 @@ _sh([
     "torch==2.5.1", "torchvision==0.20.1",
 ])
 # `--no-deps` on openenv-core: it pins a different transformers/torch
-# stack that we don't want.
+# stack that we don't want. We still need its *runtime* imports:
+# ``import forgeenv`` -> ``ForgeEnvironment`` -> ``openenv.core`` pulls in
+# ``fastmcp`` (and friends) from ``openenv.core.env_server``.
 _sh([
     sys.executable, "-m", "pip", "install", "--no-deps",
     "openenv-core>=0.2.0",
 ])
+_sh([
+    sys.executable, "-m", "pip", "install",
+    "fastmcp>=3.0.0",
+    "gradio>=4.0.0",
+    "openai>=2.7.2",
+    "tomli>=2.3.0",
+    "tomli-w>=1.2.0",
+    "websockets>=15.0.1",
+])
 _sh([
     sys.executable, "-m", "pip", "install",
     "trl==1.2.0", "peft", "accelerate", "datasets",

scripts/preflight_check.py

@@ -75,6 +75,14 @@ def t1_imports() -> str:
     return f"trl={trl.__version__} transformers={transformers.__version__}"
 
 
+def t1b_openenv_job_extras() -> str:
+    """On HF Jobs we ``pip install openenv-core --no-deps`` then add the
+    packages openenv lists as requirements so ``import openenv.core`` works."""
+    import fastmcp  # noqa: F401
+
+    return "fastmcp (required by openenv.core.env_server on import)"
+
+
 def t2_dataset_load_and_format() -> str:
     import datasets as ds
 
@@ -322,6 +330,7 @@ def t10_find_trainer_state() -> str:
 def main() -> int:
     print(f"\n=== ForgeEnv preflight (repo: {REPO_ROOT}) ===\n", flush=True)
     _run("01 imports", t1_imports, required=True)
+    _run("01b openenv extras (job: after --no-deps)", t1b_openenv_job_extras, required=True)
     _run("02 dataset load + format", t2_dataset_load_and_format, required=True)
     _run("03 TRL configs (SFT/GRPO) accept kwargs", t3_trl_configs_accept_our_kwargs, required=True)
     _run("04 reward fn returns float", t4_reward_function_returns_float, required=True)