repositories/chat_repo.py

from models.db import get_session
from models.chat import ChatMessage

class ChatRepo:
    def save(self, user, prompt, response):
        msg = ChatMessage(user=user, prompt=prompt, response=response)
        with get_session() as s:
            s.add(msg); s.commit()