from models.model_seeds import seeds, seed2str # ISSUES = ['Anxiety','Suicide'] 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 2", "CTL_llama3": "Llama 3", "CTL_mistral": "Mistral", } ENDPOINT_NAMES = { # "CTL_llama2": "texter_simulator", "CTL_llama3": "texter_simulator_llm", # 'CTL_llama2': "llama2_convo_sim", "CTL_mistral": "convo_sim_mistral" } def source2label(source): return SOURCES_LAB[source] def issue2label(issue): return seed2str.get(issue, "GCT") ENVIRON = "dev" DB_SCHEMA = 'prod_db' if ENVIRON == 'prod' else 'test_db' DB_CONVOS = 'conversations' DB_COMPLETIONS = 'comparison_completions' DB_BATTLES = 'battles' DB_ERRORS = 'completion_errors'