vrr/SurfSense
2
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2025-09-10 06:14:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: async thread on discord but

Browse source
This commit is contained in:
Muhamad Aji Wibisono 2025-06-02 19:22:04 +07:00
parent 1d67a87b82
commit a0f9efdb67
2 changed files with 46 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

14
surfsense_backend/app/tasks/connectors_indexing_tasks.py
View file

@ -15,6 +15,7 @@ from app.connectors.discord_connector import DiscordConnector
from discord import DiscordException
from slack_sdk.errors import SlackApiError
import logging
import asyncio
from app.utils.document_converters import generate_content_hash
@ -952,6 +953,9 @@ async def index_discord_messages(
if not discord_token:
return 0, "Discord token not found in connector config"
logger.info(f"Starting Discord indexing for connector {connector_id}")
logger.info(f"discord token is: {discord_token[:4]}... (truncated for security)")
# Initialize Discord client
discord_client = DiscordConnector(token=discord_token)
@ -975,13 +979,21 @@ async def index_discord_messages(
skipped_guilds = []
try:
await discord_client.start_bot()
logger.info("Starting Discord bot to fetch guilds")
discord_client._bot_task = asyncio.create_task(discord_client.start_bot())
await discord_client._wait_until_ready()
logger.info("Fetching Discord guilds")
guilds = await discord_client.get_guilds()
logger.info(f"Found {len(guilds)} guilds")
except Exception as e:
logger.error(f"Failed to get Discord guilds: {str(e)}", exc_info=True)
await discord_client.close_bot()
return 0, f"Failed to get Discord guilds: {str(e)}"
if not guilds:
logger.info("No Discord guilds found to index")
await discord_client.close_bot()
return 0, "No Discord guilds found"