QueryHelper / configProd.py
anumaurya114exp's picture
added different history and context for different logins
4099f5c
raw
history blame contribute delete
931 Bytes
import os
OPENAI_API_KEY = os.getenv("api_key")
#database credential
dbName = os.getenv("db_name")
userDB = os.getenv("user_db")
pwdDB = os.getenv("pwd_db")
host = os.getenv("host_db")
port = os.getenv("port_db")
# GPT_MODEL = "gpt-4"
GPT_MODEL = "gpt-4-1106-preview"
# GPT_MODEL = "gpt-3.5-turbo-1106"
GPT_MODEL_FINE_TUNED = "ft:gpt-3.5-turbo-1106:express-analytics::8TVCk0Ea"
HUGGING_FACE_TOKEN = os.getenv("HUGGING_FACE_TOKEN")
#gradio login
ADMIN = os.getenv("admin")
PASSWD = os.getenv("passwd")
DB_CREDS_DATA = ({"database":dbName, "user":userDB, "password":pwdDB, "host":host, "port":port})
STORAGE_DIR = os.getenv("HF_HOME", "/data")
logsDir = STORAGE_DIR
TABLES_DATA_DIR = os.path.join(STORAGE_DIR, "tablesData")
RESULT_CSV_DIR = os.path.join(STORAGE_DIR, "csvResults")
SCHEMA_INFO_FILE_PATH = os.path.join(STORAGE_DIR, "schemaInfo.pickle")
# USERS_INFO_FILE_PATH = os.path.join(STORAGE_DIR, "usersInfo.pickle")