fix(nodes): fix schema generation for output classes

All output classes need to have their properties flagged as `required` for the schema generation to work as needed.
This commit is contained in:
psychedelicious
2023-03-26 13:26:59 +11:00
parent c34ac91ff0
commit 4221cf7731
3 changed files with 46 additions and 1 deletions

View File

@ -12,3 +12,11 @@ class PromptOutput(BaseInvocationOutput):
prompt: str = Field(default=None, description="The output prompt")
#fmt: on
class Config:
schema_extra = {
'required': [
'type',
'prompt',
]
}