mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
fix(nodes): fix schema gen for GraphExecutionState
This commit is contained in:
parent
b194180f76
commit
5347c12fed
@ -773,6 +773,24 @@ class GraphExecutionState(BaseModel):
|
|||||||
default_factory=dict,
|
default_factory=dict,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Declare all fields as required; necessary for OpenAPI schema generation build.
|
||||||
|
# Technically only fields without a `default_factory` need to be listed here.
|
||||||
|
# See: https://github.com/pydantic/pydantic/discussions/4577
|
||||||
|
class Config:
|
||||||
|
schema_extra = {
|
||||||
|
'required': [
|
||||||
|
'id',
|
||||||
|
'graph',
|
||||||
|
'execution_graph',
|
||||||
|
'executed',
|
||||||
|
'executed_history',
|
||||||
|
'results',
|
||||||
|
'errors',
|
||||||
|
'prepared_source_mapping',
|
||||||
|
'source_prepared_mapping',
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
||||||
def next(self) -> BaseInvocation | None:
|
def next(self) -> BaseInvocation | None:
|
||||||
"""Gets the next node ready to execute."""
|
"""Gets the next node ready to execute."""
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user