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

tidy(processor): remove test callbacks

Browse Source
This commit is contained in:
psychedelicious 2024-05-24 11:23:26 +10:00
parent c88de180e7
commit 169b75b2b7
1 changed files with 1 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -104,40 +104,7 @@ class ApiDependencies:
names = SimpleNameService() names = SimpleNameService()
performance_statistics = InvocationStatsService() performance_statistics = InvocationStatsService()
def on_before_run_session(queue_item): session_processor = DefaultSessionProcessor(session_runner=DefaultSessionRunner())
print("BEFORE RUN SESSION", queue_item.item_id)
return True
def on_before_run_node(invocation, queue_item):
print("BEFORE RUN NODE", invocation.id)
return True
def on_after_run_node(invocation, queue_item, output):
print("AFTER RUN NODE", invocation.id)
return True
def on_node_error(invocation, queue_item, error_type, error_message, error_traceback):
print("NODE ERROR", invocation.id)
return True
def on_after_run_session(queue_item):
print("AFTER RUN SESSION", queue_item.item_id)
return True
def on_non_fatal_processor_error(queue_item, error_type, error_message, error_traceback):
print("NON FATAL PROCESSOR ERROR", error_message)
return True
session_processor = DefaultSessionProcessor(
DefaultSessionRunner(
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_callbacks=[on_non_fatal_processor_error],
)
session_queue = SqliteSessionQueue(db=db) session_queue = SqliteSessionQueue(db=db)
urls = LocalUrlService() urls = LocalUrlService()
workflow_records = SqliteWorkflowRecordsStorage(db=db) workflow_records = SqliteWorkflowRecordsStorage(db=db)