synth-ai 0.2.14__py3-none-any.whl → 0.4.1__py3-none-any.whl

synth_ai/environments/examples/crafter_classic/taskset.py

@@ -1,233 +0,0 @@
-"""Procedural Crafter taskset generation with seed filtering by world traits.
-Run this to build a TaskInstanceSet with reproducible initial snapshots.
-"""
-
-from __future__ import annotations
-
-import asyncio
-import random
-from dataclasses import asdict, dataclass, fields
-from typing import Dict, List
-from uuid import UUID, uuid4
-
-import crafter
-import numpy as np
-from crafter import objects
-
-from synth_ai.environments.tasks.core import (
-    Impetus,
-    Intent,
-    SplitInfo,
-    Task,
-    TaskInstance,
-    TaskInstanceMetadata,
-    TaskInstanceSet,
-)
-
-# ──────────────────────────────────────────────────────────────────────────────
-# Config
-# ──────────────────────────────────────────────────────────────────────────────
-TASK = Task(
-    global_premises="Procedural Crafter seed generation",
-    global_constraints="",
-    global_objectives="Survive and unlock achievements.",
-    shared_env_params={},
-)
-
-AREA = (64, 64)
-LEN = 10000
-RADIUS = 10  # Manhattan distance for local trait count
-SEED_START = 0
-NUM_INSTANCES = 50
-
-# Desired trait ranges per difficulty tier
-TRAIT_BOUNDS = {
-    "easy": {
-        "min_trees": 4,
-        "max_hostiles": 0,
-    },
-    "medium": {
-        "min_trees": 2,
-        "max_hostiles": 2,
-    },
-    "hard": {
-        "min_trees": 0,
-        "max_hostiles": 5,
-    },
-}
-
-# ──────────────────────────────────────────────────────────────────────────────
-# Metadata + instance helpers
-# ──────────────────────────────────────────────────────────────────────────────
-
-
-from typing import Optional
-
-
-@dataclass
-class CrafterTaskInstanceMetadata(TaskInstanceMetadata):
-    difficulty: str
-    seed: int
-    num_trees_radius: int
-    num_cows_radius: int
-    num_hostiles_radius: int
-    world_config: Optional[str] = "normal"  # 'easy', 'normal', 'hard', 'peaceful'
-    world_config_path: Optional[str] = None  # Path to custom JSON config
-
-
-@dataclass
-class CrafterTaskInstance(TaskInstance):
-    async def serialize(self) -> dict:  # identical to Sokoban pattern
-        data = asdict(self)
-        if isinstance(data.get("id"), UUID):
-            data["id"] = str(data["id"])
-        if "intent" in data and data["intent"] is not None:
-            data["intent"]["deterministic_eval_functions"] = []
-        return data
-
-    @classmethod
-    async def deserialize(cls, data: dict) -> "CrafterTaskInstance":
-        if "id" in data:
-            try:
-                data["id"] = UUID(str(data["id"]))
-            except Exception:
-                pass
-        if "impetus" in data and isinstance(data["impetus"], dict):
-            impetus_data = data["impetus"]
-            # Ensure instructions field exists with default if missing
-            if "instructions" not in impetus_data:
-                impetus_data["instructions"] = "Survive and unlock achievements"
-            data["impetus"] = Impetus(**impetus_data)
-        if "intent" in data and isinstance(data["intent"], dict):
-            intent_data = data["intent"]
-            # Ensure required fields exist with defaults if missing
-            if "rubric" not in intent_data:
-                intent_data["rubric"] = {"goal": "Unlock achievements"}
-            if "gold_trajectories" not in intent_data:
-                intent_data["gold_trajectories"] = None
-            if "gold_state_diff" not in intent_data:
-                intent_data["gold_state_diff"] = {}
-            intent_data["deterministic_eval_functions"] = []
-            data["intent"] = Intent(**intent_data)
-        if "metadata" in data and isinstance(data["metadata"], dict):
-            metadata_data = data["metadata"]
-            # Ensure required fields exist with defaults if missing
-            if "difficulty" not in metadata_data:
-                metadata_data["difficulty"] = "medium"
-            if "seed" not in metadata_data:
-                metadata_data["seed"] = 0
-            if "num_trees_radius" not in metadata_data:
-                metadata_data["num_trees_radius"] = 0
-            if "num_cows_radius" not in metadata_data:
-                metadata_data["num_cows_radius"] = 0
-            if "num_hostiles_radius" not in metadata_data:
-                metadata_data["num_hostiles_radius"] = 0
-            data["metadata"] = CrafterTaskInstanceMetadata(**metadata_data)
-        keep = {f.name for f in fields(cls)}
-        return cls(**{k: v for k, v in data.items() if k in keep})
-
-
-# ──────────────────────────────────────────────────────────────────────────────
-# Trait extraction util
-# ──────────────────────────────────────────────────────────────────────────────
-
-
-def world_traits(env: crafter.Env, radius: int = RADIUS) -> Dict[str, int]:
-    player = env._player  # type: ignore[attr-defined]
-    pos = np.array(player.pos)
-    counts = {"trees": 0, "cows": 0, "hostiles": 0}
-    for obj in env._world._objects:  # type: ignore[attr-defined]
-        if obj is None or obj is player:
-            continue
-        if np.abs(obj.pos - pos).sum() > radius:
-            continue
-        if isinstance(obj, objects.Plant) and getattr(obj, "kind", "") == "tree":
-            counts["trees"] += 1
-        elif isinstance(obj, objects.Cow):
-            counts["cows"] += 1
-        elif isinstance(obj, (objects.Zombie, objects.Skeleton)):
-            counts["hostiles"] += 1
-    return counts
-
-
-# ──────────────────────────────────────────────────────────────────────────────
-# Main generator
-# ──────────────────────────────────────────────────────────────────────────────
-
-
-async def create_crafter_taskset(num_instances: int = NUM_INSTANCES) -> TaskInstanceSet:
-    instances: List[CrafterTaskInstance] = []
-    seed = SEED_START
-    while len(instances) < num_instances:
-        env = crafter.Env(area=AREA, length=LEN, seed=seed)
-        _ = env.reset()
-        traits = world_traits(env)
-        # assign difficulty tier first match
-        difficulty: str | None = None
-        for diff, bounds in TRAIT_BOUNDS.items():
-            if (
-                traits["trees"] >= bounds["min_trees"]
-                and traits["hostiles"] <= bounds["max_hostiles"]
-            ):
-                difficulty = diff
-                break
-        if difficulty is None:
-            seed += 1
-            continue
-        # build instance
-        impetus = Impetus(instructions=f"Survive and unlock achievements. Difficulty={difficulty}.")
-        intent = Intent(
-            rubric={"goal": "Unlock as many achievements as possible."},
-            gold_trajectories=None,
-            gold_state_diff={},
-        )
-        metadata = CrafterTaskInstanceMetadata(
-            difficulty=difficulty,
-            seed=seed,
-            num_trees_radius=traits["trees"],
-            num_cows_radius=traits["cows"],
-            num_hostiles_radius=traits["hostiles"],
-        )
-        instance = CrafterTaskInstance(
-            id=uuid4(),
-            impetus=impetus,
-            intent=intent,
-            metadata=metadata,
-            is_reproducible=True,
-            initial_engine_snapshot=None,  # will be filled lazily when env starts
-        )
-        instances.append(instance)
-        seed += 1
-
-    # simple random split 80/10/10
-    random.shuffle(instances)
-    n = len(instances)
-    val_ids = {inst.id for inst in instances[int(0.8 * n) : int(0.9 * n)]}
-    test_ids = {inst.id for inst in instances[int(0.9 * n) :]}
-    split = SplitInfo(val_instance_ids=val_ids, test_instance_ids=test_ids, _is_split_defined=True)
-
-    return TaskInstanceSet(
-        name="Crafter Procedural TaskSet",
-        description="Crafter seeds filtered by local world traits around spawn.",
-        instances=instances,
-        split_info=split,
-    )
-
-
-# ──────────────────────────────────────────────────────────────────────────────
-# CLI example
-# ──────────────────────────────────────────────────────────────────────────────
-
-if __name__ == "__main__":
-    import json
-    import pathlib
-
-    async def _main():
-        ts = await create_crafter_taskset(30)
-        serial = await asyncio.gather(*(inst.serialize() for inst in ts.instances))
-        out = pathlib.Path("dataset/crafter_instances.json")
-        out.parent.mkdir(parents=True, exist_ok=True)
-        out.write_text(json.dumps(serial, indent=2))
-        print(f"Saved {len(serial)} instances → {out}")
-
-    asyncio.run(_main())

