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

refac: title generation

Browse Source
This commit is contained in:
Timothy J. Baek
2024-10-21 04:24:17 -07:00
parent 24885a2e38
commit e2d4a69750
4 changed files with 36 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -70,22 +70,6 @@ def replace_prompt_variable(template: str, prompt: str) -> str:
return template
def title_generation_template(
template: str, prompt: str, user: Optional[dict] = None
) -> str:
template = replace_prompt_variable(template, prompt)
template = prompt_template(
template,
**(
{"user_name": user.get("name"), "user_location": user.get("location")}
if user
else {}
),
)
return template
def replace_messages_variable(template: str, messages: list[str]) -> str:
def replacement_function(match):
full_match = match.group(0)
@@ -123,6 +107,28 @@ def replace_messages_variable(template: str, messages: list[str]) -> str:
return template
# {{prompt:middletruncate:8000}}
def title_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: