fix project

Author: yaozheng-fang

veadk/integrations/ve_viking_db_memory/ve_viking_db_memory.py

@@ -290,8 +290,8 @@ def create_collection(
         res = self.json("CreateCollection", {}, json.dumps(params))
         return json.loads(res)
 
-    def get_collection(self, collection_name):
-        params = {"CollectionName": collection_name}
+    def get_collection(self, collection_name, project="default"):
+        params = {"CollectionName": collection_name, "ProjectName": project}
         res = self.json("GetCollection", {}, json.dumps(params))
         return json.loads(res)
 

veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py

@@ -92,9 +92,9 @@ def precheck_index_naming(self):
 
     def _collection_exist(self) -> bool:
         try:
-            client = self._get_sdk_client()
+            client = self._get_client()
             client.get_collection(
-                collection_name=self.index, project_name=self.volcengine_project
+                collection_name=self.index, project=self.volcengine_project
             )
             logger.info(f"Collection {self.index} exist.")
             return True