mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
2f9ebdec69
Some tech debt related to dynamic pydantic schemas for invocations became problematic. Including the invocations and results in the event schemas was breaking pydantic's handling of ref schemas. I don't really understand why - I think it's a pydantic bug in a remote edge case that we are hitting. After many failed attempts I landed on this implementation, which is actually much tidier than what was in there before. - Create pydantic-enabled types for `AnyInvocation` and `AnyInvocationOutput` and use these in place of the janky dynamic unions. Actually, they are kinda the same, but better encapsulated. Use these in `Graph`, `GraphExecutionState`, `InvocationEventBase` and `InvocationCompleteEvent`. - Revise the custom openapi function to work with the new models. - Split out the custom openapi function to a separate file. Add a `post_transform` callback so consumers can customize the output schema. - Update makefile scripts. |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| controlnet_utils.py | ||
| custom_openapi.py | ||
| download_with_progress.py | ||
| metaenum.py | ||
| misc.py | ||
| model_exclude_null.py | ||
| profiler.py | ||
| step_callback.py | ||
| suppress_output.py | ||
| thumbnails.py | ||
| ti_utils.py | ||