├── .gitignore ├── LICENSE ├── README.md ├── datasets ├── murder_mystery.json ├── object_placements.json └── team_allocation.json ├── domain_seed ├── crime_scenes.json ├── female_names.json ├── female_relationships.json ├── male_names.json ├── male_relationships.json ├── murder_weapons.json ├── relationships.json ├── strong_motives.json └── suspicious_facts.json ├── eval ├── __init__.py ├── eval.py └── icl │ ├── __init__.py │ ├── murder_mystery_solved_ex.py │ ├── object_placements_solved_ex.py │ └── team_allocation_solved_ex.py ├── imgs ├── favicon.ico ├── logo.png ├── logo_resized.webp └── system_diagram.png ├── index.html ├── musr_dataset_scripts ├── create_murder_mysteries.py ├── create_object_placements.py └── create_team_allocation.py ├── requirements.txt ├── src ├── __init__.py ├── dataset_builder.py ├── dataset_types │ ├── __init__.py │ ├── murder_mystery_dataset.py │ ├── object_placements_dataset.py │ └── team_allocation.py ├── logic_tree │ ├── __init__.py │ └── tree.py ├── madlib │ ├── __init__.py │ └── madlib.py ├── model │ ├── __init__.py │ ├── hf.py │ ├── model.py │ └── openai.py ├── utils │ ├── __init__.py │ ├── paths.py │ └── redis_cache.py └── validators │ ├── __init__.py │ ├── types │ ├── __init__.py │ ├── forbidden_text_validator.py │ ├── model_validator.py │ └── structure_validator.py │ └── validator.py └── viewer ├── script.js ├── small_datasets ├── small_murder_mystery.json ├── small_object_placements.json └── small_team_allocation.json ├── styles.css └── treevis ├── LICENSE ├── tree.css └── tree.js /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/README.md -------------------------------------------------------------------------------- /datasets/murder_mystery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/datasets/murder_mystery.json -------------------------------------------------------------------------------- /datasets/object_placements.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/datasets/object_placements.json -------------------------------------------------------------------------------- /datasets/team_allocation.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/datasets/team_allocation.json -------------------------------------------------------------------------------- /domain_seed/crime_scenes.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/domain_seed/crime_scenes.json -------------------------------------------------------------------------------- /domain_seed/female_names.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/domain_seed/female_names.json -------------------------------------------------------------------------------- /domain_seed/female_relationships.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/domain_seed/female_relationships.json -------------------------------------------------------------------------------- /domain_seed/male_names.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/domain_seed/male_names.json -------------------------------------------------------------------------------- /domain_seed/male_relationships.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/domain_seed/male_relationships.json -------------------------------------------------------------------------------- /domain_seed/murder_weapons.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/domain_seed/murder_weapons.json -------------------------------------------------------------------------------- /domain_seed/relationships.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/domain_seed/relationships.json -------------------------------------------------------------------------------- /domain_seed/strong_motives.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/domain_seed/strong_motives.json -------------------------------------------------------------------------------- /domain_seed/suspicious_facts.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/domain_seed/suspicious_facts.json -------------------------------------------------------------------------------- /eval/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eval/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/eval/eval.py -------------------------------------------------------------------------------- /eval/icl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eval/icl/murder_mystery_solved_ex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/eval/icl/murder_mystery_solved_ex.py -------------------------------------------------------------------------------- /eval/icl/object_placements_solved_ex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/eval/icl/object_placements_solved_ex.py -------------------------------------------------------------------------------- /eval/icl/team_allocation_solved_ex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/eval/icl/team_allocation_solved_ex.py -------------------------------------------------------------------------------- /imgs/favicon.ico: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/imgs/favicon.ico -------------------------------------------------------------------------------- /imgs/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/imgs/logo.png -------------------------------------------------------------------------------- /imgs/logo_resized.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/imgs/logo_resized.webp -------------------------------------------------------------------------------- /imgs/system_diagram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/imgs/system_diagram.png -------------------------------------------------------------------------------- /index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/index.html -------------------------------------------------------------------------------- /musr_dataset_scripts/create_murder_mysteries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/musr_dataset_scripts/create_murder_mysteries.py -------------------------------------------------------------------------------- /musr_dataset_scripts/create_object_placements.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/musr_dataset_scripts/create_object_placements.py -------------------------------------------------------------------------------- /musr_dataset_scripts/create_team_allocation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/musr_dataset_scripts/create_team_allocation.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/__init__.py -------------------------------------------------------------------------------- /src/dataset_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/dataset_builder.py -------------------------------------------------------------------------------- /src/dataset_types/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/dataset_types/murder_mystery_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/dataset_types/murder_mystery_dataset.py -------------------------------------------------------------------------------- /src/dataset_types/object_placements_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/dataset_types/object_placements_dataset.py -------------------------------------------------------------------------------- /src/dataset_types/team_allocation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/dataset_types/team_allocation.py -------------------------------------------------------------------------------- /src/logic_tree/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/logic_tree/tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/logic_tree/tree.py -------------------------------------------------------------------------------- /src/madlib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/madlib/madlib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/madlib/madlib.py -------------------------------------------------------------------------------- /src/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/model/__init__.py -------------------------------------------------------------------------------- /src/model/hf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/model/hf.py -------------------------------------------------------------------------------- /src/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/model/model.py -------------------------------------------------------------------------------- /src/model/openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/model/openai.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/utils/paths.py -------------------------------------------------------------------------------- /src/utils/redis_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/utils/redis_cache.py -------------------------------------------------------------------------------- /src/validators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/validators/__init__.py -------------------------------------------------------------------------------- /src/validators/types/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/validators/types/forbidden_text_validator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/validators/types/forbidden_text_validator.py -------------------------------------------------------------------------------- /src/validators/types/model_validator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/validators/types/model_validator.py -------------------------------------------------------------------------------- /src/validators/types/structure_validator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/validators/types/structure_validator.py -------------------------------------------------------------------------------- /src/validators/validator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/src/validators/validator.py -------------------------------------------------------------------------------- /viewer/script.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/viewer/script.js -------------------------------------------------------------------------------- /viewer/small_datasets/small_murder_mystery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/viewer/small_datasets/small_murder_mystery.json -------------------------------------------------------------------------------- /viewer/small_datasets/small_object_placements.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/viewer/small_datasets/small_object_placements.json -------------------------------------------------------------------------------- /viewer/small_datasets/small_team_allocation.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/viewer/small_datasets/small_team_allocation.json -------------------------------------------------------------------------------- /viewer/styles.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/viewer/styles.css -------------------------------------------------------------------------------- /viewer/treevis/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/viewer/treevis/LICENSE -------------------------------------------------------------------------------- /viewer/treevis/tree.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/viewer/treevis/tree.css -------------------------------------------------------------------------------- /viewer/treevis/tree.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Zayne-sprague/MuSR/HEAD/viewer/treevis/tree.js --------------------------------------------------------------------------------