chore: update

xingty · Jun 25, 2024 · bfac84e · bfac84e
1 parent db4998b
commit bfac84e
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 25 deletions.
diff --git a/src/catgpt/commands/chat.py b/src/catgpt/commands/chat.py
@@ -185,32 +185,32 @@ async def do_reply(
                 else:
                     raise ae
 
-    if len(buffered) > 0:
-        delta = timeout - (time.time() - start)
-        if delta > 0:
-            await asyncio.sleep(int(delta) + 1)
-
-        text += buffered
+    # if len(buffered) > 0: for removing the endpoint info from the message
+    delta = timeout - (time.time() - start)
+    if delta > 0:
+        await asyncio.sleep(int(delta) + 1)
+
+    text += buffered
+    msg_text = escape(text)
+    if text_overflow or len(msg_text) > MAX_TEXT_LENGTH:
+        text_overflow = True
         msg_text = escape(text)
-        if text_overflow or len(msg_text) > MAX_TEXT_LENGTH:
-            text_overflow = True
-            msg_text = escape(text)
 
-        msg = await bot.edit_message_text(
-            text=msg_text,
+    msg = await bot.edit_message_text(
+        text=msg_text,
+        chat_id=reply_msg.chat.id,
+        message_id=reply_msg.message_id,
+        parse_mode="MarkdownV2",
+        disable_web_page_preview=True,
+    )
+
+    if text_overflow:
+        await bot.send_message(
             chat_id=reply_msg.chat.id,
-            message_id=reply_msg.message_id,
-            parse_mode="MarkdownV2",
-            disable_web_page_preview=True,
+            text=escape(buffered),
+            reply_to_message_id=msg.message_id,
         )
 
-        if text_overflow:
-            await bot.send_message(
-                chat_id=reply_msg.chat.id,
-                text=escape(buffered),
-                reply_to_message_id=msg.message_id,
-            )
-
     return text
 
 

diff --git a/src/catgpt/storage/tx.py b/src/catgpt/storage/tx.py
@@ -1,6 +1,7 @@
 import enum
 import os
-import time
+
+# import time
 
 from contextvars import ContextVar
 
@@ -60,7 +61,7 @@ def decorator(func):
         async def wrapper(*args, **kwargs):
             token = None
             tx = context.get()
-            start = time.time()
+            # start = time.time()
             if tx is None:
                 if tx_type == "write":
                     conn = await storage.datasource.get_write_conn()
@@ -87,8 +88,8 @@ async def wrapper(*args, **kwargs):
                 print(e)
                 raise e
             finally:
-                end = time.time()
-                print(f"Transaction time: {end} - {start} = {end - start}")
+                # end = time.time()
+                # print(f"Transaction time: {end} - {start} = {end - start}")
                 if token is not None:
                     context.reset(token)
                     tx.close()