File size: 937 Bytes
82972f8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
from typing import Union
from pydantic import BaseModel
class Prompt(BaseModel):
file_path: str
cached: Union[str, None] = None
hot_reload: bool = True
def render(self, **kwargs) -> str:
if (self.cached is None) or (self.hot_reload):
with open(self.file_path, "r") as f:
self.cached = f.read()
return self.cached.format(**kwargs)
class PromptMgr:
def __init__(self, hot_reload: bool = True, src_dir: str = "resources/prompts"):
"""Creates a prompt manager.
Args:
hot_reload: If true, reloads the prompt every time it is called.
src_dir: The directory where the prompts are stored.
"""
self.hot_reload = hot_reload
self.src_dir = src_dir
def bind(self, prompt_id: str) -> Prompt:
return Prompt(
file_path=f"{self.src_dir}/{prompt_id}.txt", hot_reload=self.hot_reload
)
|