diff --git a/bot/chatgpt/chat_gpt_bot.py b/bot/chatgpt/chat_gpt_bot.py index 52ab400..607f6c0 100644 --- a/bot/chatgpt/chat_gpt_bot.py +++ b/bot/chatgpt/chat_gpt_bot.py @@ -76,6 +76,16 @@ class ChatGPTBot(Bot): reply = Reply(ReplyType.ERROR, 'Bot不支持处理{}类型的消息'.format(context.type)) return reply + def compose_args(self): + return { + "model": conf().get("model") or "gpt-3.5-turbo", # 对话模型的名称 + "temperature":conf().get('temperature', 0.9), # 值在[0,1]之间，越大表示回复越具有不确定性 + # "max_tokens":4096, # 回复最大的字符数 + "top_p":1, + "frequency_penalty":conf().get('frequency_penalty', 0.0), # [-2,2]之间，该值越大则更倾向于产生不同的内容 + "presence_penalty":conf().get('presence_penalty', 0.0), # [-2,2]之间，该值越大则更倾向于产生不同的内容 + } + def reply_text(self, session, session_id, retry_count=0) -> dict: ''' call openai's ChatCompletion to get the answer @@ -88,13 +98,7 @@ class ChatGPTBot(Bot): if conf().get('rate_limit_chatgpt') and not self.tb4chatgpt.get_token(): return {"completion_tokens": 0, "content": "提问太快啦，请休息一下再问我吧"} response = openai.ChatCompletion.create( - model= conf().get("model") or "gpt-3.5-turbo", # 对话模型的名称 - messages=session, - temperature=conf().get('temperature', 0.9), # 值在[0,1]之间，越大表示回复越具有不确定性 - #max_tokens=4096, # 回复最大的字符数 - top_p=1, - frequency_penalty=conf().get('frequency_penalty', 0.0), # [-2,2]之间，该值越大则更倾向于产生不同的内容 - presence_penalty=conf().get('presence_penalty', 0.0), # [-2,2]之间，该值越大则更倾向于产生不同的内容 + messages=session, **self.compose_args() ) # logger.info("[ChatGPT] reply={}, total_tokens={}".format(response.choices[0]['message']['content'], response["usage"]["total_tokens"])) return {"total_tokens": response["usage"]["total_tokens"], @@ -156,53 +160,11 @@ class AzureChatGPTBot(ChatGPTBot): openai.api_type = "azure" openai.api_version = "2023-03-15-preview" - def reply_text(self, session, session_id, retry_count=0) ->dict: - ''' - call openai's ChatCompletion to get the answer - :param session: a conversation session - :param session_id: session id - :param retry_count: retry count - :return: {} - ''' - try: - if conf().get('rate_limit_chatgpt') and not self.tb4chatgpt.get_token(): - return {"completion_tokens": 0, "content": "提问太快啦，请休息一下再问我吧"} - response = openai.ChatCompletion.create( - engine=conf().get("model") or "gpt-3.5-turbo", # the model deployment name on Azure - messages=session, - temperature=conf().get('temperature', 0.9), # 值在[0,1]之间，越大表示回复越具有不确定性 - #max_tokens=4096, # 回复最大的字符数 - top_p=1, - frequency_penalty=conf().get('frequency_penalty', 0.0), # [-2,2]之间，该值越大则更倾向于产生不同的内容 - presence_penalty=conf().get('presence_penalty', 0.0), # [-2,2]之间，该值越大则更倾向于产生不同的内容 - ) - # logger.info("[ChatGPT] reply={}, total_tokens={}".format(response.choices[0]['message']['content'], response["usage"]["total_tokens"])) - return {"total_tokens": response["usage"]["total_tokens"], - "completion_tokens": response["usage"]["completion_tokens"], - "content": response.choices[0]['message']['content']} - except openai.error.RateLimitError as e: - # rate limit exception - logger.warn(e) - if retry_count < 1: - time.sleep(5) - logger.warn("[OPEN_AI] RateLimit exceed, 第{}次重试".format(retry_count+1)) - return self.reply_text(session, session_id, retry_count+1) - else: - return {"completion_tokens": 0, "content": "提问太快啦，请休息一下再问我吧"} - except openai.error.APIConnectionError as e: - # api connection exception - logger.warn(e) - logger.warn("[OPEN_AI] APIConnection failed") - return {"completion_tokens": 0, "content":"我连接不到你的网络"} - except openai.error.Timeout as e: - logger.warn(e) - logger.warn("[OPEN_AI] Timeout") - return {"completion_tokens": 0, "content":"我没有收到你的消息"} - except Exception as e: - # unknown exception - logger.exception(e) - Session.clear_session(session_id) - return {"completion_tokens": 0, "content": "请再问我一次吧"} + def compose_args(self): + args = super().compose_args() + args["engine"] = args["model"] + del(args["model"]) + return args class SessionManager(object): diff --git a/bridge/bridge.py b/bridge/bridge.py index 15a6d02..26d5440 100644 --- a/bridge/bridge.py +++ b/bridge/bridge.py @@ -13,8 +13,8 @@ class Bridge(object): def __init__(self): self.btype={ "chat": const.CHATGPT, - "voice_to_text": "openai", - "text_to_voice": "baidu" + "voice_to_text": conf().get("voice_to_text", "openai"), + "text_to_voice": conf().get("text_to_voice", "baidu") } model_type = conf().get("model") if model_type in ["text-davinci-003"]: diff --git a/config.py b/config.py index 7dd921e..e1addfd 100644 --- a/config.py +++ b/config.py @@ -4,7 +4,84 @@ import json import os from common.log import logger -config = {} +# 将所有可用的配置项写在字典里 +available_setting ={ + #openai api配置 + "open_ai_api_key": "", # openai api key + "open_ai_api_base": "https://api.openai.com/v1", # openai apibase，当use_azure_chatgpt为true时，需要设置对应的api base + "proxy": "", # openai使用的代理 + "model": "gpt-3.5-turbo", # chatgpt模型， 当use_azure_chatgpt为true时，其名称为Azure上model deployment名称 + "use_azure_chatgpt": False, # 是否使用azure的chatgpt + + #Bot触发配置 + "single_chat_prefix": ["bot", "@bot"], # 私聊时文本需要包含该前缀才能触发机器人回复 + "single_chat_reply_prefix": "[bot] ", # 私聊时自动回复的前缀，用于区分真人 + "group_chat_prefix": ["@bot"], # 群聊时包含该前缀则会触发机器人回复 + "group_name_white_list": ["ChatGPT测试群", "ChatGPT测试群2"], # 开启自动回复的群名称列表 + "group_chat_in_one_session": ["ChatGPT测试群"], # 支持会话上下文共享的群名称 + "image_create_prefix": ["画", "看", "找"], # 开启图片回复的前缀 + + #chatgpt会话参数 + "expires_in_seconds": 3600, # 无操作会话的过期时间 + "character_desc": "你是ChatGPT, 一个由OpenAI训练的大型语言模型, 你旨在回答并解决人们的任何问题，并且可以使用多种语言与人交流。", # 人格描述 + "conversation_max_tokens": 1000, # 支持上下文记忆的最多字符数 + + #chatgpt限流配置 + "rate_limit_chatgpt": 20, # chatgpt的调用频率限制 + "rate_limit_dalle": 50, # openai dalle的调用频率限制 + + + #chatgpt api参数 参考https://platform.openai.com/docs/api-reference/chat/create + "temperature": 0.9, + "top_p": 1, + "frequency_penalty": 0, + "presence_penalty": 0, + + #语音设置 + "speech_recognition": False, # 是否开启语音识别 + "voice_reply_voice": False, # 是否使用语音回复语音，需要设置对应语音合成引擎的api key + "voice_to_text": "openai", # 语音识别引擎，支持openai和google + "text_to_voice": "baidu", # 语音合成引擎，支持baidu和google + + # baidu api的配置， 使用百度语音识别和语音合成时需要 + 'baidu_app_id': "", + 'baidu_api_key': "", + 'baidu_secret_key': "", + + #服务时间限制，目前支持itchat + "chat_time_module": False, # 是否开启服务时间限制 + "chat_start_time": "00:00", # 服务开始时间 + "chat_stop_time": "24:00", # 服务结束时间 + + # itchat的配置 + "hot_reload": False, # 是否开启热重载 + + # chatgpt指令自定义触发词 + "clear_memory_commands": ['#清除记忆'], # 重置会话指令 + + +} + +class Config(dict): + def __getitem__(self, key): + if key not in available_setting: + raise Exception("key {} not in available_setting".format(key)) + return super().__getitem__(key) + + def __setitem__(self, key, value): + if key not in available_setting: + raise Exception("key {} not in available_setting".format(key)) + return super().__setitem__(key, value) + + def get(self, key, default=None): + try : + return self[key] + except KeyError as e: + return default + except Exception as e: + raise e + +config = Config() def load_config(): global config @@ -15,12 +92,14 @@ def load_config(): config_str = read_file(config_path) # 将json字符串反序列化为dict类型 - config = json.loads(config_str) + config = Config(json.loads(config_str)) # override config with environment variables. # 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. for name, value in os.environ.items(): - config[name] = value + if name in available_setting: + logger.info("[INIT] override config by environ args: {}={}".format(name, value)) + config[name] = value logger.info("[INIT] load config: {}".format(config))