mirrors/karkas
0
0
Fork 0
mirror of https://gitflic.ru/project/alt-gnome/karkas.git synced 2025-04-05 00:53:49 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: add caption handle

Browse Source
This commit is contained in:
Maxim Slipenko 2025-02-22 18:01:22 +03:00
parent 7c25c7b4a9
commit 4d25d2e83c
No known key found for this signature in database
GPG Key ID: 6DEA9FA7DE98C7D6
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/karkas_blocks/karkas_blocks/standard/spam/main.py
View File

@ -133,7 +133,9 @@ class SpamDetectionMiddleware(BaseMiddleware):
event: "Message", event: "Message",
data: Dict[str, Any], data: Dict[str, Any],
) -> Any: ) -> Any:
if USER_STATS_AVAILABLE and event.text: text = event.text or event.caption
if USER_STATS_AVAILABLE and text:
stats = await get_user_stats(event.chat.id, event.from_user.id) stats = await get_user_stats(event.chat.id, event.from_user.id)
verified_users_row = ( verified_users_row = (
@ -146,8 +148,8 @@ class SpamDetectionMiddleware(BaseMiddleware):
if verified_users_row: if verified_users_row:
if not verified_users_row["verified"]: if not verified_users_row["verified"]:
if event.text: if text:
is_spam, confidence = spam_predict(event.text) is_spam, confidence = spam_predict(text)
log( log(
"message spam confidence", "message spam confidence",

2
src/karkas_blocks/karkas_blocks/standard/statistics/main.py
View File

@ -79,7 +79,7 @@ class StatisticsMiddleware(BaseMiddleware):
"new message recieved", "new message recieved",
trace_id=trace_id, trace_id=trace_id,
chat_id=event.chat.id, chat_id=event.chat.id,
message_text=event.text, message_text=event.text or event.caption,
) )
data["trace_id"] = trace_id data["trace_id"] = trace_id