convosim-ui / app_config.py
ivnban27-ctl's picture
fixed databricks integration
42a7266
raw
history blame
984 Bytes
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'