docs/tutorials/pettingzoo_wrapper.md

How to use PettingZoo compatibility wrapper

This tutorial provides a simple example to demonstrate how to use a ChatArena environment with PettingZoo.

1. Load the environment

from chatarena.arena import Arena

arena = Arena.from_config("examples/pettingzoo_env_example.json")

2. Wrap the environment

from chatarena.pettingzoo_compatibility import PettingZooCompatibilityV0
env = PettingZooCompatibilityV0(env=arena, render_mode="human", max_turns=5)
env.reset()

print("OBS SPACE: ", env.observation_space(env.agent_selection))
print("ACT SPACE: ", env.action_space(env.agent_selection))

3. Run the environment

agent_player_mapping = {agent: player_obj
                        for agent in env.possible_agents
                        for player_obj in env._env.players}

for agent in env.agent_iter():
    observation, reward, termination, truncation, info = env.last()

    # Use chat arena agent to generate response (TODO: use manual input backend rather than this method of input)
    chatarena_agent = agent_player_mapping[agent]
    messages = env._env.environment.message_pool.get_visible_messages(agent, turn=env.current_turn)
    response = chatarena_agent(messages)

    env.step(response)
    print("---")
env.close()