new_v2_llama3

app_config.py

@@ -4,21 +4,18 @@ from models.model_seeds import seeds, seed2str
 ISSUES = [k for k,_ in seeds.items()]
 SOURCES = [
     "CTL_llama2",
-    # "CTL_llama3",
     # "CTL_mistral",
     'OA_rolemodel', 
         #    'OA_finetuned',
 ]
 SOURCES_LAB = {"OA_rolemodel":'OpenAI GPT3.5',
                "OA_finetuned":'Finetuned OpenAI',
-               "CTL_llama2": "Llama 3",
-               #"CTL_llama3": "Llama 3",
+               "CTL_llama2": "Llama",
                "CTL_mistral": "Mistral",
                }
 
 ENDPOINT_NAMES = {
-    "CTL_llama2": "texter_simulator",
-    # "CTL_llama3": "texter_simulator",
+    "CTL_llama2": "conversation_simulator",
     # 'CTL_llama2': "llama2_convo_sim",
     "CTL_mistral": "convo_sim_mistral"
 }

convosim.py

@@ -28,7 +28,7 @@ if 'texter_name' not in st.session_state:
 memories = {'memory':{"issue": st.session_state['issue'], "source": st.session_state['previous_source']}}
 
 with st.sidebar:
-    username = st.text_input("Username", value='Jasmyn', max_chars=30)
+    username = st.text_input("Username", value='Barbara', max_chars=30)
     if 'counselor_name' not in st.session_state:
         st.session_state["counselor_name"] = username #get_random_name(names_df=DEFAULT_NAMES_DF)
     # temperature = st.slider("Temperature", 0., 1., value=0.8, step=0.1)

models/model_seeds.py

@@ -6,12 +6,12 @@ texter: I need to talk to someone"""
     },
     "full_convo__seeded1": {
         # "prompt": "Your character is suffering from bullying at school. Your character has suicidal thoughs but has not harmed himself. You are talkative, and open to talk about your problems.",
-        "memory": """texter: Help
-texter: hello? is anyone there to help?"""
+        "memory": """texter: TikTok
+texter: Is someone real here?"""
     },
     "full_convo__seeded2": {
         # "prompt": "Your character is suffering anxiety from finals week. Your character does not have suicidal thoughs but has harmed herself. You are talkative, and open to talk about your problems.",
-        "memory": """texter: Help
+        "memory": """texter: HOME
 texter: I don't know what to do"""
     },
     "safety_assessment__seeded1": {
@@ -46,51 +46,6 @@ helper: You are so brave for dealing with this pain for so long.
 helper: With all this going on I can tell you are struggling and in so much pain. Have you had any thoughts of suicide in the last few days, including today?
 texter: Yes, all the time. I don't see any point in living a life without her"""
     },
-    "full_convo__seeded3": {
-        # "prompt": "",
-        "memory": """texter: Help
-texter: I am feeling so lost"""
-    },
-    "full_convo__seeded4": {
-        # "prompt": "",
-        "memory": """texter: Help
-texter: I need help"""
-    },
-    "full_convo__seeded5": {
-        # "prompt": "",
-        "memory": """texter: Help
-texter: hello? is anyone there to help?"""
-    },
-    "full_convo__seeded6": {
-        # "prompt": "",
-        "memory": """texter: Help
-texter: I think I overdosed"""
-    },
-    "full_convo__hard": {
-        # "prompt": "",
-        "memory": """texter: Help
-texter: Is there someone to talk?"""
-    },
-    "full_convo__hard__seeded1": {
-        # "prompt": "",
-        "memory": """texter: Help
-texter: hello? is anyone there to help?"""
-    },
-    "full_convo__hard__seeded2": {
-        # "prompt": "",
-        "memory": """texter: Help
-texter: I can't stand it anymore"""
-    },
-    "full_convo__hard__seeded3": {
-        # "prompt": "",
-        "memory": """texter: Help
-texter: idk what to do"""
-    },
-    "full_convo__hard__seeded4": {
-        # "prompt": "",
-        "memory": """texter: Help
-texter: I need help"""
-    },
 #     "NewSA__YesNo.2": {
 #         "prompt": "Your character is suffering from bullying at school. Your character has suicidal thoughs but has not harmed himself. You are talkative, and open to talk about your problems.",
 #         "memory": """texter: I just can’t take it anymore. Every day is just endless torment.
@@ -204,15 +159,6 @@ seed2str = {
     "full_convo__seeded2": "Parent Issues",
     "safety_assessment__seeded1": "Safety Assessment - Bullying ",
     "safety_assessment__seeded2": "Safety Assessment - Grief ",
-    "full_convo__seeded3": "LGBTQ+",
-    "full_convo__seeded4": "Relationship Issues",
-    "full_convo__seeded5": "Child Abuse",
-    "full_convo__seeded6": "Overdose",
-    "full_convo__hard": "Difficult - General",
-    "full_convo__hard__seeded1": "Difficult - Bullying",
-    "full_convo__hard__seeded2": "Difficult - Parent Issues",
-    "full_convo__hard__seeded3": "Difficult - LGBTQ+",
-    "full_convo__hard__seeded4": "Difficult - Relationship Issues"
 
     # "NewSA__NoNo.2": "No thoughts No Harm No context",
     # "NewSA__YesNo.2": "Yes thoughts No Harm No context",

utils/chain_utils.py

@@ -12,7 +12,7 @@ def get_chain(issue, language, source, memory, temperature, texter_name=""):
         seed = seeds.get(issue, "GCT")['prompt']
         template = get_template_role_models(issue, language, texter_name=texter_name, seed=seed)
         return get_role_chain(template, memory, temperature)
-    elif source in ('CTL_llama2', 'CTL_llama3'):
+    elif source in ('CTL_llama2'):
         if language == "English":
             language = "en"
         elif language == "Spanish":