UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-12-28 06:42:47 +00:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-11-30 23:50:05 -08:00
parent 746fe9ea16
commit 9b25efc3bb
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -81,12 +81,17 @@ def replace_prompt_variable(template: str, prompt: str) -> str:
return template return template
def replace_messages_variable(template: str, messages: list[str]) -> str: def replace_messages_variable(
template: str, messages: Optional[list[str]] = None
) -> str:
def replacement_function(match): def replacement_function(match):
full_match = match.group(0) full_match = match.group(0)
start_length = match.group(1) start_length = match.group(1)
end_length = match.group(2) end_length = match.group(2)
middle_length = match.group(3) middle_length = match.group(3)
# If messages is None, handle it as an empty list
if messages is None:
return ""
# Process messages based on the number of messages required # Process messages based on the number of messages required
if full_match == "{{MESSAGES}}": if full_match == "{{MESSAGES}}":
@ -221,9 +226,7 @@ def autocomplete_generation_template(
) -> str: ) -> str:
template = template.replace("{{TYPE}}", type if type else "") template = template.replace("{{TYPE}}", type if type else "")
template = replace_prompt_variable(template, prompt) template = replace_prompt_variable(template, prompt)
template = replace_messages_variable(template, messages)
if messages:
template = replace_messages_variable(template, messages)
template = prompt_template( template = prompt_template(
template, template,