Forráskód Böngészése

refactor: 移除冗余日志并添加环境变量加载功能

移除get_decode_result函数中的冗余日志输出
在llm_invoker.py中添加手动加载.env文件的功能,用于环境变量配置
max_liu 3 napja
szülő
commit
07be30417e
2 módosított fájl, 34 hozzáadás és 1 törlés
  1. 0 1
      main.py
  2. 34 0
      src/utils/llm_invoker.py

+ 0 - 1
main.py

@@ -79,7 +79,6 @@ def decode_video(param:DecodeListParam):
 def get_decode_result(param: TaskStatusParam):
     db_res = get_decode_result_by_id_db(param.task_id)
     logger.info(f"\n查询结构结果的task_id = {param.task_id}")
-    logger.info(f"\n查询结构结果 = {db_res}")
 
     if not db_res:
         return {

+ 34 - 0
src/utils/llm_invoker.py

@@ -16,6 +16,40 @@ from google import genai
 from src.utils.logger import get_logger
 from src.utils.json_extractor import JSONExtractor
 
+# 添加项目根目录到路径
+project_root = Path(__file__).parent.parent
+sys.path.insert(0, str(project_root))
+
+# 手动加载.env文件
+def load_env_file(env_path):
+    """手动加载.env文件"""
+    if not env_path.exists():
+        return False
+
+    with open(env_path, 'r') as f:
+        for line in f:
+            line = line.strip()
+            # 跳过注释和空行
+            if not line or line.startswith('#'):
+                continue
+            # 解析KEY=VALUE
+            if '=' in line:
+                key, value = line.split('=', 1)
+                os.environ[key.strip()] = value.strip()
+
+    return True
+
+env_path = project_root / ".env"
+if load_env_file(env_path):
+    print(f"✅ 已加载环境变量从: {env_path}")
+    # 验证API密钥
+    api_key = os.environ.get("GEMINI_API_KEY", "")
+    if api_key:
+        print(f"   GEMINI_API_KEY: {api_key[:10]}...")
+else:
+    print(f"⚠️  未找到.env文件: {env_path}")
+    
+
 logger = get_logger(__name__)
 
 # Gemini 文件处理相关常量