Home

Awesome

MindAgent

A minimal text-based Overcooked! 2 game.

Installation

pip install -r requirements.txt

Run LLM experiments

First, need to specify openai key on the command line

export OPENAI_API_KEY='xxx'

for example with 2 agents, with gpt-4 and level_0

Please read agents/chat_agent.py for more options.

python agents/chat_agent.py  --level level_0 gpt-4 --num_agents 2

Warning: It's very expensive!

Run RL experiment

python train/train.py --level level_0

How to collect few-shot examples using web app

  1. Launch the web-app and start to collect demonstrations

    python webpage/app.py
    
  2. Locate the the game in logs/ based off user-id

  3. Replay the log

    python agents/replay_amt.py --gameplay <game_log> | tee log_amt.txt
    
  4. Convert the log into few-shot prompt example. You may specify the number of agents used used in this log

    python assets/log_to_prompt.py --log-input log_amt.txt --prompt-output assets/amt_examples/prompt_2agent.txt --num-agents 2
    

PaLM 2

  1. Sign up here for Vertex AI and get $300 credit for free

  2. Install google cloud sdk

    Remember to run gcloud init before anything

  3. Run the following

    gcloud auth application-default login
    pip install pandas google-cloud-aiplatform[pipelines]==1.26.0 google-auth==2.17.3
    
  4. You are good to go! Take a look at completion_palm and chat_palm in llm.py.

Run & Test env

python test/test_1.py

Add new recipe

assets/recipe.json

Add new level

assets/tasks_level_final.json

unit test

pytest test

unit test with stdout

pytest test -rP

using coverage

coverage run -m pytest test
coverage report
coverage html

webserver

python webpage/app.py --debug

Minecraft Experiment

please refer to Minecraft Experiemnt