├── .flake8 ├── .github └── workflows │ └── CI.yml ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── SECRETS ├── balrog ├── __init__.py ├── agents │ ├── __init__.py │ ├── base.py │ ├── chain_of_thought.py │ ├── custom.py │ ├── dummy.py │ ├── few_shot.py │ ├── naive.py │ ├── robust_cot.py │ └── robust_naive.py ├── client.py ├── config │ └── config.yaml ├── dataset.py ├── environments │ ├── __init__.py │ ├── babaisai │ │ ├── README.md │ │ ├── __init__.py │ │ ├── babaisai_env.py │ │ └── base.py │ ├── babyai_text │ │ ├── README.md │ │ ├── __init__.py │ │ ├── babyai_env.py │ │ └── clean_lang_wrapper.py │ ├── crafter │ │ ├── __init__.py │ │ ├── crafter_env.py │ │ └── env.py │ ├── env_wrapper.py │ ├── minihack │ │ ├── __init__.py │ │ └── minihack_env.py │ ├── nle │ │ ├── Hack-Regular.ttf │ │ ├── Tiles16x16.png │ │ ├── __init__.py │ │ ├── achievements.json │ │ ├── auto_more.py │ │ ├── base.py │ │ ├── nle_env.py │ │ ├── progress.py │ │ ├── render.py │ │ ├── render_rgb.py │ │ └── tiles.pkl │ ├── textworld │ │ ├── README.md │ │ ├── __init__.py │ │ ├── base.py │ │ └── textworld_env.py │ └── wrappers │ │ ├── __init__.py │ │ ├── gym_compatibility.py │ │ └── nle_timelimit.py ├── evaluator.py ├── prompt_builder │ ├── __init__.py │ └── history.py ├── scripts │ ├── __init__.py │ └── post_install.py ├── tests │ └── test_evaluation.py └── utils.py ├── docker ├── Dockerfile └── README.md ├── docs ├── .gitignore ├── agents.md ├── contribution.md ├── envs │ ├── babaisai.md │ ├── babyai.md │ ├── crafter.md │ ├── minihack.md │ ├── nle.md │ └── textworld.md ├── evaluation.md ├── few_shot_learning.md ├── imgs │ ├── balrog_1.png │ └── balrog_banner.png └── installation.md ├── eval.py ├── pyproject.toml └── setup.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/workflows/CI.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/.github/workflows/CI.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/README.md -------------------------------------------------------------------------------- /SECRETS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/SECRETS -------------------------------------------------------------------------------- /balrog/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /balrog/agents/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/agents/__init__.py -------------------------------------------------------------------------------- /balrog/agents/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/agents/base.py -------------------------------------------------------------------------------- /balrog/agents/chain_of_thought.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/agents/chain_of_thought.py -------------------------------------------------------------------------------- /balrog/agents/custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/agents/custom.py -------------------------------------------------------------------------------- /balrog/agents/dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/agents/dummy.py -------------------------------------------------------------------------------- /balrog/agents/few_shot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/agents/few_shot.py -------------------------------------------------------------------------------- /balrog/agents/naive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/agents/naive.py -------------------------------------------------------------------------------- /balrog/agents/robust_cot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/agents/robust_cot.py -------------------------------------------------------------------------------- /balrog/agents/robust_naive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/agents/robust_naive.py -------------------------------------------------------------------------------- /balrog/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/client.py -------------------------------------------------------------------------------- /balrog/config/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/config/config.yaml -------------------------------------------------------------------------------- /balrog/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/dataset.py -------------------------------------------------------------------------------- /balrog/environments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/__init__.py -------------------------------------------------------------------------------- /balrog/environments/babaisai/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/babaisai/README.md -------------------------------------------------------------------------------- /balrog/environments/babaisai/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/babaisai/__init__.py -------------------------------------------------------------------------------- /balrog/environments/babaisai/babaisai_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/babaisai/babaisai_env.py -------------------------------------------------------------------------------- /balrog/environments/babaisai/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/babaisai/base.py -------------------------------------------------------------------------------- /balrog/environments/babyai_text/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/babyai_text/README.md -------------------------------------------------------------------------------- /balrog/environments/babyai_text/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/babyai_text/__init__.py -------------------------------------------------------------------------------- /balrog/environments/babyai_text/babyai_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/babyai_text/babyai_env.py -------------------------------------------------------------------------------- /balrog/environments/babyai_text/clean_lang_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/babyai_text/clean_lang_wrapper.py -------------------------------------------------------------------------------- /balrog/environments/crafter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/crafter/__init__.py -------------------------------------------------------------------------------- /balrog/environments/crafter/crafter_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/crafter/crafter_env.py -------------------------------------------------------------------------------- /balrog/environments/crafter/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/crafter/env.py -------------------------------------------------------------------------------- /balrog/environments/env_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/env_wrapper.py -------------------------------------------------------------------------------- /balrog/environments/minihack/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/minihack/__init__.py -------------------------------------------------------------------------------- /balrog/environments/minihack/minihack_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/minihack/minihack_env.py -------------------------------------------------------------------------------- /balrog/environments/nle/Hack-Regular.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/nle/Hack-Regular.ttf -------------------------------------------------------------------------------- /balrog/environments/nle/Tiles16x16.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/nle/Tiles16x16.png -------------------------------------------------------------------------------- /balrog/environments/nle/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/nle/__init__.py -------------------------------------------------------------------------------- /balrog/environments/nle/achievements.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/nle/achievements.json -------------------------------------------------------------------------------- /balrog/environments/nle/auto_more.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/nle/auto_more.py -------------------------------------------------------------------------------- /balrog/environments/nle/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/nle/base.py -------------------------------------------------------------------------------- /balrog/environments/nle/nle_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/nle/nle_env.py -------------------------------------------------------------------------------- /balrog/environments/nle/progress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/nle/progress.py -------------------------------------------------------------------------------- /balrog/environments/nle/render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/nle/render.py -------------------------------------------------------------------------------- /balrog/environments/nle/render_rgb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/nle/render_rgb.py -------------------------------------------------------------------------------- /balrog/environments/nle/tiles.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/nle/tiles.pkl -------------------------------------------------------------------------------- /balrog/environments/textworld/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/textworld/README.md -------------------------------------------------------------------------------- /balrog/environments/textworld/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/textworld/__init__.py -------------------------------------------------------------------------------- /balrog/environments/textworld/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/textworld/base.py -------------------------------------------------------------------------------- /balrog/environments/textworld/textworld_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/textworld/textworld_env.py -------------------------------------------------------------------------------- /balrog/environments/wrappers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/wrappers/__init__.py -------------------------------------------------------------------------------- /balrog/environments/wrappers/gym_compatibility.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/wrappers/gym_compatibility.py -------------------------------------------------------------------------------- /balrog/environments/wrappers/nle_timelimit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/environments/wrappers/nle_timelimit.py -------------------------------------------------------------------------------- /balrog/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/evaluator.py -------------------------------------------------------------------------------- /balrog/prompt_builder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/prompt_builder/__init__.py -------------------------------------------------------------------------------- /balrog/prompt_builder/history.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/prompt_builder/history.py -------------------------------------------------------------------------------- /balrog/scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /balrog/scripts/post_install.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/scripts/post_install.py -------------------------------------------------------------------------------- /balrog/tests/test_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/tests/test_evaluation.py -------------------------------------------------------------------------------- /balrog/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/balrog/utils.py -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docker/README.md -------------------------------------------------------------------------------- /docs/.gitignore: -------------------------------------------------------------------------------- 1 | api/ 2 | -------------------------------------------------------------------------------- /docs/agents.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docs/agents.md -------------------------------------------------------------------------------- /docs/contribution.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docs/contribution.md -------------------------------------------------------------------------------- /docs/envs/babaisai.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docs/envs/babaisai.md -------------------------------------------------------------------------------- /docs/envs/babyai.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docs/envs/babyai.md -------------------------------------------------------------------------------- /docs/envs/crafter.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docs/envs/crafter.md -------------------------------------------------------------------------------- /docs/envs/minihack.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docs/envs/minihack.md -------------------------------------------------------------------------------- /docs/envs/nle.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docs/envs/nle.md -------------------------------------------------------------------------------- /docs/envs/textworld.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docs/envs/textworld.md -------------------------------------------------------------------------------- /docs/evaluation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docs/evaluation.md -------------------------------------------------------------------------------- /docs/few_shot_learning.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docs/few_shot_learning.md -------------------------------------------------------------------------------- /docs/imgs/balrog_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docs/imgs/balrog_1.png -------------------------------------------------------------------------------- /docs/imgs/balrog_banner.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docs/imgs/balrog_banner.png -------------------------------------------------------------------------------- /docs/installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/docs/installation.md -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/eval.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/balrog-ai/BALROG/HEAD/setup.py --------------------------------------------------------------------------------