synth-ai 0.2.2.dev0__py3-none-any.whl → 0.2.4.dev2__py3-none-any.whl

synth_ai/environments/examples/crafter_classic/agent_demos/test_crafter_react_agent.py

@@ -23,8 +23,8 @@ from tqdm.asyncio import tqdm_asyncio
 # Add the src directory to the path
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "..", "..", "..", "src"))
 
-from synth_ai.zyk import LM
-from synth_ai.zyk.lms.tools.base import BaseTool
+from synth_ai.lm.core.main import LM
+from synth_ai.lm.tools.base import BaseTool
 import numpy as np
 
 
@@ -111,7 +111,7 @@ class CrafterConfig:
         """Add ft: regex pattern to OpenAI naming regexes if not already present."""
         try:
             import re
-            from synth_ai.zyk.lms.core import vendor_clients
+            from synth_ai.lm.core import vendor_clients
 
             # Check if ft: pattern already exists
             ft_pattern = re.compile(r"^ft:.*$")

synth_ai/environments/examples/crafter_classic/environment.py

@@ -13,7 +13,7 @@ from .config_logging import safe_compare
 # Import tracing abstractions
 from synth_ai.tracing_v3.abstractions import (
     RuntimeEvent,
-    SessionEventMessage,
+    SessionEventMarkovBlanketMessage,
     TimeRecord,
 )
 

synth_ai/environments/examples/crafter_custom/environment.py

@@ -13,7 +13,7 @@ from synth_ai.environments.examples.crafter_classic.config_logging import safe_c
 # Import tracing abstractions
 from synth_ai.tracing_v3.abstractions import (
     RuntimeEvent,
-    SessionEventMessage,
+    SessionEventMarkovBlanketMessage,
     TimeRecord,
 )
 

synth_ai/environments/examples/enron/dataset/corbt___enron_emails_sample_questions/default/0.0.0/293c9fe8170037e01cc9cf5834e0cd5ef6f1a6bb/dataset_info.json

@@ -0,0 +1 @@
+{"description": "", "citation": "", "homepage": "", "license": "", "features": {"id": {"dtype": "int32", "_type": "Value"}, "question": {"dtype": "string", "_type": "Value"}, "answer": {"dtype": "string", "_type": "Value"}, "message_ids": {"feature": {"dtype": "string", "_type": "Value"}, "_type": "Sequence"}, "how_realistic": {"dtype": "float32", "_type": "Value"}, "inbox_address": {"dtype": "string", "_type": "Value"}, "query_date": {"dtype": "string", "_type": "Value"}}, "builder_name": "parquet", "dataset_name": "enron_emails_sample_questions", "config_name": "default", "version": {"version_str": "0.0.0", "major": 0, "minor": 0, "patch": 0}, "splits": {"train": {"name": "train", "num_bytes": 1324739, "num_examples": 4390, "dataset_name": "enron_emails_sample_questions"}, "test": {"name": "test", "num_bytes": 537888, "num_examples": 1733, "dataset_name": "enron_emails_sample_questions"}}, "download_checksums": {"hf://datasets/corbt/enron_emails_sample_questions@293c9fe8170037e01cc9cf5834e0cd5ef6f1a6bb/data/train-00000-of-00001.parquet": {"num_bytes": 681490, "checksum": null}, "hf://datasets/corbt/enron_emails_sample_questions@293c9fe8170037e01cc9cf5834e0cd5ef6f1a6bb/data/test-00000-of-00001.parquet": {"num_bytes": 267520, "checksum": null}}, "download_size": 949010, "dataset_size": 1862627, "size_in_bytes": 2811637}

synth_ai/environments/examples/nethack/helpers/achievements.json

@@ -0,0 +1,64 @@
+{
+  "3.4.3": {
+    "ascension": 100,
+    "dungeons": {
+      "astral_plane": 100,
+      "earth_plane": 98,
+      "air_plane": 96,
+      "fire_plane": 94,
+      "water_plane": 92,
+      "dlvl50": 90,
+      "dlvl40": 85,
+      "dlvl30": 80,
+      "fakewiz2": 75,
+      "fakewiz1": 70,
+      "deepest_gehennom": 65,
+      "into_gehennom": 60,
+      "castle": 50,
+      "deepest_regular": 40,
+      "medusa": 35,
+      "quest_permission": 30,
+      "quest_finish": 26,
+      "quest_locate": 23,
+      "quest_portal": 20,
+      "oracle": 15,
+      "sokoban_finish": 10,
+      "sokoban": 5,
+      "dlvl10": 3,
+      "dlvl5": 1,
+      "dlvl1": 0
+    },
+    "experience_levels": {
+      "lvl30": 52.0,
+      "lvl29": 51.0,
+      "lvl28": 50.0,
+      "lvl27": 49.0,
+      "lvl26": 48.0,
+      "lvl25": 47.0,
+      "lvl24": 46.0,
+      "lvl23": 45.0,
+      "lvl22": 44.0,
+      "lvl21": 43.0,
+      "lvl20": 42.0,
+      "lvl19": 41.0,
+      "lvl18": 40.0,
+      "lvl17": 39.0,
+      "lvl16": 38.0,
+      "lvl15": 37.0,
+      "lvl14": 36.0,
+      "lvl13": 33.0,
+      "lvl12": 30.0,
+      "lvl11": 27.0,
+      "lvl10": 24.0,
+      "lvl9": 21.0,
+      "lvl8": 18.0,
+      "lvl7": 15.0,
+      "lvl6": 12.0,
+      "lvl5": 9.0,
+      "lvl4": 6.0,
+      "lvl3": 3.0,
+      "lvl2": 1.0,
+      "lvl1": 0.0
+    }
+  }
+}

