clean_agent

agents/clean_agent/tools.py

@@ -1,7 +1,6 @@
 from langchain_core.tools import tool
 from sqlalchemy.orm import Session
 from typing import Dict, Any, Tuple
-import logging
 from datetime import datetime
 import json
 import os
@@ -14,9 +13,10 @@ sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(
 from database.db import SessionLocal, get_db
 from database.models import KnowledgeParsingContent, KnowledgeExtractionContent
 from gemini import GeminiProcessor
+from utils.logging_config import get_logger
 
 # 配置日志
-logger = logging.getLogger('CleanTools')
+logger = get_logger('CleanTools')
 
 # 配置常量
 BATCH_SIZE = 5  # 分批处理大小