79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
from google import genai
|
|
from google.genai import types
|
|
from dotenv import load_dotenv
|
|
from discord import app_commands
|
|
from discord.ext import commands
|
|
from tools import searxng, open_url, run_command
|
|
import os
|
|
import io
|
|
import discord
|
|
|
|
load_dotenv()
|
|
|
|
client = genai.Client(api_key=os.getenv("GEM_API_KEY"))
|
|
config = types.GenerateContentConfig(
|
|
tools=[searxng, open_url, run_command]
|
|
)
|
|
intents = discord.Intents.default()
|
|
intents.message_content = True
|
|
bot = commands.Bot(intents=intents, command_prefix="-")
|
|
|
|
|
|
@bot.tree.command(name="test", description="Test command")
|
|
@app_commands.allowed_installs(guilds=False, users=True)
|
|
@app_commands.allowed_contexts(guilds=True, dms=True, private_channels=True)
|
|
async def test(interaction: discord.Interaction) -> None:
|
|
await interaction.response.send_message("hai :3")
|
|
|
|
|
|
@bot.tree.command(name="generation", description="generation")
|
|
@app_commands.allowed_installs(guilds=False, users=True)
|
|
@app_commands.allowed_contexts(guilds=True, dms=True, private_channels=True)
|
|
async def generation(interaction: discord.Interaction) -> None:
|
|
await interaction.response.send_message("generation")
|
|
|
|
|
|
@bot.tree.command(name="ask", description="ai thing yes 👍")
|
|
@app_commands.allowed_installs(guilds=False, users=True)
|
|
@app_commands.allowed_contexts(guilds=True, dms=True, private_channels=True)
|
|
async def ask(interaction: discord.Interaction, prompt: str) -> None:
|
|
await interaction.response.defer()
|
|
response = await client.aio.models.generate_content(
|
|
model="gemini-2.5-flash",
|
|
contents=prompt,
|
|
config=config
|
|
)
|
|
if not response.text:
|
|
await interaction.edit_original_response(content="`[E] Model returned no response`")
|
|
generation: str = response.text or ""
|
|
if len(generation) > 2000:
|
|
buffer = io.BytesIO(generation.encode("utf-8"))
|
|
buffer.seek(0)
|
|
file = discord.File(buffer, filename="output.md")
|
|
await interaction.edit_original_response(content="Sent as a file", attachments=[file])
|
|
else:
|
|
await interaction.edit_original_response(content=generation)
|
|
|
|
|
|
@bot.tree.command(name="sync", description="bot.tree.sync()")
|
|
@app_commands.allowed_installs(guilds=False, users=True)
|
|
@app_commands.allowed_contexts(guilds=True, dms=True, private_channels=True)
|
|
async def sync(interaction: discord.Interaction) -> None:
|
|
if interaction.user.id == 1139850599085645844:
|
|
await interaction.response.send_message("`[I] Syncing...`")
|
|
await bot.tree.sync()
|
|
await interaction.edit_original_response(content="`[I] Syncing... OK`")
|
|
else:
|
|
await interaction.response.send_message("`[E] 403 Forbidden`")
|
|
|
|
|
|
@bot.event
|
|
async def on_ready():
|
|
await bot.tree.sync()
|
|
print("Logged in!")
|
|
|
|
|
|
api_key: str | None = os.getenv("DSC_API_KEY")
|
|
if not api_key:
|
|
raise RuntimeError
|
|
bot.run(api_key)
|