synth-ai 0.2.9.dev4__py3-none-any.whl → 0.2.9.dev5__py3-none-any.whl

synth_ai/cli/task_apps.py

@@ -1220,12 +1220,23 @@ def _write_modal_entrypoint(
         except Exception:
             remote_file_str = None
     module_name = entry.config_factory.__module__
+
+    # Prefer a guaranteed mount for the discovered file to avoid package import issues
+    guaranteed_file_str: str | None = None
+    if original_path:
+        guaranteed_file_str = str((Path("/opt/synth_ai_repo/__local_task_app__") / Path(original_path).stem).with_suffix('.py'))
     
     dotenv_paths = [str(Path(path)) for path in (dotenv_paths or [])]
 
     pip_packages = list(modal_cfg.pip_packages)
 
     local_dirs = [(str(Path(src)), dst) for src, dst in modal_cfg.extra_local_dirs]
+    # Ensure the discovered app directory is mounted, regardless of modal_cfg
+    if original_path:
+        discovered_dir = str(Path(original_path).resolve().parent)
+        mount_dst = "/opt/synth_ai_repo/__local_task_app__"
+        if (discovered_dir, mount_dst) not in local_dirs:
+            local_dirs.append((discovered_dir, mount_dst))
     secret_names = list(modal_cfg.secret_names)
     volume_mounts = [(name, mount) for name, mount in modal_cfg.volume_mounts]
 
@@ -1244,7 +1255,7 @@ from synth_ai.task.server import create_task_app
 ENTRY_ID = {entry.app_id!r}
 MODAL_APP_NAME = {modal_name!r}
 MODULE_NAME = {module_name!r}
-MODULE_FILE = {remote_file_str!r}
+    MODULE_FILE = {guaranteed_file_str or remote_file_str!r}
 DOTENV_PATHS = {dotenv_paths!r}
 
 image = Image.debian_slim(python_version={modal_cfg.python_version!r})
@@ -1278,7 +1289,22 @@ if MODULE_FILE:
     else:
         raise RuntimeError("Failed to import task app from file: " + str(MODULE_FILE))
 else:
-    importlib.import_module(MODULE_NAME)
+    # Try module import; if that fails, attempt to load from the guaranteed mount
+    try:
+        importlib.import_module(MODULE_NAME)
+    except Exception:
+        import os
+        fallback_file = '/opt/synth_ai_repo/__local_task_app__/' + (MODULE_NAME.split('.')[-1] if MODULE_NAME else 'task_app') + '.py'
+        if os.path.exists(fallback_file):
+            spec = importlib.util.spec_from_file_location(MODULE_NAME or 'task_app_module', fallback_file)
+            if spec and spec.loader:
+                mod = importlib.util.module_from_spec(spec)
+                sys.modules[MODULE_NAME or 'task_app_module'] = mod
+                spec.loader.exec_module(mod)
+            else:
+                raise
+        else:
+            raise
 
 # Get the entry from registry (now that it's registered)
 entry = registry.get(ENTRY_ID)

{synth_ai-0.2.9.dev4.dist-info → synth_ai-0.2.9.dev5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synth-ai
-Version: 0.2.9.dev4
+Version: 0.2.9.dev5
 Summary: RL as a service SDK - Core AI functionality and tracing
 Author-email: Synth AI <josh@usesynth.ai>
 License-Expression: MIT

{synth_ai-0.2.9.dev4.dist-info → synth_ai-0.2.9.dev5.dist-info}/RECORD

@@ -81,7 +81,6 @@ examples/warming_up_to_rl/task_app/synth_envs_hosted/registry.py,sha256=pMVwiutZ
 examples/warming_up_to_rl/task_app/synth_envs_hosted/rollout.py,sha256=506KV0x0oOajTuNVVWLUxeIXG2YaZjWwfgbVg2EorcA,71477
 examples/warming_up_to_rl/task_app/synth_envs_hosted/test_agents.py,sha256=PRkQlVKGfn2-w4EOTTHc8T-tqFW6m21mS7BII6TUUd4,5599
 examples/warming_up_to_rl/task_app/synth_envs_hosted/test_service.py,sha256=PAYPP0tWU0cQUUxJuAYbDwvWeGjAcPFRHsW2vG3vOpg,5228
-examples/warming_up_to_rl/task_app/synth_envs_hosted/test_stepwise_rewards.py,sha256=OfzZ18_bykLll6-4-2-OZI4a2wFx68hnTxufZeFE5t4,1836
 examples/warming_up_to_rl/task_app/synth_envs_hosted/utils.py,sha256=Qcxfa_WASxbUCbAKejiZB6QIl8eJeeIrFPt5GU-Zb6E,2013
 examples/warming_up_to_rl/task_app/synth_envs_hosted/envs/__init__.py,sha256=5e-BmaXi_5EGcqF2hEGsnbtkHsNvPNMTX9wJ3yiz3XQ,34
 examples/warming_up_to_rl/task_app/synth_envs_hosted/envs/crafter/__init__.py,sha256=nTmSVzOfRUnjxEN1-IHjxuvPHij-AdRu_NZoH6Ozv2M,197
@@ -120,7 +119,7 @@ synth_ai/cli/recent.py,sha256=mHhM-QrR_MfjfKSzBvvPUEC-lkXTWUZrQwqYTmb2x0Y,4173
 synth_ai/cli/rl_demo.py,sha256=P09_atVrSTfGUhAs0Obe63erniJ3EDHtm51yL2xBouM,8796
 synth_ai/cli/root.py,sha256=0fuWNO5NeWTSCANzPBfoQsZKe3iUxnJ9hsSWS-SK2UM,11370
 synth_ai/cli/status.py,sha256=M_bt7U58Ubi-q-ZlrIpgCASKq9_k6uMjpx926f6kLLA,4591
-synth_ai/cli/task_apps.py,sha256=bpknAGET9vUP6iRHTJcQcA_r4TCkqsP-QMIXayc7VIo,50723
+synth_ai/cli/task_apps.py,sha256=AQmZtnvNDHMRhvUF0KOvnIF1tRbg0J-nwnJe1T49ajg,52077
 synth_ai/cli/traces.py,sha256=_QBdCR92u0Gv51U4DH0Ws1d5yCrbJRpaYKe7pmcHrHs,6484
 synth_ai/cli/watch.py,sha256=HBKbAcpUkkPhGvsPRofckbu8oILiVqp35NXHkIEpTTc,17808
 synth_ai/compound/cais.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -334,7 +333,6 @@ synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_
 synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_pull.py,sha256=ZCEr6LoSxKvVQzbRGwRBN62RAQPP5jiLoNqI8c4Ij2Q,3808
 synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_two_player.py,sha256=99yZCBsbx3i5Wiuf786TiHQjq2tMEY0OKsOEa-obhXY,3763
 synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_variations.py,sha256=VKMXRVmVdIxfa0q_T5Dfll5HHadc5QHpktRQztEIxGs,15455
-synth_ai/environments/examples/sokoban/units/astar_common.py,sha256=4vG-_JrGI-yX8HY22CTY9P3DXPqCV_AHMPwH-flr0Z4,3006
 synth_ai/environments/examples/tictactoe/__init__.py,sha256=XGoOP8GEGXjG-TI_E_-3b1jOhpGNtl_LmR9G3K59Gig,31
 synth_ai/environments/examples/tictactoe/engine.py,sha256=B2OYQ3TVi_c75BRy_zHIEWjcTSaO4vnZZQJTo6wFe7k,12430
 synth_ai/environments/examples/tictactoe/environment.py,sha256=p_ZN5-AyYKUnIJ5Df46fXMlJZ6yRerkvXC9GA1W0Ht8,9336
@@ -535,9 +533,9 @@ synth_ai/v0/tracing_v1/events/manage.py,sha256=ZDXXP-ZwLH9LCsmw7Ru9o55d7bl_diPtJ
 synth_ai/v0/tracing_v1/events/scope.py,sha256=BuBkhSpVHUJt8iGT9HJZF82rbb88mQcd2vM2shg-w2I,2550
 synth_ai/v0/tracing_v1/events/store.py,sha256=0342lvAcalyJbVEIzQFaPuMQGgwiFm7M5rE6gr-G0E8,9041
 synth_ai/zyk/__init__.py,sha256=htVLnzTYQ5rxzYpzSYBm7_o6uNKZ3pB_PrqkBrgTRS4,771
-synth_ai-0.2.9.dev4.dist-info/licenses/LICENSE,sha256=ynhjRQUfqA_RdGRATApfFA_fBAy9cno04sLtLUqxVFM,1069
-synth_ai-0.2.9.dev4.dist-info/METADATA,sha256=BH5A-8mU_wP4Sid2eqWcFQu6cGs198X5_DcmFFKON8o,5200
-synth_ai-0.2.9.dev4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-synth_ai-0.2.9.dev4.dist-info/entry_points.txt,sha256=Neq-3bT7TAijjgOIR77pKL-WYg6TWBDeO8pp_nL4vGY,91
-synth_ai-0.2.9.dev4.dist-info/top_level.txt,sha256=1moNHgctEUJ3F3eH3V-7FSMb2iTTze1V13dj1R04oUY,18
-synth_ai-0.2.9.dev4.dist-info/RECORD,,
+synth_ai-0.2.9.dev5.dist-info/licenses/LICENSE,sha256=ynhjRQUfqA_RdGRATApfFA_fBAy9cno04sLtLUqxVFM,1069
+synth_ai-0.2.9.dev5.dist-info/METADATA,sha256=-GRGwjAZPLUJpTxbLEiu-B_fZtBPodHuem1ySlmqiXk,5200
+synth_ai-0.2.9.dev5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+synth_ai-0.2.9.dev5.dist-info/entry_points.txt,sha256=Neq-3bT7TAijjgOIR77pKL-WYg6TWBDeO8pp_nL4vGY,91
+synth_ai-0.2.9.dev5.dist-info/top_level.txt,sha256=1moNHgctEUJ3F3eH3V-7FSMb2iTTze1V13dj1R04oUY,18
+synth_ai-0.2.9.dev5.dist-info/RECORD,,

examples/warming_up_to_rl/task_app/synth_envs_hosted/test_stepwise_rewards.py

@@ -1,58 +0,0 @@
-import math
-
-from backend.app.routes.clustered_training.dev.synth_envs_hosted.rollout import (
-    compute_stepwise_reward,
-)
-
-
-def test_compute_stepwise_reward_detects_unlock() -> None:
-    prev = {"build_workbench": False, "pet_cow": False}
-    new = {"build_workbench": True, "pet_cow": False}
-    actions = [{"tool": "interact", "args": {"action": "build_workbench"}}]
-
-    info, decision, stats = compute_stepwise_reward(
-        prev,
-        new,
-        decision_index=2,
-        actions_summary=actions,
-        indicator_lambda=0.75,
-    )
-
-    assert info["decision_index"] == 2
-    assert info["new_achievements"] == ["build_workbench"]
-    assert info["indicator"] == 1
-    assert math.isclose(info["reward"], 0.75)
-
-    assert decision["actions"] == actions
-    assert decision["indicator"] == 1
-    assert math.isclose(decision["r_i"], 0.75)
-
-    assert math.isclose(stats["indicator"], 1.0)
-    assert math.isclose(stats["reward"], 0.75)
-    assert math.isclose(stats["new_achievements_count"], 1.0)
-
-
-def test_compute_stepwise_reward_handles_no_unlock() -> None:
-    prev = {"collect_wheat": True, "milk_cow": False}
-    new = {"collect_wheat": True, "milk_cow": False}
-    actions = [{"tool": "interact", "args": {"action": "sleep"}}]
-
-    info, decision, stats = compute_stepwise_reward(
-        prev,
-        new,
-        decision_index=5,
-        actions_summary=actions,
-        indicator_lambda=1.2,
-    )
-
-    assert info["new_achievements"] == []
-    assert info["indicator"] == 0
-    assert math.isclose(info["reward"], 0.0)
-
-    assert decision["actions"] == actions
-    assert decision["indicator"] == 0
-    assert math.isclose(decision["r_i"], 0.0)
-
-    assert math.isclose(stats["indicator"], 0.0)
-    assert math.isclose(stats["reward"], 0.0)
-    assert math.isclose(stats["new_achievements_count"], 0.0)

synth_ai/environments/examples/sokoban/units/astar_common.py

@@ -1,95 +0,0 @@
-"""
-astar_common.py – one A* routine usable by both engine-level and
-environment-level unit tests.
-"""
-
-import heapq
-import itertools
-import json
-from typing import Any, Awaitable, Callable, List, Tuple
-
-import numpy as np
-
-
-# ---------- generic utilities ------------------------------------ #
-def _boxes_left(env_pkg) -> int:
-    """#targets – #boxes-on-targets (uses raw grids, never the counter)."""
-    return int(np.sum(env_pkg.room_fixed == 2) - np.sum(env_pkg.room_state == 3))
-
-
-def solved(obj: Any) -> bool:
-    """Expects obj to have a .package_sokoban_env attribute."""
-    return _boxes_left(obj.package_sokoban_env) == 0
-
-
-def heuristic(obj: Any) -> int:
-    """Expects obj to have a .package_sokoban_env attribute."""
-    return _boxes_left(obj.package_sokoban_env)
-
-
-# ---------- single reusable A* ----------------------------------- #
-async def astar(
-    root_obj: Any,
-    step_fn: Callable[[Any, int], Awaitable[None]],
-    deserialize_fn: Callable[[Any], Awaitable[Any]],
-    max_nodes: int = 1000,
-) -> List[int]:
-    """
-    Generic A* over Sokoban snapshots.
-
-    • `root_obj` - current engine *or* environment
-    • `step_fn(obj, action)` - async: apply one move to *obj*
-    • `deserialize_fn(snapshot)` - async: new obj from snapshot
-    """
-    start_snap = await root_obj._serialize_engine()
-
-    frontier: List[Tuple[int, int, Any, List[int]]] = []
-    counter = itertools.count()
-    frontier.append((heuristic(root_obj), next(counter), start_snap, []))
-    seen: set[str] = set()
-
-    nodes = 0
-    while frontier and nodes < max_nodes:
-        f, _, snap, path = heapq.heappop(frontier)
-        cur = await deserialize_fn(snap)
-        key = json.dumps(snap.engine_snapshot, sort_keys=True)
-        if key in seen:
-            continue
-        seen.add(key)
-        if solved(cur):
-            return path
-
-        nodes += 1
-        for action in range(cur.package_sokoban_env.action_space.n):
-            child = await deserialize_fn(snap)  # fresh copy
-            try:
-                await step_fn(child, action)
-            except Exception:  # illegal/off-board
-                continue
-
-            child_snap = await child._serialize_engine()
-            g = len(path) + 1
-            heapq.heappush(
-                frontier,
-                (g + heuristic(child), next(counter), child_snap, path + [action]),
-            )
-    return []
-
-
-# convenience lambdas for the two concrete APIs
-async def _engine_step(e, a):  # `SokobanEngine`
-    await e._step_engine(a)
-
-
-async def _env_step(env, a):  # `SokobanEnvironment` (expects Move wrapper)
-    from synth_ai.environments.examples.sokoban.units.test_sokoban_environment import Move
-
-    await env.step([[Move(a)]])
-
-
-ENGINE_ASTAR = lambda eng, **kw: astar(eng, _engine_step, eng.__class__._deserialize_engine, **kw)
-ENV_ASTAR = lambda env, **kw: astar(
-    env.engine, _env_step, env.engine.__class__._deserialize_engine, **kw
-)
-
-# ----------------------------------------------------------------- #