|
@@ -73,6 +73,7 @@ from open_webui.config import (
|
|
ENABLE_MODEL_FILTER,
|
|
ENABLE_MODEL_FILTER,
|
|
ENABLE_OLLAMA_API,
|
|
ENABLE_OLLAMA_API,
|
|
ENABLE_OPENAI_API,
|
|
ENABLE_OPENAI_API,
|
|
|
|
+ ENABLE_TAGS_GENERATION,
|
|
ENV,
|
|
ENV,
|
|
FRONTEND_BUILD_DIR,
|
|
FRONTEND_BUILD_DIR,
|
|
MODEL_FILTER_LIST,
|
|
MODEL_FILTER_LIST,
|
|
@@ -199,6 +200,7 @@ app.state.config.TASK_MODEL = TASK_MODEL
|
|
app.state.config.TASK_MODEL_EXTERNAL = TASK_MODEL_EXTERNAL
|
|
app.state.config.TASK_MODEL_EXTERNAL = TASK_MODEL_EXTERNAL
|
|
app.state.config.TITLE_GENERATION_PROMPT_TEMPLATE = TITLE_GENERATION_PROMPT_TEMPLATE
|
|
app.state.config.TITLE_GENERATION_PROMPT_TEMPLATE = TITLE_GENERATION_PROMPT_TEMPLATE
|
|
app.state.config.TAGS_GENERATION_PROMPT_TEMPLATE = TAGS_GENERATION_PROMPT_TEMPLATE
|
|
app.state.config.TAGS_GENERATION_PROMPT_TEMPLATE = TAGS_GENERATION_PROMPT_TEMPLATE
|
|
|
|
+app.state.config.ENABLE_TAGS_GENERATION = ENABLE_TAGS_GENERATION
|
|
app.state.config.SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE = (
|
|
app.state.config.SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE = (
|
|
SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE
|
|
SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE
|
|
)
|
|
)
|
|
@@ -1473,6 +1475,7 @@ async def get_task_config(user=Depends(get_verified_user)):
|
|
"TASK_MODEL_EXTERNAL": app.state.config.TASK_MODEL_EXTERNAL,
|
|
"TASK_MODEL_EXTERNAL": app.state.config.TASK_MODEL_EXTERNAL,
|
|
"TITLE_GENERATION_PROMPT_TEMPLATE": app.state.config.TITLE_GENERATION_PROMPT_TEMPLATE,
|
|
"TITLE_GENERATION_PROMPT_TEMPLATE": app.state.config.TITLE_GENERATION_PROMPT_TEMPLATE,
|
|
"TAGS_GENERATION_PROMPT_TEMPLATE": app.state.config.TAGS_GENERATION_PROMPT_TEMPLATE,
|
|
"TAGS_GENERATION_PROMPT_TEMPLATE": app.state.config.TAGS_GENERATION_PROMPT_TEMPLATE,
|
|
|
|
+ "ENABLE_TAGS_GENERATION": app.state.config.ENABLE_TAGS_GENERATION,
|
|
"ENABLE_SEARCH_QUERY": app.state.config.ENABLE_SEARCH_QUERY,
|
|
"ENABLE_SEARCH_QUERY": app.state.config.ENABLE_SEARCH_QUERY,
|
|
"SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE": app.state.config.SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE,
|
|
"SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE": app.state.config.SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE,
|
|
"TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE": app.state.config.TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE,
|
|
"TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE": app.state.config.TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE,
|
|
@@ -1484,6 +1487,7 @@ class TaskConfigForm(BaseModel):
|
|
TASK_MODEL_EXTERNAL: Optional[str]
|
|
TASK_MODEL_EXTERNAL: Optional[str]
|
|
TITLE_GENERATION_PROMPT_TEMPLATE: str
|
|
TITLE_GENERATION_PROMPT_TEMPLATE: str
|
|
TAGS_GENERATION_PROMPT_TEMPLATE: str
|
|
TAGS_GENERATION_PROMPT_TEMPLATE: str
|
|
|
|
+ ENABLE_TAGS_GENERATION: bool
|
|
SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE: str
|
|
SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE: str
|
|
ENABLE_SEARCH_QUERY: bool
|
|
ENABLE_SEARCH_QUERY: bool
|
|
TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE: str
|
|
TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE: str
|
|
@@ -1499,6 +1503,7 @@ async def update_task_config(form_data: TaskConfigForm, user=Depends(get_admin_u
|
|
app.state.config.TAGS_GENERATION_PROMPT_TEMPLATE = (
|
|
app.state.config.TAGS_GENERATION_PROMPT_TEMPLATE = (
|
|
form_data.TAGS_GENERATION_PROMPT_TEMPLATE
|
|
form_data.TAGS_GENERATION_PROMPT_TEMPLATE
|
|
)
|
|
)
|
|
|
|
+ app.state.config.ENABLE_TAGS_GENERATION = form_data.ENABLE_TAGS_GENERATION
|
|
|
|
|
|
app.state.config.SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE = (
|
|
app.state.config.SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE = (
|
|
form_data.SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE
|
|
form_data.SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE
|
|
@@ -1513,6 +1518,7 @@ async def update_task_config(form_data: TaskConfigForm, user=Depends(get_admin_u
|
|
"TASK_MODEL_EXTERNAL": app.state.config.TASK_MODEL_EXTERNAL,
|
|
"TASK_MODEL_EXTERNAL": app.state.config.TASK_MODEL_EXTERNAL,
|
|
"TITLE_GENERATION_PROMPT_TEMPLATE": app.state.config.TITLE_GENERATION_PROMPT_TEMPLATE,
|
|
"TITLE_GENERATION_PROMPT_TEMPLATE": app.state.config.TITLE_GENERATION_PROMPT_TEMPLATE,
|
|
"TAGS_GENERATION_PROMPT_TEMPLATE": app.state.config.TAGS_GENERATION_PROMPT_TEMPLATE,
|
|
"TAGS_GENERATION_PROMPT_TEMPLATE": app.state.config.TAGS_GENERATION_PROMPT_TEMPLATE,
|
|
|
|
+ "ENABLE_TAGS_GENERATION": app.state.config.ENABLE_TAGS_GENERATION,
|
|
"SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE": app.state.config.SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE,
|
|
"SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE": app.state.config.SEARCH_QUERY_GENERATION_PROMPT_TEMPLATE,
|
|
"ENABLE_SEARCH_QUERY": app.state.config.ENABLE_SEARCH_QUERY,
|
|
"ENABLE_SEARCH_QUERY": app.state.config.ENABLE_SEARCH_QUERY,
|
|
"TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE": app.state.config.TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE,
|
|
"TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE": app.state.config.TOOLS_FUNCTION_CALLING_PROMPT_TEMPLATE,
|
|
@@ -1602,6 +1608,12 @@ Artificial Intelligence in Healthcare
|
|
@app.post("/api/task/tags/completions")
|
|
@app.post("/api/task/tags/completions")
|
|
async def generate_chat_tags(form_data: dict, user=Depends(get_verified_user)):
|
|
async def generate_chat_tags(form_data: dict, user=Depends(get_verified_user)):
|
|
print("generate_chat_tags")
|
|
print("generate_chat_tags")
|
|
|
|
+ if not app.state.config.ENABLE_TAGS_GENERATION:
|
|
|
|
+ return JSONResponse(
|
|
|
|
+ status_code=status.HTTP_200_OK,
|
|
|
|
+ content={"detail": "Tags generation is disabled"},
|
|
|
|
+ )
|
|
|
|
+
|
|
model_id = form_data["model"]
|
|
model_id = form_data["model"]
|
|
if model_id not in app.state.MODELS:
|
|
if model_id not in app.state.MODELS:
|
|
raise HTTPException(
|
|
raise HTTPException(
|