ml_api/schemas.py

from ninja import Schema


class ChatInput(Schema):
    text: str


class ChatOutput(Schema):
    result: str