Commit 0383efa2 authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

refac: pipelines

parent e231333b
...@@ -245,6 +245,7 @@ class PipelineMiddleware(BaseHTTPMiddleware): ...@@ -245,6 +245,7 @@ class PipelineMiddleware(BaseHTTPMiddleware):
model model
for model in app.state.MODELS.values() for model in app.state.MODELS.values()
if "pipeline" in model if "pipeline" in model
and "type" in model["pipeline"]
and model["pipeline"]["type"] == "filter" and model["pipeline"]["type"] == "filter"
and ( and (
model["pipeline"]["pipelines"] == ["*"] model["pipeline"]["pipelines"] == ["*"]
...@@ -279,11 +280,10 @@ class PipelineMiddleware(BaseHTTPMiddleware): ...@@ -279,11 +280,10 @@ class PipelineMiddleware(BaseHTTPMiddleware):
if key != "": if key != "":
headers = {"Authorization": f"Bearer {key}"} headers = {"Authorization": f"Bearer {key}"}
r = requests.post( r = requests.post(
f"{url}/filter", f"{url}/{filter['id']}/filter",
headers=headers, headers=headers,
json={ json={
"user": user, "user": user,
"model": filter["id"],
"body": data, "body": data,
}, },
) )
...@@ -448,7 +448,7 @@ async def get_models(user=Depends(get_verified_user)): ...@@ -448,7 +448,7 @@ async def get_models(user=Depends(get_verified_user)):
models = [ models = [
model model
for model in models for model in models
if "pipeline" not in model or model["pipeline"]["type"] != "filter" if "pipeline" not in model or model["pipeline"].get("type", None) != "filter"
] ]
if app.state.config.ENABLE_MODEL_FILTER: if app.state.config.ENABLE_MODEL_FILTER:
......
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