models/prompt.py

from pydantic import BaseModel


class Prompt(BaseModel):
    prompt:str

class PromptResponse(Prompt):
    response:str