python: always check status code of HTTP responses (#1502)

replace-pkg-resources
cebtenzzre 12 months ago committed by GitHub
parent afaa291eab
commit aed2068342
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -113,7 +113,10 @@ class GPT4All:
Returns:
Model list in JSON format.
"""
return requests.get("https://gpt4all.io/models/models2.json").json()
resp = requests.get("https://gpt4all.io/models/models2.json")
if resp.status_code != 200:
raise ValueError(f'Request failed: HTTP {resp.status_code} {resp.reason}')
return resp.json()
@staticmethod
def retrieve_model(
@ -215,6 +218,9 @@ class GPT4All:
download_url = get_download_url(model_filename)
response = requests.get(download_url, stream=True)
if response.status_code != 200:
raise ValueError(f'Request failed: HTTP {response.status_code} {response.reason}')
total_size_in_bytes = int(response.headers.get("content-length", 0))
block_size = 2**20 # 1 MB

Loading…
Cancel
Save