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

minor

Browse Source
This commit is contained in:
Ying Sheng
2024-02-08 04:35:25 +00:00
parent 405f26b00b
commit a6aa46dd3f
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/sglang/utils.py
View File

@@ -91,6 +91,8 @@ class HttpResponse:
def http_request(url, json=None, stream=False, auth_token=None): def http_request(url, json=None, stream=False, auth_token=None):
"""A faster version of requests.post with low-level urllib API.""" """A faster version of requests.post with low-level urllib API."""
if stream: if stream:
if auth_token is None:
return requests.post(url, json=json, stream=True)
headers = { headers = {
"Content-Type": "application/json", "Content-Type": "application/json",
"Authentication": f"Bearer {auth_token}" "Authentication": f"Bearer {auth_token}"
@@ -99,6 +101,7 @@ def http_request(url, json=None, stream=False, auth_token=None):
else: else:
req = urllib.request.Request(url) req = urllib.request.Request(url)
req.add_header("Content-Type", "application/json; charset=utf-8") req.add_header("Content-Type", "application/json; charset=utf-8")
if auth_token is not None:
req.add_header("Authentication", f"Bearer {auth_token}") req.add_header("Authentication", f"Bearer {auth_token}")
if json is None: if json is None:
data = None data = None