fix: fix knowledgebase cridential get

Author: yaozheng-fang

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "veadk-python"
-version = "0.5.5"
+version = "0.5.6"
 description = "Volcengine agent development kit, integrations with Volcengine cloud services."
 readme = "README.md"
 requires-python = ">=3.10"

veadk/knowledgebase/backends/vikingdb_knowledge_backend.py

@@ -136,14 +136,6 @@ class VikingDBKnowledgeBackend(BaseKnowledgebaseBackend):
     def model_post_init(self, __context: Any) -> None:
         self._set_service_info()
 
-        self._viking_sdk_client = VikingKnowledgeBaseService(
-            host=self.host,
-            ak=self.volcengine_access_key,
-            sk=self.volcengine_secret_key,
-            sts_token=self.session_token,
-            scheme=self.schema,
-        )
-
         self.precheck_index_naming()
 
         # check whether collection exist, if not, create it
@@ -552,6 +544,16 @@ def _search_knowledge(
             "chunk_diffusion_count": chunk_diffusion_count,
         }
 
+        self._set_service_info()
+
+        self._viking_sdk_client = VikingKnowledgeBaseService(
+            host=self.host,
+            ak=self.volcengine_access_key,
+            sk=self.volcengine_secret_key,
+            sts_token=self.session_token,
+            scheme=self.schema,
+        )
+
         response = self._viking_sdk_client.search_knowledge(
             collection_name=self.index,
             project=self.volcengine_project,
@@ -606,6 +608,8 @@ def _do_request(
     ) -> dict:
         full_path = f"{self.base_url}{path}"
 
+        self._set_service_info()
+
         request = build_vikingdb_knowledgebase_request(
             path=path,
             volcengine_access_key=self.volcengine_access_key,

veadk/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-VERSION = "0.5.5"
+VERSION = "0.5.6"