fix: remove file after summary

This commit is contained in:
zhayujie
2023-09-25 16:42:36 +08:00
parent 5b5dbcd78b
commit a5a234ed97
2 changed files with 5 additions and 2 deletions

View File

@@ -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进行计算

View File

@@ -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)