Browse Source

fix: remove file after summary

develop
zhayujie 1 year ago
parent
commit
a5a234ed97
2 changed files with 5 additions and 2 deletions
  1. +1
    -1
      plugins/linkai/README.md
  2. +4
    -1
      plugins/linkai/linkai.py

+ 1
- 1
plugins/linkai/README.md 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进行计算

+ 4
- 1
plugins/linkai/linkai.py 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)

Loading…
Cancel
Save