UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-24 13:07:25 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

use filters envvars

Browse Source
This commit is contained in:
Michael Poluektov 2024-08-19 10:46:52 +01:00
parent ce7a1a73ac
commit fd422d2e3c
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
backend/main.py
View File

@ -118,6 +118,8 @@ from config import (
WEBUI_SESSION_COOKIE_SAME_SITE,
WEBUI_SESSION_COOKIE_SECURE,
ENABLE_ADMIN_CHAT_ACCESS,
ENABLE_TOOLS_FILTER,
ENABLE_FILES_FILTER,
AppConfig,
)
@ -443,6 +445,10 @@ async def get_content_from_response(response) -> Optional[str]:
async def chat_completion_tools_handler(
body: dict, user: UserModel, extra_params: dict
) -> tuple[dict, dict]:
log.debug(f"{ENABLE_TOOLS_FILTER=}")
if not ENABLE_TOOLS_FILTER:
return body, {}
skip_files = False
contexts = []
citations = []
@ -533,6 +539,10 @@ async def chat_completion_tools_handler(
async def chat_completion_files_handler(body) -> tuple[dict, dict[str, list]]:
log.debug(f"{ENABLE_FILES_FILTER=}")
if not ENABLE_FILES_FILTER:
return body, {}
contexts = []
citations = []