modify: 插件配置读取修改

doubao/__init__.py

@@ -48,10 +48,26 @@ async def create_workers(session, config, count=1, worker_manager=None):
             raise
 
         base_url = config.get("base_url", "https://ark.cn-beijing.volces.com/api/v3")
-        os.environ["DOUBAO_BASE_URL"] = base_url
-        logger.info("豆包 AICC base_url: %s", base_url)
+        # ra_url 为远程证明（安全令牌）端点地址，可能与推理 base_url 不同
+        # 若不单独配置则默认使用 base_url 的域名 + /v1/security/token
+        ra_url = config.get("ra_url", base_url)
+        ak = config.get("ak", "")
+        sk = config.get("sk", "")
+        ra_service_name = config.get("ra_service_name", "AICC.ConfidentialInference")
+        ra_policy_id = config.get("ra_policy_id", "router_policy")
 
-        aicc_config = AICCConfig(ra_url=base_url, auth_token=api_key)
+        os.environ["DOUBAO_BASE_URL"] = base_url
+        logger.info("豆包 AICC base_url: %s, ra_url: %s, ra_service_name: %s, ra_policy_id: %s",
+                    base_url, ra_url, ra_service_name, ra_policy_id)
+
+        aicc_config = AICCConfig(
+            ra_url=ra_url,
+            auth_token=api_key,
+            ak=ak,
+            sk=sk,
+            aicc_llm_service_name=ra_service_name,
+            policy_id=ra_policy_id,
+        )
         client = AsyncOpenAI(
             api_key=api_key,
             base_url=base_url,