4次修改

src/llm_models/utils.py

@@ -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}, "