From da06009fc13d0a7d27cd9ab36a55c55c5bae9870 Mon Sep 17 00:00:00 2001
From: Alex Pascal <apascal07@gmail.com>
Date: Wed, 26 Jun 2024 16:03:35 -0700
Subject: [PATCH] Update vertexai.go

---
 go/plugins/vertexai/vertexai.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/go/plugins/vertexai/vertexai.go b/go/plugins/vertexai/vertexai.go
index 562ae3e4c5..4ec920488e 100644
--- a/go/plugins/vertexai/vertexai.go
+++ b/go/plugins/vertexai/vertexai.go
@@ -112,6 +112,7 @@ func Init(ctx context.Context, projectID, location string) error {
 	if err != nil {
 		return err
 	}
+	state.initted = true
 	for model, caps := range knownCaps {
 		if _, err := DefineModel(model, &caps); err != nil {
 			return fmt.Errorf("vertexai.Init: failed to define known model %q: %w", model, err)
@@ -120,7 +121,6 @@ func Init(ctx context.Context, projectID, location string) error {
 	for _, e := range knownEmbedders {
 		DefineEmbedder(e)
 	}
-	state.initted = true
 	return nil
 }