diff --git a/README.md b/README.md index 6d8543eea..2e409a89e 100644 --- a/README.md +++ b/README.md @@ -8,7 +8,7 @@ - [x] **基础对话：** 私聊及群聊的消息智能回复，支持多轮会话上下文记忆，支持 GPT-3.5, GPT-4, claude, 文心一言, 讯飞星火 - [x] **语音识别：** 可识别语音消息，通过文字或语音回复，支持 azure, baidu, google, openai等多种语音模型 - [x] **图片生成：** 支持图片生成 和 图生图（如照片修复），可选择 Dell-E, stable diffusion, replicate, midjourney模型 -- [x] **丰富插件：** 支持个性化插件扩展，已实现多角色切换、文字冒险、敏感词过滤、聊天记录总结等插件 +- [x] **丰富插件：** 支持个性化插件扩展，已实现多角色切换、文字冒险、敏感词过滤、聊天记录总结、文档总结和对话等插件 - [X] **Tool工具：** 与操作系统和互联网交互，支持最新信息搜索、数学计算、天气和资讯查询、网页总结，基于 [chatgpt-tool-hub](https://github.com/goldfishh/chatgpt-tool-hub) 实现 - [x] **知识库：** 通过上传知识库文件自定义专属机器人，可作为数字分身、领域知识库、智能客服使用，基于 [LinkAI](https://chat.link-ai.tech/console) 实现 @@ -28,7 +28,7 @@ Demo made by [Visionn](https://www.wangpc.cc/) # 更新日志 ->**2023.09.26：** 插件增加 文件/文章链接 一键总结和对话的功能，使用参考：插件说明 +>**2023.09.26：** 插件增加 文件/文章链接 一键总结和对话的功能，使用参考：[插件说明](https://github.com/zhayujie/chatgpt-on-wechat/tree/master/plugins/linkai#3%E6%96%87%E6%A1%A3%E6%80%BB%E7%BB%93%E5%AF%B9%E8%AF%9D%E5%8A%9F%E8%83%BD) >**2023.08.08：** 接入百度文心一言模型，通过 [插件](https://github.com/zhayujie/chatgpt-on-wechat/tree/master/plugins/linkai) 支持 Midjourney 绘图 @@ -44,8 +44,6 @@ Demo made by [Visionn](https://www.wangpc.cc/) >**2023.03.09：** 基于 `whisper API`(后续已接入更多的语音`API`服务) 实现对微信语音消息的解析和回复，添加配置项 `"speech_recognition":true` 即可启用，使用参考 [#415](https://github.com/zhayujie/chatgpt-on-wechat/issues/415)。(contributed by [wanggang1987](https://github.com/wanggang1987) in [#385](https://github.com/zhayujie/chatgpt-on-wechat/pull/385)) ->**2023.03.02：** 接入[ChatGPT API](https://platform.openai.com/docs/guides/chat) (gpt-3.5-turbo)，默认使用该模型进行对话，需升级openai依赖 (`pip3 install --upgrade openai`)。网络问题参考 [#351](https://github.com/zhayujie/chatgpt-on-wechat/issues/351) - >**2023.02.09：** 扫码登录存在账号限制风险，请谨慎使用，参考[#58](https://github.com/AutumnWhj/ChatGPT-wechat-bot/issues/158) # 快速开始 diff --git a/channel/wework/wework_message.py b/channel/wework/wework_message.py index dd24aa46e..e95dfb194 100644 --- a/channel/wework/wework_message.py +++ b/channel/wework/wework_message.py @@ -8,6 +8,7 @@ from bridge.context import ContextType from channel.chat_message import ChatMessage from common.log import logger +from ntwork.const import send_type def get_with_retry(get_func, max_retries=5, delay=5): @@ -41,15 +42,37 @@ def cdn_download(wework, message, file_name): data = message["data"] aes_key = data["cdn"]["aes_key"] file_size = data["cdn"]["size"] - file_type = 2 - file_id = data["cdn"]["file_id"] # 获取当前工作目录，然后与文件名拼接得到保存路径 current_dir = os.getcwd() save_path = os.path.join(current_dir, "tmp", file_name) - result = wework.c2c_cdn_download(file_id, aes_key, file_size, file_type, save_path) - logger.debug(result) + # 下载保存图片到本地 + if "url" in data["cdn"].keys() and "auth_key" in data["cdn"].keys(): + url = data["cdn"]["url"] + auth_key = data["cdn"]["auth_key"] + # result = wework.wx_cdn_download(url, auth_key, aes_key, file_size, save_path) # ntwork库本身接口有问题，缺失了aes_key这个参数 + """ + 下载wx类型的cdn文件，以https开头 + """ + data = { + 'url': url, + 'auth_key': auth_key, + 'aes_key': aes_key, + 'size': file_size, + 'save_path': save_path + } + result = wework._WeWork__send_sync(send_type.MT_WXCDN_DOWNLOAD_MSG, data) # 直接用wx_cdn_download的接口内部实现来调用 + elif "file_id" in data["cdn"].keys(): + file_type = 2 + file_id = data["cdn"]["file_id"] + result = wework.c2c_cdn_download(file_id, aes_key, file_size, file_type, save_path) + else: + logger.error(f"something is wrong, data: {data}") + return + + # 输出下载结果 + logger.debug(f"result: {result}") def c2c_download_and_convert(wework, message, file_name):