UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-17 18:46:57 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refac: pipelines

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-28 11:43:48 -07:00
parent e231333bcd
commit 0383efa207
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/main.py
View File

@ -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):
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)):
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: