convo2db

models/ta_models/config.py

@@ -28,9 +28,9 @@ BP_LAB2STR = {
 }
 
 QUESTION2PHASE = {
-    "question_1": ["0_ActiveEngagement","1_Explore"],
+    "question_1": ["0_ActiveEngagement"],
     "question_4": ["1_Explore"],
-    "question_5": ["0_ActiveEngagement", "1_Explore"],
+    "question_5": ["1_Explore"],
     # "question_7": ["1_Explore"],
     # "question_9": ["4_SP&NS"],
     "question_10": ["4_SP&NS"],

models/ta_models/cpc_utils.py

@@ -2,6 +2,7 @@ import streamlit as st
 from streamlit.logger import get_logger
 import requests
 import os
+from langchain_core.messages import HumanMessage
 from .config import model_name_or_path
 from transformers import AutoTokenizer
 from utils.mongo_utils import new_cpc_comparison
@@ -16,6 +17,13 @@ HEADERS = {
     "Content-Type": "application/json",
 }
 
+def modify_last_human_message(memory, new_phase):
+    # Travel list backwards
+    for msg in memory.chat_memory.messages[::-1]:
+        if type(msg) == HumanMessage:
+            msg.response_metadata = {"phase":new_phase}
+            break
+
 def cpc_predict_message(context, input):
     # context = memory.load_memory_variables({})[memory.memory_key]
     encoding = tokenizer(

pages/convosim.py

@@ -2,13 +2,13 @@ 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, are_models_alive
 from utils.memory_utils import clear_memory, push_convo2db
 from utils.chain_utils import get_chain, custom_chain_predict
 from app_config import ISSUES, SOURCES, source2label, issue2label, MAX_MSG_COUNT, WARN_MSG_COUT
 from models.ta_models.config import CPC_LBL_OPTS, cpc_label2str, BP_LAB2STR, BP_LBL_OPTS
-from models.ta_models.cpc_utils import cpc_push2db
+from models.ta_models.cpc_utils import cpc_push2db, modify_last_human_message
 from models.ta_models.bp_utils import bp_predict_message, bp_push2db
 
 logger = get_logger(__name__)
@@ -99,6 +99,9 @@ def sent_request_llm(llm_chain, prompt):
     responses = custom_chain_predict(llm_chain, prompt, stopper)
     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.dialog("Bad Practice Detected")
 # def confirm_bp(bp_prediction, prompt):
@@ -158,6 +161,7 @@ with st.sidebar:
     # st.markdown()
     def on_change_cpc():
         cpc_push2db(False)
+        modify_last_human_message(memoryA, st.session_state['sel_phase'])
         st.session_state.changed_cpc = True
     def on_change_bp():
         bp_push2db()

utils/app_utils.py

@@ -96,8 +96,8 @@ def is_model_alive(name, timeout=2, model_type="classificator"):
         endpoint_url="https://api.openai.com/v1/models"
         headers = {"Authorization": f"Bearer {os.environ['OPENAI_API_KEY']}",}
         try:
-            _ = requests.get(url=endpoint_url, headers=headers, timeout=1)
-            return "200"
+            response = requests.get(url=endpoint_url, headers=headers, timeout=1)
+            return str(response.status_code)
         except:
             return "404" 
 
@@ -107,4 +107,5 @@ def are_models_alive():
     for config in ENDPOINT_NAMES.values():
         models_alive.append(is_model_alive(**config))
     openai = is_model_alive("openai", model_type="openai")
-    return all([x=="200" for x in models_alive + [openai]])
+    models_alive.append(openai)
+    return all([x=="200" for x in models_alive])

utils/mongo_utils.py

@@ -41,6 +41,17 @@ def new_convo(client, issue, language, username, is_comparison, model_one, model
     logger.debug(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