Merge pull request drudilorenzo#10 from crcresearch/search-and-rescue

Search and Rescue branch merge
jackboyla · Oct 21, 2024 · 3b400c7 · 3b400c7
2 parents adaf0bd + 85ef4dd
commit 3b400c7
Show file tree

Hide file tree

Showing 41 changed files with 943 additions and 709 deletions.
diff --git a/.gitignore b/.gitignore
@@ -124,4 +124,4 @@ ENV/
 # Rope project settings
 .ropeproject
 
-.history/
+.history/
diff --git a/README.md b/README.md
@@ -13,6 +13,7 @@ _______________________________________
 1. [Setup](#setting-up-the-environment)
 2. [Execution](#running-a-simulation)
 3. [Cost-Tracking](#cost-tracking)
+4. [Customizing the Map](#customizing-the-map)
 _______________________________________
 
 ## Setting Up The Environment
@@ -126,7 +127,7 @@ The following script offer a range of enhanced features:
 - **Headless Mode**: The scripts support running simulations in headless mode, enabling execution on a server without a UI.
 - **Configurable Port Number**: You can configure the port number as needed.
 
-For more details, refer to: [run_backend_automatic.sh](https://github.com/drudilorenzo/generative_agents/blob/fix-and-improve/run_backend_automatic.sh) and [automatic_execution.py](https://github.com/drudilorenzo/generative_agents/blob/fix-and-improve/reverie/backend_server/automatic_execution.py).
+For more details, refer to: [run_backend_automatic.sh](run_backend_automatic.sh) and [automatic_execution.py](reverie/backend_server/automatic_execution.py).
 ```bash
     ./run_backend_automatic.sh -o <ORIGIN> -t <TARGET> -s <STEP> --ui <True|None|False> -p <PORT> --browser_path <BROWSER-PATH>
 ```
@@ -156,7 +157,7 @@ For a more detailed explanation see the [original readme](README_origin.md).
 
 For the cost tracking is used the package "[openai-cost-logger](https://github.com/drudilorenzo/openai-cost-logger)". Given the possible high cost of a simulation,  you can set a cost upperbound in the config file to be able to raise an exception and stop the execution when it is reached.
 
-See all the details of your expenses using the notebook "[cost_viz.ipynb](https://github.com/drudilorenzo/generative_agents/blob/main/cost_viz.ipynb)."
+See all the details of your expenses using the notebook "[cost_viz.ipynb](cost_viz.ipynb)."
 
 ## Cost Assessment
 
@@ -170,7 +171,7 @@ See all the details of your expenses using the notebook "[cost_viz.ipynb](https:
 
 ### 2. base_the_ville_n25
 
-- See the simulation saved: [skip-morning-s-14](https://github.com/drudilorenzo/generative_agents/tree/fix-and-improve/environment/frontend_server/storage/skip-morning-s-14)
+- See the simulation saved: [skip-morning-s-14](environment/frontend_server/storage/skip-morning-s-14)
 - **Model**: "gpt-3.5-turbo-0125"
 - **Embeddings**: "text-embedding-3-small"
 - **N. Agents**: 25
@@ -184,3 +185,44 @@ See all the details of your expenses using the notebook "[cost_viz.ipynb](https:
 - **N. Agents**: 25
 - **Steps**: ~8650 (full day)
 - **Final Cost**: ~18.5 USD
+
+## Customizing the Map
+
+The default simulation map, "The Ville", is a small town with locations such as a college, apartments, a cafe, a market, etc. The full list of locations and objects in this map are contained in the following files: [`sector_blocks.csv`](environment/frontend_server/static_dirs/assets/the_ville/matrix/special_blocks/sector_blocks.csv), [`arena_blocks.csv`](environment/frontend_server/static_dirs/assets/the_ville/matrix/special_blocks/arena_blocks.csv), and [`game_object_blocks.csv`](environment/frontend_server/static_dirs/assets/the_ville/matrix/special_blocks/game_object_blocks.csv). These are organized in a rough hierarchy: sector blocks roughly define buildings, arena blocks define rooms in buildings, and game object blocks define objects or areas in rooms.
+
+To fully overhaul the map for your own customized simulation, you'd probably need to use the Tiled map editor as described in the [original repo's README](README_origin.md). This repo has added a shortcut method of customizing the map: renaming locations and objects that already exist in the Ville map.
+
+To use this feature, add a `block_remaps` property to your simulation's `meta.json` file. Here's an example of remapping the supply store to be a fire station instead:
+
+```json
+{
+  "fork_sim_code": "base_the_ville_isabella_maria_klaus",
+  "start_date": "February 13, 2023",
+  "curr_time": "February 13, 2023, 00:00:00",
+  "sec_per_step": 10,
+  "maze_name": "the_ville",
+  "persona_names": [
+    "Isabella Rodriguez",
+    "Maria Lopez",
+    "Klaus Mueller"
+  ],
+  "step": 0,
+  "block_remaps": {
+    "sector": {
+      "Harvey Oak Supply Store": "Fire station"
+    },
+    "arena": {
+      "supply store": "fire station"
+    },
+    "game_object": {
+      "supply store product shelf": "fire truck",
+      "supply store counter": "common area",
+      "behind the supply store counter": "bunks"
+    }
+  }
+}
+```
+
+When using this feature, reference the existing blocks in the blocks CSVs listed above. **Make sure to spell and case everything exactly as they are in those files!**
+
+After remapping locations and objects in `meta.json`, you'll also need to rename them in each agent's `spatial_memory.json` file too, if they're referenced. This file defines what locations the agent is already aware of when the simulation starts. For instance, here's a link to Isabella's spatial memory file for the `base_the_ville_isabella_maria_klaus` simulation: [`spatial_memory.json`](environment/frontend_server/storage/base_the_ville_isabella_maria_klaus/personas/Isabella%20Rodriguez/bootstrap_memory/spatial_memory.json). Also, if a particular location is referenced in an agent's `scratch.json`, you'll need to update that too: [`scratch.json`](environment/frontend_server/storage/base_the_ville_isabella_maria_klaus/personas/Isabella%20Rodriguez/bootstrap_memory/scratch.json).
diff --git a/environment/frontend_server/storage/base_search_and_rescue/environment/0.json b/environment/frontend_server/storage/base_search_and_rescue/environment/0.json
@@ -0,0 +1,17 @@
+{
+  "Isabella Rodriguez": {
+    "maze": "the_ville",
+    "x": 72,
+    "y": 14
+  },
+  "Klaus Mueller": {
+    "maze": "the_ville",
+    "x": 126,
+    "y": 46
+  },
+  "Maria Lopez": {
+    "maze": "the_ville",
+    "x": 123,
+    "y": 57
+  }
+}
diff --git a/...nd_rescue/personas/Isabella Rodriguez/bootstrap_memory/associative_memory/embeddings.json b/...nd_rescue/personas/Isabella Rodriguez/bootstrap_memory/associative_memory/embeddings.json
@@ -0,0 +1 @@
+{}
diff --git a/...d_rescue/personas/Isabella Rodriguez/bootstrap_memory/associative_memory/kw_strength.json b/...d_rescue/personas/Isabella Rodriguez/bootstrap_memory/associative_memory/kw_strength.json
@@ -0,0 +1,2 @@
+{"kw_strength_event": {},
+ "kw_strength_thought": {}}
diff --git a/...rch_and_rescue/personas/Isabella Rodriguez/bootstrap_memory/associative_memory/nodes.json b/...rch_and_rescue/personas/Isabella Rodriguez/bootstrap_memory/associative_memory/nodes.json
@@ -0,0 +1 @@
+{}
diff --git a/.../storage/base_search_and_rescue/personas/Isabella Rodriguez/bootstrap_memory/scratch.json b/.../storage/base_search_and_rescue/personas/Isabella Rodriguez/bootstrap_memory/scratch.json
@@ -0,0 +1,51 @@
+{
+  "vision_r": 8, 
+  "att_bandwidth": 8, 
+  "retention": 8, 
+  "curr_time": null, 
+  "curr_tile": null, 
+  "daily_plan_req": "Isabella Rodriguez works at the fire station from 1:30am to 1:30pm.", 
+  "name": "Isabella Rodriguez", 
+  "first_name": "Isabella", 
+  "last_name": "Rodriguez", 
+  "age": 34, 
+  "innate": "responsible, trustworthy, stable", 
+  "learned": "Isabella Rodriguez is a firefighter. She works at the fire station and responds to any kind of emergency around town. Klaus is her coworker.", 
+  "currently": "Isabella is responding to a distress call about someone trapped in a building in town. She doesn't know exactly who or where this person is, so she's searching around town for them.", 
+  "lifestyle": "Isabella Rodriguez goes to bed around 6pm and awakes around 1am.", 
+  "living_area": "the Ville:Isabella Rodriguez's apartment:main room",
+  "concept_forget": 100,
+  "daily_reflection_time": 180,
+  "daily_reflection_size": 5,
+  "overlap_reflect_th": 4,
+  "kw_strg_event_reflect_th": 10, 
+  "kw_strg_thought_reflect_th": 9, 
+
+  "recency_w": 1,
+  "relevance_w": 1,
+  "importance_w": 1,
+  "recency_decay": 0.995,
+  "importance_trigger_max": 150,
+  "importance_trigger_curr": 150,
+  "importance_ele_n": 0, 
+  "thought_count": 5, 
+
+  "daily_req": [], 
+  "f_daily_schedule": [],
+  "f_daily_schedule_hourly_org": [],
+  "act_address": null, 
+  "act_start_time": null, 
+  "act_duration": null, 
+  "act_description": null, 
+  "act_pronunciatio": null, 
+  "act_event": ["Isabella Rodriguez", null, null], 
+  "act_obj_description": null, 
+  "act_obj_pronunciatio": null, 
+  "act_obj_event": [null, null, null], 
+  "chatting_with": null, 
+  "chat": null,
+  "chatting_with_buffer": {}, 
+  "chatting_end_time": null,
+  "act_path_set": false, 
+  "planned_path": []
+}
diff --git a/...e/base_search_and_rescue/personas/Isabella Rodriguez/bootstrap_memory/spatial_memory.json b/...e/base_search_and_rescue/personas/Isabella Rodriguez/bootstrap_memory/spatial_memory.json
@@ -0,0 +1,66 @@
+{
+  "the Ville": {
+    "Hobbs Cafe": {
+      "cafe": [
+        "refrigerator",
+        "cafe customer seating",
+        "cooking area",
+        "kitchen sink",
+        "behind the cafe counter",
+        "piano"
+      ]
+    },
+    "Isabella Rodriguez's apartment": {
+      "main room": [
+        "bed",
+        "desk",
+        "refrigerator",
+        "closet",
+        "shelf"
+      ]
+    },
+    "The Rose and Crown Pub": {
+      "pub": [
+        "shelf",
+        "refrigerator",
+        "bar customer seating",
+        "behind the bar counter",
+        "kitchen sink",
+        "cooking area",
+        "microphone"
+      ]
+    },
+    "Fire station": {
+      "fire station": [
+        "fire truck",
+        "bunks",
+        "common area"
+      ]
+    },
+    "The Willows Market and Pharmacy": {
+      "store": [
+        "behind the pharmacy counter",
+        "pharmacy store shelf",
+        "pharmacy store counter",
+        "grocery store shelf",
+        "behind the grocery counter",
+        "grocery store counter"
+      ]
+    },
+    "Dorm for Oak Hill College": {
+      "garden": [
+        "dorm garden"
+      ],
+      "common room": [
+        "common room sofa",
+        "pool table",
+        "common room table"
+      ]
+    },
+    "Johnson Park": {
+      "park": [
+        "park garden"
+      ]
+    }
+  }
+}
diff --git a/...rch_and_rescue/personas/Klaus Mueller/bootstrap_memory/associative_memory/embeddings.json b/...rch_and_rescue/personas/Klaus Mueller/bootstrap_memory/associative_memory/embeddings.json
@@ -0,0 +1 @@
+{}
diff --git a/...ch_and_rescue/personas/Klaus Mueller/bootstrap_memory/associative_memory/kw_strength.json b/...ch_and_rescue/personas/Klaus Mueller/bootstrap_memory/associative_memory/kw_strength.json
@@ -0,0 +1,2 @@
+{"kw_strength_event": {},
+ "kw_strength_thought": {}}
diff --git a/...e_search_and_rescue/personas/Klaus Mueller/bootstrap_memory/associative_memory/nodes.json b/...e_search_and_rescue/personas/Klaus Mueller/bootstrap_memory/associative_memory/nodes.json
@@ -0,0 +1 @@
+{}
diff --git a/...erver/storage/base_search_and_rescue/personas/Klaus Mueller/bootstrap_memory/scratch.json b/...erver/storage/base_search_and_rescue/personas/Klaus Mueller/bootstrap_memory/scratch.json
@@ -0,0 +1,51 @@
+{
+  "vision_r": 8, 
+  "att_bandwidth": 8, 
+  "retention": 8, 
+  "curr_time": null, 
+  "curr_tile": null, 
+  "daily_plan_req": "Klaus Mueller works at the fire station from 1:30am to 1:30pm.", 
+  "name": "Klaus Mueller", 
+  "first_name": "Klaus", 
+  "last_name": "Mueller", 
+  "age": 20,
+  "innate": "hardworking, resourceful, and friendly", 
+  "learned": "Klaus Mueller is a firefighter. He works at the fire station and responds to any kind of emergency around town. Isabella is his coworker.", 
+  "currently": "Klaus is responding to a distress call about someone trapped in a building in town. He doesn't know exactly who or where this person is, so he's searching around town for them.", 
+  "lifestyle": "Klaus Mueller goes to bed around 6pm and awakes around 1am.", 
+  "living_area": "the Ville:Dorm for Oak Hill College:Klaus Mueller's room",
+  "concept_forget": 100,
+  "daily_reflection_time": 180,
+  "daily_reflection_size": 5,
+  "overlap_reflect_th": 4,
+  "kw_strg_event_reflect_th": 10, 
+  "kw_strg_thought_reflect_th": 9, 
+
+  "recency_w": 1,
+  "relevance_w": 1,
+  "importance_w": 1,
+  "recency_decay": 0.99,
+  "importance_trigger_max": 150,
+  "importance_trigger_curr": 150,
+  "importance_ele_n": 0, 
+  "thought_count": 5, 
+
+  "daily_req": [], 
+  "f_daily_schedule": [],
+  "f_daily_schedule_hourly_org": [],
+  "act_address": null, 
+  "act_start_time": null, 
+  "act_duration": null, 
+  "act_description": null, 
+  "act_pronunciatio": null, 
+  "act_event": ["Klaus Mueller", null, null], 
+  "act_obj_description": null, 
+  "act_obj_pronunciatio": null, 
+  "act_obj_event": [null, null, null], 
+  "chatting_with": null, 
+  "chat": null,
+  "chatting_with_buffer": {}, 
+  "chatting_end_time": null,
+  "act_path_set": false, 
+  "planned_path": []
+}
diff --git a/...torage/base_search_and_rescue/personas/Klaus Mueller/bootstrap_memory/spatial_memory.json b/...torage/base_search_and_rescue/personas/Klaus Mueller/bootstrap_memory/spatial_memory.json
@@ -0,0 +1,86 @@
+{
+  "the Ville": {
+    "Oak Hill College": {
+      "hallway": [],
+      "library": [
+        "library sofa",
+        "library table",
+        "bookshelf"
+      ],
+      "classroom": [
+        "blackboard",
+        "classroom podium",
+        "classroom student seating"
+      ]
+    },
+    "Dorm for Oak Hill College": {
+      "garden": [
+        "dorm garden"
+      ],
+      "Klaus Mueller's room": [
+        "bed",
+        "game console",
+        "closet",
+        "desk"
+      ],
+      "woman's bathroom": [
+        "toilet",
+        "shower",
+        "bathroom sink"
+      ],
+      "common room": [
+        "common room sofa",
+        "pool table",
+        "common room table"
+      ],
+      "man's bathroom": [
+        "shower",
+        "bathroom sink",
+        "toilet"
+      ]
+    },
+    "The Willows Market and Pharmacy": {
+      "store": [
+        "grocery store shelf",
+        "behind the grocery counter",
+        "grocery store counter",
+        "pharmacy store shelf",
+        "pharmacy store counter",
+        "behind the pharmacy counter"
+      ]
+    },
+    "Fire station": {
+      "fire station": [
+        "fire truck",
+        "bunks",
+        "common area"
+      ]
+    },
+    "Johnson Park": {
+      "park": [
+        "park garden"
+      ]
+    },
+    "The Rose and Crown Pub": {
+      "pub": [
+        "shelf",
+        "refrigerator",
+        "bar customer seating",
+        "behind the bar counter",
+        "kitchen sink",
+        "cooking area",
+        "microphone"
+      ]
+    },
+    "Hobbs Cafe": {
+      "cafe": [
+        "refrigerator",
+        "cafe customer seating",
+        "cooking area",
+        "kitchen sink",
+        "behind the cafe counter",
+        "piano"
+      ]
+    }
+  }
+}
diff --git a/...earch_and_rescue/personas/Maria Lopez/bootstrap_memory/associative_memory/embeddings.json b/...earch_and_rescue/personas/Maria Lopez/bootstrap_memory/associative_memory/embeddings.json
@@ -0,0 +1 @@
+{}
diff --git a/...arch_and_rescue/personas/Maria Lopez/bootstrap_memory/associative_memory/kw_strength.json b/...arch_and_rescue/personas/Maria Lopez/bootstrap_memory/associative_memory/kw_strength.json
@@ -0,0 +1,2 @@
+{"kw_strength_event": {},
+ "kw_strength_thought": {}}
diff --git a/...ase_search_and_rescue/personas/Maria Lopez/bootstrap_memory/associative_memory/nodes.json b/...ase_search_and_rescue/personas/Maria Lopez/bootstrap_memory/associative_memory/nodes.json
@@ -0,0 +1 @@
+{}
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		{"kw_strength_event": {},
		"kw_strength_thought": {}}