utils to save transcript in mongodb

convosim.py

@@ -2,7 +2,7 @@ import os
 import streamlit as st
 from streamlit.logger import get_logger
 from langchain.schema.messages import HumanMessage
-from utils.mongo_utils import get_db_client
+from utils.mongo_utils import get_db_client, update_convo
 from utils.app_utils import create_memory_add_initial_message, get_random_name,  DEFAULT_NAMES_DF
 from utils.memory_utils import clear_memory, push_convo2db
 from utils.chain_utils import get_chain, custom_chain_predict
@@ -87,6 +87,8 @@ if prompt := st.chat_input(disabled=st.session_state['total_messages'] > MAX_MSG
     # response = update_memory_completion(prompt, st.session_state["memory"], OA_engine, temperature)
     for response in responses:
         st.chat_message("assistant").write(response)
+    transcript = memoryA.load_memory_variables({})[memoryA.memory_key]
+    update_convo(st.session_state["db_client"], st.session_state["convo_id"], transcript)
 
 st.session_state['total_messages'] = len(memoryA.chat_memory.messages)
 if st.session_state['total_messages'] >= MAX_MSG_COUNT:

utils/mongo_utils.py

@@ -41,6 +41,16 @@ def new_convo(client, issue, language, username, is_comparison, model_one, model
     logger.info(f"DBUTILS: new convo id is {convo_id}")
     st.session_state['convo_id'] = convo_id
 
+def update_convo(client, convo_id, transcript=""):
+    from bson.objectid import ObjectId
+    db = client[DB_SCHEMA]
+    convos = db[DB_CONVOS]
+    myquery = { "_id": ObjectId(convo_id) }
+    newvalues = { "$set": { "transcript": transcript } }
+    result = convos.update_one(myquery, newvalues)
+    if result.matched_count == 1:
+        logger.debug(f"Updated conversation {convo_id}")
+
 def new_comparison(client, prompt_timestamp, completion_timestamp, 
                    chat_history, prompt, completionA, completionB,
                    source="webapp", subset=None