synth_ai/environments/examples/crafter_classic/trace_hooks_v3.py

@@ -1,228 +0,0 @@
-"""
-Trace hooks for Crafter Classic environment - v3 version.
-Updated to use the v3 tracing architecture with async support.
-"""
-
-from datetime import datetime
-from typing import Any, Dict, Optional, Set
-
-from synth_ai.tracing_v3.abstractions import BaseEvent, EnvironmentEvent
-from synth_ai.tracing_v3.hooks import HookManager
-
-# Achievement categorization by difficulty
-EASY_ACHIEVEMENTS = {
-    "collect_wood",
-    "collect_stone",
-    "collect_sapling",
-    "collect_drink",
-    "place_stone",
-    "place_table",
-    "wake_up",
-    "eat_plant",
-}
-
-MEDIUM_ACHIEVEMENTS = {
-    "make_wood_pickaxe",
-    "make_wood_sword",
-    "place_furnace",
-    "place_plant",
-    "collect_coal",
-    "collect_iron",
-    "eat_cow",
-}
-
-HARD_ACHIEVEMENTS = {
-    "make_stone_pickaxe",
-    "make_stone_sword",
-    "make_iron_pickaxe",
-    "make_iron_sword",
-    "collect_diamond",
-    "defeat_skeleton",
-    "defeat_zombie",
-}
-
-
-async def check_easy_achievements(event_obj: BaseEvent, **kwargs) -> Optional[Dict[str, Any]]:
-    """Hook that fires when an easy achievement is unlocked."""
-    # Only process EnvironmentEvents
-    if not isinstance(event_obj, EnvironmentEvent):
-        return None
-
-    # Get achievements before and after
-    before_achievements = {}
-    after_achievements = {}
-
-    state_before = event_obj.system_state_before
-    state_after = event_obj.system_state_after
-
-    if isinstance(state_before, dict):
-        before_achievements = state_before.get("public_state", {}).get("achievements_status", {})
-    if isinstance(state_after, dict):
-        after_achievements = state_after.get("public_state", {}).get("achievements_status", {})
-
-    # Find new easy achievements
-    new_easy_achievements = []
-    for achievement, status in after_achievements.items():
-        if status and not before_achievements.get(achievement, False):
-            if achievement in EASY_ACHIEVEMENTS:
-                new_easy_achievements.append(achievement)
-
-    if new_easy_achievements:
-        # print(f"🎯 Easy achievement(s) unlocked: {', '.join(new_easy_achievements)}")  # Disabled for clean output
-        return {
-            "achievements": new_easy_achievements,
-            "difficulty": "easy",
-            "timestamp": datetime.now().isoformat(),
-        }
-    return None
-
-
-async def check_medium_achievements(event_obj: BaseEvent, **kwargs) -> Optional[Dict[str, Any]]:
-    """Hook that fires when a medium achievement is unlocked."""
-    # Only process EnvironmentEvents
-    if not isinstance(event_obj, EnvironmentEvent):
-        return None
-
-    # Get achievements before and after
-    before_achievements = {}
-    after_achievements = {}
-
-    state_before = event_obj.system_state_before
-    state_after = event_obj.system_state_after
-
-    if isinstance(state_before, dict):
-        before_achievements = state_before.get("public_state", {}).get("achievements_status", {})
-    if isinstance(state_after, dict):
-        after_achievements = state_after.get("public_state", {}).get("achievements_status", {})
-
-    # Find new medium achievements
-    new_medium_achievements = []
-    for achievement, status in after_achievements.items():
-        if status and not before_achievements.get(achievement, False):
-            if achievement in MEDIUM_ACHIEVEMENTS:
-                new_medium_achievements.append(achievement)
-
-    if new_medium_achievements:
-        # print(f"⭐ Medium achievement(s) unlocked: {', '.join(new_medium_achievements)}")  # Disabled for clean output
-        return {
-            "achievements": new_medium_achievements,
-            "difficulty": "medium",
-            "timestamp": datetime.now().isoformat(),
-        }
-    return None
-
-
-async def check_hard_achievements(event_obj: BaseEvent, **kwargs) -> Optional[Dict[str, Any]]:
-    """Hook that fires when a hard achievement is unlocked."""
-    # Only process EnvironmentEvents
-    if not isinstance(event_obj, EnvironmentEvent):
-        return None
-
-    # Get achievements before and after
-    before_achievements = {}
-    after_achievements = {}
-
-    state_before = event_obj.system_state_before
-    state_after = event_obj.system_state_after
-
-    if isinstance(state_before, dict):
-        before_achievements = state_before.get("public_state", {}).get("achievements_status", {})
-    if isinstance(state_after, dict):
-        after_achievements = state_after.get("public_state", {}).get("achievements_status", {})
-
-    # Find new hard achievements
-    new_hard_achievements = []
-    for achievement, status in after_achievements.items():
-        if status and not before_achievements.get(achievement, False):
-            if achievement in HARD_ACHIEVEMENTS:
-                new_hard_achievements.append(achievement)
-
-    if new_hard_achievements:
-        # print(f"🏆 Hard achievement(s) unlocked: {', '.join(new_hard_achievements)}")  # Disabled for clean output
-        return {
-            "achievements": new_hard_achievements,
-            "difficulty": "hard",
-            "timestamp": datetime.now().isoformat(),
-        }
-    return None
-
-
-async def log_invalid_actions(event_obj: BaseEvent, **kwargs) -> Optional[Dict[str, Any]]:
-    """Hook that logs invalid actions."""
-    from synth_ai.tracing_v3.abstractions import RuntimeEvent
-
-    # Only process RuntimeEvents
-    if not isinstance(event_obj, RuntimeEvent):
-        return None
-
-    # Check if action was invalid
-    if event_obj.metadata.get("valid") is False:
-        action_name = event_obj.metadata.get("action_name", "unknown")
-        # print(f"⚠️  Invalid action attempted: {action_name}")  # Disabled for clean output
-        return {
-            "action": action_name,
-            "valid": False,
-            "timestamp": datetime.now().isoformat(),
-        }
-    return None
-
-
-async def track_reward_milestones(event_obj: BaseEvent, **kwargs) -> Optional[Dict[str, Any]]:
-    """Hook that tracks significant reward events."""
-    # Only process EnvironmentEvents
-    if not isinstance(event_obj, EnvironmentEvent):
-        return None
-
-    reward = event_obj.reward
-    if reward and reward >= 1.0:  # Significant positive reward
-        # print(f"💰 High reward received: {reward}")  # Disabled for clean output
-        return {
-            "reward": reward,
-            "timestamp": datetime.now().isoformat(),
-        }
-    return None
-
-
-# Create the global CRAFTER_HOOKS instance
-CRAFTER_HOOKS = HookManager()
-
-# Register all hooks
-CRAFTER_HOOKS.register(
-    "event_recorded",
-    check_easy_achievements,
-    name="easy_achievements",
-    priority=10,
-    event_types=["environment"],
-)
-
-CRAFTER_HOOKS.register(
-    "event_recorded",
-    check_medium_achievements,
-    name="medium_achievements",
-    priority=10,
-    event_types=["environment"],
-)
-
-CRAFTER_HOOKS.register(
-    "event_recorded",
-    check_hard_achievements,
-    name="hard_achievements",
-    priority=10,
-    event_types=["environment"],
-)
-
-CRAFTER_HOOKS.register(
-    "event_recorded",
-    log_invalid_actions,
-    name="invalid_actions",
-    priority=5,
-    event_types=["runtime"],
-)
-
-CRAFTER_HOOKS.register(
-    "event_recorded",
-    track_reward_milestones,
-    name="reward_milestones",
-    priority=5,
-    event_types=["environment"],
-)