mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
feat(api): add ModelIdentifierField to openapi schema
- Also add `ProgressImage`
This commit is contained in:
parent
92b0d13d0e
commit
daeb766468
@ -3,6 +3,8 @@
|
|||||||
# values from the command line or config file.
|
# values from the command line or config file.
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
|
from invokeai.app.invocations.model import ModelIdentifierField
|
||||||
|
from invokeai.app.services.session_processor.session_processor_common import ProgressImage
|
||||||
from invokeai.version.invokeai_version import __version__
|
from invokeai.version.invokeai_version import __version__
|
||||||
|
|
||||||
from .services.config import InvokeAIAppConfig
|
from .services.config import InvokeAIAppConfig
|
||||||
@ -156,17 +158,19 @@ def custom_openapi() -> dict[str, Any]:
|
|||||||
openapi_schema["components"]["schemas"][schema_key] = output_schema
|
openapi_schema["components"]["schemas"][schema_key] = output_schema
|
||||||
openapi_schema["components"]["schemas"][schema_key]["class"] = "output"
|
openapi_schema["components"]["schemas"][schema_key]["class"] = "output"
|
||||||
|
|
||||||
# Add Node Editor UI helper schemas
|
# Some models don't end up in the schemas as standalone definitions
|
||||||
ui_config_schemas = models_json_schema(
|
additional_schemas = models_json_schema(
|
||||||
[
|
[
|
||||||
(UIConfigBase, "serialization"),
|
(UIConfigBase, "serialization"),
|
||||||
(InputFieldJSONSchemaExtra, "serialization"),
|
(InputFieldJSONSchemaExtra, "serialization"),
|
||||||
(OutputFieldJSONSchemaExtra, "serialization"),
|
(OutputFieldJSONSchemaExtra, "serialization"),
|
||||||
|
(ModelIdentifierField, "serialization"),
|
||||||
|
(ProgressImage, "serialization"),
|
||||||
],
|
],
|
||||||
ref_template="#/components/schemas/{model}",
|
ref_template="#/components/schemas/{model}",
|
||||||
)
|
)
|
||||||
for schema_key, ui_config_schema in ui_config_schemas[1]["$defs"].items():
|
for schema_key, schema_json in additional_schemas[1]["$defs"].items():
|
||||||
openapi_schema["components"]["schemas"][schema_key] = ui_config_schema
|
openapi_schema["components"]["schemas"][schema_key] = schema_json
|
||||||
|
|
||||||
# Add a reference to the output type to additionalProperties of the invoker schema
|
# Add a reference to the output type to additionalProperties of the invoker schema
|
||||||
for invoker in all_invocations:
|
for invoker in all_invocations:
|
||||||
|
Loading…
Reference in New Issue
Block a user