average-stab

This commit is contained in:
Untone 2024-09-27 13:34:44 +03:00
parent 21591df427
commit 773a01cdb1
2 changed files with 17 additions and 14 deletions

View File

@ -48,12 +48,10 @@ async def messages_routing(msg, state):
reply_to_msg_id = reply_msg.get("message_id")
if not reply_to_msg_id and latest_toxic_message_id:
reply_to_msg_id = int(latest_toxic_message_id)
try:
# count average between all of messages
toxic_pattern = f"toxic:{cid}:{uid}:*"
toxic_score = await get_average_pattern(toxic_pattern)
except Exception:
pass
# current mesasage toxicity
if reply_to_msg_id:

View File

@ -24,6 +24,8 @@ async def get_all_pattern(uid):
async def get_average_pattern(pattern):
scores = []
toxic_score = 0
try:
async for key in redis.scan_iter(pattern):
scr = await redis.get(str(key))
scr = int(scr)
@ -31,4 +33,7 @@ async def get_average_pattern(pattern):
scores.append(scr)
logger.debug(f"found {len(scores)} messages")
toxic_score = math.floor(sum(scores) / len(scores)) if scores else 0
except Exception:
pass
finally:
return toxic_score