Nevar pievienot vairāk kā 25 tēmas Tēmai ir jāsākas ar burtu vai ciparu, tā var saturēt domu zīmes ('-') un var būt līdz 35 simboliem gara.

pirms 2 gadiem
pirms 1 gada
pirms 1 gada
pirms 1 gada
pirms 1 gada
pirms 1 gada
pirms 1 gada
pirms 1 gada
pirms 1 gada
pirms 1 gada
pirms 1 gada
pirms 1 gada
pirms 1 gada
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227
  1. # encoding:utf-8
  2. import json
  3. import logging
  4. import os
  5. import pickle
  6. from common.log import logger
  7. # 将所有可用的配置项写在字典里, 请使用小写字母
  8. # 此处的配置值无实际意义,程序不会读取此处的配置,仅用于提示格式,请将配置加入到config.json中
  9. available_setting = {
  10. # openai api配置
  11. "open_ai_api_key": "", # openai api key
  12. # openai apibase,当use_azure_chatgpt为true时,需要设置对应的api base
  13. "open_ai_api_base": "https://api.openai.com/v1",
  14. "proxy": "", # openai使用的代理
  15. # chatgpt模型, 当use_azure_chatgpt为true时,其名称为Azure上model deployment名称
  16. "model": "gpt-3.5-turbo",
  17. "use_azure_chatgpt": False, # 是否使用azure的chatgpt
  18. "azure_deployment_id": "", # azure 模型部署名称
  19. # Bot触发配置
  20. "single_chat_prefix": ["bot", "@bot"], # 私聊时文本需要包含该前缀才能触发机器人回复
  21. "single_chat_reply_prefix": "[bot] ", # 私聊时自动回复的前缀,用于区分真人
  22. "group_chat_prefix": ["@bot"], # 群聊时包含该前缀则会触发机器人回复
  23. "group_chat_reply_prefix": "", # 群聊时自动回复的前缀
  24. "group_chat_keyword": [], # 群聊时包含该关键词则会触发机器人回复
  25. "group_at_off": False, # 是否关闭群聊时@bot的触发
  26. "group_name_white_list": ["ChatGPT测试群", "ChatGPT测试群2"], # 开启自动回复的群名称列表
  27. "group_name_keyword_white_list": [], # 开启自动回复的群名称关键词列表
  28. "group_chat_in_one_session": ["ChatGPT测试群"], # 支持会话上下文共享的群名称
  29. "trigger_by_self": False, # 是否允许机器人触发
  30. "image_create_prefix": ["画", "看", "找"], # 开启图片回复的前缀
  31. "concurrency_in_session": 1, # 同一会话最多有多少条消息在处理中,大于1可能乱序
  32. "image_create_size": "256x256", # 图片大小,可选有 256x256, 512x512, 1024x1024
  33. # chatgpt会话参数
  34. "expires_in_seconds": 3600, # 无操作会话的过期时间
  35. "character_desc": "你是ChatGPT, 一个由OpenAI训练的大型语言模型, 你旨在回答并解决人们的任何问题,并且可以使用多种语言与人交流。", # 人格描述
  36. "conversation_max_tokens": 1000, # 支持上下文记忆的最多字符数
  37. # chatgpt限流配置
  38. "rate_limit_chatgpt": 20, # chatgpt的调用频率限制
  39. "rate_limit_dalle": 50, # openai dalle的调用频率限制
  40. # chatgpt api参数 参考https://platform.openai.com/docs/api-reference/chat/create
  41. "temperature": 0.9,
  42. "top_p": 1,
  43. "frequency_penalty": 0,
  44. "presence_penalty": 0,
  45. "request_timeout": 60, # chatgpt请求超时时间,openai接口默认设置为600,对于难问题一般需要较长时间
  46. "timeout": 120, # chatgpt重试超时时间,在这个时间内,将会自动重试
  47. # 语音设置
  48. "speech_recognition": False, # 是否开启语音识别
  49. "group_speech_recognition": False, # 是否开启群组语音识别
  50. "voice_reply_voice": False, # 是否使用语音回复语音,需要设置对应语音合成引擎的api key
  51. "always_reply_voice": False, # 是否一直使用语音回复
  52. "voice_to_text": "openai", # 语音识别引擎,支持openai,baidu,google,azure
  53. "text_to_voice": "baidu", # 语音合成引擎,支持baidu,google,pytts(offline),azure
  54. # baidu 语音api配置, 使用百度语音识别和语音合成时需要
  55. "baidu_app_id": "",
  56. "baidu_api_key": "",
  57. "baidu_secret_key": "",
  58. # 1536普通话(支持简单的英文识别) 1737英语 1637粤语 1837四川话 1936普通话远场
  59. "baidu_dev_pid": "1536",
  60. # azure 语音api配置, 使用azure语音识别和语音合成时需要
  61. "azure_voice_api_key": "",
  62. "azure_voice_region": "japaneast",
  63. # 服务时间限制,目前支持itchat
  64. "chat_time_module": False, # 是否开启服务时间限制
  65. "chat_start_time": "00:00", # 服务开始时间
  66. "chat_stop_time": "24:00", # 服务结束时间
  67. # 翻译api
  68. "translate": "baidu", # 翻译api,支持baidu
  69. # baidu翻译api的配置
  70. "baidu_translate_app_id": "", # 百度翻译api的appid
  71. "baidu_translate_app_key": "", # 百度翻译api的秘钥
  72. # itchat的配置
  73. "hot_reload": False, # 是否开启热重载
  74. # wechaty的配置
  75. "wechaty_puppet_service_token": "", # wechaty的token
  76. # wechatmp的配置
  77. "wechatmp_token": "", # 微信公众平台的Token
  78. "wechatmp_port": 8080, # 微信公众平台的端口,需要端口转发到80或443
  79. "wechatmp_app_id": "", # 微信公众平台的appID
  80. "wechatmp_app_secret": "", # 微信公众平台的appsecret
  81. "wechatmp_aes_key": "", # 微信公众平台的EncodingAESKey,加密模式需要
  82. # wechatcom的通用配置
  83. "wechatcom_corp_id": "", # 企业微信公司的corpID
  84. # wechatcomapp的配置
  85. "wechatcomapp_token": "", # 企业微信app的token
  86. "wechatcomapp_port": 9898, # 企业微信app的服务端口,不需要端口转发
  87. "wechatcomapp_secret": "", # 企业微信app的secret
  88. "wechatcomapp_agent_id": "", # 企业微信app的agent_id
  89. "wechatcomapp_aes_key": "", # 企业微信app的aes_key
  90. # chatgpt指令自定义触发词
  91. "clear_memory_commands": ["#清除记忆"], # 重置会话指令,必须以#开头
  92. # channel配置
  93. "channel_type": "wx", # 通道类型,支持:{wx,wxy,terminal,wechatmp,wechatmp_service,wechatcom_app}
  94. "subscribe_msg": "", # 订阅消息, 支持: wechatmp, wechatmp_service, wechatcom_app
  95. "debug": False, # 是否开启debug模式,开启后会打印更多日志
  96. "appdata_dir": "", # 数据目录
  97. # 插件配置
  98. "plugin_trigger_prefix": "$", # 规范插件提供聊天相关指令的前缀,建议不要和管理员指令前缀"#"冲突
  99. # 应用市场配置
  100. "app_market_api_key": "",
  101. "app_code": ""
  102. }
  103. class Config(dict):
  104. def __init__(self, d=None):
  105. super().__init__()
  106. if d is None:
  107. d = {}
  108. for k, v in d.items():
  109. self[k] = v
  110. # user_datas: 用户数据,key为用户名,value为用户数据,也是dict
  111. self.user_datas = {}
  112. def __getitem__(self, key):
  113. if key not in available_setting:
  114. raise Exception("key {} not in available_setting".format(key))
  115. return super().__getitem__(key)
  116. def __setitem__(self, key, value):
  117. if key not in available_setting:
  118. raise Exception("key {} not in available_setting".format(key))
  119. return super().__setitem__(key, value)
  120. def get(self, key, default=None):
  121. try:
  122. return self[key]
  123. except KeyError as e:
  124. return default
  125. except Exception as e:
  126. raise e
  127. # Make sure to return a dictionary to ensure atomic
  128. def get_user_data(self, user) -> dict:
  129. if self.user_datas.get(user) is None:
  130. self.user_datas[user] = {}
  131. return self.user_datas[user]
  132. def load_user_datas(self):
  133. try:
  134. with open(os.path.join(get_appdata_dir(), "user_datas.pkl"), "rb") as f:
  135. self.user_datas = pickle.load(f)
  136. logger.info("[Config] User datas loaded.")
  137. except FileNotFoundError as e:
  138. logger.info("[Config] User datas file not found, ignore.")
  139. except Exception as e:
  140. logger.info("[Config] User datas error: {}".format(e))
  141. self.user_datas = {}
  142. def save_user_datas(self):
  143. try:
  144. with open(os.path.join(get_appdata_dir(), "user_datas.pkl"), "wb") as f:
  145. pickle.dump(self.user_datas, f)
  146. logger.info("[Config] User datas saved.")
  147. except Exception as e:
  148. logger.info("[Config] User datas error: {}".format(e))
  149. config = Config()
  150. def load_config():
  151. global config
  152. config_path = "./config.json"
  153. if not os.path.exists(config_path):
  154. logger.info("配置文件不存在,将使用config-template.json模板")
  155. config_path = "./config-template.json"
  156. config_str = read_file(config_path)
  157. logger.debug("[INIT] config str: {}".format(config_str))
  158. # 将json字符串反序列化为dict类型
  159. config = Config(json.loads(config_str))
  160. # override config with environment variables.
  161. # 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.
  162. for name, value in os.environ.items():
  163. name = name.lower()
  164. if name in available_setting:
  165. logger.info("[INIT] override config by environ args: {}={}".format(name, value))
  166. try:
  167. config[name] = eval(value)
  168. except:
  169. if value == "false":
  170. config[name] = False
  171. elif value == "true":
  172. config[name] = True
  173. else:
  174. config[name] = value
  175. if config.get("debug", False):
  176. logger.setLevel(logging.DEBUG)
  177. logger.debug("[INIT] set log level to DEBUG")
  178. logger.info("[INIT] load config: {}".format(config))
  179. config.load_user_datas()
  180. def get_root():
  181. return os.path.dirname(os.path.abspath(__file__))
  182. def read_file(path):
  183. with open(path, mode="r", encoding="utf-8") as f:
  184. return f.read()
  185. def conf():
  186. return config
  187. def get_appdata_dir():
  188. data_path = os.path.join(get_root(), conf().get("appdata_dir", ""))
  189. if not os.path.exists(data_path):
  190. logger.info("[INIT] data path not exists, create it: {}".format(data_path))
  191. os.makedirs(data_path)
  192. return data_path
  193. def subscribe_msg():
  194. trigger_prefix = conf().get("single_chat_prefix", [""])[0]
  195. msg = conf().get("subscribe_msg", "")
  196. return msg.format(trigger_prefix=trigger_prefix)