Merge pull request #44 from epsilla-cloud/upsert

0.3.3 release

Author: richard-epsilla

pyepsilla/cloud/client.py

@@ -177,6 +177,17 @@ def insert(self, table_name: str, records: list[dict]):
         res.close()
         return status_code, body
 
+    def upsert(self, table_name: str, records: list[dict]):
+        req_url = "{}/data/insert".format(self._baseurl)
+        req_data = {"table": table_name, "data": records, "upsert": True}
+        res = requests.post(
+            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
+
     # Query data from table
     def query(
         self,

pyepsilla/enterprise/client.py

@@ -222,6 +222,21 @@ def insert(self, table_name: str, records: list[dict]):
         res.close()
         return status_code, body
 
+    def upsert(self, table_name: str, records: list[dict]):
+        if self._db_id is None:
+            raise Exception("[ERROR] db_id is None!")
+        if records is None:
+            records = []
+        req_url = "{}/data/insert".format(self._baseurl)
+        req_data = {"table": table_name, "data": records, "upsert": True}
+        res = requests.post(
+            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
+
     # Query data from table
     def query(
         self,

pyepsilla/vectordb/client.py

@@ -166,6 +166,21 @@ def insert(self, table_name: str, records: list = None):
         res.close()
         return status_code, body
 
+    def upsert(self, table_name: str, records: list = None):
+        if self._db is None:
+            raise Exception("[ERROR] Please use_db() first!")
+        if records is None:
+            records = []
+        req_url = "{}/api/{}/data/insert".format(self._baseurl, self._db)
+        req_data = {"table": table_name, "data": records, "upsert": True}
+        res = requests.post(
+            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 delete(
         self,
         table_name: str,

pyepsilla/vectordb/version.py

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