ssube
/
taleweave-ai
1
0
Fork 0
Code Actions

clean up some generator prompts, make larger worlds

This commit is contained in:
Sean Sube 2024-05-07 20:39:58 -05:00
parent 3e27dfb129
commit 469ae79a9c
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
adventure/context.py
View File

@ -134,3 +134,7 @@ def get_dungeon_master() -> Agent:
) )
return dungeon_master return dungeon_master
def has_dungeon_master():
return dungeon_master is not None

21
adventure/generate.py
View File

@ -35,7 +35,7 @@ def generate_room(
name = loop_retry( name = loop_retry(
agent, agent,
"Generate one room, area, or location that would make sense in the world of {world_theme}. " "Generate one room, area, or location that would make sense in the world of {world_theme}. "
"Only respond with the room name, do not include the description or any other text. " "Only respond with the room name in title case, do not include the description or any other text. "
'Do not prefix the name with "the", do not wrap it in quotes. The existing rooms are: {existing_rooms}', 'Do not prefix the name with "the", do not wrap it in quotes. The existing rooms are: {existing_rooms}',
context={ context={
"world_theme": world_theme, "world_theme": world_theme,
@ -86,9 +86,10 @@ def generate_item(
name = loop_retry( name = loop_retry(
agent, agent,
"Generate one item or object that would make sense in the world of {world_theme}. {dest_note}. " "Generate one item or object that would make sense in the world of {world_theme}. {dest_note}. "
"Only respond with the item name, do not include a description or any other text. Do not prefix the " "Only respond with the item name in title case, do not include a description or any other text. Do not prefix the "
'name with "the", do not wrap it in quotes. Do not include the name of the room. ' 'name with "the", do not wrap it in quotes. Do not include the name of the room. Use a unique name. '
"Do not create any duplicate items in the same room. Do not give characters any duplicate items. The existing items are: {existing_items}", "Do not create any duplicate items in the same room. Do not give characters any duplicate items. "
"The existing items are: {existing_items}",
context={ context={
"dest_note": dest_note, "dest_note": dest_note,
"existing_items": existing_items, "existing_items": existing_items,
@ -125,8 +126,10 @@ def generate_actor(
name = loop_retry( name = loop_retry(
agent, agent,
"Generate one person or creature that would make sense in the world of {world_theme}. The character will be placed in the {dest_room} room. " "Generate one person or creature that would make sense in the world of {world_theme}. "
'Only respond with the character name, do not include a description or any other text. Do not prefix the name with "the", do not wrap it in quotes. ' "The character will be placed in the {dest_room} room. "
"Only respond with the character name in title case, do not include a description or any other text. "
'Do not prefix the name with "the", do not wrap it in quotes. '
"Do not include the name of the room. Do not give characters any duplicate names." "Do not include the name of the room. Do not give characters any duplicate names."
"Do not create any duplicate characters. The existing characters are: {existing_actors}", "Do not create any duplicate characters. The existing characters are: {existing_actors}",
context={ context={
@ -186,7 +189,7 @@ def generate_world(
rooms.append(room) rooms.append(room)
existing_rooms.append(room.name) existing_rooms.append(room.name)
item_count = randint(0, 3) item_count = randint(1, 3)
if callable(callback): if callable(callback):
callback(f"Generating {item_count} items for room: {room.name}") callback(f"Generating {item_count} items for room: {room.name}")
@ -202,7 +205,7 @@ def generate_world(
room.items.append(item) room.items.append(item)
existing_items.append(item.name) existing_items.append(item.name)
actor_count = randint(0, 3) actor_count = randint(1, 3)
if callable(callback): if callable(callback):
callback(f"Generating {actor_count} actors for room: {room.name}") callback(f"Generating {actor_count} actors for room: {room.name}")
@ -219,7 +222,7 @@ def generate_world(
existing_actors.append(actor.name) existing_actors.append(actor.name)
# generate the actor's inventory # generate the actor's inventory
item_count = randint(0, 3) item_count = randint(0, 2)
if callable(callback): if callable(callback):
callback(f"Generating {item_count} items for actor {actor.name}") callback(f"Generating {item_count} items for actor {actor.name}")