From 7e0fbd600fcc70c57dfef7d350d857fef41b4048 Mon Sep 17 00:00:00 2001
From: zhayujie <yjzha1996@163.com>
Date: Mon, 29 Jan 2024 11:46:00 +0800
Subject: [PATCH] feat: add media send limit and interval

---
 bot/linkai/link_ai_bot.py | 8 ++++++++
 config.py                 | 4 +++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/bot/linkai/link_ai_bot.py b/bot/linkai/link_ai_bot.py
index c93d5c4..efd571a 100644
--- a/bot/linkai/link_ai_bot.py
+++ b/bot/linkai/link_ai_bot.py
@@ -386,8 +386,14 @@ class LinkAIBot(Bot):
     def _send_image(self, channel, context, image_urls):
         if not image_urls:
             return
+        max_send_num = conf().get("max_media_send_count")
+        send_interval = conf().get("media_send_interval")
         try:
+            i = 0
             for url in image_urls:
+                if max_send_num and i >= max_send_num:
+                    continue
+                i += 1
                 if url.endswith(".mp4"):
                     reply_type = ReplyType.VIDEO_URL
                 elif url.endswith(".pdf") or url.endswith(".doc") or url.endswith(".docx"):
@@ -399,6 +405,8 @@ class LinkAIBot(Bot):
                     reply_type = ReplyType.IMAGE_URL
                 reply = Reply(reply_type, url)
                 channel.send(reply, context)
+                if send_interval:
+                    time.sleep(send_interval)
         except Exception as e:
             logger.error(e)
 
diff --git a/config.py b/config.py
index f17db3e..c8068a2 100644
--- a/config.py
+++ b/config.py
@@ -148,7 +148,9 @@ available_setting = {
     "plugin_trigger_prefix": "$",  # 规范插件提供聊天相关指令的前缀，建议不要和管理员指令前缀"#"冲突
     # 是否使用全局插件配置
     "use_global_plugin_config": False,
-    # 知识库平台配置
+    "max_media_send_count": 3,     # 单次最大发送媒体资源的个数
+    "media_send_interval": 1,  # 发送图片的事件间隔，单位秒
+    # LinkAI平台配置
     "use_linkai": False,
     "linkai_api_key": "",
     "linkai_app_code": "",