added an error message

This commit is contained in:
Jonathan Dunn 2024-03-06 10:39:45 -05:00
parent 4562f0564b
commit 19dddd9ffd

View File

@ -562,6 +562,11 @@ class Setup:
model = model.strip() model = model.strip()
if model: if model:
# Write or update the DEFAULT_MODEL in env_file # Write or update the DEFAULT_MODEL in env_file
allModels = self.claudeList + self.fullOllamaList + self.gptlist
if model not in allModels:
print(
f"Error: {model} is not a valid model. Please run fabric --listmodels to see the available models.")
sys.exit()
if os.path.exists(self.env_file): if os.path.exists(self.env_file):
with open(self.env_file, "r") as f: with open(self.env_file, "r") as f:
lines = f.readlines() lines = f.readlines()