From 93a3e001c88f455884995f5362ac349b6e7759f6 Mon Sep 17 00:00:00 2001
From: yihong0618 <zouzou0208@gmail.com>
Date: Mon, 18 Mar 2024 15:16:50 +0800
Subject: [PATCH] fix: stream end

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
---
 handlers/claude.py | 11 +++++++++++
 tg.py              |  2 +-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/handlers/claude.py b/handlers/claude.py
index 1aaf141..254acb5 100644
--- a/handlers/claude.py
+++ b/handlers/claude.py
@@ -155,6 +155,17 @@ def claude_pro_handler(message: Message, bot: TeleBot) -> None:
                         )
                     except:
                         pass
+        try:
+            # maybe not complete
+            # maybe the same message
+            bot.edit_message_text(
+                message_id=reply_id.message_id,
+                chat_id=reply_id.chat.id,
+                text=convert(s),
+                parse_mode="MarkdownV2",
+            )
+        except:
+            pass
 
         player_message.append(
             {
diff --git a/tg.py b/tg.py
index 528ba88..427ca29 100644
--- a/tg.py
+++ b/tg.py
@@ -31,7 +31,7 @@ def main():
 
     # Start bot
     print("Starting tg collections bot.")
-    bot.infinity_polling(timeout=10, long_polling_timeout=5)
+    bot.infinity_polling(timeout=40, long_polling_timeout=20)
 
 
 if __name__ == "__main__":