'commit'

dsLightRag/Volcengine/VikingDBMemoryService.py

@@ -15,17 +15,19 @@ from volcengine.auth.SignerV4 import SignerV4
 from volcengine.base.Service import Service
 from volcenginesdkarkruntime import Ark
 
+from Config.Config import VOLC_SECRETKEY, VOLC_ACCESSKEY, VOLC_API_KEY
+
 
 def initialize_services(ak=None, sk=None, ark_api_key=None):
     """初始化记忆数据库服务和LLM客户端"""
     load_dotenv()
     # 如果参数未提供，尝试从环境变量获取
     if not ak:
-        ak = os.getenv("VOLC_ACCESSKEY")
+        ak = VOLC_ACCESSKEY
     if not sk:
-        sk = os.getenv("VOLC_SECRETKEY")
+        sk = VOLC_SECRETKEY
     if not ark_api_key:
-        ark_api_key = os.getenv("VOLC_API_KEY")
+        ark_api_key = VOLC_API_KEY
 
     if not all([ak, sk, ark_api_key]):
         raise ValueError("必须提供 VOLC_ACCESSKEY, VOLC_SECRETKEY, 和 VOLC_API_KEY。")