gallyga commited on
Commit
e43fde8
·
verified ·
1 Parent(s): d19dce9

Update src/key_manager.py

Browse files
Files changed (1) hide show
  1. src/key_manager.py +5 -5
src/key_manager.py CHANGED
@@ -10,7 +10,7 @@ from typing import Dict, List, Optional, Any, Union, Tuple, Callable
10
  logger = logging.getLogger("sora-api.key_manager")
11
 
12
  class KeyManager:
13
- def __init__(self, storage_file: str = "api_keys.json"):
14
  """
15
  初始化密钥管理器
16
 
@@ -686,12 +686,12 @@ class KeyManager:
686
  # 返回最后一次尝试的结果
687
  return False, result, current_key
688
 
689
- # 创建全局密钥管理器实例
690
- storage_file = os.getenv("KEYS_STORAGE_FILE", "api_keys.json")
691
  # 如果提供了绝对路径则直接使用,否则使用相对路径
692
  if not os.path.isabs(storage_file):
693
- base_dir = os.getenv("BASE_DIR", os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
694
- storage_file = os.path.join(base_dir, storage_file)
695
 
696
  key_manager = KeyManager(storage_file=storage_file)
697
  logger.info(f"初始化全局密钥管理器,存储文件: {storage_file}")
 
10
  logger = logging.getLogger("sora-api.key_manager")
11
 
12
  class KeyManager:
13
+ def __init__(self, storage_file: str = "/tmp/api_keys.json"):
14
  """
15
  初始化密钥管理器
16
 
 
686
  # 返回最后一次尝试的结果
687
  return False, result, current_key
688
 
689
+ # 配置存储文件路径
690
+ storage_file = os.getenv("KEYS_STORAGE_FILE", "/tmp/api_keys.json")
691
  # 如果提供了绝对路径则直接使用,否则使用相对路径
692
  if not os.path.isabs(storage_file):
693
+ # 对于非绝对路径,始终将其保存到/tmp目录
694
+ storage_file = os.path.join("/tmp", os.path.basename(storage_file))
695
 
696
  key_manager = KeyManager(storage_file=storage_file)
697
  logger.info(f"初始化全局密钥管理器,存储文件: {storage_file}")