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

4次修改

Browse Source
This commit is contained in:
tt-P607
2025-09-20 11:57:22 +08:00
parent 898208f425
commit a8992cdd51
6 changed files with 69 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/llm_models/utils.py
View File

@@ -146,7 +146,7 @@ class LLMUsageRecorder:
"""
@staticmethod
def record_usage_to_database(
async def record_usage_to_database(
model_info: ModelInfo,
model_usage: UsageRecord,
user_id: str,
@@ -161,7 +161,7 @@ class LLMUsageRecorder:
session = None
try:
# 使用 SQLAlchemy 会话创建记录
with get_db_session() as session:
async with get_db_session() as session:
usage_record = LLMUsage(
model_name=model_info.model_identifier,
model_assign_name=model_info.name,
@@ -179,7 +179,7 @@ class LLMUsageRecorder:
)
session.add(usage_record)
session.commit()
await session.commit()
logger.debug(
f"Token使用情况 - 模型: {model_usage.model_name}, "

6
src/llm_models/utils_model.py
View File

@@ -202,7 +202,7 @@ class LLMRequest:
content, extracted_reasoning = self._extract_reasoning(content)
reasoning_content = extracted_reasoning
if usage := response.usage:
llm_usage_recorder.record_usage_to_database(
await llm_usage_recorder.record_usage_to_database(
model_info=model_info,
model_usage=usage,
user_id="system",
@@ -367,7 +367,7 @@ class LLMRequest:
# 成功获取响应
if usage := response.usage:
llm_usage_recorder.record_usage_to_database(
await llm_usage_recorder.record_usage_to_database(
model_info=model_info,
model_usage=usage,
time_cost=time.time() - start_time,
@@ -442,7 +442,7 @@ class LLMRequest:
embedding = response.embedding
if usage := response.usage:
llm_usage_recorder.record_usage_to_database(
await llm_usage_recorder.record_usage_to_database(
model_info=model_info,
time_cost=time.time() - start_time,
model_usage=usage,