mirror of
https://github.com/hwchase17/langchain.git
synced 2025-05-22 07:27:45 +00:00
4bc209c6f7
This notebook shows how the DialogueAgent and DialogueSimulator class make it easy to extend the [Two-Player Dungeons & Dragons example](https://python.langchain.com/en/latest/use_cases/agent_simulations/two_player_dnd.html) to multiple players. The main difference between simulating two players and multiple players is in revising the schedule for when each agent speaks To this end, we augment DialogueSimulator to take in a custom function that determines the schedule of which agent speaks. In the example below, each character speaks in round-robin fashion, with the storyteller interleaved between each player.
494 lines
23 KiB
Plaintext
494 lines
23 KiB
Plaintext
{
|
|
"cells": [
|
|
{
|
|
"cell_type": "markdown",
|
|
"metadata": {},
|
|
"source": [
|
|
"# Multi-Player Dungeons & Dragons\n",
|
|
"\n",
|
|
"This notebook shows how the `DialogueAgent` and `DialogueSimulator` class make it easy to extend the [Two-Player Dungeons & Dragons example](https://python.langchain.com/en/latest/use_cases/agent_simulations/two_player_dnd.html) to multiple players.\n",
|
|
"\n",
|
|
"The main difference between simulating two players and multiple players is in revising the schedule for when each agent speaks\n",
|
|
"\n",
|
|
"To this end, we augment `DialogueSimulator` to take in a custom function that determines the schedule of which agent speaks. In the example below, each character speaks in round-robin fashion, with the storyteller interleaved between each player."
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Import LangChain related modules "
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 1,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"from typing import List, Dict, Callable\n",
|
|
"from langchain.chat_models import ChatOpenAI\n",
|
|
"from langchain.schema import (\n",
|
|
" AIMessage,\n",
|
|
" HumanMessage,\n",
|
|
" SystemMessage,\n",
|
|
" BaseMessage,\n",
|
|
")"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"metadata": {},
|
|
"source": [
|
|
"## `DialogueAgent` class\n",
|
|
"The `DialogueAgent` class is a simple wrapper around the `ChatOpenAI` model that stores the message history from the `dialogue_agent`'s point of view by simply concatenating the messages as strings.\n",
|
|
"\n",
|
|
"It exposes two methods: \n",
|
|
"- `send()`: applies the chatmodel to the message history and returns the message string\n",
|
|
"- `receive(name, message)`: adds the `message` spoken by `name` to message history"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 2,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"class DialogueAgent():\n",
|
|
"\n",
|
|
" def __init__(\n",
|
|
" self,\n",
|
|
" name,\n",
|
|
" system_message: SystemMessage,\n",
|
|
" model: ChatOpenAI,\n",
|
|
" ) -> None:\n",
|
|
" self.name = name\n",
|
|
" self.system_message = system_message\n",
|
|
" self.model = model\n",
|
|
" self.message_history = f\"\"\"Here is the conversation so far.\n",
|
|
" \"\"\"\n",
|
|
" self.prefix = f'\\n{self.name}:'\n",
|
|
" \n",
|
|
" def send(self) -> str:\n",
|
|
" \"\"\"\n",
|
|
" Applies the chatmodel to the message history\n",
|
|
" and returns the message string\n",
|
|
" \"\"\"\n",
|
|
" message = self.model(\n",
|
|
" [self.system_message, \n",
|
|
" HumanMessage(content=self.message_history+self.prefix)])\n",
|
|
" return message.content\n",
|
|
" \n",
|
|
" def receive(self, name: str, message: str) -> None:\n",
|
|
" \"\"\"\n",
|
|
" Concatenates {message} spoken by {name} into message history\n",
|
|
" \"\"\"\n",
|
|
" self.message_history += f'\\n{name}: {message}'"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"metadata": {},
|
|
"source": [
|
|
"## `DialogueSimulator` class\n",
|
|
"The `DialogueSimulator` class takes a list of agents. At each step, it performs the following:\n",
|
|
"1. Select the next speaker\n",
|
|
"2. Calls the next speaker to send a message \n",
|
|
"3. Broadcasts the message to all other agents\n",
|
|
"4. Update the step counter.\n",
|
|
"The selection of the next speaker can be implemented as any function, but in this case we simply loop through the agents."
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 3,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"class DialogueSimulator():\n",
|
|
" \n",
|
|
" def __init__(\n",
|
|
" self, \n",
|
|
" agents: List[DialogueAgent], \n",
|
|
" selection_function: Callable[[int, List[DialogueAgent]], int]\n",
|
|
" ) -> None:\n",
|
|
" self.agents = agents\n",
|
|
" self._step = 0\n",
|
|
" self.select_next_speaker = selection_function\n",
|
|
" \n",
|
|
" def reset(self, name: str, message: str):\n",
|
|
" \"\"\"\n",
|
|
" Initiates the conversation with a {message} from {name}\n",
|
|
" \"\"\"\n",
|
|
" for agent in self.agents:\n",
|
|
" agent.receive(name, message)\n",
|
|
" \n",
|
|
" # increment time\n",
|
|
" self._step += 1\n",
|
|
" \n",
|
|
" def step(self) -> tuple[str, str]:\n",
|
|
" # 1. choose the next speaker\n",
|
|
" speaker_idx = self.select_next_speaker(self._step, self.agents)\n",
|
|
" speaker = self.agents[speaker_idx]\n",
|
|
" \n",
|
|
" # 2. next speaker sends message\n",
|
|
" message = speaker.send()\n",
|
|
" \n",
|
|
" # 3. everyone receives message\n",
|
|
" for receiver in self.agents:\n",
|
|
" receiver.receive(speaker.name, message)\n",
|
|
" \n",
|
|
" # 4. increment time\n",
|
|
" self._step += 1\n",
|
|
" \n",
|
|
" return speaker.name, message"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Define roles and quest"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 4,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"character_names = [\"Harry Potter\", \"Ron Weasley\", \"Hermione Granger\", \"Argus Filch\"]\n",
|
|
"storyteller_name = \"Dungeon Master\"\n",
|
|
"quest = \"Find all of Lord Voldemort's seven horcruxes.\"\n",
|
|
"word_limit = 50 # word limit for task brainstorming"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Ask an LLM to add detail to the game description"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 5,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"game_description = f\"\"\"Here is the topic for a Dungeons & Dragons game: {quest}.\n",
|
|
" The characters are: {*character_names,}.\n",
|
|
" The story is narrated by the storyteller, {storyteller_name}.\"\"\"\n",
|
|
"\n",
|
|
"player_descriptor_system_message = SystemMessage(\n",
|
|
" content=\"You can add detail to the description of a Dungeons & Dragons player.\")\n",
|
|
"\n",
|
|
"def generate_character_description(character_name):\n",
|
|
" character_specifier_prompt = [\n",
|
|
" player_descriptor_system_message,\n",
|
|
" HumanMessage(content=\n",
|
|
" f\"\"\"{game_description}\n",
|
|
" Please reply with a creative description of the character, {character_name}, in {word_limit} words or less. \n",
|
|
" Speak directly to {character_name}.\n",
|
|
" Do not add anything else.\"\"\"\n",
|
|
" )\n",
|
|
" ]\n",
|
|
" character_description = ChatOpenAI(temperature=1.0)(character_specifier_prompt).content\n",
|
|
" return character_description\n",
|
|
"\n",
|
|
"def generate_character_system_message(character_name, character_description):\n",
|
|
" return SystemMessage(content=(\n",
|
|
" f\"\"\"{game_description}\n",
|
|
" Your name is {character_name}. \n",
|
|
" Your character description is as follows: {character_description}.\n",
|
|
" You will propose actions you plan to take and {storyteller_name} will explain what happens when you take those actions.\n",
|
|
" Speak in the first person from the perspective of {character_name}.\n",
|
|
" For describing your own body movements, wrap your description in '*'.\n",
|
|
" Do not change roles!\n",
|
|
" Do not speak from the perspective of anyone else.\n",
|
|
" Remember you are {character_name}.\n",
|
|
" Stop speaking the moment you finish speaking from your perspective.\n",
|
|
" Never forget to keep your response to {word_limit} words!\n",
|
|
" Do not add anything else.\n",
|
|
" \"\"\"\n",
|
|
" ))\n",
|
|
"\n",
|
|
"character_descriptions = [generate_character_description(character_name) for character_name in character_names]\n",
|
|
"character_system_messages = [generate_character_system_message(character_name, character_description) for character_name, character_description in zip(character_names, character_descriptions)]\n",
|
|
"\n",
|
|
"storyteller_specifier_prompt = [\n",
|
|
" player_descriptor_system_message,\n",
|
|
" HumanMessage(content=\n",
|
|
" f\"\"\"{game_description}\n",
|
|
" Please reply with a creative description of the storyteller, {storyteller_name}, in {word_limit} words or less. \n",
|
|
" Speak directly to {storyteller_name}.\n",
|
|
" Do not add anything else.\"\"\"\n",
|
|
" )\n",
|
|
"]\n",
|
|
"storyteller_description = ChatOpenAI(temperature=1.0)(storyteller_specifier_prompt).content\n",
|
|
"\n",
|
|
"storyteller_system_message = SystemMessage(content=(\n",
|
|
"f\"\"\"{game_description}\n",
|
|
"You are the storyteller, {storyteller_name}. \n",
|
|
"Your description is as follows: {storyteller_description}.\n",
|
|
"The other players will propose actions to take and you will explain what happens when they take those actions.\n",
|
|
"Speak in the first person from the perspective of {storyteller_name}.\n",
|
|
"Do not change roles!\n",
|
|
"Do not speak from the perspective of anyone else.\n",
|
|
"Remember you are the storyteller, {storyteller_name}.\n",
|
|
"Stop speaking the moment you finish speaking from your perspective.\n",
|
|
"Never forget to keep your response to {word_limit} words!\n",
|
|
"Do not add anything else.\n",
|
|
"\"\"\"\n",
|
|
"))"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 6,
|
|
"metadata": {},
|
|
"outputs": [
|
|
{
|
|
"name": "stdout",
|
|
"output_type": "stream",
|
|
"text": [
|
|
"Storyteller Description:\n",
|
|
"Dungeon Master, your vivid imagination conjures a world of wonder and danger. Will you lead our triumphant trio or be the ultimate foil to their quest to rid the world of Voldemort's horcruxes? The fate of both the muggle and wizarding worlds rests in your hands.\n",
|
|
"Harry Potter Description:\n",
|
|
"Harry Potter, the boy who lived, you hold the fate of the wizarding world in your hands. Your bravery and loyalty to your friends are unmatched. The burden you carry is heavy, but with the power of love by your side, you can overcome any obstacle. The hunt for the horcruxes begins now.\n",
|
|
"Ron Weasley Description:\n",
|
|
"Ron Weasley, you are Harry Potter's loyal and brave best friend. You have a great sense of humor and always bring joy to the team. Your skills with magic and strategy make you a valuable asset in the fight against Voldemort. Your love for food and your family keeps you grounded and motivated.\n",
|
|
"Hermione Granger Description:\n",
|
|
"Hermione Granger, you are the brightest witch of your age. Your quick wit and vast knowledge are essential in our quest to find the horcruxes. Trust in your abilities and remember, knowledge is power.\n",
|
|
"Argus Filch Description:\n",
|
|
"Argus Filch, you are a bitter and cruel caretaker of the Hogwarts School of Witchcraft and Wizardry. Your harsh mannerisms and love for punishing the students know no bounds. Your loyalty to the Wizarding World and disdain for magic-wielders makes it surprising that you would join Harry, Ron, and Hermione in their quest to defeat Voldemort.\n"
|
|
]
|
|
}
|
|
],
|
|
"source": [
|
|
"print('Storyteller Description:')\n",
|
|
"print(storyteller_description)\n",
|
|
"for character_name, character_description in zip(character_names, character_descriptions):\n",
|
|
" print(f'{character_name} Description:')\n",
|
|
" print(character_description)"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Use an LLM to create an elaborate quest description"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 7,
|
|
"metadata": {},
|
|
"outputs": [
|
|
{
|
|
"name": "stdout",
|
|
"output_type": "stream",
|
|
"text": [
|
|
"Original quest:\n",
|
|
"Find all of Lord Voldemort's seven horcruxes.\n",
|
|
"\n",
|
|
"Detailed quest:\n",
|
|
"You have discovered that one of Voldemort's horcruxes is hidden deep in the Forbidden Forest. You must navigate the dangerous terrain, avoid the creatures lurking within, and find the horcrux before the full moon rises, unleashing a pack of hungry werewolves. Remember, time is of the essence!\n",
|
|
"\n"
|
|
]
|
|
}
|
|
],
|
|
"source": [
|
|
"quest_specifier_prompt = [\n",
|
|
" SystemMessage(content=\"You can make a task more specific.\"),\n",
|
|
" HumanMessage(content=\n",
|
|
" f\"\"\"{game_description}\n",
|
|
" \n",
|
|
" You are the storyteller, {storyteller_name}.\n",
|
|
" Please make the quest more specific. Be creative and imaginative.\n",
|
|
" Please reply with the specified quest in {word_limit} words or less. \n",
|
|
" Speak directly to the characters: {*character_names,}.\n",
|
|
" Do not add anything else.\"\"\"\n",
|
|
" )\n",
|
|
"]\n",
|
|
"specified_quest = ChatOpenAI(temperature=1.0)(quest_specifier_prompt).content\n",
|
|
"\n",
|
|
"print(f\"Original quest:\\n{quest}\\n\")\n",
|
|
"print(f\"Detailed quest:\\n{specified_quest}\\n\")"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Main Loop"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 8,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"characters = []\n",
|
|
"for character_name, character_system_message in zip(character_names, character_system_messages):\n",
|
|
" characters.append(DialogueAgent(\n",
|
|
" name=character_name,\n",
|
|
" system_message=character_system_message, \n",
|
|
" model=ChatOpenAI(temperature=0.2)))\n",
|
|
"storyteller = DialogueAgent(name=storyteller_name,\n",
|
|
" system_message=storyteller_system_message, \n",
|
|
" model=ChatOpenAI(temperature=0.2))"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 9,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"def select_next_speaker(step: int, agents: List[DialogueAgent]) -> int:\n",
|
|
" \"\"\"\n",
|
|
" If the step is even, then select the storyteller\n",
|
|
" Otherwise, select the other characters in a round-robin fashion.\n",
|
|
" \n",
|
|
" For example, with three characters with indices: 1 2 3\n",
|
|
" The storyteller is index 0.\n",
|
|
" Then the selected index will be as follows:\n",
|
|
"\n",
|
|
" step: 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16\n",
|
|
"\n",
|
|
" idx: 0 1 0 2 0 3 0 1 0 2 0 3 0 1 0 2 0\n",
|
|
" \"\"\"\n",
|
|
" if step % 2 == 0:\n",
|
|
" idx = 0\n",
|
|
" else:\n",
|
|
" idx = (step//2) % (len(agents)-1) + 1\n",
|
|
" return idx"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 10,
|
|
"metadata": {},
|
|
"outputs": [
|
|
{
|
|
"name": "stdout",
|
|
"output_type": "stream",
|
|
"text": [
|
|
"(Dungeon Master): You have discovered that one of Voldemort's horcruxes is hidden deep in the Forbidden Forest. You must navigate the dangerous terrain, avoid the creatures lurking within, and find the horcrux before the full moon rises, unleashing a pack of hungry werewolves. Remember, time is of the essence!\n",
|
|
"\n",
|
|
"\n",
|
|
"(Harry Potter): I take out my wand and cast a Lumos spell to light our way through the dark forest. We need to move quickly and quietly to avoid any unwanted attention from the creatures. Ron, Hermione, and I will lead the way while Argus Filch keeps watch behind us. Let's go!\n",
|
|
"\n",
|
|
"\n",
|
|
"(Dungeon Master): As you make your way through the forest, you hear the rustling of leaves and the snapping of twigs. Suddenly, a group of acromantulas, giant spiders, appear in front of you, blocking your path. What do you do?\n",
|
|
"\n",
|
|
"\n",
|
|
"(Ron Weasley): I quickly cast a spell to create a wall of fire between us and the acromantulas. Hopefully, the flames will deter them from attacking us. We need to keep moving forward and find that horcrux before it's too late.\n",
|
|
"\n",
|
|
"\n",
|
|
"(Dungeon Master): The acromantulas hiss and retreat from the wall of fire, allowing you to pass. As you continue deeper into the forest, you come across a clearing with a small pond. In the center of the pond, you see a glowing object. It must be the horcrux! But how do you get to it? What do you do?\n",
|
|
"\n",
|
|
"\n",
|
|
"(Hermione Granger): I take out my wand and cast a spell to conjure a small boat. We can use it to reach the center of the pond and retrieve the horcrux. But we need to be careful, there could be traps or other obstacles in our way. Ron, Harry, let's row the boat while Argus Filch keeps watch from the shore.\n",
|
|
"\n",
|
|
"\n",
|
|
"(Dungeon Master): As you row towards the center of the pond, you hear a loud hissing sound. Suddenly, a giant serpent emerges from the water, blocking your path. It looks angry and ready to attack. What do you do?\n",
|
|
"\n",
|
|
"\n",
|
|
"(Argus Filch): I take out my crossbow and aim it at the serpent. I may not be a wizard, but I know how to handle a weapon. I'll shoot it if it comes any closer. We can't let this serpent stop us from getting that horcrux.\n",
|
|
"\n",
|
|
"\n",
|
|
"(Dungeon Master): The serpent lunges towards the boat, but Argus Filch's crossbow bolt hits it in the head, causing it to retreat back into the water. You reach the center of the pond and retrieve the glowing object, which turns out to be a locket. Congratulations, you have found one of Voldemort's horcruxes! But there are still six more to find. What challenges will you face next?\n",
|
|
"\n",
|
|
"\n",
|
|
"(Harry Potter): We need to regroup and figure out our next move. We should head back to Hogwarts and consult with Professor Dumbledore's portrait. He may have some insight on where the other horcruxes could be hidden. We can't waste any time, Voldemort is getting stronger every day. Let's go!\n",
|
|
"\n",
|
|
"\n",
|
|
"(Dungeon Master): As you make your way back to Hogwarts, you hear a loud roar coming from the Forbidden Forest. It sounds like a werewolf. You must hurry before it catches up to you. You arrive at Dumbledore's office and he tells you that the next horcrux is hidden in a dangerous location. Are you ready for the next challenge?\n",
|
|
"\n",
|
|
"\n",
|
|
"(Ron Weasley): I'm always ready for a challenge! What's the location and what do we need to do to get there? We can't let Voldemort win, we have to find all of the horcruxes and destroy them. Let's do this!\n",
|
|
"\n",
|
|
"\n",
|
|
"(Dungeon Master): Dumbledore tells you that the next horcrux is hidden in the depths of Gringotts Bank. You must break into the bank, navigate its treacherous security measures, and find the horcrux before the goblins catch you. Are you ready to face the challenge of a lifetime? The fate of the wizarding world rests in your hands.\n",
|
|
"\n",
|
|
"\n",
|
|
"(Hermione Granger): I suggest we do some research on Gringotts Bank and its security measures before we attempt to break in. We need to be prepared and have a solid plan in place. We can also gather any necessary tools or potions that may help us along the way. Let's not rush into this blindly.\n",
|
|
"\n",
|
|
"\n",
|
|
"(Dungeon Master): As you research and plan your break-in to Gringotts Bank, you discover that the bank is heavily guarded by goblins, dragons, and other dangerous creatures. You'll need to be stealthy and quick to avoid detection. Are you ready to put your plan into action and face the dangers that await you? The clock is ticking, Voldemort's power grows stronger with each passing day.\n",
|
|
"\n",
|
|
"\n",
|
|
"(Argus Filch): I'll make sure to keep watch outside the bank while you all go in. I may not be able to help with the magic, but I can make sure no one interferes with our mission. We can't let anyone stop us from finding that horcrux and defeating Voldemort. Let's go!\n",
|
|
"\n",
|
|
"\n",
|
|
"(Dungeon Master): As you approach Gringotts Bank, you see the imposing structure looming before you. You sneak past the guards and make your way inside, navigating the twisting corridors and avoiding the traps set to catch intruders. Finally, you reach the vault where the horcrux is hidden. But it's guarded by a fierce dragon. What do you do?\n",
|
|
"\n",
|
|
"\n",
|
|
"(Harry Potter): I remember the time when I faced a dragon during the Triwizard Tournament. I take out my wand and cast a spell to distract the dragon while Ron and Hermione retrieve the horcrux. We need to work together and be quick. Time is running out and we can't afford to fail.\n",
|
|
"\n",
|
|
"\n",
|
|
"(Dungeon Master): The dragon roars and breathes fire, but Harry's spell distracts it long enough for Ron and Hermione to retrieve the horcrux. You make your way out of Gringotts Bank, but the goblins are hot on your trail. You must escape before they catch you. Congratulations, you have found another horcrux. But there are still five more to go. What challenges will you face next?\n",
|
|
"\n",
|
|
"\n",
|
|
"(Ron Weasley): We need to regroup and figure out our next move. We should consult with Professor Dumbledore's portrait again and see if he has any information on the next horcrux. We also need to be prepared for whatever challenges come our way. Voldemort won't make it easy for us, but we can't give up. Let's go!\n",
|
|
"\n",
|
|
"\n",
|
|
"(Dungeon Master): As you make your way back to Hogwarts, you hear a loud explosion coming from the direction of Hogsmeade. You arrive to find that Death Eaters have attacked the village and are wreaking havoc. You must fight off the Death Eaters and protect the innocent villagers. Are you ready to face this unexpected challenge and defend the wizarding world? The fate of both muggles and wizards rests in your hands.\n",
|
|
"\n",
|
|
"\n"
|
|
]
|
|
}
|
|
],
|
|
"source": [
|
|
"max_iters = 20\n",
|
|
"n = 0\n",
|
|
"\n",
|
|
"simulator = DialogueSimulator(\n",
|
|
" agents=[storyteller] + characters,\n",
|
|
" selection_function=select_next_speaker\n",
|
|
")\n",
|
|
"simulator.reset(storyteller_name, specified_quest)\n",
|
|
"print(f\"({storyteller_name}): {specified_quest}\")\n",
|
|
"print('\\n')\n",
|
|
"\n",
|
|
"while n < max_iters:\n",
|
|
" name, message = simulator.step()\n",
|
|
" print(f\"({name}): {message}\")\n",
|
|
" print('\\n')\n",
|
|
" n += 1"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": []
|
|
}
|
|
],
|
|
"metadata": {
|
|
"kernelspec": {
|
|
"display_name": "Python 3 (ipykernel)",
|
|
"language": "python",
|
|
"name": "python3"
|
|
},
|
|
"language_info": {
|
|
"codemirror_mode": {
|
|
"name": "ipython",
|
|
"version": 3
|
|
},
|
|
"file_extension": ".py",
|
|
"mimetype": "text/x-python",
|
|
"name": "python",
|
|
"nbconvert_exporter": "python",
|
|
"pygments_lexer": "ipython3",
|
|
"version": "3.9.16"
|
|
}
|
|
},
|
|
"nbformat": 4,
|
|
"nbformat_minor": 2
|
|
}
|