update with res.close()

Author: eric-epsilla

pyepsilla/cloud/client.py

@@ -15,7 +15,9 @@ def __init__(self, project_id: str, api_key: str):
 
     def validate(self):
         res = requests.get(url=self._baseurl, data=None, headers=self._header)
-        return res.json()
+        data = res.json()
+        res.close()
+        return data
 
 
     def get_db_list(self):
@@ -26,6 +28,7 @@ def get_db_list(self):
         body = res.json()
         if status_code == 200 and body["statusCode"] == 200:
             db_list = [ db_id for db_id in res.json()["result"] ]
+        res.close()
         return db_list
 
 
@@ -34,6 +37,7 @@ def get_db_info(self, db_id: str):
         res = requests.get(url=req_url, data=None, headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
 
@@ -94,6 +98,7 @@ def insert(self, table_name: str, records: list[dict]):
         res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
 
@@ -117,6 +122,7 @@ def query(self, table_name: str, query_field: str = None, query_vector: list = N
         res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
 
@@ -145,6 +151,7 @@ def delete(self, table_name: str, primary_keys: Optional[list[Union[str,int]]] =
         res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
 
@@ -176,6 +183,7 @@ def get(self, table_name: str, response_fields: Optional[list] = None, primary_k
         res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
 

pyepsilla/vectordb/client.py

@@ -35,6 +35,7 @@ def welcome(self):
         res = requests.get(url=req_url, data=json.dumps(req_data), headers=self._header, timeout=self._timeout)
         status_code = res.status_code
         body = res.text
+        res.close()
         return status_code, body
 
     def state(self):
@@ -43,6 +44,7 @@ def state(self):
         res = requests.get(url=req_url, data=json.dumps(req_data), headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
     def use_db(self, db_name: str):
@@ -58,6 +60,7 @@ def load_db(self, db_name: str, db_path: str, vector_scale: int = None, wal_enab
         res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
 
@@ -66,6 +69,7 @@ def unload_db(self, db_name: str):
         res = requests.post(url=req_url, data=None, headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
     def create_table(self, table_name: str, table_fields: list[str] = None):
@@ -78,6 +82,7 @@ def create_table(self, table_name: str, table_fields: list[str] = None):
         res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
     def list_tables(self):
@@ -87,6 +92,7 @@ def list_tables(self):
         res = requests.get(url=req_url, headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
 
@@ -100,6 +106,7 @@ def insert(self, table_name: str, records: list = None):
         res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
     def delete(self, table_name: str, primary_keys: list[Union[str,int]] = None, ids: list[Union[str,int]] = None, filter: Optional[str] = None):
@@ -128,6 +135,7 @@ def delete(self, table_name: str, primary_keys: list[Union[str,int]] = None, ids
         res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
 
@@ -141,6 +149,7 @@ def rebuild(self, timeout: int = 7200):
         print("[INFO] Start Time:{}\n       End   Time:{}".format(start_time, end_time))
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
     def query(
@@ -171,6 +180,7 @@ def query(
         res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
     def get(self, table_name: str, response_fields: Optional[list] = None, primary_keys: Optional[list[Union[str,int]]] = None, ids: Optional[list[Union[str,int]]] = None, filter: Optional[str] = None, skip: Optional[int] = None, limit: Optional[int] = None):
@@ -206,6 +216,7 @@ def get(self, table_name: str, response_fields: Optional[list] = None, primary_k
         res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
 
@@ -217,11 +228,13 @@ def drop_table(self, table_name: str = None):
         res = requests.delete(url=req_url, data=json.dumps(req_data), headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body
 
     def drop_db(self, db_name: str):
         req_url = "{}/api/{}/drop".format(self._baseurl, db_name)
         res = requests.delete(url=req_url, data=None, headers=self._header)
         status_code = res.status_code
         body = res.json()
+        res.close()
         return status_code, body

pyepsilla/vectordb/version.py

@@ -1 +1 @@
-__version__ = '0.1.21'
+__version__ = '0.1.22'