EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Add SSL Cert Functionality (#224)

Browse Source
This commit is contained in:
Srinivas Billa
2024-03-03 09:41:41 +00:00
committed by GitHub
parent dfb13ac455
commit 01b07ea3ac
2 changed files with 15 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/sglang/utils.py
View File

@@ -88,16 +88,16 @@ class HttpResponse:
return self.resp.status
def http_request(url, json=None, stream=False, auth_token=None):
def http_request(url, json=None, stream=False, auth_token=None, verify=None):
"""A faster version of requests.post with low-level urllib API."""
if stream:
if auth_token is None:
return requests.post(url, json=json, stream=True)
return requests.post(url, json=json, stream=True, verify=verify)
headers = {
"Content-Type": "application/json",
"Authentication": f"Bearer {auth_token}",
}
return requests.post(url, json=json, stream=True, headers=headers)
return requests.post(url, json=json, stream=True, headers=headers, verify=verify)
else:
req = urllib.request.Request(url)
req.add_header("Content-Type", "application/json; charset=utf-8")
@@ -107,7 +107,7 @@ def http_request(url, json=None, stream=False, auth_token=None):
data = None
else:
data = bytes(dumps(json), encoding="utf-8")
resp = urllib.request.urlopen(req, data=data)
resp = urllib.request.urlopen(req, data=data, cafile=verify)
return HttpResponse(resp)