Archives/fabric
2
0
Fork 0
mirror of https://github.com/danielmiessler/fabric synced 2024-11-10 07:10:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use safer method to get data from exception

Browse Source 
Signed-off-by: zestysoft <ian@zestysoft.com>
This commit is contained in:
zestysoft 2024-03-12 03:21:01 -07:00
parent 8e871028ad
commit cdde4b8307
No known key found for this signature in database
GPG Key ID: D9923DD6508A15B9
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
installer/client/cli/utils.py
View File

@ -253,13 +253,13 @@ class Standalone:
try:
models = [model.id for model in self.client.models.list().data]
except APIConnectionError as e:
if e.__cause__.args[0] == "Illegal header value b'Bearer '":
if getattr(e.__cause__, 'args', [''])[0] == "Illegal header value b'Bearer '":
print("Error: Cannot connect to the OpenAI API Server because the API key is not set. Please run fabric --setup and add a key.")
else:
print(f'{e.message} trying to access {e.request.url}: {e.__cause__.args}')
print(f'{e.message} trying to access {e.request.url}: {getattr(e.__cause__, 'args', [''])}')
sys.exit()
except Exception as e:
print(f"Error: {e.__context__.args[0]}")
print(f"Error: {getattr(e.__context__, 'args', [''])[0]}")
sys.exit()
if "/" in models[0] or "\\" in models[0]:
# lmstudio returns full paths to models. Iterate and truncate everything before and including the last slash
@ -426,13 +426,13 @@ class Setup:
try:
models = [model.id for model in self.client.models.list().data]
except APIConnectionError as e:
if e.__cause__.args[0] == "Illegal header value b'Bearer '":
if getattr(e.__cause__, 'args', [''])[0] == "Illegal header value b'Bearer '":
print("Error: Cannot connect to the OpenAI API Server because the API key is not set. Please run fabric --setup and add a key.")
else:
print(f'{e.message} trying to access {e.request.url}: {e.__cause__.args}')
print(f'{e.message} trying to access {e.request.url}: {getattr(e.__cause__, 'args', [''])}')
sys.exit()
except Exception as e:
print(f"Error: {e.__context__.args[0]}")
print(f"Error: {getattr(e.__context__, 'args', [''])[0]}")
sys.exit()
if "/" in models[0] or "\\" in models[0]:
# lmstudio returns full paths to models. Iterate and truncate everything before and including the last slash