mirror of
https://github.com/Paillat-dev/viralfactory.git
synced 2026-01-02 01:06:19 +00:00
✨ feat(prompting): Add new function get_prompts for loading multiple prompts from one yaml file at a time
This commit is contained in:
@@ -25,3 +25,21 @@ def get_prompt(
|
||||
with open(path, "r") as file:
|
||||
prompt: Prompt = yaml.safe_load(file)
|
||||
return prompt["system"], prompt["chat"]
|
||||
|
||||
|
||||
def get_prompts(
|
||||
name, *, location: str = "src/chore/prompts", by_file_location: str = None
|
||||
) -> dict[str, Prompt]:
|
||||
if by_file_location:
|
||||
path = os.path.join(
|
||||
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):
|
||||
raise FileNotFoundError(f"Prompt file {path} does not exist.")
|
||||
with open(path, "r") as file:
|
||||
prompts: dict[str, Prompt] = yaml.safe_load(file)
|
||||
return prompts
|
||||
|
||||
Reference in New Issue
Block a user