Skip to content

Commit 82d2686

Browse files
eric-epsillaeric-epsilla
committed
fix the error of [SSL: CERTIFICATE_VERIFY_FAILED] certificate verify failed: self signed certificate
1 parent c5d2163 commit 82d2686

2 files changed

Lines changed: 25 additions & 27 deletions

File tree

‎pyepsilla/cloud/client.py‎

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -3,18 +3,19 @@
33
import json, datetime, socket, requests, json, pprint
44
from typing import Union, Optional
55
import sentry_sdk
6+
requests.packages.urllib3.disable_warnings()
67

78
class Client(object):
89
def __init__(self, project_id: str, api_key: str):
910
self._project_id = project_id
1011
self._apikey = api_key
1112
self._baseurl = "https://dispatch.epsilla.com/api/v2/project/{}".format(self._project_id)
1213
self._timeout = 10
13-
self._header = {'Content-type': 'application/json', 'X-API-Key': api_key}
14+
self._header = {'Content-type': 'application/json', "Connection": "close", 'X-API-Key': api_key}
1415

1516

1617
def validate(self):
17-
res = requests.get(url=self._baseurl, data=None, headers=self._header)
18+
res = requests.get(url=self._baseurl, data=None, headers=self._header, verify=False)
1819
data = res.json()
1920
res.close()
2021
return data
@@ -23,7 +24,7 @@ def validate(self):
2324
def get_db_list(self):
2425
db_list = []
2526
req_url = "{}/vectordb/list".format(self._baseurl)
26-
res = requests.get(url=req_url, data=None, headers=self._header)
27+
res = requests.get(url=req_url, data=None, headers=self._header, verify=False)
2728
status_code = res.status_code
2829
body = res.json()
2930
if status_code == 200 and body["statusCode"] == 200:
@@ -34,7 +35,7 @@ def get_db_list(self):
3435

3536
def get_db_info(self, db_id: str):
3637
req_url = "{}/vectordb/{}".format(self._baseurl, db_id)
37-
res = requests.get(url=req_url, data=None, headers=self._header)
38+
res = requests.get(url=req_url, data=None, headers=self._header, verify=False)
3839
status_code = res.status_code
3940
body = res.json()
4041
res.close()
@@ -95,7 +96,7 @@ def drop_table(self, table_name: str):
9596
def insert(self, table_name: str, records: list[dict]):
9697
req_url = "{}/data/insert".format(self._baseurl)
9798
req_data = {"table": table_name, "data": records}
98-
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
99+
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header, verify=False)
99100
status_code = res.status_code
100101
body = res.json()
101102
res.close()
@@ -119,7 +120,7 @@ def query(self, table_name: str, query_field: str = None, query_vector: list = N
119120
if with_distance != None:
120121
req_data["withDistance"] = with_distance
121122

122-
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
123+
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header, verify=False)
123124
status_code = res.status_code
124125
body = res.json()
125126
res.close()
@@ -148,7 +149,7 @@ def delete(self, table_name: str, primary_keys: Optional[list[Union[str,int]]] =
148149
if filter != None:
149150
req_data["filter"] = filter
150151

151-
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
152+
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header, verify=False)
152153
status_code = res.status_code
153154
body = res.json()
154155
res.close()
@@ -180,7 +181,7 @@ def get(self, table_name: str, response_fields: Optional[list] = None, primary_k
180181
req_data["limit"] = limit
181182

182183
req_url = "{}/data/get".format(self._baseurl)
183-
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
184+
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header, verify=False)
184185
status_code = res.status_code
185186
body = res.json()
186187
res.close()

‎pyepsilla/vectordb/client.py‎

Lines changed: 16 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,8 @@
11
#!/usr/bin/env python
22
# -*- coding:utf-8 -*-
3-
import json
4-
import requests
5-
import socket
6-
import datetime, time
7-
import sentry_sdk
3+
import json, requests, socket, datetime, time, sentry_sdk
84
from typing import Union, Optional
5+
requests.packages.urllib3.disable_warnings()
96

107

