tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Releases 2 Packages Activity Actions

fix: only edit body with whitelisted paths

Browse source
This commit is contained in:
Timothy J. Baek 2024-03-08 22:51:42 -08:00
parent 6c58bb59be
commit dfcc314283
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
backend/main.py
View file

@ -62,8 +62,12 @@ async def on_startup():
class RAGMiddleware(BaseHTTPMiddleware): class RAGMiddleware(BaseHTTPMiddleware):
async def dispatch(self, request: Request, call_next): async def dispatch(self, request: Request, call_next):
"chat/completions" in request.url.path
print(request.url.path) print(request.url.path)
if request.method == "POST": if request.method == "POST" and (
"/api/chat" in request.url.path or "/chat/completions" in request.url.path
):
# Read the original request body # Read the original request body
body = await request.body() body = await request.body()
# Decode body to string # Decode body to string