From 5b5dbcd78b62317a7c55db442545ed409842d2ac Mon Sep 17 00:00:00 2001 From: zhayujie Date: Sun, 24 Sep 2023 14:33:39 +0800 Subject: [PATCH] feat: remove file word calc and support url link --- plugins/linkai/README.md | 7 +++---- plugins/linkai/config.json.template | 1 - plugins/linkai/linkai.py | 3 ++- plugins/linkai/summary.py | 14 ++++++-------- 4 files changed, 11 insertions(+), 14 deletions(-) diff --git a/plugins/linkai/README.md b/plugins/linkai/README.md index 319e7c4..838cfb8 100644 --- a/plugins/linkai/README.md +++ b/plugins/linkai/README.md @@ -25,8 +25,7 @@ "summary": { "enabled": true, # 文档总结和对话功能开关 "group_enabled": true, # 是否支持群聊开启 - "max_summary_words": 50000, # 文章的最大字数，超过字数则直接忽略 - "max_file_size": 15000 # 文件的大小限制，单位KB，超过该大小直接忽略 + "max_file_size": 10000 # 文件的大小限制，单位KB，默认为10M，超过该大小直接忽略 } } ``` @@ -85,7 +84,7 @@ 注意事项： 1. 使用 `$mj open` 和 `$mj close` 指令可以快速打开和关闭绘图功能 -2. 海外环境部署请将 `img_proxy` 设置为 `False` +2. 海外环境部署请将 `img_proxy` 设置为 `false` 3. 开启 `use_image_create_prefix` 配置后可直接复用全局画图触发词，以"画"开头便可以生成图片。 4. 提示词内容中包含敏感词或者参数格式错误可能导致绘画失败，生成失败不消耗积分 5. 若未收到图片可能有两种可能，一种是收到了图片但微信发送失败，可以在后台日志查看有没有获取到图片url，一般原因是受到了wx限制，可以稍后重试或更换账号尝试；另一种情况是图片提示词存在疑似违规，mj不会直接提示错误但会在画图后删掉原图导致程序无法获取，这种情况不消耗积分。 @@ -104,6 +103,6 @@ #### 限制 - 1. 文件目前 支持 `txt`, `docx`, `pdf`, `md`, `csv`格式，文件大小由 `max_file_size` 限制，最大不超过15M，文件字数由 `max_summary_words` 配置限制，最多可支持百万字的文件。但不建议上传字数过多的文件，一是token消耗过大，二是摘要很难覆盖到全部内容，但可以通过多轮对话来了解细节。 + 1. 文件目前 支持 `txt`, `docx`, `pdf`, `md`, `csv`格式，文件大小由 `max_file_size` 限制，最大不超过15M，文件字数最多可支持百万字的文件。但不建议上传字数过多的文件，一是token消耗过大，二是摘要很难覆盖到全部内容，但可以通过多轮对话来了解细节。 2. 分享链接 目前仅支持 公众号文章，后续会支持更多文章类型及视频链接等 3. 总结及对话的 费用与 LinkAI 3.5-4K 模型的计费方式相同，按文档内容的tokens进行计算 diff --git a/plugins/linkai/config.json.template b/plugins/linkai/config.json.template index 6302b7e..30517e0 100644 --- a/plugins/linkai/config.json.template +++ b/plugins/linkai/config.json.template @@ -14,7 +14,6 @@ "summary": { "enabled": true, "group_enabled": true, - "max_summary_words": 50000, "max_file_size": 15000 } } diff --git a/plugins/linkai/linkai.py b/plugins/linkai/linkai.py index 475b5c8..8e2f50b 100644 --- a/plugins/linkai/linkai.py +++ b/plugins/linkai/linkai.py @@ -61,7 +61,8 @@ class LinkAI(Plugin): _set_reply_text(res.get("summary") + "\n\n💬 发送 \"开启对话\" 可以开启与文件内容的对话", e_context, level=ReplyType.TEXT) return - if context.type == ContextType.SHARING and self._is_summary_open(context): + if (context.type == ContextType.SHARING and self._is_summary_open(context)) or \ + (context.type == ContextType.TEXT and LinkSummary().check_url(context.content)): if not LinkSummary().check_url(context.content): return _send_info(e_context, "正在为你加速生成摘要，请稍后") diff --git a/plugins/linkai/summary.py b/plugins/linkai/summary.py index db32ea2..c95f58d 100644 --- a/plugins/linkai/summary.py +++ b/plugins/linkai/summary.py @@ -65,13 +65,9 @@ class LinkSummary: def check_file(self, file_path: str, sum_config: dict) -> bool: file_size = os.path.getsize(file_path) // 1000 - with open(file_path, 'r') as f: - content = f.read() - word_count = len(content) - if (sum_config.get("max_file_size") and file_size > sum_config.get("max_file_size")) or file_size > 15000\ - or (sum_config.get("max_summary_words") and word_count > sum_config.get("max_summary_words")): - logger.warn(f"[LinkSum] file size exceeds limit, No processing, file_size={file_size}KB, word_count={word_count}") + if (sum_config.get("max_file_size") and file_size > sum_config.get("max_file_size")) or file_size > 15000: + logger.warn(f"[LinkSum] file size exceeds limit, No processing, file_size={file_size}KB") return True suffix = file_path.split(".")[-1] @@ -83,9 +79,11 @@ class LinkSummary: return True def check_url(self, url: str): - support_list = ["mp.weixin.qq.com"] + if not url: + return False + support_list = ["http://mp.weixin.qq.com", "https://mp.weixin.qq.com"] for support_url in support_list: - if support_url in url: + if url.strip().startswith(support_url): return True logger.warn("[LinkSum] unsupported url") return False