automigration.py

from dotenv import dotenv_values, load_dotenv
from utils import cprint
import os
import sys
import json

def verify_environment():
  env_filename = os.environ.get('BOTALITY_ENV_FILE', '.env')
  assert os.path.exists(env_filename), \
    f"Specified configuration file ({env_filename}) does not exist, please make a copy of .env.example and edit it."
  assert not env_filename.endswith('.env.example'), \
    "You should not use example file as your configuration, please copy it to .env file."
  assert os.path.exists('.env.example'), \
    f".env.example does not exist, please recover it, the file is used to automatically add new config options after updating"
  assert env_filename.endswith('.env'), \
    "Configuration files must have .env extension"
  
  system_env = dotenv_values(env_filename)
  demo_env = dotenv_values('.env.example')
  
  def check_new_keys_in_example_env():
    towrite = '''
'''
    for key in demo_env:
      if key not in system_env:
        towrite += f"{key}='{demo_env[key]}'\n"
        print('New config key added', key)
  
    if len(towrite) != 1:
      with open(env_filename, 'a') as file:
        file.write(towrite)
  
  DEPRECATED_KEYS = [
    'llm_active_model_type', 
    'sd_available_loras', 
    'tts_so_vits_svc_code_path',
    'tts_enable_so_vits_svc',
    'tts_so_vits_svc_base_tts_provider',
    'lang_main',
    'stt_autoreply_mode',
    'stt_autoreply_voice'
  ]
  DEPRECATED_KVS = {
    'llm_assistant_chronicler': ['gpt4all', 'minchatgpt', 'alpaca'],
    'llm_python_model_type': 'cerebras_gpt'
  }
  
  def check_deprecated_keys_in_dotenv():
    for key in system_env:
      if key in DEPRECATED_KEYS:
        cprint(f'Warning! The key "{key}" has been deprecated! See CHANGELOG.md.', color='red')
      value = system_env[key]
      if len(value) < 1:
        continue
      if (value[0] != '[' and value[0] != '{'):
        value = f'"{value}"'
      value = json.loads(value)
      if type(value) is str:
        if key in DEPRECATED_KVS and value in DEPRECATED_KVS[key]:
          cprint(f'Warning! The value "{value}" of "{key}" has been deprecated! See CHANGELOG.md.', color='yellow')
  
  check_new_keys_in_example_env()
  check_deprecated_keys_in_dotenv()
  load_dotenv(env_filename, override=True)