UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 21:42:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: filter pipeline

Browse Source
This commit is contained in:
Timothy J. Baek 2024-06-12 13:31:05 -07:00
parent c794d59fd5
commit c6c0bc19d8
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
backend/main.py
View File

@ -472,13 +472,10 @@ def filter_pipeline(payload, user):
if r is not None: if r is not None:
try: try:
res = r.json() res = r.json()
if "detail" in res:
return JSONResponse(
status_code=r.status_code,
content=res,
)
except: except:
pass pass
if "detail" in res:
raise Exception(r.status_code, res["detail"])
else: else:
pass pass
@ -489,6 +486,7 @@ def filter_pipeline(payload, user):
if "title" in payload: if "title" in payload:
del payload["title"] del payload["title"]
return payload return payload
@ -510,7 +508,14 @@ class PipelineMiddleware(BaseHTTPMiddleware):
user = get_current_user( user = get_current_user(
get_http_authorization_cred(request.headers.get("Authorization")) get_http_authorization_cred(request.headers.get("Authorization"))
) )
data = filter_pipeline(data, user)
try:
data = filter_pipeline(data, user)
except Exception as e:
return JSONResponse(
status_code=e.args[0],
content={"detail": e.args[1]},
)
modified_body_bytes = json.dumps(data).encode("utf-8") modified_body_bytes = json.dumps(data).encode("utf-8")
# Replace the request body with the modified one # Replace the request body with the modified one