Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 20:07:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat: autocomplete backend endpoint

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2024-11-28 23:53:52 -08:00
parent 28ce102a79
commit 0e8e9820d0
4 changed files with 139 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
backend/open_webui/utils/task.py
View File

@@ -212,6 +212,29 @@ def emoji_generation_template(
return template
def autocomplete_generation_template(
template: str,
messages: list[dict],
context: Optional[str] = None,
user: Optional[dict] = None,
) -> str:
prompt = get_last_user_message(messages)
template = template.replace("{{CONTEXT}}", context if context else "")
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 query_generation_template(
template: str, messages: list[dict], user: Optional[dict] = None
) -> str: