Manage location on VertexAI instead of model (#12630)

This commit is contained in:
Paul Beusterien authored and Andrew Heard committed 2024-05-13 12:14:00 -04:00
1 parent fdc2985678
commit 0f1fdc888c
6 files changed
+33 -37

No files matched your search

@@ -36,10 +36,7 @@ class ConversationViewModel: ObservableObject {
private var chatTask: Task<Void, Never>?
init() {
model = VertexAI.vertexAI().generativeModel(
modelName: "gemini-1.0-pro",
location: "us-central1"
)
model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
chat = model.startChat()
}