from typing import Literal

from pydantic.fields import Field

from .baseinvocation import BaseInvocationOutput


class PromptOutput(BaseInvocationOutput):
    """Base class for invocations that output a prompt"""
    #fmt: off
    type: Literal["prompt"] = "prompt"

    prompt: str = Field(default=None, description="The output prompt")
    #fmt: on