Browse Source

fix(decode_task): 添加环境变量检查并记录日志

检查GEMINI_API_KEY或GOOGLE_API_KEY环境变量是否存在
并在缺失时记录错误日志
max_liu 2 days ago
parent
commit
f604d263a1
1 changed files with 6 additions and 0 deletions
  1. 6 0
      decode_task/decodeTask.py

+ 6 - 0
decode_task/decodeTask.py

@@ -1,4 +1,5 @@
 import json
+import os
 from loguru import logger
 import sys
 import time
@@ -40,6 +41,11 @@ def get_decode_result_by_id(task_id:str):
 
 def  decode_task_status_handler():
     # 从数据库中获取任务,每次获取一个
+    api_key = os.getenv("GEMINI_API_KEY") or os.getenv("GOOGLE_API_KEY")
+    
+    logger.error(f"❌ GEMINI_API_KEY = {api_key}")
+        
+    
     sql = "SELECT * FROM decode_record WHERE task_status = 0 ORDER BY create_timestamp ASC LIMIT 1"
     task = mysql.fetchone(sql)