From 26f578c6871a5894a326b60cb414649ea8c88d54 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 4 Jun 2024 08:57:35 -0500 Subject: [PATCH] fix missing generator prompt --- prompts/llama-base.yml | 2 ++ taleweave/generate.py | 2 +- taleweave/utils/world.py | 2 +- 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/prompts/llama-base.yml b/prompts/llama-base.yml index 96c5519..4c05ac2 100644 --- 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 index 2c0d1dc..cc9e3a9 100644 --- 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 index 6aa2161..c68c42d 100644 --- 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