InvokeAI/invokeai/app/invocations/prompt.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

15 lines
352 B
Python
Raw Normal View History

from typing import Literal
2023-03-03 06:02:00 +00:00
from pydantic.fields import Field
2023-03-03 06:02:00 +00:00
from .baseinvocation import BaseInvocationOutput
2023-03-03 06:02:00 +00:00
class PromptOutput(BaseInvocationOutput):
"""Base class for invocations that output a prompt"""
#fmt: off
2023-03-03 06:02:00 +00:00
type: Literal["prompt"] = "prompt"
prompt: str = Field(default=None, description="The output prompt")
#fmt: on