Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

[nodes] Add cancelation to the API

Browse Source
This commit is contained in:
Kyle Schouviller
2023-03-16 20:05:36 -07:00
committed by psychedelicious
parent 076fac07eb
commit bd7e515290
4 changed files with 64 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
invokeai/app/api/routers/sessions.py
View File

@ -270,3 +270,18 @@ async def invoke_session(
ApiDependencies.invoker.invoke(session, invoke_all=all)
return Response(status_code=202)
@session_router.delete(
"/{session_id}/invoke",
operation_id="cancel_session_invoke",
responses={
202: {"description": "The invocation is canceled"}
},
)
async def cancel_session_invoke(
session_id: str = Path(description="The id of the session to cancel"),
) -> None:
"""Invokes a session"""
ApiDependencies.invoker.cancel(session_id)
return Response(status_code=202)