From 870c65cbd84ae2d0563ae6f9fa3cd2bbc6407ff6 Mon Sep 17 00:00:00 2001
From: HexatomicRing <54496918+HexatomicRing@users.noreply.github.com>
Date: Wed, 9 Apr 2025 10:42:21 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=B6=88=E6=81=AF=E7=BB=9F?=
 =?UTF-8?q?=E8=AE=A1=E5=AF=B9=E8=BF=9C=E5=8F=A4=E7=89=88=E6=9C=AC=E6=A0=BC?=
 =?UTF-8?q?=E5=BC=8F=E7=9A=84=E6=94=AF=E6=8C=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/plugins/utils/statistic.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/plugins/utils/statistic.py b/src/plugins/utils/statistic.py
index 4b9afff3..01d0dff2 100644
--- a/src/plugins/utils/statistic.py
+++ b/src/plugins/utils/statistic.py
@@ -142,12 +142,23 @@ class LLMStatistics:
             group_info = chat_info.get("group_info") if chat_info else {}
             # print(f"group_info: {group_info}")
             group_name = None
+            group_id = None
             if group_info:
                 group_id = f"g{group_info.get('group_id')}"
-                group_name = group_info.get("group_name", f"群{group_info.get('group_id')}")
+                group_name = group_info.get("group_name", f"群{group_id}")
             if user_info and not group_name:
                 group_id = f"u{user_info['user_id']}"
                 group_name = user_info["user_nickname"]
+            # 添加对远古版本格式的支持，防止炸飞
+            if group_id is None:
+                if "group_id" in doc:
+                    group_id = f"g{doc.get('group_id')}"
+                    group_name = doc.get("group_name", f"群{group_id}")
+                elif "user_id" in doc:
+                    group_id = f"g{doc.get('user_id')}"
+                    group_name = doc.get("user_nickname", "")
+                else:
+                    continue
             if self.name_dict.get(group_id):
                 if message_time > self.name_dict.get(group_id)[1]:
                     self.name_dict[group_id] = [group_name, message_time]