fix missing generator prompt

ssube · Jun 4, 2024 · 26f578c · 26f578c
1 parent d4bf336
commit 26f578c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/prompts/llama-base.yml b/prompts/llama-base.yml
@@ -232,6 +232,8 @@ prompts:
     Generating {{item_count}} items for room: {{name}}
   world_generate_room_broadcast_characters: |
     Generating {{character_count}} characters for room: {{name}}
+  world_generate_room_broadcast_portals: |
+    Generating {{portal_count}} portals for room: {{name}}
 
   world_generate_portal_name_outgoing: |
     Generate the name of a portal that leads from the {{source_room}} room to the {{dest_room}} room and fits the world theme of {{world_theme}}.

diff --git a/taleweave/generate.py b/taleweave/generate.py
@@ -522,7 +522,7 @@ def link_rooms(
         broadcast_generated(
             format_prompt(
                 "world_generate_room_broadcast_portals",
-                num_portals=num_portals,
+                portal_count=num_portals,
                 name=room.name,
             )
         )

diff --git a/taleweave/utils/world.py b/taleweave/utils/world.py
@@ -12,7 +12,7 @@ def describe_character(
     perspective: FormatPerspective = FormatPerspective.SECOND_PERSON,
 ) -> str:
     attribute_descriptions = format_attributes(character, perspective=perspective)
-    logger.info("describing character: %s, %s", character, attribute_descriptions)
+    logger.info("describing character: %s, %s", character.name, attribute_descriptions)
 
     if perspective == FormatPerspective.SECOND_PERSON:
         character_description = character.backstory