initial code

main.py

@@ -0,0 +1,79 @@
+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
+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]
+)
+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)