"...composable_kernel.git" did not exist on "cee9295124049be2b81cb4df3e8ecbd12662cd3a"
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): ...@@ -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): ...@@ -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): ...@@ -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
......
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