UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 21:42:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac: pipeline valves -> filters

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-27 19:34:05 -07:00
parent 4fac99c5b3
commit 966f10e715
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/main.py
View File

@ -253,7 +253,7 @@ class PipelineMiddleware(BaseHTTPMiddleware):
model
for model in app.state.MODELS.values()
if "pipeline" in model
and model["pipeline"]["type"] == "valve"
and model["pipeline"]["type"] == "filter"
and model_id
in [
target_model["id"]
@ -285,7 +285,7 @@ class PipelineMiddleware(BaseHTTPMiddleware):
if key != "":
headers = {"Authorization": f"Bearer {key}"}
r = requests.post(
f"{url}/valve",
f"{url}/filter",
headers=headers,
json={
"user": user,
@ -428,11 +428,11 @@ async def get_all_models():
async def get_models(user=Depends(get_verified_user)):
models = await get_all_models()
# Filter out valve models
# Filter out filter pipelines
models = [
model
for model in models
if "pipeline" not in model or model["pipeline"]["type"] != "valve"
if "pipeline" not in model or model["pipeline"]["type"] != "filter"
]
if app.state.config.ENABLE_MODEL_FILTER: