update with statistics

Author: eric-epsilla

pyepsilla/cloud/client.py

@@ -61,7 +61,7 @@ def get_db_info(self, db_id: str):
         res.close()
         return status_code, body
 
-    def get_db_statistics(self):
+    def get_db_statistics(self, db_id: str):
         req_url = "{}/vectordb/{}/statistics".format(self._baseurl, db_id)
         req_data = None
         res = requests.get(
@@ -98,7 +98,14 @@ def vectordb(self, db_id: str):
 
 
 class Vectordb(Client):
-    def __init__(self, project_id: str, db_id: str, api_key: str, public_endpoint: str, headers: dict = None):
+    def __init__(
+        self,
+        project_id: str,
+        db_id: str,
+        api_key: str,
+        public_endpoint: str,
+        headers: dict = None,
+    ):
         self._project_id = project_id
         self._db_id = db_id
         self._api_key = api_key
@@ -122,7 +129,12 @@ def list_tables(self):
         return status_code, body
 
     # Create table
-    def create_table(self, table_name: str, table_fields: list[dict] = None, indices: list[dict] = None):
+    def create_table(
+        self,
+        table_name: str,
+        table_fields: list[dict] = None,
+        indices: list[dict] = None,
+    ):
         if self._db_id is None:
             raise Exception("[ERROR] db_id is None!")
         if table_fields is None:

pyepsilla/vectordb/client.py

@@ -119,7 +119,6 @@ def statistics(self):
         res.close()
         return status_code, body
 
-
     def create_table(
         self,
         table_name: str,
@@ -316,19 +315,6 @@ def get(
         res.close()
         return status_code, body
 
-    def statistics(self):
-        if self._db is None:
-            raise Exception("[ERROR] Please use_db() first!")
-        req_url = "{}/api/{}/statistics".format(self._baseurl, self._db)
-        req_data = None
-        res = requests.get(
-            url=req_url, data=json.dumps(req_data), headers=self._header, verify=False
-        )
-        status_code = res.status_code
-        body = res.json()
-        res.close()
-        return status_code, body
-
     def drop_table(self, table_name: str = None):
         if self._db is None:
             raise Exception("[ERROR] Please use_db() first!")