Merge pull request #42 from epsilla-cloud/dev

upgrade python client

Author: eric-epsilla

ISSUE_TEMPLATE/bug_report.md

@@ -0,0 +1,17 @@
+---
+name: Bug report
+about: Create a report to help us improve
+title: ''
+labels: bug
+assignees: ''
+
+---
+
+**Describe the bug**
+A clear and concise description of what the bug is.
+
+**Screenshots**
+- If applicable, add screenshots to help explain your problem.
+
+**Additional context**
+- Add any other context about the problem here.

ISSUE_TEMPLATE/feature_request.md

@@ -0,0 +1,17 @@
+---
+name: Feature request
+about: Suggest an idea for this project
+title: ''
+labels: feature
+assignees: ''
+
+---
+
+**Describe the feature**
+- to be filled
+
+**Motivation and use case**
+- to be filled
+
+**Additional context**
+- to be filled

pyepsilla/cloud/client.py

@@ -61,6 +61,17 @@ def get_db_info(self, db_id: str):
         res.close()
         return status_code, body
 
+    def get_db_statistics(self, db_id: str):
+        req_url = "{}/vectordb/{}/statistics".format(self._baseurl, db_id)
+        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 vectordb(self, db_id: str):
         # validate project_id and api_key
         res = self.validate()
@@ -87,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
@@ -111,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

@@ -106,6 +106,19 @@ def unload_db(self, db_name: str):
         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 create_table(
         self,
         table_name: str,
@@ -302,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!")

pyepsilla/vectordb/version.py

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