Adjustments for llama3

app_config.py

@@ -4,18 +4,21 @@ 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",
+               "CTL_llama2": "Llama 3",
+               #"CTL_llama3": "Llama 3",
                "CTL_mistral": "Mistral",
                }
 
 ENDPOINT_NAMES = {
-    "CTL_llama2": "conversation_simulator",
+    "CTL_llama2": "texter_simulator",
+    # "CTL_llama3": "texter_simulator",
     # 'CTL_llama2': "llama2_convo_sim",
     "CTL_mistral": "convo_sim_mistral"
 }
@@ -26,7 +29,7 @@ def source2label(source):
 def issue2label(issue):
     return seed2str.get(issue, "GCT")
 
-ENVIRON = "prod"
+ENVIRON = "dev"
 
 DB_SCHEMA = 'prod_db' if ENVIRON == 'prod' else 'test_db' 
 DB_CONVOS = 'conversations'

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'):
+    elif source in ('CTL_llama2', 'CTL_llama3'):
         if language == "English":
             language = "en"
         elif language == "Spanish":