refactor(knowledgebase): 添加STS token支持并优化代码格式

本次提交主要完成了以下改进：
1. 在多个数据API类中添加了STS token的支持
2. 对长参数列表进行了适当的换行处理，提高代码可读性
3. 更新相关单元测试以验证新的STS token功能

这些改动增强了系统的安全性和灵活性，同时保持了良好的代码风格。

Co-developed-by: Aone Copilot <noreply@alibaba-inc.com>
Signed-off-by: Sodawyx <sodawyx@126.com>

Author: Sodawyx

agentrun/knowledgebase/api/__data_async_template.py

@@ -595,6 +595,7 @@ def _build_agent_storage_client(
         return AgentStorageClient(
             access_key_id=cfg.get_access_key_id(),
             access_key_secret=cfg.get_access_key_secret(),
+            sts_token=cfg.get_security_token(),
             ots_endpoint=ots_endpoint,
             ots_instance_name=self.provider_settings.ots_instance_name,
         )

agentrun/knowledgebase/api/data.py

@@ -854,6 +854,7 @@ def _build_agent_storage_client(
         return AgentStorageClient(
             access_key_id=cfg.get_access_key_id(),
             access_key_secret=cfg.get_access_key_secret(),
+            sts_token=cfg.get_security_token(),
             ots_endpoint=ots_endpoint,
             ots_instance_name=self.provider_settings.ots_instance_name,
         )

tests/unittests/knowledgebase/test_ots_knowledgebase.py

@@ -600,6 +600,7 @@ def test_build_client(self, mock_client_class):
         mock_config.get_region_id.return_value = "cn-hangzhou"
         mock_config.get_access_key_id.return_value = "test-ak"
         mock_config.get_access_key_secret.return_value = "test-sk"
+        mock_config.get_security_token.return_value = "test-sts"
 
         with patch.object(Config, "with_configs", return_value=mock_config):
             api = OTSDataAPI(
@@ -613,6 +614,7 @@ def test_build_client(self, mock_client_class):
         mock_client_class.assert_called_once_with(
             access_key_id="test-ak",
             access_key_secret="test-sk",
+            sts_token="test-sts",
             ots_endpoint="http://ots-cn-hangzhou.aliyuncs.com",
             ots_instance_name="test-instance",
         )