Enable passing reader/retriever configuration

eea · Oct 18, 2021 · 3b8ffc9 · 3b8ffc9
1 parent fc6965d
commit 3b8ffc9
Showing 1 changed file with 29 additions and 1 deletion.
diff --git a/src/SearchBlock/schema.js b/src/SearchBlock/schema.js
@@ -19,7 +19,15 @@ export const SearchBlockSchema = ({ formData = {} }) => ({
           {
             id: 'nlp',
             title: 'NLP Capabilities Settings',
-            fields: ['use_qa_dp', 'qa_queryTypes', 'cutoffScore'],
+            fields: [
+              'use_qa_dp',
+              'qa_queryTypes',
+              'cutoffScore',
+              'rawIndex',
+              'dprIndex',
+              'topKRetriever',
+              'topKReader',
+            ],
           },
         ]
       : []),
@@ -78,6 +86,26 @@ export const SearchBlockSchema = ({ formData = {} }) => ({
       default: 0.1,
       configPath: 'nlp.qa.cutoffScore',
     },
+    rawIndex: {
+      title: 'ElasticSearch Index',
+      configPath: 'nlp.qa.raw_index',
+    },
+    dprIndex: {
+      title: 'DPR ElasticSearch Index',
+      configPath: 'nlp.qa.dpr_index',
+    },
+    topKRetriever: {
+      title: 'TopK Retriever',
+      type: 'number',
+      defaultValue: 10,
+      configPath: 'nlp.qa.topk_retriever',
+    },
+    topKReader: {
+      title: 'TopK Reader',
+      type: 'number',
+      defaultValue: 10,
+      configPath: 'nlp.qa.topk_reader',
+    },
   },
 
   required: ['appName'],