Paillat-dev/viralfactory
1
0
Fork 0
mirror of https://github.com/Paillat-dev/viralfactory.git synced 2026-01-02 01:06:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix dict key

Browse Source
This commit is contained in:
Paillat
2024-02-15 17:54:21 +01:00
parent 45a48cfa49
commit e3deafbf75
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/utils/prompting.py
View File

@@ -2,13 +2,16 @@ import yaml
import os import os
from typing import TypedDict from typing import TypedDict
class Prompt(TypedDict): class Prompt(TypedDict):
system: str system: str
chat: str chat: str
def get_prompt(name, *, location = "src/chore/prompts") -> Prompt:
def get_prompt(name, *, location="src/chore/prompts") -> tuple[str, str]:
path = os.path.join(os.getcwd(), location, f"{name}.yaml") path = os.path.join(os.getcwd(), location, f"{name}.yaml")
if not os.path.exists(path): if not os.path.exists(path):
raise FileNotFoundError(f"Prompt file {path} does not exist.") raise FileNotFoundError(f"Prompt file {path} does not exist.")
with open(path, "r") as file: with open(path, "r") as file:
return yaml.safe_load(file) prompt: Prompt = yaml.safe_load(file)
return prompt["system"], prompt["chat"]