add statistics

Author: eric-epsilla

examples/hello_epsilla.py

@@ -73,6 +73,11 @@
 status_code, response = client.get(table_name="MyTable", limit=2)
 print(response)
 
+# Get Statistics
+status_code, response = client.statistics()
+print(response)
+
+# Delete Vectors
 # status_code, response =  client.delete(table_name="MyTable", ids=[3])
 status_code, response = client.delete(table_name="MyTable", primary_keys=[3, 4])
 # status_code, response =  client.delete(table_name="MyTable", filter="Doc <> 'San Francisco'")

pyepsilla/vectordb/client.py

@@ -17,7 +17,11 @@
 
 class Client:
     def __init__(
-        self, protocol: str = "http", host: str = "localhost", port: str = "8888", headers: dict = None
+        self,
+        protocol: str = "http",
+        host: str = "localhost",
+        port: str = "8888",
+        headers: dict = None,
     ):
         self._protocol = protocol
         self._host = host
@@ -102,7 +106,12 @@ def unload_db(self, db_name: str):
         res.close()
         return status_code, body
 
-    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 is None:
             raise Exception("[ERROR] Please use_db() first!")
         if table_fields is None:
@@ -293,6 +302,19 @@ 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!")

pyepsilla/vectordb/version.py

@@ -1 +1 @@
-__version__ = "0.3.0"
+__version__ = "0.3.1"