数据知识回调入库

app/services/import_analysis.py

@@ -23,6 +23,7 @@ from app.models import (
     LLMRole,
 )
 from app.settings import DEFAULT_IMPORT_MODEL, get_supported_import_models
+from app.utils.llm_usage import extract_usage
 
 logger = logging.getLogger(__name__)
 
@@ -375,18 +376,6 @@ async def dispatch_import_analysis_job(
 
     return result
 
-# 兼容处理多模型的使用量字段提取
-def extract_usage(resp_json: dict) -> dict:
-    usage = resp_json.get("usage") or resp_json.get("usageMetadata") or {}
-    return {
-        "prompt_tokens": usage.get("prompt_tokens") or usage.get("input_tokens") or usage.get("promptTokenCount"),
-        "completion_tokens": usage.get("completion_tokens") or usage.get("output_tokens") or usage.get("candidatesTokenCount"),
-        "total_tokens": usage.get("total_tokens") or usage.get("totalTokenCount") or (
-            (usage.get("prompt_tokens") or usage.get("input_tokens") or 0)
-            + (usage.get("completion_tokens") or usage.get("output_tokens") or 0)
-        )
-    }
-
 async def notify_import_analysis_callback(
     callback_url: str,
     payload: Dict[str, Any],