app/models/prediction_models.py

from pydantic import BaseModel
from typing import List


class PredictionRequest(BaseModel):
    invoiceId: List[int]
    country: str


class PredictionResponse(BaseModel):
    invoiceId: int
    prediction: float