crime_env/__init__.py

"""AI Crime Investigation World — Multi-Agent RL Environment."""

from crime_env.case_generator import generate_case
from crime_env.consistency_tracker import ConsistencyTracker
from crime_env.reward_calculator import RewardCalculator
from crime_env.agent_prompts import build_system_prompt
from crime_env.environment import CrimeInvestigationEnv
from crime_env.constants import (
    SUSPECT_A,
    SUSPECT_B,
    WITNESS_1,
    VALID_TARGETS,
)

__all__ = [
    "generate_case",
    "ConsistencyTracker",
    "RewardCalculator",
    "build_system_prompt",
    "CrimeInvestigationEnv",
    "SUSPECT_A",
    "SUSPECT_B",
    "WITNESS_1",
    "VALID_TARGETS",
]