chelsea/Synculous-2
1
0
Fork 0
Code Issues 8 Pull Requests Actions Packages Projects Releases Wiki Activity

Fix presence tracking and med reminder bugs

Browse Source
This commit is contained in:
chelsea
2026-02-19 20:31:43 -06:00
parent a19e30db68
commit cc1aace73d
3 changed files with 136 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
bot/bot.py
View File

@@ -128,6 +128,7 @@ class JurySystem:
async def retrieve(self, query_text, top_k=5):
"""Async retrieval — returns list of {metadata, score} dicts."""
import asyncio
return await asyncio.to_thread(self._retrieve_sync, query_text, top_k)
async def query(self, query_text):
@@ -147,6 +148,7 @@ If the answer is not in the context, say you don't know based on the provided te
Be concise, compassionate, and practical."""
from ai.jury_council import generate_rag_answer
return await generate_rag_answer(query_text, context_text, system_prompt)
except Exception as e:
return f"Error querying DBT knowledge base: {e}"
@@ -181,7 +183,9 @@ def apiRequest(method, endpoint, token=None, data=None, _retried=False):
if resp.status_code == 401 and not _retried:
new_token = _try_refresh_token_for_session(token)
if new_token:
return apiRequest(method, endpoint, token=new_token, data=data, _retried=True)
return apiRequest(
method, endpoint, token=new_token, data=data, _retried=True
)
try:
return resp.json(), resp.status_code
except ValueError:
@@ -201,9 +205,12 @@ def _try_refresh_token_for_session(expired_token):
if cached:
refresh_token = cached.get("refresh_token")
if refresh_token:
result, status = apiRequest("post", "/api/refresh",
data={"refresh_token": refresh_token},
_retried=True)
result, status = apiRequest(
"post",
"/api/refresh",
data={"refresh_token": refresh_token},
_retried=True,
)
if status == 200 and "token" in result:
new_token = result["token"]
session["token"] = new_token
@@ -258,7 +265,8 @@ def negotiateToken(discord_id, username, password):
# Try refresh token first (avoids sending password)
if cached and cached.get("refresh_token"):
result, status = apiRequest(
"post", "/api/refresh",
"post",
"/api/refresh",
data={"refresh_token": cached["refresh_token"]},
_retried=True,
)
@@ -279,7 +287,9 @@ def negotiateToken(discord_id, username, password):
and cached.get("hashed_password")
and verifyPassword(password, cached.get("hashed_password"))
):
result, status = apiRequest("post", "/api/login", data=login_data, _retried=True)
result, status = apiRequest(
"post", "/api/login", data=login_data, _retried=True
)
if status == 200 and "token" in result:
token = result["token"]
payload = decodeJwtPayload(token)
@@ -530,7 +540,9 @@ async def handleDBTQuery(message):
if not jury_result.safe_questions:
return True
await message.channel.send("🔍 Searching knowledge base with approved questions...")
await message.channel.send(
"🔍 Searching knowledge base with approved questions..."
)
# Step 3: Multi-query retrieval — deduplicated by chunk ID
seen_ids = set()
@@ -544,7 +556,9 @@ async def handleDBTQuery(message):
context_chunks.append(r["metadata"]["text"])
if not context_chunks:
await message.channel.send("⚠️ No relevant content found in the knowledge base.")
await message.channel.send(
"⚠️ No relevant content found in the knowledge base."
)
return True
context = "\n\n---\n\n".join(context_chunks)
@@ -644,7 +658,8 @@ def _restore_sessions_from_cache():
if not refresh_token:
continue
result, status = apiRequest(
"post", "/api/refresh",
"post",
"/api/refresh",
data={"refresh_token": refresh_token},
_retried=True,
)
@@ -705,15 +720,20 @@ async def update_presence_tracking():
import core.adaptive_meds as adaptive_meds
import core.postgres as postgres
print(f"[DEBUG] Running presence tracking. Guilds: {len(client.guilds)}", flush=True)
print(
f"[DEBUG] Running presence tracking. Guilds: {len(client.guilds)}",
flush=True,
)
for guild in client.guilds:
print(f"[DEBUG] Guild: {guild.name} ({guild.id}) - Members: {guild.member_count}")
print(
f"[DEBUG] Guild: {guild.name} ({guild.id}) - Members: {guild.member_count}"
)
# Get all users with presence tracking enabled
settings = postgres.select(
"adaptive_med_settings", {"presence_tracking_enabled": True}
)
print(f"[DEBUG] Found {len(settings)} users with presence tracking enabled")
for setting in settings:
@@ -733,27 +753,46 @@ async def update_presence_tracking():
# Get the member from a shared guild (needed for presence data)
try:
member = None
target_id = int(discord_user_id)
try:
target_id = int(discord_user_id)
except (ValueError, TypeError):
print(
f"[DEBUG] Invalid Discord ID for user {user_uuid}: {discord_user_id}",
flush=True,
)
continue
# Search through all guilds the bot is in
for guild in client.guilds:
member = guild.get_member(target_id)
print(f"[DEBUG] Checked guild {guild.name}, member: {member}", flush=True)
print(
f"[DEBUG] Checked guild {guild.name}, member: {member}",
flush=True,
)
if member:
break
if not member:
print(f"[DEBUG] User {discord_user_id} not found in any shared guild", flush=True)
print(
f"[DEBUG] User {discord_user_id} not found in any shared guild",
flush=True,
)
continue
# Check if user is online
is_online = member.status != discord.Status.offline
print(f"[DEBUG] User status: {member.status}, is_online: {is_online}", flush=True)
print(
f"[DEBUG] User status: {member.status}, is_online: {is_online}",
flush=True,
)
# Get current presence from DB
presence = adaptive_meds.get_user_presence(user_uuid)
was_online = presence.get("is_currently_online") if presence else False
print(f"[DEBUG] Previous state: {was_online}, Current: {is_online}", flush=True)
print(
f"[DEBUG] Previous state: {was_online}, Current: {is_online}",
flush=True,
)
# Update presence if changed
if is_online != was_online:
@@ -788,6 +827,7 @@ async def presenceTrackingLoop():
except Exception as e:
print(f"[ERROR] presenceTrackingLoop failed: {e}", flush=True)
import traceback
traceback.print_exc()