|
|
@@ -33,6 +33,7 @@ from open_webui.config import (
|
|
|
DEFAULT_AUTOCOMPLETE_GENERATION_PROMPT_TEMPLATE,
|
|
|
DEFAULT_EMOJI_GENERATION_PROMPT_TEMPLATE,
|
|
|
DEFAULT_MOA_GENERATION_PROMPT_TEMPLATE,
|
|
|
+ DEFAULT_VOICE_MODE_PROMPT_TEMPLATE,
|
|
|
)
|
|
|
from open_webui.env import SRC_LOG_LEVELS
|
|
|
|
|
|
@@ -68,6 +69,7 @@ async def get_task_config(request: Request, user=Depends(get_verified_user)):
|
|
|
"ENABLE_RETRIEVAL_QUERY_GENERATION": request.app.state.config.ENABLE_RETRIEVAL_QUERY_GENERATION,
|
|
|
"QUERY_GENERATION_PROMPT_TEMPLATE": request.app.state.config.QUERY_GENERATION_PROMPT_TEMPLATE,
|
|
|
"TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE": request.app.state.config.TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE,
|
|
|
+ "VOICE_MODE_PROMPT_TEMPLATE": request.app.state.config.VOICE_MODE_PROMPT_TEMPLATE,
|
|
|
}
|
|
|
|
|
|
|
|
|
@@ -87,6 +89,7 @@ class TaskConfigForm(BaseModel):
|
|
|
ENABLE_RETRIEVAL_QUERY_GENERATION: bool
|
|
|
QUERY_GENERATION_PROMPT_TEMPLATE: str
|
|
|
TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE: str
|
|
|
+ VOICE_MODE_PROMPT_TEMPLATE: Optional[str]
|
|
|
|
|
|
|
|
|
@router.post("/config/update")
|
|
|
@@ -136,6 +139,10 @@ async def update_task_config(
|
|
|
form_data.TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE
|
|
|
)
|
|
|
|
|
|
+ request.app.state.config.VOICE_MODE_PROMPT_TEMPLATE = (
|
|
|
+ form_data.VOICE_MODE_PROMPT_TEMPLATE
|
|
|
+ )
|
|
|
+
|
|
|
return {
|
|
|
"TASK_MODEL": request.app.state.config.TASK_MODEL,
|
|
|
"TASK_MODEL_EXTERNAL": request.app.state.config.TASK_MODEL_EXTERNAL,
|
|
|
@@ -152,6 +159,7 @@ async def update_task_config(
|
|
|
"ENABLE_RETRIEVAL_QUERY_GENERATION": request.app.state.config.ENABLE_RETRIEVAL_QUERY_GENERATION,
|
|
|
"QUERY_GENERATION_PROMPT_TEMPLATE": request.app.state.config.QUERY_GENERATION_PROMPT_TEMPLATE,
|
|
|
"TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE": request.app.state.config.TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE,
|
|
|
+ "VOICE_MODE_PROMPT_TEMPLATE": request.app.state.config.VOICE_MODE_PROMPT_TEMPLATE,
|
|
|
}
|
|
|
|
|
|
|