From a5a234ed9754e224b77f44e7349afa8f3f919ea5 Mon Sep 17 00:00:00 2001 From: zhayujie Date: Mon, 25 Sep 2023 16:42:36 +0800 Subject: [PATCH] fix: remove file after summary --- plugins/linkai/README.md | 2 +- plugins/linkai/linkai.py | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/plugins/linkai/README.md b/plugins/linkai/README.md index 838cfb8..bfcd1ca 100644 --- a/plugins/linkai/README.md +++ b/plugins/linkai/README.md @@ -103,6 +103,6 @@ #### 限制 - 1. 文件目前 支持 `txt`, `docx`, `pdf`, `md`, `csv`格式，文件大小由 `max_file_size` 限制，最大不超过15M，文件字数最多可支持百万字的文件。但不建议上传字数过多的文件，一是token消耗过大，二是摘要很难覆盖到全部内容，但可以通过多轮对话来了解细节。 + 1. 文件目前 支持 `txt`, `docx`, `pdf`, `md`, `csv`格式，文件大小由 `max_file_size` 限制，最大不超过15M，文件字数最多可支持百万字的文件。但不建议上传字数过多的文件，一是token消耗过大，二是摘要很难覆盖到全部内容，只能通过多轮对话来了解细节。 2. 分享链接 目前仅支持 公众号文章，后续会支持更多文章类型及视频链接等 3. 总结及对话的 费用与 LinkAI 3.5-4K 模型的计费方式相同，按文档内容的tokens进行计算 diff --git a/plugins/linkai/linkai.py b/plugins/linkai/linkai.py index 8e2f50b..910dbb8 100644 --- a/plugins/linkai/linkai.py +++ b/plugins/linkai/linkai.py @@ -8,6 +8,7 @@ from .summary import LinkSummary from bridge import bridge from common.expired_dict import ExpiredDict from common import const +import os @plugins.register( @@ -59,6 +60,7 @@ class LinkAI(Plugin): return USER_FILE_MAP[_find_user_id(context) + "-sum_id"] = res.get("summary_id") _set_reply_text(res.get("summary") + "\n\n💬 发送 \"开启对话\" 可以开启与文件内容的对话", e_context, level=ReplyType.TEXT) + os.remove(file_path) return if (context.type == ContextType.SHARING and self._is_summary_open(context)) or \ @@ -281,4 +283,5 @@ def _find_sum_id(context): def _find_file_id(context): return USER_FILE_MAP.get(_find_user_id(context) + "-file_id") -USER_FILE_MAP = ExpiredDict(60 * 60) + +USER_FILE_MAP = ExpiredDict(conf().get("expires_in_seconds") or 60 * 60)