File size: 528 Bytes
975a927
75b7dbb
 
 
975a927
75b7dbb
 
 
975a927
 
 
 
75b7dbb
975a927
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
ISSUES = ['Anxiety','Suicide']
SOURCES = ['OA_rolemodel', 
        #    'OA_finetuned',
           "CTL_llama2"]
SOURCES_LAB = {"OA_rolemodel":'OpenAI GPT3.5',
               "OA_finetuned":'Finetuned OpenAI',
               "CTL_llama2": "Custom CTL"
               }

def source2label(source):
    return SOURCES_LAB[source]

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'