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

feat: valves on_startup & on_shutdown

Browse Source
This commit is contained in:
Timothy J. Baek
2024-05-28 15:39:28 -07:00
parent a8348a3197
commit 88f3d59fcb
20 changed files with 56 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@@ -255,6 +255,7 @@ async def update_valves(pipeline_id: str, form_data: dict):
pipeline_module = PIPELINE_MODULES[pipeline_id]
await pipeline_module.on_shutdown()
try:
ValvesModel = pipeline_module.valves.__class__
valves = ValvesModel(**form_data)
@@ -265,6 +266,7 @@ async def update_valves(pipeline_id: str, form_data: dict):
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
detail=f"{str(e)}",
)
await pipeline_module.on_startup()
return pipeline_module.valves