synth-ai 0.2.6.dev1__py3-none-any.whl → 0.2.6.dev2__py3-none-any.whl

synth_ai/cli/rl_demo.py

@@ -34,13 +34,17 @@ def register(cli):
     def rl_demo():
         """RL Demo commands (separate from legacy demo)."""
 
-    @rl_demo.command("check")
+    # Help pyright understand dynamic Click group attributes
+    from typing import Any, cast as _cast
+    _rlg = _cast(Any, rl_demo)
+
+    @_rlg.command("check")
     def rl_check():
         _forward(["rl_demo.check"])  # reuse same implementation
 
     # (prepare command removed; consolidated into configure)
 
-    @rl_demo.command("deploy")
+    @_rlg.command("deploy")
     @click.option("--local", is_flag=True, help="Run local FastAPI instead of Modal deploy")
     @click.option("--app", type=click.Path(), default=None, help="Path to Modal app.py for uv run modal deploy")
     @click.option("--name", type=str, default="synth-math-demo", help="Modal app name")
@@ -57,11 +61,19 @@ def register(cli):
             args.extend(["--script", script])
         _forward(args)
 
-    @rl_demo.command("configure")
+    @_rlg.command("configure")
     def rl_configure():
         _forward(["rl_demo.configure"]) 
 
-    @rl_demo.command("run")
+    @_rlg.command("init")
+    @click.option("--force", is_flag=True, help="Overwrite existing files in CWD")
+    def rl_init(force: bool):
+        args = ["rl_demo.init"]
+        if force:
+            args.append("--force")
+        _forward(args)
+
+    @_rlg.command("run")
     @click.option("--config", type=click.Path(), default=None, help="Path to TOML config (skip prompt)")
     @click.option("--batch-size", type=int, default=None)
     @click.option("--group-size", type=int, default=None)
@@ -84,6 +96,20 @@ def register(cli):
             args.append("--dry-run")
         _forward(args)
 
+    @_rlg.command("eval")
+    @click.option("--config", type=click.Path(), default=None, help="Path to TOML config (optional)")
+    @click.option("--model", type=str, default=None, help="Model to evaluate (default Qwen/Qwen3-0.6B)")
+    @click.option("--timeout", type=int, default=300, help="Seconds to wait for metrics")
+    def rl_eval(config: str | None, model: str | None, timeout: int):
+        args = ["rl_demo.eval"]
+        if config:
+            args.extend(["--config", config])
+        if model:
+            args.extend(["--model", model])
+        if timeout is not None:
+            args.extend(["--timeout", str(timeout)])
+        _forward(args)
+
     # Dotted aliases (top-level) for convenience: rl_demo.check etc.
     @cli.command("rl_demo.check")
     def rl_check_alias():
@@ -112,6 +138,14 @@ def register(cli):
     def rl_configure_alias():
         _forward(["rl_demo.configure"]) 
 
+    @cli.command("rl_demo.init")
+    @click.option("--force", is_flag=True, help="Overwrite existing files in CWD")
+    def rl_init_alias(force: bool):
+        args = ["rl_demo.init"]
+        if force:
+            args.append("--force")
+        _forward(args)
+
     @cli.command("rl_demo.run")
     @click.option("--config", type=click.Path(), default=None, help="Path to TOML config (skip prompt)")
     @click.option("--batch-size", type=int, default=None)
@@ -135,3 +169,17 @@ def register(cli):
             args.append("--dry-run")
         _forward(args)
 
+    @cli.command("rl_demo.eval")
+    @click.option("--config", type=click.Path(), default=None, help="Path to TOML config (optional)")
+    @click.option("--model", type=str, default=None, help="Model to evaluate (default Qwen/Qwen3-0.6B)")
+    @click.option("--timeout", type=int, default=300, help="Seconds to wait for metrics")
+    def rl_eval_alias(config: str | None, model: str | None, timeout: int):
+        args = ["rl_demo.eval"]
+        if config:
+            args.extend(["--config", config])
+        if model:
+            args.extend(["--model", model])
+        if timeout is not None:
+            args.extend(["--timeout", str(timeout)])
+        _forward(args)
+