cdryzun/tg_bot_collections
1
0
Fork 0
mirror of https://github.com/cdryzun/tg_bot_collections.git synced 2025-05-05 13:18:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request from F4ria/chatgpt

Browse Source 
feat: support ChatGPT
This commit is contained in:
yihong 2024-04-18 09:26:31 +08:00 committed by GitHub
commit c8725e27a1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 248 additions and 11 deletions

13
README.md

@ -41,6 +41,15 @@ Note, if you are using third party service, you need to `export ANTHROPIC_BASE_U
![image](https://github.com/yihong0618/tg_bot_collections/assets/15976103/11d96dde-447b-4b7e-886d-c3564e27b0d6)
## Bot -> ChatGPT
1. visit https://platform.openai.com/account/api-keys get the key
2. export OPENAI_API_KEY=${the_key}
3. use `chatgpt: ${message}` to ask
Note, if you are using third party service, you need to `export OPENAI_API_BASE=${the_url}` to change the url.
## HOW TO Install and Run
### Manually install
@ -58,7 +67,8 @@ Note, if you are using third party service, you need to `export ANTHROPIC_BASE_U
`docker run -d --name tg_bot_collections -e ANTHROPIC_API_KEY='${ANTHROPIC_API_KEY}' -e TELEGRAM_BOT_TOKEN='${TELEGRAM_BOT_TOKEN}' --network host tg_bot_collections`
#### Run lingyiwanwu
`docker run -d --name tg_bot_collections -e YI_API_KEY='${YI_API_KEY}' -e YI_BASE_URL='${YI_BASE_URL}' -e TELEGRAM_BOT_TOKEN='${TELEGRAM_BOT_TOKEN}' --network host tg_bot_collections`
#### Run ChatGPT
`docker run -d --name tg_bot_collections -e OPENAI_API_KEY='${CHATGPT_API_KEY}' -e TELEGRAM_BOT_TOKEN='${TELEGRAM_BOT_TOKEN}' --network host tg_bot_collections`
### Run as shell
@ -86,6 +96,7 @@ Note, this may break your system config -> check this https://github.com/yihong0
- Gemini use -> https://github.com/google/generative-ai-python
- Telegram markdownV2 change code copy from https://github.com/yym68686/md2tgmd/blob/main/src/md2tgmd.py thanks a lot.
- Telegram markdownV2 change to telegramify-markdown
- ChatGPT use -> https://github.com/openai/openai-python
## Appreciation

14
handlers/__init__.py

@ -1,5 +1,6 @@
from __future__ import annotations
import base64
import importlib
import re
import traceback
@ -186,5 +187,16 @@ def enrich_text_with_urls(text: str) -> str:
return text
def image_to_data_uri(file_path):
with open(file_path, "rb") as image_file:
encoded_image = base64.b64encode(image_file.read()).decode("utf-8")
return f"data:image/png;base64,{encoded_image}"
# `import *` will give you these
__all__ = ["bot_reply_first", "bot_reply_markdown", "enrich_text_with_urls"]
__all__ = [
"bot_reply_first",
"bot_reply_markdown",
"enrich_text_with_urls",
"image_to_data_uri",
]

222
handlers/chatgpt.py Normal file

