Gemini CLI with Discord

services:
  gemini:
    build: .
    container_name: gemini
    command: python3 /root/bot.py
    restart: always
    environment:
      - DISCORD_TOKEN=xxxxxxxxxxxxxxxxxxxxxxxxxxxxx
      - MODELS=gemini-2.5-flash
    stdin_open: true
    tty: true
    volumes:
      - ./home:/root
      - ./app:/app

FROM python:3.11-slim

RUN apt-get update && apt-get install -y curl \
  && curl -fsSL https://deb.nodesource.com/setup_20.x | bash - \
  && apt-get install -y nodejs \
  && npm install -g @google/gemini-cli

RUN set -eux; \
  npm install -g @google/gemini-cli > /dev/null 2>&1; \
  apt-get update -qq; \
  apt-get install -y --no-install-recommends \
    python3 \
    make \
    g++ \
    man-db \
    curl \
    dnsutils \
    less \
    jq \
    bc \
    gh \
    git \
    unzip \
    rsync \
    ripgrep \
    procps \
    psmisc \
    lsof \
    socat \
    ca-certificates \
    lsb-release \
    wget > /dev/null 2>&1; \
  install -m 0755 -d /etc/apt/keyrings; \
  wget -qO - https://packages.adoptium.net/artifactory/api/gpg/key/public | tee /etc/apt/keyrings/adoptium.asc > /dev/null; \
  echo "deb [signed-by=/etc/apt/keyrings/adoptium.asc] https://packages.adoptium.net/artifactory/deb $(lsb_release -cs) main" \
    | tee /etc/apt/sources.list.d/adoptium.list > /dev/null; \
  apt-get update -qq; \
  apt-get install -y --no-install-recommends temurin-17-jdk > /dev/null 2>&1; \
  apt-get clean; \
  rm -rf /var/lib/apt/lists/*


COPY home/requirements.txt .

RUN pip install --no-cache-dir -r requirements.txt

WORKDIR /app

CMD ["python", "~/bot.py"]

discord.py
python-dotenv

import discord
from discord.ext import commands
from dotenv import load_dotenv
import os
import shlex
import asyncio

load_dotenv()

DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
if not DISCORD_TOKEN:
    raise ValueError("DISCORD_TOKEN is not set in environment variables.")

MODELS = os.getenv("MODELS", "gemini-2.5-pro")

intents = discord.Intents.default()
intents.message_content = True

bot = commands.Bot(command_prefix="/", intents=intents)

@bot.command()
async def gemini(ctx, *, question: str):
    session_id = f"thread-{ctx.channel.id}"  # unused
    escaped_question = shlex.quote(question)
    cmd = f"gemini -m {MODELS} -p {escaped_question} --yolo"

    try:
        await ctx.message.add_reaction("🤖")
    except discord.HTTPException:
        pass

    try:
        print(f"[gemini] Running command: {cmd}")
        proc = await asyncio.create_subprocess_shell(
            cmd,
            stdout=asyncio.subprocess.PIPE,
            stderr=asyncio.subprocess.PIPE,
        )
        stdout, stderr = await proc.communicate()

        if stderr:
            error_msg = stderr.decode().strip()
            raise RuntimeError(error_msg)

        output = stdout.decode().strip()
        print(output)
        if not output:
            raise RuntimeError("No output from gemini.")

        # Discordの2000文字制限に対応して分割送信
        for i in range(0, len(output), 2000):
            await ctx.send(output[i:i+2000])

        try:
            await ctx.message.remove_reaction("🤖", bot.user)
            await ctx.message.add_reaction("✅")
        except discord.HTTPException:
            pass

    except Exception as e:
        try:
            await ctx.message.remove_reaction("🤖", bot.user)
            await ctx.message.add_reaction("❌")
        except discord.HTTPException:
            pass

        error_msg = str(e)
        if len(error_msg) <= 2000:
            await ctx.send(f"Error: {error_msg}")
        else:
            await ctx.send("Error occurred. Output too long to display.")

bot.run(DISCORD_TOKEN)

.
├── Dockerfile
├── README.md
├── app
│   └── simple-api  # 適当なJavaアプリケーション
├── docker-compose.yaml
└── home
    ├── bot.py
    └── requirements.txt

72 directories, 50 files