Spaces:
Sleeping
Sleeping
File size: 1,164 Bytes
a3386d3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
from src.config.databaseConfig import DATABASE
from src.config.appConfig import ENV_VAR
from bson import ObjectId
async def save_question_and_answer_to_chat_history(username, history_obj):
try:
existing_chat_history = DATABASE.client[ENV_VAR.MONGO_DB_NAME_CHATS][
"chathistories"
].find_one({"username": username})
history_obj["_id"] = ObjectId()
if not existing_chat_history:
new_chat_history = {
"username": username,
"history": [history_obj],
"historyCount": 1,
}
DATABASE.client[ENV_VAR.MONGO_DB_NAME_CHATS]["chathistories"].insert_one(
new_chat_history
)
else:
DATABASE.client[ENV_VAR.MONGO_DB_NAME_CHATS]["chathistories"].update_one(
{"username": username},
{
"$push": {"history": {"$each": [history_obj], "$position": 0}},
"$inc": {"historyCount": 1},
},
)
except Exception as error:
raise ValueError(f"Error saving question and answer to chat history: {error}")
|