fixed something with models i broke yesterday

This commit is contained in:
Jonathan Dunn 2024-03-14 10:37:06 -04:00
parent 996d44a9b8
commit 3e5423abfe

View File

@ -46,9 +46,9 @@ class Standalone:
self.pattern = pattern
self.args = args
self.model = None
try:
if args.model:
self.model = args.model
except:
else:
try:
self.model = os.environ["DEFAULT_MODEL"]
except:
@ -278,8 +278,8 @@ class Standalone:
gptlist = []
fullOllamaList = []
claudeList = ['claude-3-opus-20240229',
'claude-3-sonnet-20240229',
'claude-3-haiku-20240307',
'claude-3-sonnet-20240229',
'claude-3-haiku-20240307',
'claude-2.1']
try:
models = [model.id.strip()