File size: 482 Bytes
316a678
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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()