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

feat(app): support multiple processor lifecycle callbacks

Browse Source
This commit is contained in:
psychedelicious
2024-05-22 19:05:49 +10:00
parent 7f70cde038
commit 1d973f92ff
2 changed files with 34 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/app/api/dependencies.py
View File

@ -130,13 +130,13 @@ class ApiDependencies:
session_processor = DefaultSessionProcessor(
DefaultSessionRunner(
on_before_run_session=on_before_run_session,
on_before_run_node=on_before_run_node,
on_after_run_node=on_after_run_node,
on_node_error=on_node_error,
on_after_run_session=on_after_run_session,
on_before_run_session_callbacks=[on_before_run_session],
on_before_run_node_callbacks=[on_before_run_node],
on_after_run_node_callbacks=[on_after_run_node],
on_node_error_callbacks=[on_node_error],
on_after_run_session_callbacks=[on_after_run_session],
),
on_non_fatal_processor_error,
on_non_fatal_processor_error_callbacks=[on_non_fatal_processor_error],
)
session_queue = SqliteSessionQueue(db=db)
urls = LocalUrlService()