Add 'enabled' setting to vector settings to globally enable/disable v…

…ector services
grafana · Sep 14, 2023 · bbfa61c · bbfa61c
1 parent 8e6985b
commit bbfa61c
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 9 deletions.
diff --git a/pkg/plugin/app.go b/pkg/plugin/app.go
@@ -45,14 +45,18 @@ func NewApp(appSettings backend.AppInstanceSettings) (instancemgmt.Instance, err
 	settings := loadSettings(appSettings)
 	var err error
 
-	log.DefaultLogger.Debug("Creating vector service")
-	app.vectorService, err = vector.NewService(settings.Vector, appSettings.DecryptedSecureJSONData)
-	if err != nil {
-		log.DefaultLogger.Error("Error creating vector service", "err", err)
-		return nil, err
+	if settings.Vector.Enabled {
+		log.DefaultLogger.Debug("Creating vector service")
+		app.vectorService, err = vector.NewService(
+			settings.Vector,
+			appSettings.DecryptedSecureJSONData,
+		)
+		if err != nil {
+			log.DefaultLogger.Error("Error creating vector service", "err", err)
+			return nil, err
+		}
 	}
 
-	log.DefaultLogger.Debug("App instance created")
 	return &app, nil
 }
 

diff --git a/pkg/plugin/vector/service.go b/pkg/plugin/vector/service.go
@@ -17,9 +17,10 @@ type Service interface {
 }
 
 type VectorSettings struct {
-	Model string         `json:"model"`
-	Embed embed.Settings `json:"embed"`
-	Store store.Settings `json:"store"`
+	Enabled bool           `json:"enabled"`
+	Model   string         `json:"model"`
+	Embed   embed.Settings `json:"embed"`
+	Store   store.Settings `json:"store"`
 }
 
 type vectorService struct {

diff --git a/provisioning/plugins/grafana-llm-app.yaml b/provisioning/plugins/grafana-llm-app.yaml
@@ -4,6 +4,7 @@ apps:
   - type: grafana-llm-app
     jsonData:
       vector:
+        enabled: true
         model: text-embedding-ada-002
         embed:
           type: openai