kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-07-21 22:27:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Linkai bot supports more file types.

Browse Source
This commit is contained in:
6vision
2024-07-15 18:57:58 +08:00
parent 2711fa1b1b
commit 993853757b
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
bot/linkai/link_ai_bot.py
View File

@@ -399,6 +399,7 @@ class LinkAIBot(Bot):
return return
max_send_num = conf().get("max_media_send_count") max_send_num = conf().get("max_media_send_count")
send_interval = conf().get("media_send_interval") send_interval = conf().get("media_send_interval")
file_type = (".pdf", ".doc", ".docx", ".csv", ".xls", ".xlsx", ".txt", ".rtf", ".ppt", ".pptx")
try: try:
i = 0 i = 0
for url in image_urls: for url in image_urls:
@@ -407,7 +408,7 @@ class LinkAIBot(Bot):
i += 1 i += 1
if url.endswith(".mp4"): if url.endswith(".mp4"):
reply_type = ReplyType.VIDEO_URL reply_type = ReplyType.VIDEO_URL
elif url.endswith(".pdf") or url.endswith(".doc") or url.endswith(".docx") or url.endswith(".csv"): elif url.endswith(file_type):
reply_type = ReplyType.FILE reply_type = ReplyType.FILE
url = _download_file(url) url = _download_file(url)
if not url: if not url: