File size: 837 Bytes
18f6362
9ff00d4
 
 
18f6362
 
bdca921
18f6362
9ff00d4
18f6362
975a927
9ff00d4
20b3b4a
 
9ff00d4
975a927
20b3b4a
 
 
 
 
975a927
 
 
18f6362
 
 
238098c
975a927
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
from models.model_seeds import seeds, seed2str

# ISSUES = ['Anxiety','Suicide']
ISSUES = [k for k,_ in seeds.items()]
SOURCES = [
    "CTL_llama2",
    # "CTL_mistral",
    'OA_rolemodel', 
        #    'OA_finetuned',
]
SOURCES_LAB = {"OA_rolemodel":'OpenAI GPT3.5',
               "OA_finetuned":'Finetuned OpenAI',
               "CTL_llama2": "Llama",
               "CTL_mistral": "Mistral",
               }

ENDPOINT_NAMES = {
    '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 = "prod"

DB_SCHEMA = 'prod_db' if ENVIRON == 'prod' else 'test_db' 
DB_CONVOS = 'conversations'
DB_COMPLETIONS = 'comparison_completions'
DB_BATTLES = 'battles'
DB_ERRORS = 'completion_errors'