convosim-ui / app_config.py
ivnban27-ctl's picture
Added MongoDB functionality
0f381ca
raw
history blame
430 Bytes
ISSUES = ['Anxiety','Suicide']
SOURCES = ['OA_rolemodel', 'OA_finetuned']
SOURCES_LAB = {"OA_rolemodel":'OpenAI GPT3.5',
"OA_finetuned":'Finetuned OpenAI'}
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'