feat: llama3 bot

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong0618
2024-04-19 16:22:43 +08:00
parent cd2185c7d6
commit 450c08ba54
7 changed files with 282 additions and 87 deletions

View File

@ -205,16 +205,20 @@ def chatgpt_photo_handler(message: Message, bot: TeleBot) -> None:
bot_reply_markdown(reply_id, who, "answer wrong", bot)
def register(bot: TeleBot) -> None:
bot.register_message_handler(chatgpt_handler, commands=["gpt"], pass_bot=True)
bot.register_message_handler(chatgpt_handler, regexp="^gpt:", pass_bot=True)
bot.register_message_handler(
chatgpt_pro_handler, commands=["gpt_pro"], pass_bot=True
)
bot.register_message_handler(chatgpt_pro_handler, regexp="^gpt_pro:", pass_bot=True)
bot.register_message_handler(
chatgpt_photo_handler,
content_types=["photo"],
func=lambda m: m.caption and m.caption.startswith(("gpt:", "/gpt")),
pass_bot=True,
)
if CHATGPT_API_KEY:
def register(bot: TeleBot) -> None:
bot.register_message_handler(chatgpt_handler, commands=["gpt"], pass_bot=True)
bot.register_message_handler(chatgpt_handler, regexp="^gpt:", pass_bot=True)
bot.register_message_handler(
chatgpt_pro_handler, commands=["gpt_pro"], pass_bot=True
)
bot.register_message_handler(
chatgpt_pro_handler, regexp="^gpt_pro:", pass_bot=True
)
bot.register_message_handler(
chatgpt_photo_handler,
content_types=["photo"],
func=lambda m: m.caption and m.caption.startswith(("gpt:", "/gpt")),
pass_bot=True,
)