fix: file size calc bug

zhayujie · Sep 28, 2023 · 2fa1df2 · 2fa1df2
1 parent b45eea5
commit 2fa1df2
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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
@@ -13,7 +13,7 @@ def summary_file(self, file_path: str):
             "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 @@ def check_file(self, file_path: str, sum_config: dict) -> bool:
 
         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"]