enh: autocompletion

This commit is contained in:
Timothy Jaeryang Baek
2024-11-30 00:29:27 -08:00
parent ba6dc71810
commit 1f53e0922e
7 changed files with 77 additions and 15 deletions

View File

@@ -214,13 +214,17 @@ def emoji_generation_template(
def autocomplete_generation_template(
template: str,
prompt: Optional[str] = None,
prompt: str,
messages: Optional[list[dict]] = None,
type: Optional[str] = None,
user: Optional[dict] = None,
) -> str:
template = template.replace("{{TYPE}}", type if type else "")
template = replace_prompt_variable(template, prompt)
if messages:
template = replace_messages_variable(template, messages)
template = prompt_template(
template,
**(