cdryzun/tg_bot_collections
1
0
Fork 0
mirror of https://github.com/cdryzun/tg_bot_collections.git synced 2025-11-04 16:56:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update the MarkdownV2 message

Browse Source
This commit is contained in:
Yi Zhao
2024-03-29 23:01:34 +08:00
parent 27444d6fa1
commit 98909312f2
4 changed files with 13 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
handlers/__init__.py
View File

@ -20,6 +20,12 @@ T = TypeVar("T", bound=Callable)
BOT_MESSAGE_LENGTH = 4000 BOT_MESSAGE_LENGTH = 4000
def bot_reply_first(message: Message, who: str, bot: TeleBot) -> Message:
""" Create the first reply message which make user feel the bot is working. """
return bot.reply_to(message,
f"**{who}** is __thinking__ \.\.\.",
parse_mode="MarkdownV2"
)
def bot_reply_markdown(reply_id: Message, who: str, text: str, bot: TeleBot) -> None: def bot_reply_markdown(reply_id: Message, who: str, text: str, bot: TeleBot) -> None:
""" """
@ -144,4 +150,4 @@ def list_available_commands() -> list[str]:
# `import *` will give you these # `import *` will give you these
__all__ = ["bot_reply_markdown", "extract_prompt"] __all__ = ["bot_reply_first", "bot_reply_markdown", "extract_prompt"]

7
handlers/claude.py
View File

@ -6,7 +6,7 @@ from anthropic import Anthropic, APITimeoutError
from telebot import TeleBot from telebot import TeleBot
from telebot.types import Message from telebot.types import Message
from . import bot_reply_markdown from . import *
from telegramify_markdown import convert from telegramify_markdown import convert
from telegramify_markdown.customize import markdown_symbol from telegramify_markdown.customize import markdown_symbol
@ -48,10 +48,7 @@ def claude_handler(message: Message, bot: TeleBot) -> None:
return return
# show something, make it more responsible # show something, make it more responsible
reply_id = bot.reply_to(message, reply_id = bot_reply_first(message, "Claude", bot)
"**Claude** is __thinking__...",
parse_mode="MarkdownV2"
)
player_message.append({"role": "user", "content": m}) player_message.append({"role": "user", "content": m})
# keep the last 5, every has two ask and answer. # keep the last 5, every has two ask and answer.

7
handlers/gemini.py
View File

@ -6,7 +6,7 @@ from google.generativeai.types.generation_types import StopCandidateException
from telebot import TeleBot from telebot import TeleBot
from telebot.types import Message from telebot.types import Message
from . import bot_reply_markdown from . import *
GOOGLE_GEMINI_KEY = environ.get("GOOGLE_GEMINI_KEY") GOOGLE_GEMINI_KEY = environ.get("GOOGLE_GEMINI_KEY")
@ -59,10 +59,7 @@ def gemini_handler(message: Message, bot: TeleBot) -> None:
return return
# show something, make it more responsible # show something, make it more responsible
reply_id = bot.reply_to(message, reply_id = bot_reply_first(message, "Gemini", bot)
"**Gemini** is __thinking__...",
parse_mode="MarkdownV2"
)
# keep the last 5, every has two ask and answer. # keep the last 5, every has two ask and answer.
if len(player.history) > 10: if len(player.history) > 10:

7
handlers/yi.py
View File

@ -7,7 +7,7 @@ import requests
from telebot import TeleBot from telebot import TeleBot
from telebot.types import Message from telebot.types import Message
from . import bot_reply_markdown from . import *
YI_BASE_URL = environ.get("YI_BASE_URL") YI_BASE_URL = environ.get("YI_BASE_URL")
YI_API_KEY = environ.get("YI_API_KEY") YI_API_KEY = environ.get("YI_API_KEY")
@ -45,10 +45,7 @@ def yi_handler(message: Message, bot: TeleBot) -> None:
return return
# show something, make it more responsible # show something, make it more responsible
reply_id = bot.reply_to(message, reply_id = bot_reply_first(message, "Yi", bot)
"**Yi** is __thinking__...",
parse_mode="MarkdownV2"
)
player_message.append({"role": "user", "content": m}) player_message.append({"role": "user", "content": m})
# keep the last 5, every has two ask and answer. # keep the last 5, every has two ask and answer.