OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-06-26 18:15:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-28 15:47:21 -07:00
parent cabd666152
commit 0541e71368
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
main.py
View File

@ -159,6 +159,7 @@ app.add_middleware(
@app.middleware("http") @app.middleware("http")
async def check_url(request: Request, call_next): async def check_url(request: Request, call_next):
start_time = int(time.time()) start_time = int(time.time())
app.state.PIPELINES = get_all_pipelines()
response = await call_next(request) response = await call_next(request)
process_time = int(time.time()) - start_time process_time = int(time.time()) - start_time
response.headers["X-Process-Time"] = str(process_time) response.headers["X-Process-Time"] = str(process_time)
@ -202,8 +203,6 @@ async def get_models():
@app.get("/{pipeline_id}/valves") @app.get("/{pipeline_id}/valves")
async def get_valves(pipeline_id: str): async def get_valves(pipeline_id: str):
app.state.PIPELINES = get_all_pipelines()
if pipeline_id not in app.state.PIPELINES: if pipeline_id not in app.state.PIPELINES:
raise HTTPException( raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND, status_code=status.HTTP_404_NOT_FOUND,