From 6ae99c9c3e70f0b05b91abcb5a05e5c46512a606 Mon Sep 17 00:00:00 2001
From: Ben Sully <ben.sully@grafana.com>
Date: Wed, 27 Sep 2023 16:55:22 +0100
Subject: [PATCH] Add 'Enabled' switch for vector services to configuration UI

---
 src/components/AppConfig/Vector.tsx | 11 +++++++++++
 src/components/testIds.ts           |  1 +
 2 files changed, 12 insertions(+)

diff --git a/src/components/AppConfig/Vector.tsx b/src/components/AppConfig/Vector.tsx
index 2b23c8f4..0b0d78b6 100644
--- a/src/components/AppConfig/Vector.tsx
+++ b/src/components/AppConfig/Vector.tsx
@@ -48,6 +48,17 @@ interface Props<T> {
 export function VectorConfig({ settings, onChange }: Props<VectorSettings>) {
   return (
     <FieldSet label="Vector Settings">
+
+      <Field label="Enabled" description="Enable vector database powered features.">
+        <Switch
+          name="enabled"
+          data-testid={testIds.appConfig.vectorEnabled}
+          defaultChecked={settings?.enabled}
+          checked={settings?.enabled}
+          onChange={e => onChange({ ...settings, enabled: e.currentTarget.checked })}
+        />
+      </Field>
+
       <Field label="Model" description="The model used by the embedder and for embeddings stored in the store">
         <Input
           width={60}
diff --git a/src/components/testIds.ts b/src/components/testIds.ts
index f2751cc1..d33b7c64 100644
--- a/src/components/testIds.ts
+++ b/src/components/testIds.ts
@@ -5,6 +5,7 @@ export const testIds = {
     openAIKey: 'data-testid ac-openai-api-key',
     openAIOrganizationID: 'data-testid ac-openai-api-organization-id',
     openAIUrl: 'data-testid ac-openai-api-url',
+    vectorEnabled: 'data-testid ac-vector-enabled',
     qdrantAddress: 'data-testid ac-qdrant-address',
     qdrantSecure: 'data-testid ac-qdrant-secure',
     grafanaVectorApiUrl: 'data-testid ac-grafana-vector-api-url',