synth_ai/environments/examples/red/units/test_exploration_strategy.py

@@ -79,7 +79,7 @@ async def test_exploration_when_stuck():
         obs_payload = await env.initialize()
 
         if "error" in obs_payload:
-            pytest.fail(f"Environment initialization failed: {obs_payload['error']}")
+            pytest.fail(f"Environment initialization failed: {obs_payload['error']}")  # type: ignore[no-untyped-call]
 
         print("[DEBUG] Environment initialized successfully")
 

synth_ai/environments/examples/red/units/test_menu_bug_reproduction.py

@@ -134,7 +134,7 @@ async def test_menu_close_bug_reproduction():
         print("Initializing environment...")
         obs_payload = await env.initialize()
         if "error" in obs_payload:
-            pytest.fail(f"Environment initialization failed: {obs_payload['error']}")
+            pytest.fail(f"Environment initialization failed: {obs_payload['error']}")  # type: ignore[no-untyped-call]
 
         initial_menu_state = obs_payload.get("menu_state")
         initial_screen_hash = obs_payload.get("screen_hash")
@@ -238,7 +238,7 @@ async def test_menu_close_bug_reproduction():
             print("\n⚠️ Test inconclusive: No menu was open initially")
 
     except Exception as e:
-        pytest.fail(f"Test failed with exception: {e}")
+        pytest.fail(f"Test failed with exception: {e}")  # type: ignore[no-untyped-call]
 
 
 @pytest.mark.asyncio
@@ -337,7 +337,7 @@ async def test_menu_close_bug_fix_verification():
         print("Initializing environment...")
         obs_payload = await env.initialize()
         if "error" in obs_payload:
-            pytest.fail(f"Environment initialization failed: {obs_payload['error']}")
+            pytest.fail(f"Environment initialization failed: {obs_payload['error']}")  # type: ignore[no-untyped-call]
 
         initial_menu_state = obs_payload.get("menu_state")
         initial_screen_hash = obs_payload.get("screen_hash")
@@ -359,7 +359,7 @@ async def test_menu_close_bug_fix_verification():
         step_result = await env.step(PressButtonCall("B"))
 
         if "error" in step_result:
-            pytest.fail(f"Step failed: {step_result['error']}")
+            pytest.fail(f"Step failed: {step_result['error']}")  # type: ignore[no-untyped-call]
 
         final_menu_state = step_result.get("menu_state")
         final_screen_hash = step_result.get("screen_hash")
@@ -388,7 +388,7 @@ async def test_menu_close_bug_fix_verification():
             print("   This might indicate an actual menu was open and closed")
 
     except Exception as e:
-        pytest.fail(f"Test failed with exception: {e}")
+        pytest.fail(f"Test failed with exception: {e}")  # type: ignore[no-untyped-call]
 
 
 @pytest.mark.asyncio

synth_ai/environments/examples/red/units/test_movement_debug.py

@@ -127,7 +127,7 @@ async def test_deterministic_left_movement():
         obs_payload = await env.initialize()
 
         if "error" in obs_payload:
-            pytest.fail(f"Environment initialization failed: {obs_payload['error']}")
+            pytest.fail(f"Environment initialization failed: {obs_payload['error']}")  # type: ignore[no-untyped-call]
 
         print("[DEBUG] Environment initialized successfully")
         print(f"[DEBUG] Initial observation keys: {list(obs_payload.keys())}")
@@ -158,7 +158,7 @@ async def test_deterministic_left_movement():
             step_result = await env.step([[PressButtonCall("LEFT")]])
 
             if "error" in step_result:
-                pytest.fail(f"Environment step {step} failed: {step_result['error']}")
+                pytest.fail(f"Environment step {step} failed: {step_result['error']}")  # type: ignore[no-untyped-call]
 
             # Get new state
             new_pub = step_result["public"]

synth_ai/environments/examples/red/units/test_retry_movement.py

@@ -89,7 +89,7 @@ async def test_movement_with_retry():
         obs_payload = await env.initialize()
 
         if "error" in obs_payload:
