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