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

synth_ai/__init__.py

@@ -5,39 +5,32 @@ Synth AI - Software for aiding the best and multiplying the will.
 # Environment exports - moved from synth-env
 from synth_ai.environments import *  # noqa
 import synth_ai.environments as environments  # expose module name for __all__
-from synth_ai.lm.core.main import LM  # Moved from zyk to lm for better organization
-from synth_ai.lm.provider_support.anthropic import Anthropic, AsyncAnthropic
+try:
+    from synth_ai.lm.core.main import LM  # Moved from zyk to lm for better organization
+except Exception:  # allow minimal imports (e.g., tracing) without LM stack
+    LM = None  # type: ignore
+try:
+    from synth_ai.lm.provider_support.anthropic import Anthropic, AsyncAnthropic
+except Exception:  # optional in minimal environments
+    Anthropic = AsyncAnthropic = None  # type: ignore
 
 # Provider support exports - moved from synth-sdk to synth_ai/lm
-from synth_ai.lm.provider_support.openai import AsyncOpenAI, OpenAI
+try:
+    from synth_ai.lm.provider_support.openai import AsyncOpenAI, OpenAI
+except Exception:
+    AsyncOpenAI = OpenAI = None  # type: ignore
 
-# Tracing exports - moved from synth-sdk (deprecated v1)
-from synth_ai.tracing_v1 import *  # noqa
-import synth_ai.tracing_v1 as tracing  # expose module name for __all__
-from synth_ai.tracing_v1.abstractions import (
-    EventPartitionElement,
-    RewardSignal,
-    SystemTrace,
-    TrainingQuestion,
-)
-from synth_ai.tracing_v1.decorators import trace_event_async, trace_event_sync
-from synth_ai.tracing_v1.upload import upload
+# Legacy tracing v1 is not required for v3 usage and can be unavailable in minimal envs.
+tracing = None  # type: ignore
+EventPartitionElement = RewardSignal = SystemTrace = TrainingQuestion = None  # type: ignore
+trace_event_async = trace_event_sync = upload = None  # type: ignore
 
-__version__ = "0.2.6"
+__version__ = "0.2.6.dev1"
 __all__ = [
     "LM",
-    "tracing",
     "OpenAI",
     "AsyncOpenAI",
     "Anthropic",
     "AsyncAnthropic",
     "environments",
-    # v1 tracing legacy API re-exports
-    "EventPartitionElement",
-    "RewardSignal",
-    "SystemTrace",
-    "TrainingQuestion",
-    "trace_event_async",
-    "trace_event_sync",
-    "upload",
-]  # Explicitly define public API
+]  # Explicitly define public API (v1 tracing omitted in minimal env)

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)
+