2023-03-03 01:02:00 -05:00

14 lines
326 B
Python

from typing import Literal
from pydantic.fields import Field
from .baseinvocation import BaseInvocationOutput
class PromptOutput(BaseInvocationOutput):
"""Base class for invocations that output a prompt"""
type: Literal["prompt"] = "prompt"
prompt: str = Field(default=None, description="The output prompt")