"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "0be2803fb9f998b04d33e8d75f0cbdde6141715a"
Commit c6c0bc19 authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

fix: filter pipeline

parent c794d59f
......@@ -472,13 +472,10 @@ def filter_pipeline(payload, user):
if r is not None:
try:
res = r.json()
if "detail" in res:
return JSONResponse(
status_code=r.status_code,
content=res,
)
except:
pass
if "detail" in res:
raise Exception(r.status_code, res["detail"])
else:
pass
......@@ -489,6 +486,7 @@ def filter_pipeline(payload, user):
if "title" in payload:
del payload["title"]
return payload
......@@ -510,7 +508,14 @@ class PipelineMiddleware(BaseHTTPMiddleware):
user = get_current_user(
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")
# Replace the request body with the modified one
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment