74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
import os
|
|
import discord
|
|
from discord.ext import commands
|
|
from dotenv import load_dotenv
|
|
from google import genai
|
|
from google.genai import types
|
|
|
|
# Загружаем переменные из .env
|
|
load_dotenv()
|
|
DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
|
|
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
|
|
|
|
# Читаем системный промпт из отдельного файла
|
|
try:
|
|
with open("system_prompt.txt", "r", encoding="utf-8") as f:
|
|
SYSTEM_PROMPT = f.read().strip()
|
|
except FileNotFoundError:
|
|
SYSTEM_PROMPT = "Ты помощник."
|
|
|
|
# Инициализируем клиент Gemini
|
|
ai_client = genai.Client(api_key=GEMINI_API_KEY)
|
|
|
|
# Настройка интентов Discord
|
|
intents = discord.Intents.default()
|
|
intents.message_content = True
|
|
|
|
bot = commands.Bot(command_prefix="!", intents=intents)
|
|
|
|
@bot.event
|
|
async def on_ready():
|
|
print(f"Logged in as {bot.user.name} (ID: {bot.user.id}) 🚀")
|
|
|
|
@bot.event
|
|
async def on_message(message):
|
|
# Игнорируем сообщения от самого бота
|
|
if message.author == bot.user:
|
|
return
|
|
|
|
# Проверяем пинг бота
|
|
if bot.user in message.mentions:
|
|
# Очищаем пинг из текста
|
|
prompt = message.content.replace(f"<@{bot.user.id}>", "").replace(f"<@!{bot.user.id}>", "").strip()
|
|
|
|
if not prompt:
|
|
await message.reply("вась")
|
|
return
|
|
|
|
async with message.channel.typing():
|
|
try:
|
|
# Передаем загруженный из файла системный промпт
|
|
response = ai_client.models.generate_content(
|
|
model="gemini-2.5-flash",
|
|
contents=prompt,
|
|
config=types.GenerateContentConfig(
|
|
system_instruction=SYSTEM_PROMPT
|
|
)
|
|
)
|
|
|
|
reply_text = response.text
|
|
|
|
# Делим длинные ответы (лимит Discord — 2000 символов)
|
|
if len(reply_text) > 2000:
|
|
for i in range(0, len(reply_text), 2000):
|
|
await message.channel.send(reply_text[i:i+2000])
|
|
else:
|
|
await message.reply(reply_text)
|
|
|
|
except Exception as e:
|
|
print(f"Error communicating with Gemini: {e}")
|
|
await message.reply("damn, something broke trying to reach the AI 😭")
|
|
|
|
await bot.process_commands(message)
|
|
|
|
bot.run(DISCORD_TOKEN) |