UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-24 21:13:59 +00:00
Code Issues Actions 5 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
for model in app.state.MODELS.values()
if "pipeline" in model
and "type" in model["pipeline"]
and model["pipeline"]["type"] == "filter"
and (
model["pipeline"]["pipelines"] == ["*"]
@ -279,11 +280,10 @@ class PipelineMiddleware(BaseHTTPMiddleware):
if key != "":
headers = {"Authorization": f"Bearer {key}"}
r = requests.post(
f"{url}/filter",
f"{url}/{filter['id']}/filter",
headers=headers,
json={
"user": user,
"model": filter["id"],
"body": data,
},
)
@ -448,7 +448,7 @@ async def get_models(user=Depends(get_verified_user)):
models = [
model
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: