Pārlūkot izejas kodu

fix: emoji generation

Timothy J. Baek 6 mēneši atpakaļ
vecāks
revīzija
424062d75f
2 mainītis faili ar 18 papildinājumiem un 1 dzēšanām
  1. 2 1
      backend/open_webui/main.py
  2. 16 0
      backend/open_webui/utils/task.py

+ 2 - 1
backend/open_webui/main.py

@@ -122,6 +122,7 @@ from open_webui.utils.task import (
     moa_response_generation_template,
     tags_generation_template,
     search_query_generation_template,
+    emoji_generation_template,
     title_generation_template,
     tools_function_calling_generation_template,
 )
@@ -1718,7 +1719,7 @@ Your task is to reflect the speaker's likely facial expression through a fitting
 
 Message: """{{prompt}}"""
 '''
-    content = title_generation_template(
+    content = emoji_generation_template(
         template,
         form_data["prompt"],
         {

+ 16 - 0
backend/open_webui/utils/task.py

@@ -147,6 +147,22 @@ def tags_generation_template(
     return template
 
 
+def emoji_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 search_query_generation_template(
     template: str, messages: list[dict], user: Optional[dict] = None
 ) -> str: