Create repositories/chat_repo.py

repositories/chat_repo.py

@@ -0,0 +1,8 @@
+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()