fix(app): add dynamic validator to AnyInvocation & AnyInvocationOutput

This fixes the tests and slightly changes output types.
This commit is contained in:
psychedelicious
2024-05-29 21:05:42 +10:00
parent 50d3030471
commit ac56ab79a7
2 changed files with 18 additions and 7 deletions

View File

@ -1,5 +1,6 @@
import pytest
from pydantic import TypeAdapter
from pydantic.json_schema import models_json_schema
from invokeai.app.invocations.baseinvocation import (
BaseInvocation,
@ -713,4 +714,4 @@ def test_iterate_accepts_collection():
def test_graph_can_generate_schema():
# Not throwing on this line is sufficient
# NOTE: if this test fails, it's PROBABLY because a new invocation type is breaking schema generation
_ = Graph.model_json_schema()
models_json_schema([(Graph, "serialization")])