fixed gemini support

This commit is contained in:
xssdoctor 2024-05-14 18:52:00 -04:00
parent e2231b3504
commit ca9fb0f65a

View File

@ -732,8 +732,8 @@ class Setup:
model = model.strip() model = model.strip()
env = os.path.expanduser("~/.config/fabric/.env") env = os.path.expanduser("~/.config/fabric/.env")
standalone = Standalone(args=[], pattern="") standalone = Standalone(args=[], pattern="")
gpt, ollama, claude = standalone.fetch_available_models() gpt, ollama, claude, google = standalone.fetch_available_models()
allmodels = gpt + ollama + claude allmodels = gpt + ollama + claude + google
if model not in allmodels: if model not in allmodels:
print( print(
f"Error: {model} is not a valid model. Please run fabric --listmodels to see the available models.") f"Error: {model} is not a valid model. Please run fabric --listmodels to see the available models.")
@ -793,6 +793,9 @@ class Setup:
print("Please enter your claude API key. If you do not have one, or if you have already entered it, press enter.\n") print("Please enter your claude API key. If you do not have one, or if you have already entered it, press enter.\n")
claudekey = input() claudekey = input()
self.claude_key(claudekey) self.claude_key(claudekey)
print("Please enter your Google API key. If you do not have one, or if you have already entered it, press enter.\n")
googlekey = input()
self.google_key(googlekey)
print("Please enter your YouTube API key. If you do not have one, or if you have already entered it, press enter.\n") print("Please enter your YouTube API key. If you do not have one, or if you have already entered it, press enter.\n")
youtubekey = input() youtubekey = input()
self.youtube_key(youtubekey) self.youtube_key(youtubekey)