chat_history.py

import sqlalchemy
from sqlalchemy import text
from sqlalchemy.orm import Session


def insert_chat_history(conn: sqlalchemy.engine.Connection, query: str, answer: str):
    with Session(conn) as conn:
        conn.execute(
            text("INSERT INTO chat_history (query, answer) VALUES (:query, :answer);"),
            [
                {
                    "query": query,
                    "answer": answer,
                }
            ],
        )
        conn.commit()