gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

better：优化了统计信息，会在控制台显示统计信息

Browse Source
This commit is contained in:
SengokuCola
2025-03-31 23:26:38 +08:00
parent ab3413e24b
commit 6f3cc2cb55
6 changed files with 134 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/heart_flow/observation.py
View File

@@ -33,7 +33,7 @@ class ChattingObservation(Observation):
self.sub_observe = None
self.llm_summary = LLM_request(
model=global_config.llm_observation, temperature=0.7, max_tokens=300, request_type="outer_world"
model=global_config.llm_observation, temperature=0.7, max_tokens=300, request_type="chat_observation"
)
# 进行一次观察 返回观察结果observe_info

4
src/plugins/chat/message_sender.py
View File

@@ -192,7 +192,7 @@ class MessageManager:
print(thinking_time)
if (
message_earliest.is_head
and message_earliest.update_thinking_time() > 8
and message_earliest.update_thinking_time() > 18
and not message_earliest.is_private_message() # 避免在私聊时插入reply
):
logger.debug(f"设置回复消息{message_earliest.processed_plain_text}")
@@ -219,7 +219,7 @@ class MessageManager:
# print(msg.is_private_message())
if (
msg.is_head
and msg.update_thinking_time() > 8
and msg.update_thinking_time() > 18
and not msg.is_private_message() # 避免在私聊时插入reply
):
logger.debug(f"设置回复消息{msg.processed_plain_text}")

2
src/plugins/config/config.py
View File

@@ -25,7 +25,7 @@ logger = get_module_logger("config", config=config_config)
#考虑到，实际上配置文件中的mai_version是不会自动更新的,所以采用硬编码
mai_version_main = "0.6.0"
mai_version_fix = "mmc-2"
mai_version_fix = "mmc-3"
mai_version = f"{mai_version_main}-{mai_version_fix}"
def update_config():

126
src/plugins/utils/statistic.py
View File

@@ -20,6 +20,7 @@ class LLMStatistics:
self.output_file = output_file
self.running = False
self.stats_thread = None
self.console_thread = None
self._init_database()
def _init_database(self):
@@ -32,15 +33,22 @@ class LLMStatistics:
"""启动统计线程"""
if not self.running:
self.running = True
# 启动文件统计线程
self.stats_thread = threading.Thread(target=self._stats_loop)
self.stats_thread.daemon = True
self.stats_thread.start()
# 启动控制台输出线程
self.console_thread = threading.Thread(target=self._console_output_loop)
self.console_thread.daemon = True
self.console_thread.start()
def stop(self):
"""停止统计线程"""
self.running = False
if self.stats_thread:
self.stats_thread.join()
if self.console_thread:
self.console_thread.join()
def _record_online_time(self):
"""记录在线时间"""
@@ -126,10 +134,19 @@ class LLMStatistics:
messages_cursor = db.messages.find({"time": {"$gte": start_time.timestamp()}})
for doc in messages_cursor:
stats["total_messages"] += 1
user_id = str(doc.get("user_info", {}).get("user_id", "unknown"))
chat_id = str(doc.get("chat_id", "unknown"))
stats["messages_by_user"][user_id] += 1
stats["messages_by_chat"][chat_id] += 1
# user_id = str(doc.get("user_info", {}).get("user_id", "unknown"))
chat_info = doc.get("chat_info", {})
user_info = doc.get("user_info", {})
group_info = chat_info.get("group_info") if chat_info else {}
# print(f"group_info: {group_info}")
group_name = "unknown"
if group_info:
group_name = group_info["group_name"]
if user_info and group_name == "unknown":
group_name = user_info["user_nickname"]
# print(f"group_name: {group_name}")
stats["messages_by_user"][user_id] += 1
stats["messages_by_chat"][group_name] += 1
return stats
@@ -201,17 +218,74 @@ class LLMStatistics:
)
output.append("")
# 添加消息统计
output.append("消息统计:")
output.append(("用户ID 消息数量"))
for user_id, count in sorted(stats["messages_by_user"].items()):
output.append(f"{user_id[:32]:<32} {count:>10}")
# 添加聊天统计
output.append("群组统计:")
output.append(("群组名称 消息数量"))
for group_name, count in sorted(stats["messages_by_chat"].items()):
output.append(f"{group_name[:32]:<32} {count:>10}")
return "\n".join(output)
def _format_stats_section_lite(self, stats: Dict[str, Any], title: str) -> str:
"""格式化统计部分的输出"""
output = []
output.append("\n" + "-" * 84)
output.append(f"{title}")
output.append("-" * 84)
# output.append(f"总请求数: {stats['total_requests']}")
if stats["total_requests"] > 0:
# output.append(f"总Token数: {stats['total_tokens']}")
output.append(f"总花费: {stats['total_cost']:.4f}¥")
# output.append(f"在线时间: {stats['online_time_minutes']}分钟")
output.append(f"总消息数: {stats['total_messages']}\n")
data_fmt = "{:<32} {:>10} {:>14} {:>13.4f} ¥"
# 按模型统计
output.append("按模型统计:")
output.append(("模型名称 调用次数 Token总量 累计花费"))
for model_name, count in sorted(stats["requests_by_model"].items()):
tokens = stats["tokens_by_model"][model_name]
cost = stats["costs_by_model"][model_name]
output.append(
data_fmt.format(model_name[:32] + ".." if len(model_name) > 32 else model_name, count, tokens, cost)
)
output.append("")
output.append("聊天统计:")
output.append(("聊天ID 消息数量"))
for chat_id, count in sorted(stats["messages_by_chat"].items()):
output.append(f"{chat_id[:32]:<32} {count:>10}")
# 按请求类型统计
# output.append("按请求类型统计:")
# output.append(("模型名称 调用次数 Token总量 累计花费"))
# for req_type, count in sorted(stats["requests_by_type"].items()):
# tokens = stats["tokens_by_type"][req_type]
# cost = stats["costs_by_type"][req_type]
# output.append(
# data_fmt.format(req_type[:22] + ".." if len(req_type) > 24 else req_type, count, tokens, cost)
# )
# output.append("")
# 修正用户统计列宽
# output.append("按用户统计:")
# output.append(("用户ID 调用次数 Token总量 累计花费"))
# for user_id, count in sorted(stats["requests_by_user"].items()):
# tokens = stats["tokens_by_user"][user_id]
# cost = stats["costs_by_user"][user_id]
# output.append(
# data_fmt.format(
# user_id[:22], # 不再添加省略号，保持原始ID
# count,
# tokens,
# cost,
# )
# )
# output.append("")
# 添加聊天统计
output.append("群组统计:")
output.append(("群组名称 消息数量"))
for group_name, count in sorted(stats["messages_by_chat"].items()):
output.append(f"{group_name[:32]:<32} {count:>10}")
return "\n".join(output)
@@ -237,8 +311,30 @@ class LLMStatistics:
with open(self.output_file, "w", encoding="utf-8") as f:
f.write("\n".join(output))
def _console_output_loop(self):
"""控制台输出循环，每5分钟输出一次最近1小时的统计"""
while self.running:
# 等待5分钟
for _ in range(30): # 5分钟 = 300秒
if not self.running:
break
time.sleep(1)
try:
# 收集最近1小时的统计数据
now = datetime.now()
hour_stats = self._collect_statistics_for_period(now - timedelta(hours=1))
# 使用logger输出
stats_output = self._format_stats_section_lite(hour_stats, "最近1小时统计：详细信息见根目录文件：llm_statistics.txt")
logger.info("\n" + stats_output + "\n" + "=" * 50)
except Exception:
logger.exception("控制台统计数据输出失败")
def _stats_loop(self):
"""统计循环，每1分钟运行一次"""
"""统计循环，每5分钟运行一次"""
while self.running:
try:
# 记录在线时间
@@ -250,7 +346,7 @@ class LLMStatistics:
logger.exception("统计数据处理失败")
# 等待5分钟
for _ in range(30): # 5分钟 = 300秒
for _ in range(300): # 5分钟 = 300秒
if not self.running:
break
time.sleep(1)