mirror of
https://github.com/invoke-ai/InvokeAI
synced 2026-04-16 11:55:08 +02:00
4b334be7d0
When a queue item is popped for processing, we need to retrieve its session from the DB. Pydantic serializes the graph at this stage. It's possible for a graph to have been made invalid during the graph preparation stage (e.g. an ancestor node executes, and its output is not valid for its successor node's input field). When this occurs, the session in the DB will fail validation, but we don't have a chance to find out until it is retrieved and parsed by pydantic. This logic was previously not wrapped in any exception handling. Just after retrieving a session, we retrieve the specific invocation to execute from the session. It's possible that this could also have some sort of error, though it should be impossible for it to be a pydantic validation error (that would have been caught during session validation). There was also no exception handling here. When either of these processes fail, the processor gets soft-locked because the processor's cleanup logic is never run. (I didn't dig deeper into exactly what cleanup is not happening, because the fix is to just handle the exceptions.) This PR adds exception handling to both the session retrieval and node retrieval and events for each: `session_retrieval_error` and `invocation_retrieval_error`. These events are caught and displayed in the UI as toasts, along with the type of the python exception (e.g. `Validation Error`). The events are also logged to the browser console. |
||
|---|---|---|
| .. | ||
| models | ||
| __init__.py | ||
| board_image_record_storage.py | ||
| board_images.py | ||
| board_record_storage.py | ||
| boards.py | ||
| config.py | ||
| default_graphs.py | ||
| events.py | ||
| graph.py | ||
| image_file_storage.py | ||
| image_record_storage.py | ||
| images.py | ||
| invocation_queue.py | ||
| invocation_services.py | ||
| invoker.py | ||
| item_storage.py | ||
| latent_storage.py | ||
| model_manager_service.py | ||
| processor.py | ||
| resource_name.py | ||
| sqlite.py | ||
| urls.py | ||