From 2fa1df29bede99153f2beae83bd4251a28a02e39 Mon Sep 17 00:00:00 2001
From: zhayujie <yjzha1996@163.com>
Date: Thu, 28 Sep 2023 16:26:53 +0800
Subject: [PATCH] fix: file size calc bug

---
 README.md                           | 2 +-
 plugins/linkai/README.md            | 2 +-
 plugins/linkai/config.json.template | 2 +-
 plugins/linkai/summary.py           | 4 ++--
 4 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index 96e12eb..6d8543e 100644
--- a/README.md
+++ b/README.md
@@ -28,7 +28,7 @@ Demo made by [Visionn](https://www.wangpc.cc/)
 
 # 更新日志
 
->**2023.09.01：** 增加 [企微个人号](https://github.com/zhayujie/chatgpt-on-wechat/pull/1385) 通道，[claude](https://github.com/zhayujie/chatgpt-on-wechat/pull/1382) 模型
+>**2023.09.26：** 插件增加 文件/文章链接 一键总结和对话的功能，使用参考：插件说明
 
 >**2023.08.08：** 接入百度文心一言模型，通过 [插件](https://github.com/zhayujie/chatgpt-on-wechat/tree/master/plugins/linkai) 支持 Midjourney 绘图
 
diff --git a/plugins/linkai/README.md b/plugins/linkai/README.md
index 0ff5699..b41a63b 100644
--- a/plugins/linkai/README.md
+++ b/plugins/linkai/README.md
@@ -25,7 +25,7 @@
     "summary": {
         "enabled": true,              # 文档总结和对话功能开关
         "group_enabled": true,        # 是否支持群聊开启
-        "max_file_size": 10000        # 文件的大小限制，单位KB，默认为10M，超过该大小直接忽略
+        "max_file_size": 5000        # 文件的大小限制，单位KB，默认为5M，超过该大小直接忽略
     }
 }
 ```
diff --git a/plugins/linkai/config.json.template b/plugins/linkai/config.json.template
index 30517e0..ccd8967 100644
--- a/plugins/linkai/config.json.template
+++ b/plugins/linkai/config.json.template
@@ -14,6 +14,6 @@
     "summary": {
         "enabled": true,
         "group_enabled": true,
-        "max_file_size": 15000
+        "max_file_size": 5000
     }
 }
diff --git a/plugins/linkai/summary.py b/plugins/linkai/summary.py
index efa2911..e7f64be 100644
--- a/plugins/linkai/summary.py
+++ b/plugins/linkai/summary.py
@@ -13,7 +13,7 @@ class LinkSummary:
             "file": open(file_path, "rb"),
             "name": file_path.split("/")[-1],
         }
-        res = requests.post(url=self.base_url() + "/v1/summary/file", headers=self.headers(), files=file_body, timeout=(5, 180))
+        res = requests.post(url=self.base_url() + "/v1/summary/file", headers=self.headers(), files=file_body, timeout=(5, 300))
         return self._parse_summary_res(res)
 
     def summary_url(self, url: str):
@@ -68,7 +68,7 @@ class LinkSummary:
 
         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
+            return False
 
         suffix = file_path.split(".")[-1]
         support_list = ["txt", "csv", "docx", "pdf", "md"]