mirror of
https://gitflic.ru/project/maks1ms/ocab.git
synced 2024-12-23 16:23:01 +03:00
Множественные улучшения.
This commit is contained in:
armatik
parent
e197e81beb
commit
3e5a3f2f74
@ -59,9 +59,6 @@ def openai_response(message_formated_text):
|
||||
#запись ошибки в лог с указанием времени и даты
|
||||
with open(os.path.join(mother_path, 'src/OpenAI/GPT35turbo/log.txt'), 'a') as log_file:
|
||||
log_file.write('\n' + time.strftime("%d.%m.%Y %H:%M:%S") + ' ' + str(ex))
|
||||
#Проверка на то что ответ не содержит ошибку
|
||||
count_length = 0
|
||||
|
||||
return response
|
||||
|
||||
def sort_message_from_user(message_formated_text, message_id):
|
||||
@ -137,4 +134,65 @@ def openai_message_processing(message_id):
|
||||
else:
|
||||
return f"Сообщение слишком длинное, максимальная длина сообщения \
|
||||
{max_token_count - len(message_formated_text[0]['content'])} символов, укоротите его на \
|
||||
{len(str(cursor.execute('SELECT message_text FROM message_list WHERE message_id'))) - max_token_count} символов"
|
||||
{len(str(cursor.execute('SELECT message_text FROM message_list WHERE message_id'))) - max_token_count} символов"
|
||||
|
||||
def openai_collecting_history_context(start_id, end_id, message_formated_text, message_id = 0):
|
||||
# собираем список сообщений для OpenAI длинной до 14500 символов начиная с end_id и заканчивая start_id
|
||||
if message_id == 0:
|
||||
message_id = end_id
|
||||
message_formated_text = openai_collecting_history_context(start_id, end_id, message_formated_text, message_id)
|
||||
elif message_id > start_id:
|
||||
message_formated_text = openai_collecting_history_context(start_id, end_id, message_formated_text, (message_id - 1))
|
||||
try:
|
||||
message_formated_text.append({
|
||||
"role": "user",
|
||||
"content": str(*(cursor.execute("SELECT message_text FROM message_list WHERE message_id = ?",
|
||||
(message_id,)).fetchone()))
|
||||
})
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
return message_formated_text
|
||||
|
||||
def openai_message_history_processing(start_id, end_id):
|
||||
message_formated_text = []
|
||||
message_formated_text = openai_collecting_history_context(start_id, end_id, message_formated_text)
|
||||
|
||||
max_token_count_history = 15000
|
||||
min_token_for_answer_history = 1500
|
||||
count_length = 0
|
||||
try:
|
||||
for message in message_formated_text:
|
||||
count_length += len(message['content'])
|
||||
while count_length > max_token_count_history - min_token_for_answer_history:
|
||||
message_formated_text.pop(1)
|
||||
count_length = 0
|
||||
for message in message_formated_text:
|
||||
count_length += len(message['content'])
|
||||
except IndexError:
|
||||
message_formated_text = [
|
||||
{
|
||||
"role": "system",
|
||||
"content": "Выведи сообщение об ошибке."
|
||||
}
|
||||
]
|
||||
message_formated_text.append({
|
||||
"role": "user",
|
||||
"content": "Сделай пересказ всей истории сообщений без потери смысла от 3-его лица."
|
||||
})
|
||||
try:
|
||||
response = openai.ChatCompletion.create(
|
||||
model="gpt-3.5-turbo-16k",
|
||||
messages=message_formated_text,
|
||||
max_tokens=max_token_count_history - count_length
|
||||
)
|
||||
except OpenAIError as ex:
|
||||
if 'on requests per min. Limit: 3 / min. Please try again' in str(ex):
|
||||
response = ('Извини мой процессор перегрелся, дай мне минутку отдохнуть')
|
||||
elif 'Bad gateway.' in str(ex):
|
||||
response = (
|
||||
'Ой, где я? Кажется кто то перерзал мой интернет кабель, подожди немного пока я его починю')
|
||||
# запись ошибки в лог с указанием времени и даты
|
||||
with open(os.path.join(mother_path, 'src/OpenAI/GPT35turbo/log.txt'), 'a') as log_file:
|
||||
log_file.write('\n' + time.strftime("%d.%m.%Y %H:%M:%S") + ' ' + str(ex))
|
||||
return response
|
||||
@ -62,6 +62,7 @@ cursor.execute("""CREATE TABLE IF NOT EXISTS user_list (
|
||||
user_name TEXT NOT NULL,
|
||||
user_role INTEGER,
|
||||
user_stats INTEGER
|
||||
user_rep INTEGER
|
||||
)""")
|
||||
#запись информации о чате в базу данных
|
||||
|
||||
@ -69,7 +70,6 @@ async def empty_role(id):
|
||||
cursor.execute("UPDATE user_list SET user_role = ? WHERE user_id = ?", (0, id))
|
||||
database.commit()
|
||||
|
||||
|
||||
async def check(id):
|
||||
#проверка что у человека есть роль
|
||||
user_role = cursor.execute("SELECT user_role FROM user_list WHERE user_id = ?", (id,)).fetchone()[0]
|
||||
@ -92,6 +92,7 @@ async def get_role(id):
|
||||
user_role = cursor.execute("SELECT user_role FROM user_list WHERE user_id = ?", (id,)).fetchone()[0]
|
||||
return await get_role_name(user_role)
|
||||
|
||||
|
||||
async def check_admin(id, chat_id):
|
||||
#Проверка что человек есть в списке администраторов чата
|
||||
chat_admins = await bot.get_chat_administrators(chat_id)
|
||||
@ -167,7 +168,7 @@ async def mute(message: types.Message):
|
||||
#получаем id отправителя сообщение на которое отвечает message
|
||||
target_id = message.reply_to_message.from_user.id
|
||||
target_name = cursor.execute("SELECT user_name FROM user_list WHERE user_id = ?", (target_id,)).fetchone()[0]
|
||||
#Проверка что человек пользователь
|
||||
#Проверка, что человек пользователь
|
||||
if cursor.execute("SELECT user_role FROM user_list WHERE user_id = ?", (target_id,)).fetchone()[0] == 0:
|
||||
#ограничения прав пользователя по отправке сообщений на time секунд
|
||||
time_sec = await time_to_seconds(message.text.split(' ')[1])
|
||||
@ -247,20 +248,6 @@ async def new_chat_members(message: types.Message):
|
||||
#Заносим пользователя в базу данных
|
||||
cursor.execute("INSERT INTO user_list VALUES (?, ?, ?, ?)", (message.new_chat_members[0].id, message.new_chat_members[0].username, 0, 0))
|
||||
|
||||
# @dp.message_handler(commands=['mail'])
|
||||
# async def mail(message: types.Message):
|
||||
# #получаем тег пользователя которому должны написать в лс
|
||||
# target_tag = message.text.split(' ')[1]
|
||||
# target_tag = target_tag[1:]
|
||||
# print(target_tag)
|
||||
# #получаем id пользователя которому должны написать в лс
|
||||
# target_id = int(cursor.execute("SELECT user_id FROM user_list WHERE user_name = ?", (target_tag,)).fetchone()[0])
|
||||
# print(target_id)
|
||||
# #пишем пользователю в лс, привет {его имя}
|
||||
# await bot.send_message(target_id, f"Привет {target_tag}! Я бот чата {message.chat.title}. {message.from_user.first_name} хочет с тобой связаться. Напиши ему в лс.")
|
||||
# #пишем пользователю в чат, что его сообщение отправлено
|
||||
# await message.reply(f"Сообщение отправлено пользователю {target_tag}")
|
||||
|
||||
|
||||
@dp.message_handler(commands=['chatinfo'])
|
||||
async def chat_info(message: types.Message):
|
||||
@ -273,11 +260,12 @@ async def chat_info(message: types.Message):
|
||||
chat_moderators = cursor.execute("SELECT COUNT(user_id) FROM user_list WHERE user_role = 1").fetchone()[0]
|
||||
chat_messages = cursor.execute("SELECT chat_stats FROM chat_list WHERE chat_id = ?", (chat_id,)).fetchone()[0]
|
||||
await message.reply(
|
||||
f"Название чата: {chat_title}\n"
|
||||
f"Кол-во пользователей: {chat_members}\n"
|
||||
f"Кол-во администраторов: {len(chat_admins)}\n"
|
||||
f"Кол-во модераторов: {chat_moderators}\n"
|
||||
f"Кол-во сообщений: {chat_messages}")
|
||||
f"<b>Название чата:</b> {chat_title}\n"
|
||||
f"<b>Количество пользователей:</b> {chat_members}\n"
|
||||
f"<b>Количество администраторов:</b> {len(chat_admins)}\n"
|
||||
f"<b>Количество модераторов:</b> {chat_moderators}\n"
|
||||
f"<b>Количество сообщений:</b> {chat_messages}",
|
||||
parse_mode="HTML")
|
||||
await top10(message)
|
||||
else:
|
||||
await message.reply(
|
||||
@ -302,12 +290,18 @@ async def start(message: types.Message):
|
||||
# Проверка наличия столбца имени пользователя в базе данных, если его нет, то добавить.
|
||||
try:
|
||||
cursor.execute("SELECT user_name FROM user_list")
|
||||
# sqlite3.OperationalError: no such column: user_name
|
||||
except sqlite3.OperationalError:
|
||||
cursor.execute("ALTER TABLE user_list ADD COLUMN user_name TEXT")
|
||||
database.commit()
|
||||
await message.reply("База данных пользователей реструктурирована.")
|
||||
# Проверка что все участники чата есть в базе данных
|
||||
# Проверка наличия столбца репутации пользователя в базе данных, если его нет, то добавить.
|
||||
try:
|
||||
cursor.execute("SELECT user_rep FROM user_list")
|
||||
except sqlite3.OperationalError:
|
||||
cursor.execute("ALTER TABLE user_list ADD COLUMN user_rep INTEGER")
|
||||
database.commit()
|
||||
await message.reply("База данных пользователей реструктурирована.")
|
||||
|
||||
|
||||
|
||||
@dp.message_handler(commands=['top10'])
|
||||
@ -323,17 +317,6 @@ async def top10(message: types.Message):
|
||||
f"{config['Telegram']['top10_answer']}\n{top10_message}")
|
||||
|
||||
|
||||
@dp.message_handler(commands=['stats'])
|
||||
async def stat(message: types.Message):
|
||||
#статистика пользователя в формате: Имя пользователя - количество сообщений
|
||||
user_stats = cursor.execute("SELECT user_stats FROM user_list WHERE user_id = ?", (message.from_user.id,)).fetchone()
|
||||
your_id = message.from_id
|
||||
your_name = message.from_user.username
|
||||
await message.reply(
|
||||
f"[{your_name}](tg://user?id={str(your_id)}) - {user_stats[0]}",
|
||||
parse_mode="Markdown")
|
||||
|
||||
|
||||
@dp.message_handler(commands=['aboutme'])
|
||||
async def aboutme(message: types.Message):
|
||||
your_id = message.from_id
|
||||
@ -341,6 +324,10 @@ async def aboutme(message: types.Message):
|
||||
your_name = message.from_user.username
|
||||
user_stats = cursor.execute("SELECT user_stats FROM user_list WHERE user_id = ?", (message.from_user.id,)).fetchone()
|
||||
rolenum = cursor.execute("SELECT user_role FROM user_list WHERE user_id = ?", (message.from_user.id,)).fetchone()
|
||||
user_rep = cursor.execute("SELECT user_rep FROM user_list WHERE user_id = ?", (message.from_user.id,)).fetchone()
|
||||
if "None" in str(user_rep):
|
||||
user_rep = 0
|
||||
cursor.execute("UPDATE user_list SET user_rep = ? WHERE user_id = ?", (user_rep, your_id))
|
||||
role = await get_role_name(rolenum[0])
|
||||
#Имя пользователя на следующей строке статистика сообщений на следующей строке права пользователя
|
||||
#если пользователь есть в списке администраторов чата, то выдаём роль администратор
|
||||
@ -352,39 +339,43 @@ async def aboutme(message: types.Message):
|
||||
if admin.user.id == your_id:
|
||||
if default_for_admin == 0:
|
||||
await message.reply(
|
||||
f"Вы обнаружены в списке администраторов чата! \n"
|
||||
f"Вам будет выдана роль: {config['Roles']['user']}\n"
|
||||
f"Имя: [{your_name}](tg://user?id={str(your_id)})\n"
|
||||
f"Кол-во сообщений: {user_stats[0]}\n"
|
||||
f"Роль: {role}",
|
||||
parse_mode="Markdown")
|
||||
f"<b>Вы обнаружены в списке администраторов чата!</b> \n"
|
||||
f"<b>Вам будет выдана роль:</b> {config['Roles']['user']}\n"
|
||||
f"<b>Имя:</b> <a href=\"tg://user?id={str(your_id)}\">{your_name}</a>\n"
|
||||
f"<b>Кол-во сообщений:</b> {user_stats[0]}\n"
|
||||
f"<b>Репутация:</b> {user_rep[0]}\n"
|
||||
f"<b>Роль:</b> {role}",
|
||||
parse_mode="HTML")
|
||||
no_answer = True
|
||||
elif default_for_admin == 1:
|
||||
await message.reply(
|
||||
f"Вы обнаружены в списке администраторов чата.\n"
|
||||
f"Вам будет выдана роль: {config['Roles']['moderator']}\n"
|
||||
f"Имя: [{your_name}](tg://user?id={str(your_id)})\n"
|
||||
f"Кол-во сообщений: {user_stats[0]}\n"
|
||||
f"Роль: {role}",
|
||||
parse_mode="Markdown")
|
||||
f"<b>Вы обнаружены в списке администраторов чата!</b> \n"
|
||||
f"<b>Вам будет выдана роль:</b> {config['Roles']['moderator']}\n"
|
||||
f"<b>Имя:</b> <a href=\"tg://user?id={str(your_id)}\">{your_name}</a>\n"
|
||||
f"<b>Кол-во сообщений:</b> {user_stats[0]}\n"
|
||||
f"<b>Репутация:</b> {user_rep[0]}\n"
|
||||
f"<b>Роль:</b> {role}",
|
||||
parse_mode="HTML")
|
||||
no_answer = True
|
||||
elif default_for_admin == 2:
|
||||
await message.reply(
|
||||
f"Вы обнаружены в списке администраторов чата.\n"
|
||||
f"Вам будет выдана роль: {config['Roles']['admin']}\n"
|
||||
f"Имя: [{your_name}](tg://user?id={str(your_id)})\n"
|
||||
f"Кол-во сообщений: {user_stats[0]}\n"
|
||||
f"Роль: {role}",
|
||||
parse_mode="Markdown")
|
||||
f"<b>Вы обнаружены в списке администраторов чата!</b> \n"
|
||||
f"<b>Вам будет выдана роль:</b> {config['Roles']['admin']}\n"
|
||||
f"<b>Имя:</b> <a href=\"tg://user?id={str(your_id)}\">{your_name}</a>\n"
|
||||
f"<b>Кол-во сообщений:</b> {user_stats[0]}\n"
|
||||
f"<b>Репутация:</b> {user_rep[0]}\n"
|
||||
f"<b>Роль:</b> {role}",
|
||||
parse_mode="HTML")
|
||||
no_answer = True
|
||||
cursor.execute("UPDATE user_list SET user_role = ? WHERE user_id = ?", (default_for_admin, your_id))
|
||||
database.commit()
|
||||
if no_answer == False:
|
||||
await message.reply(
|
||||
f"Имя: [{your_name}](tg://user?id={str(your_id)})\n"
|
||||
f"Кол-во сообщений: {user_stats[0]}\n"
|
||||
f"Роль: {role}",
|
||||
parse_mode="Markdown")
|
||||
f"<b>Имя:</b> <a href=\"tg://user?id={str(your_id)}\">{your_name}</a>\n"
|
||||
f"<b>Кол-во сообщений:</b> {user_stats[0]}\n"
|
||||
f"<b>Репутация:</b> {user_rep[0]}\n"
|
||||
f"<b>Роль:</b> {role}",
|
||||
parse_mode="HTML")
|
||||
|
||||
@dp.message_handler(commands=['setrole'])
|
||||
async def setrole(message: types.Message):
|
||||
@ -439,6 +430,10 @@ async def about(message: types.Message):
|
||||
target_name = target_name[1:]
|
||||
target_id = cursor.execute("SELECT user_id FROM user_list WHERE user_name = ?", (target_name,)).fetchone()
|
||||
target_id = target_id[0]
|
||||
user_rep = cursor.execute("SELECT user_rep FROM user_list WHERE user_id = ?", (target_id,)).fetchone()[0]
|
||||
if "None" in str(user_rep):
|
||||
user_rep = 0
|
||||
cursor.execute("UPDATE user_list SET user_rep = ? WHERE user_id = ?", (user_rep, target_id))
|
||||
await check(target_id)
|
||||
except:
|
||||
await message.reply(
|
||||
@ -448,13 +443,43 @@ async def about(message: types.Message):
|
||||
user_role = cursor.execute("SELECT user_role FROM user_list WHERE user_id = ?", (target_id,)).fetchone()[0]
|
||||
user_role = await get_role_name(user_role)
|
||||
await message.reply(
|
||||
f"Имя: {target_name}\n"
|
||||
f"Кол-во сообщений: {user_stats}\n"
|
||||
f"Роль: {user_role}")
|
||||
f"<b>Имя:</b> {target_name}\n"
|
||||
f"<b>Кол-во сообщений:</b> {user_stats}\n"
|
||||
f"<b>Репутация:</b> {user_rep}\n"
|
||||
f"<b>Роль:</b> {user_role}", parse_mode="HTML")
|
||||
else:
|
||||
await message.reply(
|
||||
f"Ошибка! Проверьте правильность написания команды.")
|
||||
|
||||
from src.OpenAI.GPT35turbo.OA_processing import openai_message_history_processing
|
||||
|
||||
@dp.message_handler(commands=['history'])
|
||||
async def history(message: types.Message):
|
||||
if message.reply_to_message is not None:
|
||||
# Получаем id сообщения на которое отвечает message
|
||||
start_message_id = message.reply_to_message.message_id
|
||||
# Получаем id сообщения message
|
||||
end_message_id = message.message_id
|
||||
elif message.text.split()[1] != '':
|
||||
end_message_id = message.message_id
|
||||
start_message_id = end_message_id - int(message.text.split()[1])
|
||||
else:
|
||||
await message.reply("Ошибка! Проверьте правильность написания команды.")
|
||||
return
|
||||
temp_message = await message.reply("Подождите немного, я обрабатываю историю сообщений... Функция ЭКСПЕРЕМЕНТАЛЬНАЯ "
|
||||
"и может работать некорректно.")
|
||||
# Получаем историю сообщений
|
||||
response = openai_message_history_processing(start_message_id, end_message_id)
|
||||
if response is None:
|
||||
await message.reply("Я не понял тебя, попробуй перефразировать")
|
||||
else:
|
||||
bot_message_id = await message.reply(response['choices'][0]['message']['content'], parse_mode="markdown")
|
||||
# заносим сообщение в базу данных в качестве message_id мы пишем id сообщения в bot_message_id
|
||||
cursor.execute("INSERT INTO message_list VALUES (?, ?, ?, ?)",
|
||||
(bot_message_id.message_id, response['choices'][0]['message']['content'], 0, message.message_id))
|
||||
# очищаем сообщение
|
||||
asyncio.create_task(delete_message(temp_message, 0))
|
||||
|
||||
from src.OpenAI.GPT35turbo.OA_processing import openai_message_processing
|
||||
|
||||
|
||||
@ -493,6 +518,18 @@ async def in_message(message: types.Message):
|
||||
send_answer = True
|
||||
typing_mode = False
|
||||
|
||||
if send_answer == False:
|
||||
# Если сообщение отвечает на другое сообщение, то проверяем наличие слова спасибо
|
||||
if message.reply_to_message is not None:
|
||||
if message.text.startswith("Спасибо"):
|
||||
target_id = message.reply_to_message.from_user.id
|
||||
target_name = cursor.execute("SELECT user_name FROM user_list WHERE user_id = ?", (target_id,)).fetchone()[0]
|
||||
cursor.execute("UPDATE user_list SET user_rep = user_rep + 1 WHERE user_id = ?", (target_id,))
|
||||
database.commit()
|
||||
await message.reply(
|
||||
f"Спасибо, [{target_name}](tg://user?id={str(target_id)}) за ваш ответ!",
|
||||
parse_mode="Markdown")
|
||||
|
||||
if send_answer:
|
||||
if typing_mode is False:
|
||||
your_id = message.from_id
|
||||
|
||||
Loading…
Reference in New Issue
Block a user