diff --git a/src/banks/extensions/generate.py b/src/banks/extensions/generate.py index 8dce66c..65845c7 100644 --- a/src/banks/extensions/generate.py +++ b/src/banks/extensions/generate.py @@ -14,7 +14,7 @@ SYSTEM_PROMPT = Prompt("{{canary_word}} You are a helpful assistant.") -def generate(model_name: str): +def generate(model_name: str): # noqa """ `generate` can be used to call the LiteLLM API passing the tag text as a prompt and get back some content. diff --git a/src/banks/prompt.py b/src/banks/prompt.py index bc0f16a..9e26037 100644 --- a/src/banks/prompt.py +++ b/src/banks/prompt.py @@ -47,7 +47,7 @@ def from_template(cls, name: str) -> "BasePrompt": ``` . └── templates -    └── foo.jinja + └── foo.jinja ``` The code would be the following: