重构数据库访问,替换为统一的数据库实例引用

This commit is contained in:
晴猫
2025-03-12 22:27:59 +09:00
parent 49082267bb
commit 8be087dcad
19 changed files with 138 additions and 284 deletions

View File

@@ -5,7 +5,7 @@ from datetime import datetime, timedelta
from typing import Any, Dict
from loguru import logger
from ...common.database import Database
from ...common.database import db
class LLMStatistics:
@@ -15,7 +15,6 @@ class LLMStatistics:
Args:
output_file: 统计结果输出文件路径
"""
self.db = Database.get_instance()
self.output_file = output_file
self.running = False
self.stats_thread = None
@@ -53,7 +52,7 @@ class LLMStatistics:
"costs_by_model": defaultdict(float)
}
cursor = self.db.llm_usage.find({
cursor = db.llm_usage.find({
"timestamp": {"$gte": start_time}
})