diff --git a/agentrun/knowledgebase/api/__data_async_template.py b/agentrun/knowledgebase/api/__data_async_template.py
index fce66a9..a068ff5 100644
--- a/agentrun/knowledgebase/api/__data_async_template.py
+++ b/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,
         )
diff --git a/agentrun/knowledgebase/api/data.py b/agentrun/knowledgebase/api/data.py
index 2dacece..2b4fd55 100644
--- a/agentrun/knowledgebase/api/data.py
+++ b/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,
         )
diff --git a/tests/unittests/knowledgebase/test_ots_knowledgebase.py b/tests/unittests/knowledgebase/test_ots_knowledgebase.py
index ad7ea09..907d86c 100644
--- a/tests/unittests/knowledgebase/test_ots_knowledgebase.py
+++ b/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",
         )