删除一个logger.trace防炸

This commit is contained in:
UnCLAS-Prommer
2025-07-29 10:28:10 +08:00
parent fa58889905
commit 5b2914f48f

View File

@@ -42,14 +42,13 @@ class ModelUsageStatistic:
# 确保表已经创建
try:
from src.common.database.database import db
db.create_tables([LLMUsage], safe=True)
except Exception as e:
logger.error(f"创建LLMUsage表失败: {e}")
@staticmethod
def _calculate_cost(
prompt_tokens: int, completion_tokens: int, model_info: ModelInfo
) -> float:
def _calculate_cost(prompt_tokens: int, completion_tokens: int, model_info: ModelInfo) -> float:
"""计算API调用成本
使用模型的pri_in和pri_out价格计算输入和输出的成本
@@ -101,11 +100,11 @@ class ModelUsageStatistic:
timestamp=datetime.now(),
)
logger.trace(
f"创建了一条模型使用情况记录 - 模型: {model_name}, "
f"子任务: {task_name}, 类型: {request_type.value}, "
f"用户: {user_id}, 记录ID: {usage_record.id}"
)
# logger.trace(
# f"创建了一条模型使用情况记录 - 模型: {model_name}, "
# f"子任务: {task_name}, 类型: {request_type.value}, "
# f"用户: {user_id}, 记录ID: {usage_record.id}"
# )
return usage_record.id
except Exception as e:
@@ -150,10 +149,8 @@ class ModelUsageStatistic:
prompt_tokens=prompt_tokens,
completion_tokens=completion_tokens,
total_tokens=total_tokens,
cost=self._calculate_cost(
prompt_tokens, completion_tokens, model_info
) if usage_data else 0.0,
).where(LLMUsage.id == record_id)
cost=self._calculate_cost(prompt_tokens, completion_tokens, model_info) if usage_data else 0.0,
).where(LLMUsage.id == record_id) # type: ignore
updated_count = update_query.execute()