Added Google Vertex AI support

This commit is contained in:
based
2023-12-11 03:44:49 +10:00
parent 782864d6cc
commit 266bc976ea
5 changed files with 83 additions and 8 deletions
+5
View File
@@ -36,6 +36,10 @@ class APIKey:
self.deployments = []
self.unfiltered = False
elif provider == Provider.VERTEXAI:
self.project_id = ""
pass
class Provider(Enum):
OPENAI = 1,
@@ -44,3 +48,4 @@ class Provider(Enum):
PALM = 4
AWS = 5
AZURE = 6
VERTEXAI = 7