@ -0,0 +1,222 @@
from os import environ
import time
from openai import OpenAI
from telebot import TeleBot
from telebot.types import Message
from . import *
from telegramify_markdown import convert
from telegramify_markdown.customize import markdown_symbol
markdown_symbol.head_level_1 = "📌" # If you want, Customizing the head level 1 symbol
markdown_symbol.link = "🔗" # If you want, Customizing the link symbol
CHATGPT_API_KEY = environ.get("OPENAI_API_KEY")
CHATGPT_BASE_URL = environ.get("OPENAI_API_BASE") or "https://api.openai.com/v1"
CHATGPT_MODEL = "gpt-3.5-turbo"
CHATGPT_PRO_MODEL = "gpt-4-turbo"
client = OpenAI(api_key=CHATGPT_API_KEY, base_url=CHATGPT_BASE_URL, timeout=20)
# Global history cache
chatgpt_player_dict = {}
chatgpt_pro_player_dict = {}
def chatgpt_handler(message: Message, bot: TeleBot) -> None:
"""chatgpt : /chatgpt <question>"""
m = message.text.strip()
player_message = []
# restart will lose all TODO
if str(message.from_user.id) not in chatgpt_player_dict:
chatgpt_player_dict[str(message.from_user.id)] = (
player_message # for the imuutable list
)
else:
player_message = chatgpt_player_dict[str(message.from_user.id)]
if m.strip() == "clear":
bot.reply_to(
message,
"just clear your chatgpt messages history",
)
player_message.clear()
return
if m[:4].lower() == "new ":
m = m[4:].strip()
player_message.clear()
m = enrich_text_with_urls(m)
who = "ChatGPT"
# show something, make it more responsible
reply_id = bot_reply_first(message, who, bot)
player_message.append({"role": "user", "content": m})
# keep the last 5, every has two ask and answer.
if len(player_message) > 10:
player_message = player_message[2:]
chatgpt_reply_text = ""
try:
r = client.chat.completions.create(
messages=player_message, max_tokens=1024, model=CHATGPT_MODEL
)
content = r.choices[0].message.content.encode("utf8").decode()
if not content:
chatgpt_reply_text = f"{who} did not answer."
player_message.pop()
else:
chatgpt_reply_text = content
player_message.append(
{
"role": "assistant",
"content": chatgpt_reply_text,
}
)
except Exception as e:
print(e)
bot_reply_markdown(reply_id, who, "answer wrong", bot)
# pop my user
player_message.pop()
return
# reply back as Markdown and fallback to plain text if failed.
bot_reply_markdown(reply_id, who, chatgpt_reply_text, bot)
def chatgpt_pro_handler(message: Message, bot: TeleBot) -> None:
"""chatgpt_pro : /chatgpt_pro <question>"""
m = message.text.strip()
player_message = []
# restart will lose all TODO
if str(message.from_user.id) not in chatgpt_player_dict:
chatgpt_player_dict[str(message.from_user.id)] = (
player_message # for the imuutable list
)
else:
player_message = chatgpt_player_dict[str(message.from_user.id)]
if m.strip() == "clear":
bot.reply_to(
message,
"just clear your chatgpt messages history",
)
player_message.clear()
return
if m[:4].lower() == "new ":
m = m[4:].strip()
player_message.clear()
m = enrich_text_with_urls(m)
who = "ChatGPT Pro"
reply_id = bot_reply_first(message, who, bot)
player_message.append({"role": "user", "content": m})
# keep the last 5, every has two ask and answer.
if len(player_message) > 10:
player_message = player_message[2:]
try:
r = client.chat.completions.create(
messages=player_message,
max_tokens=2048,
model=CHATGPT_PRO_MODEL,
stream=True,
)
s = ""
start = time.time()
for chunk in r:
if chunk.choices[0].delta.content is None:
break
s += chunk.choices[0].delta.content
if time.time() - start > 2.0:
start = time.time()
bot_reply_markdown(reply_id, who, s, bot, split_text=False)
if not bot_reply_markdown(reply_id, who, s, bot):
# maybe not complete
# maybe the same message
player_message.clear()
return
player_message.append(
{
"role": "assistant",
"content": convert(s),
}
)
except Exception as e:
print(e)
bot_reply_markdown(reply_id, who, "answer wrong", bot)
player_message.clear()
return
def chatgpt_photo_handler(message: Message, bot: TeleBot) -> None:
s = message.caption
prompt = s.strip()
who = "ChatGPT Vision"
# show something, make it more responsible
reply_id = bot_reply_first(message, who, bot)
# get the high quaility picture.
max_size_photo = max(message.photo, key=lambda p: p.file_size)
file_path = bot.get_file(max_size_photo.file_id).file_path
downloaded_file = bot.download_file(file_path)
with open("chatgpt_temp.jpg", "wb") as temp_file:
temp_file.write(downloaded_file)
try:
r = client.chat.completions.create(
max_tokens=2048,
messages=[
{
"role": "user",
"content": [
{"type": "text", "text": prompt},
{
"type": "image_url",
"image_url": {"url": image_to_data_uri("chatgpt_temp.jpg")},
},
],
}
],
model=CHATGPT_PRO_MODEL,
stream=True,
)
s = ""
start = time.time()
for chunk in r:
if chunk.choices[0].delta.content is None:
break
s += chunk.choices[0].delta.content
if time.time() - start > 2.0:
start = time.time()
bot_reply_markdown(reply_id, who, s, bot, split_text=False)
bot_reply_markdown(reply_id, who, s, bot)
except Exception as e:
print(e)
bot_reply_markdown(reply_id, who, "answer wrong", bot)
def register(bot: TeleBot) -> None:
bot.register_message_handler(chatgpt_handler, commands=["chatgpt"], pass_bot=True)
bot.register_message_handler(chatgpt_handler, regexp="^chatgpt:", pass_bot=True)
bot.register_message_handler(
chatgpt_pro_handler, commands=["chatgpt_pro"], pass_bot=True
)
bot.register_message_handler(
chatgpt_pro_handler, regexp="^chatgpt_pro:", pass_bot=True
)
bot.register_message_handler(
chatgpt_photo_handler,
content_types=["photo"],
func=lambda m: m.caption and m.caption.startswith(("chatgpt:", "/chatgpt")),
pass_bot=True,
)

10
handlers/yi.py

@ -1,6 +1,4 @@
import base64
from os import environ
from pathlib import Path
from openai import OpenAI
import requests
@ -88,12 +86,6 @@ def yi_handler(message: Message, bot: TeleBot) -> None:
bot_reply_markdown(reply_id, who, yi_reply_text, bot)
def _image_to_data_uri(file_path):
with open(file_path, "rb") as image_file:
encoded_image = base64.b64encode(image_file.read()).decode("utf-8")
return f"data:image/png;base64,{encoded_image}"
def yi_photo_handler(message: Message, bot: TeleBot) -> None:
s = message.caption
prompt = s.strip()
@ -121,7 +113,7 @@ def yi_photo_handler(message: Message, bot: TeleBot) -> None:
{"type": "text", "text": prompt},
{
"type": "image_url",
"image_url": {"url": _image_to_data_uri("yi_temp.jpg")},
"image_url": {"url": image_to_data_uri("yi_temp.jpg")},
},
],
}