feat: follow ups

This commit is contained in:
Timothy Jaeryang Baek
2025-06-03 18:07:29 +04:00
parent f8b941fb96
commit 9e49fbc8bf
6 changed files with 176 additions and 0 deletions

View File

@@ -207,6 +207,24 @@ def title_generation_template(
return template
def follow_up_generation_template(
template: str, messages: list[dict], user: Optional[dict] = None
) -> str:
prompt = get_last_user_message(messages)
template = replace_prompt_variable(template, prompt)
template = replace_messages_variable(template, messages)
template = prompt_template(
template,
**(
{"user_name": user.get("name"), "user_location": user.get("location")}
if user
else {}
),
)
return template
def tags_generation_template(
template: str, messages: list[dict], user: Optional[dict] = None
) -> str: