您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

141 行
5.3KB

  1. # encoding:utf-8
  2. import json
  3. import os
  4. from common.log import logger
  5. # 将所有可用的配置项写在字典里, 请使用小写字母
  6. available_setting = {
  7. # openai api配置
  8. "open_ai_api_key": "", # openai api key
  9. # openai apibase,当use_azure_chatgpt为true时,需要设置对应的api base
  10. "open_ai_api_base": "https://api.openai.com/v1",
  11. "proxy": "", # openai使用的代理
  12. # chatgpt模型, 当use_azure_chatgpt为true时,其名称为Azure上model deployment名称
  13. "model": "gpt-3.5-turbo",
  14. "use_azure_chatgpt": False, # 是否使用azure的chatgpt
  15. # Bot触发配置
  16. "single_chat_prefix": ["bot", "@bot"], # 私聊时文本需要包含该前缀才能触发机器人回复
  17. "single_chat_reply_prefix": "[bot] ", # 私聊时自动回复的前缀,用于区分真人
  18. "group_chat_prefix": ["@bot"], # 群聊时包含该前缀则会触发机器人回复
  19. "group_chat_reply_prefix": "", # 群聊时自动回复的前缀
  20. "group_chat_keyword": [], # 群聊时包含该关键词则会触发机器人回复
  21. "group_at_off": False, # 是否关闭群聊时@bot的触发
  22. "group_name_white_list": ["ChatGPT测试群", "ChatGPT测试群2"], # 开启自动回复的群名称列表
  23. "group_name_keyword_white_list": [], # 开启自动回复的群名称关键词列表
  24. "group_chat_in_one_session": ["ChatGPT测试群"], # 支持会话上下文共享的群名称
  25. "image_create_prefix": ["画", "看", "找"], # 开启图片回复的前缀
  26. # chatgpt会话参数
  27. "expires_in_seconds": 3600, # 无操作会话的过期时间
  28. "character_desc": "你是ChatGPT, 一个由OpenAI训练的大型语言模型, 你旨在回答并解决人们的任何问题,并且可以使用多种语言与人交流。", # 人格描述
  29. "conversation_max_tokens": 1000, # 支持上下文记忆的最多字符数
  30. # chatgpt限流配置
  31. "rate_limit_chatgpt": 20, # chatgpt的调用频率限制
  32. "rate_limit_dalle": 50, # openai dalle的调用频率限制
  33. # chatgpt api参数 参考https://platform.openai.com/docs/api-reference/chat/create
  34. "temperature": 0.9,
  35. "top_p": 1,
  36. "frequency_penalty": 0,
  37. "presence_penalty": 0,
  38. # 语音设置
  39. "speech_recognition": False, # 是否开启语音识别
  40. "group_speech_recognition": False, # 是否开启群组语音识别
  41. "voice_reply_voice": False, # 是否使用语音回复语音,需要设置对应语音合成引擎的api key
  42. "voice_to_text": "openai", # 语音识别引擎,支持openai和google
  43. "text_to_voice": "baidu", # 语音合成引擎,支持baidu和google
  44. # baidu api的配置, 使用百度语音识别和语音合成时需要
  45. "baidu_app_id": "",
  46. "baidu_api_key": "",
  47. "baidu_secret_key": "",
  48. # 1536普通话(支持简单的英文识别) 1737英语 1637粤语 1837四川话 1936普通话远场
  49. "baidu_dev_pid": "1536",
  50. # 服务时间限制,目前支持itchat
  51. "chat_time_module": False, # 是否开启服务时间限制
  52. "chat_start_time": "00:00", # 服务开始时间
  53. "chat_stop_time": "24:00", # 服务结束时间
  54. # itchat的配置
  55. "hot_reload": False, # 是否开启热重载
  56. # wechaty的配置
  57. "wechaty_puppet_service_token": "", # wechaty的token
  58. # chatgpt指令自定义触发词
  59. "clear_memory_commands": ['#清除记忆'], # 重置会话指令
  60. "channel_type": "wx", # 通道类型,支持wx,wxy和terminal
  61. }
  62. class Config(dict):
  63. def __getitem__(self, key):
  64. if key not in available_setting:
  65. raise Exception("key {} not in available_setting".format(key))
  66. return super().__getitem__(key)
  67. def __setitem__(self, key, value):
  68. if key not in available_setting:
  69. raise Exception("key {} not in available_setting".format(key))
  70. return super().__setitem__(key, value)
  71. def get(self, key, default=None):
  72. try:
  73. return self[key]
  74. except KeyError as e:
  75. return default
  76. except Exception as e:
  77. raise e
  78. config = Config()
  79. def load_config():
  80. global config
  81. config_path = "./config.json"
  82. if not os.path.exists(config_path):
  83. logger.info('配置文件不存在,将使用config-template.json模板')
  84. config_path = "./config-template.json"
  85. config_str = read_file(config_path)
  86. logger.debug("[INIT] config str: {}".format(config_str))
  87. # 将json字符串反序列化为dict类型
  88. config = Config(json.loads(config_str))
  89. # override config with environment variables.
  90. # Some online deployment platforms (e.g. Railway) deploy project from github directly. So you shouldn't put your secrets like api key in a config file, instead use environment variables to override the default config.
  91. for name, value in os.environ.items():
  92. name = name.lower()
  93. if name in available_setting:
  94. logger.info(
  95. "[INIT] override config by environ args: {}={}".format(name, value))
  96. try:
  97. config[name] = eval(value)
  98. except:
  99. config[name] = value
  100. logger.info("[INIT] load config: {}".format(config))
  101. def get_root():
  102. return os.path.dirname(os.path.abspath(__file__))
  103. def read_file(path):
  104. with open(path, mode='r', encoding='utf-8') as f:
  105. return f.read()
  106. def conf():
  107. return config