github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-07-17 17:02:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
cebtenzzre 2023-10-11 18:11:28 -04:00 committed by GitHub
parent afaa291eab
commit aed2068342
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
gpt4all-bindings/python/gpt4all/gpt4all.py
View File

@ -113,7 +113,10 @@ class GPT4All:
Returns: Returns:
Model list in JSON format. 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 @staticmethod
def retrieve_model( def retrieve_model(
@ -215,6 +218,9 @@ class GPT4All:
download_url = get_download_url(model_filename) download_url = get_download_url(model_filename)
response = requests.get(download_url, stream=True) 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)) total_size_in_bytes = int(response.headers.get("content-length", 0))
block_size = 2**20 # 1 MB block_size = 2**20 # 1 MB