klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-06 13:22:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Do not try downloading model from GPT4All if the user is not connected to the internet

Browse Source
This commit is contained in:
sabaimran
2023-08-19 19:08:09 -07:00
parent 3ff4e19dd2
commit f9e09ba490
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/khoj/processor/conversation/gpt4all/utils.py
View File

@@ -29,6 +29,12 @@ def download_model(model_name: str):
filename = os.path.expanduser(f"~/.cache/gpt4all/{model_name}") filename = os.path.expanduser(f"~/.cache/gpt4all/{model_name}")
if os.path.exists(filename): if os.path.exists(filename):
# Check if the user is connected to the internet
try:
requests.get("https://www.google.com/", timeout=5)
except:
logger.debug("User is offline. Disabling allowed download flag")
return GPT4All(model_name, allow_download=False)
return GPT4All(model_name) return GPT4All(model_name)
# Download the model to a tmp file. Once the download is completed, move the tmp file to the actual file # Download the model to a tmp file. Once the download is completed, move the tmp file to the actual file