Add feature flag to control query assistant like features

Signed-off-by: Songkan Tang <[email protected]>
opensearch-project · Jul 18, 2024 · baf8f20 · baf8f20
1 parent 2421ad3
commit baf8f20
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 0 deletions.
diff --git a/common/types/config.ts b/common/types/config.ts
@@ -14,6 +14,9 @@ export const configSchema = schema.object({
   incontextInsight: schema.object({
     enabled: schema.boolean({ defaultValue: true }),
   }),
+  next: schema.object({
+    enabled: schema.boolean({ defaultValue: false }),
+  }),
 });
 
 export type ConfigSchema = TypeOf<typeof configSchema>;
diff --git a/public/plugin.tsx b/public/plugin.tsx
@@ -147,6 +147,7 @@ export class AssistantPlugin
         actionExecutors[actionType] = execute;
       },
       chatEnabled: () => this.config.chat.enabled,
+      nextEnabled: () => this.config.next.enabled,
       assistantActions,
       registerIncontextInsight: this.incontextInsightRegistry.register.bind(
         this.incontextInsightRegistry

diff --git a/public/types.ts b/public/types.ts
@@ -46,6 +46,10 @@ export interface AssistantSetup {
    * Returns true if chat UI is enabled.
    */
   chatEnabled: () => boolean;
+  /**
+   * Returns true if contextual assistant is enabled.
+   */
+  nextEnabled: () => boolean;
   assistantActions: Omit<AssistantActions, 'executeAction'>;
   registerIncontextInsight: IncontextInsightRegistry['register'];
   renderIncontextInsight: (component: React.ReactNode) => React.ReactNode;

diff --git a/server/index.ts b/server/index.ts
@@ -11,6 +11,7 @@ export const config: PluginConfigDescriptor<ConfigSchema> = {
   exposeToBrowser: {
     chat: true,
     incontextInsight: true,
+    next: true,
   },
   schema: configSchema,
 };