118
class Client():
@@ -16,7 +13,7 @@ def __init__(self, protocol: str = 'http', host: str = 'localhost', port: str =
1613
self._baseurl = "{}://{}:{}".format(self._protocol, self._host, self._port)
1714
self._db = None
1815
self._timeout = 10
19-
self._header = {'Content-type': 'application/json'}
16+
self._header = {'Content-type': 'application/json', "Connection": "close"}
2017
self.check_networking()
2118

2219
def check_networking(self):
@@ -32,7 +29,7 @@ def check_networking(self):
3229
def welcome(self):
3330
req_url = "{}/".format(self._baseurl)
3431
req_data = None
35-
res = requests.get(url=req_url, data=json.dumps(req_data), headers=self._header, timeout=self._timeout)
32+
res = requests.get(url=req_url, data=json.dumps(req_data), headers=self._header, timeout=self._timeout, verify=False)
3633
status_code = res.status_code
3734
body = res.text
3835
res.close()
@@ -41,7 +38,7 @@ def welcome(self):
4138
def state(self):
4239
req_url = "{}/state".format(self._baseurl)
4340
req_data = None
44-
res = requests.get(url=req_url, data=json.dumps(req_data), headers=self._header)
41+
res = requests.get(url=req_url, data=json.dumps(req_data), headers=self._header, verify=False)
4542
status_code = res.status_code
4643
body = res.json()
4744
res.close()
@@ -57,7 +54,7 @@ def load_db(self, db_name: str, db_path: str, vector_scale: int = None, wal_enab
5754
req_data["vectorScale"] = vector_scale
5855
if wal_enabled is not None:
5956
req_data["walEnabled"] = wal_enabled
60-
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
57+
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header, verify=False)
6158
status_code = res.status_code
6259
body = res.json()
6360
res.close()
@@ -66,7 +63,7 @@ def load_db(self, db_name: str, db_path: str, vector_scale: int = None, wal_enab
6663

6764
def unload_db(self, db_name: str):
6865
req_url = "{}/api/{}/unload".format(self._baseurl, db_name)
69-
res = requests.post(url=req_url, data=None, headers=self._header)
66+
res = requests.post(url=req_url, data=None, headers=self._header, verify=False)
7067
status_code = res.status_code
7168
body = res.json()
7269
res.close()
@@ -79,7 +76,7 @@ def create_table(self, table_name: str, table_fields: list[str] = None):
7976
table_fields = []
8077
req_url = "{}/api/{}/schema/tables".format(self._baseurl, self._db)
8178
req_data = {"name": table_name, "fields": table_fields}
82-
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
79+
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header, verify=False)
8380
status_code = res.status_code
8481
body = res.json()
8582
res.close()
@@ -89,7 +86,7 @@ def list_tables(self):
8986
if self._db is None:
9087
raise Exception("[ERROR] Please use_db() first!")
9188
req_url = "{}/api/{}/schema/tables/show".format(self._baseurl, self._db)
92-
res = requests.get(url=req_url, headers=self._header)
89+
res = requests.get(url=req_url, headers=self._header, verify=False)
9390
status_code = res.status_code
9491
body = res.json()
9592
res.close()
@@ -103,7 +100,7 @@ def insert(self, table_name: str, records: list = None):
103100
records = []
104101
req_url = "{}/api/{}/data/insert".format(self._baseurl, self._db)
105102
req_data = {"table": table_name, "data": records}
106-
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
103+
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header, verify=False)
107104
status_code = res.status_code
108105
body = res.json()
109106
res.close()
@@ -132,7 +129,7 @@ def delete(self, table_name: str, primary_keys: list[Union[str,int]] = None, ids
132129
req_data["primaryKeys"] = primary_keys
133130
if filter != None:
134131
req_data["filter"] = filter
135-
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
132+
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header, verify=False)
136133
status_code = res.status_code
137134
body = res.json()
138135
res.close()
@@ -144,7 +141,7 @@ def rebuild(self, timeout: int = 7200):
144141
req_data = None
145142
print("[INFO] waiting until rebuild is finished ...")
146143
start_time = datetime.datetime.now().strftime("%Y-%m-%dT%H:%M:%S")
147-
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header, timeout=timeout)
144+
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header, timeout=timeout, verify=False)
148145
end_time = datetime.datetime.now().strftime("%Y-%m-%dT%H:%M:%S")
149146
print("[INFO] Start Time:{}\n End Time:{}".format(start_time, end_time))
150147
status_code = res.status_code
@@ -177,7 +174,7 @@ def query(
177174
"filter": filter,
178175
"withDistance": with_distance
179176
}
180-
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
177+
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header, verify=False)
181178
status_code = res.status_code
182179
body = res.json()
183180
res.close()
@@ -213,7 +210,7 @@ def get(self, table_name: str, response_fields: Optional[list] = None, primary_k
213210
req_data["limit"] = filter
214211

215212
req_url = "{}/api/{}/data/get".format(self._baseurl, self._db)
216-
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header)
213+
res = requests.post(url=req_url, data=json.dumps(req_data), headers=self._header, verify=False)
217214
status_code = res.status_code
218215
body = res.json()
219216
res.close()
@@ -225,15 +222,15 @@ def drop_table(self, table_name: str = None):
225222
raise Exception("[ERROR] Please use_db() first!")
226223
req_url = "{}/api/{}/schema/tables/{}".format(self._baseurl, self._db, table_name)
227224
req_data = None
228-
res = requests.delete(url=req_url, data=json.dumps(req_data), headers=self._header)
225+
res = requests.delete(url=req_url, data=json.dumps(req_data), headers=self._header, verify=False)
229226
status_code = res.status_code
230227
body = res.json()
231228
res.close()
232229
return status_code, body
233230

234231
def drop_db(self, db_name: str):
235232
req_url = "{}/api/{}/drop".format(self._baseurl, db_name)
236-
res = requests.delete(url=req_url, data=None, headers=self._header)
233+
res = requests.delete(url=req_url, data=None, headers=self._header, verify=False)
237234
status_code = res.status_code
238235
body = res.json()
239236
res.close()

0 commit comments

Comments
 (0)