-            pytest.fail(f"Environment initialization failed: {obs_payload['error']}")
+            pytest.fail(f"Environment initialization failed: {obs_payload['error']}")  # type: ignore[no-untyped-call]
 
         print("[DEBUG] Environment initialized successfully")
 

synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/available_envs.json

@@ -0,0 +1,122 @@
+[
+  {
+    "id": "Sokoban-v0",
+    "entry_point": "gym_sokoban.envs:SokobanEnv1"
+  },
+  {
+    "id": "Sokoban-v1",
+    "entry_point": "gym_sokoban.envs:SokobanEnv"
+  },
+  {
+    "id": "Sokoban-v2",
+    "entry_point": "gym_sokoban.envs:SokobanEnv2"
+  },
+  {
+    "id": "Sokoban-small-v0",
+    "entry_point": "gym_sokoban.envs:SokobanEnv_Small0"
+  },
+  {
+    "id": "Sokoban-small-v1",
+    "entry_point": "gym_sokoban.envs:SokobanEnv_Small1"
+  },
+  {
+    "id": "Sokoban-large-v0",
+    "entry_point": "gym_sokoban.envs:SokobanEnv_Large0"
+  },
+  {
+    "id": "Sokoban-large-v1",
+    "entry_point": "gym_sokoban.envs:SokobanEnv_Large1"
+  },
+  {
+    "id": "Sokoban-large-v2",
+    "entry_point": "gym_sokoban.envs:SokobanEnv_Large2"
+  },
+  {
+    "id": "Sokoban-huge-v0",
+    "entry_point": "gym_sokoban.envs:SokobanEnv_Huge0"
+  },
+  {
+    "id": "FixedTarget-Sokoban-v0",
+    "entry_point": "gym_sokoban.envs:FixedTargets_Env_v0"
+  },
+  {
+    "id": "FixedTarget-Sokoban-v1",
+    "entry_point": "gym_sokoban.envs:FixedTargets_Env_v1"
+  },
+  {
+    "id": "FixedTarget-Sokoban-v2",
+    "entry_point": "gym_sokoban.envs:FixedTargets_Env_v2"
+  },
+  {
+    "id": "FixedTarget-Sokoban-v3",
+    "entry_point": "gym_sokoban.envs:FixedTargets_Env_v3"
+  },
+  {
+    "id": "PushAndPull-Sokoban-v0",
+    "entry_point": "gym_sokoban.envs:PushAndPull_Env_v0"
+  },
+  {
+    "id": "PushAndPull-Sokoban-v1",
+    "entry_point": "gym_sokoban.envs:PushAndPull_Env_v1"
+  },
+  {
+    "id": "PushAndPull-Sokoban-v2",
+    "entry_point": "gym_sokoban.envs:PushAndPull_Env_v2"
+  },
+  {
+    "id": "PushAndPull-Sokoban-v3",
+    "entry_point": "gym_sokoban.envs:PushAndPull_Env_v3"
+  },
+  {
+    "id": "PushAndPull-Sokoban-v4",
+    "entry_point": "gym_sokoban.envs:PushAndPull_Env_v4"
+  },
+  {
+    "id": "PushAndPull-Sokoban-v5",
+    "entry_point": "gym_sokoban.envs:PushAndPull_Env_v5"
+  },
+  {
+    "id": "TwoPlayer-Sokoban-v0",
+    "entry_point": "gym_sokoban.envs:TwoPlayer_Env0"
+  },
+  {
+    "id": "TwoPlayer-Sokoban-v1",
+    "entry_point": "gym_sokoban.envs:TwoPlayer_Env1"
+  },
+  {
+    "id": "TwoPlayer-Sokoban-v2",
+    "entry_point": "gym_sokoban.envs:TwoPlayer_Env2"
+  },
+  {
+    "id": "TwoPlayer-Sokoban-v3",
+    "entry_point": "gym_sokoban.envs:TwoPlayer_Env3"
+  },
+  {
+    "id": "TwoPlayer-Sokoban-v4",
+    "entry_point": "gym_sokoban.envs:TwoPlayer_Env4"
+  },
+  {
+    "id": "TwoPlayer-Sokoban-v5",
+    "entry_point": "gym_sokoban.envs:TwoPlayer_Env5"
+  },
+  {
+    "id": "Boxoban-Train-v0",
+    "entry_point": "gym_sokoban.envs:Boxban_Env0"
+  },
+  {
+    "id": "Boxoban-Test-v0",
+    "entry_point": "gym_sokoban.envs:Boxban_Env0_test"
+  },
+  {
+    "id": "Boxoban-Val-v0",
+    "entry_point": "gym_sokoban.envs:Boxban_Env0_val"
+  },
+  {
+    "id": "Boxoban-Train-v1",
+    "entry_point": "gym_sokoban.envs:Boxban_Env1"
+  },
+  {
+    "id": "Boxoban-Val-v1",
+    "entry_point": "gym_sokoban.envs:Boxban_Env1_val"
+  }
+]