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

Formatting

Browse Source
This commit is contained in:
Paillat
2024-02-20 14:47:54 +01:00
parent 3c3761ba7b
commit 280df1dd67
8 changed files with 58 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/utils/prompting.py
View File

@@ -8,11 +8,15 @@ class Prompt(TypedDict):
chat: str
def get_prompt(name, *, location: str="src/chore/prompts", by_file_location: str = None) -> tuple[str, str]:
def get_prompt(
name, *, location: str = "src/chore/prompts", by_file_location: str = None
) -> tuple[str, str]:
if by_file_location:
path = os.path.join(
os.path.dirname(os.path.abspath(by_file_location)), "prompts", f"{name}.yaml"
)
os.path.dirname(os.path.abspath(by_file_location)),
"prompts",
f"{name}.yaml",
)
else:
path = os.path.join(os.getcwd(), location, f"{name}.yaml")
if not os.path.exists(path):