dan/memory
1
0
Fork 0
mirror of https://github.com/mruwnik/memory.git synced 2025-11-13 00:04:05 +01:00
Code Issues 1 Packages Projects Releases Wiki Activity

tool to set up discord bot

Browse Source
This commit is contained in:
mruwnik 2025-11-03 11:11:19 +00:00
parent 2944a0bce1
commit 2d3dc06fdf
3 changed files with 106 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/memory/discord/collector.py
View File

@ -246,8 +246,8 @@ class MessageCollector(commands.Bot):
async def on_ready(self):
"""Called when bot connects to Discord"""
logger.info(f"Discord collector connected as {self.user}")
logger.info(f"Connected to {len(self.guilds)} servers")
logger.error(f"Discord collector connected as {self.user}")
logger.error(f"Connected to {len(self.guilds)} servers")
# Sync server and channel metadata
await self.sync_servers_and_channels()
@ -257,7 +257,7 @@ class MessageCollector(commands.Bot):
except Exception as exc: # pragma: no cover - defensive
logger.error("Failed to sync slash commands: %s", exc)
logger.info("Discord message collector ready")
logger.error("Discord message collector ready")
async def on_message(self, message: discord.Message):
"""Queue incoming message for database storage"""

2
src/memory/discord/messages.py
View File

@ -269,7 +269,7 @@ def call_llm(
return provider.run_with_tools(
messages=provider.as_messages(message_content),
tools=tools,
system_prompt=bot_user.system_prompt + "\n\n" + system_prompt,
system_prompt=(bot_user.system_prompt or "") + "\n\n" + (system_prompt or ""),
mcp_servers=mcp_servers,
max_iterations=settings.DISCORD_MAX_TOOL_CALLS,
).response

108
tools/discord_setup.py
View File

@ -1,8 +1,14 @@
import argparse
import click
import requests
def make_invite(client_id: str | int) -> str:
permissions = 2048 + 16 + 1024 # = 3088
invite_url = f"https://discord.com/oauth2/authorize?client_id={str(client_id)}&scope=bot&permissions={permissions}"
return invite_url
@click.command()
@click.option("--bot-token", type=str, required=True)
def generate_bot_invite_url(bot_token: str):
@ -25,12 +31,8 @@ def generate_bot_invite_url(bot_token: str):
except Exception as e:
raise ValueError(f"Could not get bot info: {e}")
# Permissions needed: Send Messages (2048) + Manage Channels (16) + View Channels (1024)
permissions = 2048 + 16 + 1024 # = 3088
invite_url = f"https://discord.com/oauth2/authorize?client_id={client_id}&scope=bot&permissions={permissions}"
invite_url = make_invite(client_id)
click.echo(f"Bot invite URL: {invite_url}")
return invite_url
@click.command()
@ -43,6 +45,99 @@ def create_channels():
click.echo("Discord channels setup completed.")
@click.command()
@click.option("--bot-token", type=str, required=True, help="Discord bot token")
def add_bot_user(bot_token: str):
"""Add a Discord bot user to the system by fetching bot info from Discord API."""
from memory.common.db.connection import make_session
from memory.common.db.models import DiscordUser, DiscordServer, DiscordBotUser
# Fetch bot information from Discord API
try:
headers = {"Authorization": f"Bot {bot_token}"}
response = requests.get(
"https://discord.com/api/v10/users/@me", headers=headers
)
response.raise_for_status()
bot_info = response.json()
discord_id = int(bot_info["id"])
username = bot_info["username"]
display_name = bot_info.get("global_name")
discriminator = bot_info.get("discriminator", "0")
# Use discriminator if it's not "0" (new username system uses "0")
if discriminator != "0":
username = f"{username}#{discriminator}"
except Exception as e:
click.echo(f"Error: Could not fetch bot info from Discord API: {e}")
return
# Create email and name from Discord info
email = f"{username.replace('#', '_')}@discord.bot"
name = display_name or username
with make_session() as session:
# Get or create DiscordUser
discord_user = (
session.query(DiscordUser).filter(DiscordUser.id == discord_id).first()
)
if discord_user:
click.echo(f"Found existing Discord user: {discord_user.username}")
# Update username and display_name in case they changed
discord_user.username = username
discord_user.display_name = display_name
else:
discord_user = DiscordUser(
id=discord_id,
username=username,
display_name=display_name,
)
session.add(discord_user)
click.echo(f"Created new Discord user: {username}")
# Get or create DiscordBotUser (search by api_key)
bot_user = (
session.query(DiscordBotUser)
.filter(DiscordBotUser.api_key == bot_token)
.first()
)
if bot_user:
click.echo(f"Found existing bot user: {bot_user.name}")
# Update email and name in case they changed
bot_user.email = email
bot_user.name = name
# Ensure they're connected
if discord_user not in bot_user.discord_users:
bot_user.discord_users.append(discord_user)
click.echo("Linked Discord user to bot user")
else:
bot_user = DiscordBotUser.create_with_api_key(
discord_users=[discord_user],
name=name,
email=email,
api_key=bot_token,
)
session.add(bot_user)
click.echo(f"Created new bot user: {name}")
click.echo("Linked Discord user to bot user")
session.commit()
click.echo("\n✓ Successfully configured Discord bot user:")
click.echo(f" Bot Name: {bot_user.name}")
click.echo(f" Bot Email: {bot_user.email}")
click.echo(f" API Key: {bot_user.api_key}")
click.echo(f" Discord ID: {discord_user.id}")
click.echo(f" Discord Username: {discord_user.username}")
if display_name:
click.echo(f" Discord Display Name: {display_name}")
click.echo("\n\nTo add the bot to your server, click the link below:")
click.echo(f"Bot invite URL: {make_invite(discord_id)}")
@click.group()
def cli():
"""Discord setup utilities."""
@ -51,6 +146,7 @@ def cli():
cli.add_command(generate_bot_invite_url, name="generate-invite")
cli.add_command(create_channels, name="create-channels")
cli.add_command(add_bot_user, name="add-bot-user")
if __name__ == "__main__":