Spaces:
Paused
Paused
Update src/key_manager.py
Browse files- 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 |
-
|
| 694 |
-
storage_file = os.path.join(
|
| 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}")
|