fix: add sts token in viking memory sdk client

Author: ethfoo

veadk/integrations/ve_viking_db_memory/ve_viking_db_memory.py

@@ -95,6 +95,9 @@ def setHeader(self, header):
                 api_info[key].header[item] = header[item]
         self.api_info = api_info
 
+    def get_host(self):
+        return self.service_info.host
+
     @staticmethod
     def get_service_info(host, region, scheme, connection_timeout, socket_timeout):
         service_info = ServiceInfo(

veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py

@@ -122,13 +122,15 @@ def _get_client(self) -> VikingDBMemoryClient:
         )
 
     def _get_sdk_client(self) -> VikingMem:
+        client = self._get_client()
         return VikingMem(
-            host="api-knowledgebase.mlp.cn-beijing.volces.com",
-            region="cn-beijing",
+            host=client.get_host(),
+            region=self.region,
             auth=IAM(
                 ak=self.volcengine_access_key,
                 sk=self.volcengine_secret_key,
             ),
+            sts_token=self.session_token,
         )
 
     @override