class FeedbackRepository:
    def __init__(self, db):
        self._db = db

    def save_feedback(
        self,
        user_id: str,
        message: str,
        response: str,
        rating: int,
        langsmith_run_id: str | None = None,
    ) -> None:
        self._db.execute_write(
            """
            INSERT INTO td_feedback (user_id, message, response, rating, langsmith_run_id)
            VALUES (%s, %s, %s, %s, %s)
            """,
            (user_id, message, response, rating, langsmith_run_id),
        )