Brain/src/model/feedback_model.py

import datetime


# feedback model to manage user's feedback (thumb up and down)
from Brain.src.model.basic_model import BasicModel


class FeedbackModel:
    def __init__(
        self, uuid: str, prompt: BasicModel, completion: BasicModel, rating: int
    ):
        self.uuid = uuid
        self.prompt = prompt
        self.completion = completion
        self.rating = rating
        self.timestamp = datetime.datetime.now().timestamp()

    def to_json(self):
        return {
            "uuid": self.uuid,
            "prompt": self.prompt.to_json(),
            "completion": self.completion.to_json(),
            "rating": self.rating,
            "timestamp": self.timestamp,
        }