1

tools/indentify/image_identifier.py

@@ -16,11 +16,11 @@ from PIL import Image
 import requests
 from io import BytesIO
 from concurrent.futures import ThreadPoolExecutor, as_completed
-from utils.logging_config import get_logger
 
 # 导入自定义模块
 sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
 from llm.openrouter import OpenRouterProcessor, OpenRouterModel
+from utils.logging_config import get_logger
 
 # 创建 logger
 logger = get_logger('ImageIdentifier')
@@ -53,7 +53,7 @@ class ImageIdentifier:
     def _ensure_configured(self):
         """确保Gemini已配置"""
         if not self._configured:
-            self.api_key = os.getenv('GEMINI_API_KEY')
+            self.api_key = os.getenv('GEMINI_API_KEY_1')
             
             if not self.api_key:
                 raise ValueError("请在环境变量中设置 GEMINI_API_KEY")
@@ -121,6 +121,7 @@ class ImageIdentifier:
                     # 检查响应状态
                     if response.candidates and len(response.candidates) > 0:
                         candidate = response.candidates[0]
+                        logger.info(f"图片 {url} 被安全过滤器阻止: {candidate.finish_reason}")
                         if candidate.finish_reason == 1:  # SAFETY
                             logger.warning(f"图片 {url} 被安全过滤器阻止")
                             return {"idx": idx, "url": url, "content": "", "success": False, "error": "内容被安全过滤器阻止"}

tools/indentify/video_identifier.py

@@ -16,13 +16,13 @@ import uuid
 import requests
 from typing import Dict, Any, List, Optional
 from dotenv import load_dotenv
+
+# 导入自定义模块
+sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
 from utils.logging_config import get_logger
 
 # 创建 logger
 logger = get_logger('VideoIdentifier')
-
-# 导入自定义模块
-sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
 from llm.openrouter import OpenRouterProcessor, OpenRouterModel
 
 # 导入Google Generative AI