prompts.py

from string import Template
from langchain import PromptTemplate
from langchain.prompts import load_prompt
from utils import (
    request_openai,
    yaml_from_str,
    save_yaml_from_data,
    populate_dataclass_with_dicts,
    # debug,
)
import openai
from prompt_toolkit import prompt
import yaml

from logging import debug


new_world_state_request = Template(
    """$tick_rate ${tick_type}s have passed. Describe only the new world state and the world's temperature. \
If something's changed in the world or there was some event(s), please describe them. Output it as a python string \
assigned to `world_state` var in the code block and int `temperature` var in the same code block. No matter what \
- DON'T output any comments, NPC's changes, notes and additional descriptions.
"""
)

new_npc_state_request = Template(
    """Here is the NPC:
$npc_prompt
Describe ${npc_name}'s state. \
Output it as a python string assigned to `current_state_prompt` var in the code block.
In the same code block, output the NPC's change attributes (delta) as a python list assigned to `attributes` var. \
If the attribute is not changed, output 0 for it. \
If the attribute is increased, output a positive number for it. \
If the attribute is decreased, output a negative number for it. \
NPCs can interact with each other and with the world. NPCs are usually awake during the day. \
And they usually sleep at night. Take this into account in `current_state_prompt` \
No matter what - DON'T output any comments, NPC's changes, notes, additional descriptions and explanations.
"""
)

# World
world_init_state = load_prompt("data/prompts/world/init_state.yaml")
world_current_state = load_prompt("data/prompts/world/current_state.yaml")
world_previous_state = load_prompt("data/prompts/world/previous_state.yaml")
world_parameters = load_prompt("data/prompts/world/parameters.yaml")
world_new_state = load_prompt("data/prompts/world/new_state_request.yaml")


# NPC
create_npc_request = load_prompt("data/prompts/npc/create_npc_request.yaml")
create_global_goals = load_prompt("data/prompts/npc/create_global_goals.yaml")
create_social_connections = load_prompt(
    "data/prompts/npc/create_social_connections_request.yaml"
)
npc_new_state = load_prompt("data/prompts/npc/new_state_request.yaml")
generate_npc_image = load_prompt("data/prompts/npc/generate_npc_image.yaml")

if __name__ == "__main__":
    from classes import World, Npc, Settings

    openai.api_key = prompt("Provide OpenAI API key: ", is_password=True)

    settings = Settings()
    settings.load(path="./settings.yaml")
    from utils import (
        load_yaml_to_dataclass,
        hour_to_pm_am,
        hour_to_daytime,
        int_to_month,
        hour_to_iso,
        minute_to_iso,
        second_to_iso,
        YamlDumperDoubleQuotes,
        dataclass_to_dict_copy,
    )
    from resources_paths import *

    world = World()
    load_yaml_to_dataclass(
        world,
        "/mnt/Shared/Demania/repos/world-gpt/data/games/Test Game/worlds/Test_World/world_tick_0.yaml",
    )

    with open(YAML_TEMPLATES_PATH / "npc_social_connections.yaml", "r") as f:
        npc_social_connections = yaml.safe_load(f)

    npc_0 = Npc()
    npc_1 = Npc()
    npc_2 = Npc()
    npc_3 = Npc()

    load_yaml_to_dataclass(
        npc_0,
        "/mnt/Shared/Demania/repos/world-gpt/data/games/Test Game6/worlds/Test_World/npcs/Vasile Boieru/npc_tick_0.yaml",
    )

    load_yaml_to_dataclass(
        npc_1,
        "/mnt/Shared/Demania/repos/world-gpt/data/games/Test Game6/worlds/Test_World/npcs/Andrei Popescu/npc_tick_0.yaml",
    )
    load_yaml_to_dataclass(
        npc_2,
        "/mnt/Shared/Demania/repos/world-gpt/data/games/Test Game6/worlds/Test_World/npcs/Gheorghe Popescu/npc_tick_0.yaml",
    )
    load_yaml_to_dataclass(
        npc_3,
        "/mnt/Shared/Demania/repos/world-gpt/data/games/Test Game6/worlds/Test_World/npcs/Radu Carp/npc_tick_0.yaml",
    )

    original_npcs = [npc_1, npc_2, npc_3]
    other_npcs = []

    keys_to_delete = [
        key
        for key in npc_0.__dict__.keys()
        if key not in ["name", "current_state_prompt"]
    ]

    for npc in original_npcs:
        npc_dict = dataclass_to_dict_copy(npc, keys_to_delete)
        other_npcs.append([npc_dict])

    create_social_connections_request = create_social_connections.format(
        world_general_description=world.current_state_prompt,
        current_npc_name=npc_0.name,
        current_npc_state=npc_0.current_state_prompt,
        max_npc_social_connections=world.attributes["max_npc_social_connections"],
        npc_social_connection_yaml_template=npc_social_connections,
        other_npcs=[
            yaml.dump(npc, sort_keys=False, Dumper=YamlDumperDoubleQuotes)
            for npc in other_npcs
        ],
    )

    world_new_state_request = world_new_state.format(
        init_state=world.current_state_prompt,
        previous_state=world.current_state_prompt,
        current_state=world.current_state_prompt,
        attributes=world.attributes,
        date=world.time,
        tick_rate=world.tick_rate,
        tick_type=world.tick_type,
    )

    npc_0.social_connections = ["Radu Carp"]

    npc_new_state_request = npc_new_state.format(
        world_general_description=world.current_state_prompt,
        world_current_state="The mid-sized village by the river in medieval Moldova is in a state of panic as a group of bandits have been spotted approaching the village. The militia has been called upon to defend the village, and the villagers are preparing for battle. Fear and uncertainty grip the community as they face this dangerous threat.",
        world_attributes=world.attributes,
        date=world.time,
        current_npc=yaml.dump(npc_0, sort_keys=False, Dumper=YamlDumperDoubleQuotes),
        other_npcs=other_npcs,
        tick_rate=world.tick_rate,
        tick_type=world.tick_type,
    )

    debug(npc_new_state_request)