[elsa] add knowledgeExtensions params of knowledgeRetrievalNode option struct. (#270)

Co-authored-by: RonnyChan <[email protected]>

Author: wuayee

framework/elsa/fit-elsa-react/src/common/Consts.js

@@ -250,4 +250,21 @@ export const DEFAULT_KNOWLEDGE_NODE_RERANK_TOP_N = {
   type: DATA_TYPES.INTEGER,
   from: FROM_TYPE.INPUT,
   value: 3,
+};
+
+export const DEFAULT_KNOWLEDGE_RETRIEVAL_NODE_EXTENSIONS = {
+    id: `knowledgeExtensions_${uuidv4()}`,
+    name: 'extensions',
+    type: DATA_TYPES.OBJECT,
+    from: FROM_TYPE.EXPAND,
+    value: [{
+        id: uuidv4(),
+        name: VIRTUAL_CONTEXT_NODE_VARIABLES.USER_ID,
+        type: DATA_TYPES.STRING,
+        from: FROM_TYPE.REFERENCE,
+        referenceId: VIRTUAL_CONTEXT_NODE_VARIABLES.USER_ID,
+        referenceKey: VIRTUAL_CONTEXT_NODE_VARIABLES.USER_ID,
+        referenceNode: VIRTUAL_CONTEXT_NODE.id,
+        value: [VIRTUAL_CONTEXT_NODE_VARIABLES.USER_ID],
+    }],
 };

framework/elsa/fit-elsa-react/src/components/knowledgeRetrieval/knowledgeRetrievalComponent.jsx

@@ -14,6 +14,7 @@ import {
   DEFAULT_KNOWLEDGE_REPO_GROUP,
   DEFAULT_KNOWLEDGE_RETRIEVAL_NODE_KNOWLEDGE_CONFIG_ID,
   FROM_TYPE,
+  DEFAULT_KNOWLEDGE_RETRIEVAL_NODE_EXTENSIONS,
 } from '@/common/Consts.js';
 import {
   ChangeAccessInfoReducer,
@@ -152,6 +153,7 @@ export const knowledgeRetrievalComponent = (jadeConfig, shape) => {
             value: DEFAULT_KNOWLEDGE_REPO_GROUP,
           },
           JSON.parse(JSON.stringify(DEFAULT_KNOWLEDGE_RETRIEVAL_NODE_KNOWLEDGE_CONFIG_ID)),
+          JSON.parse(JSON.stringify(DEFAULT_KNOWLEDGE_RETRIEVAL_NODE_EXTENSIONS)),
         ],
       }],
       outputParams: [{

framework/elsa/fit-elsa-react/src/flow/compatibility/compatibilityProcessors.js

@@ -13,6 +13,7 @@ import {
   DEFAULT_KNOWLEDGE_NODE_RERANK_TOP_N,
   DEFAULT_KNOWLEDGE_REPO_GROUP_STRUCT,
   DEFAULT_KNOWLEDGE_RETRIEVAL_NODE_KNOWLEDGE_CONFIG_ID,
+  DEFAULT_KNOWLEDGE_RETRIEVAL_NODE_EXTENSIONS,
   DEFAULT_LLM_KNOWLEDGE_BASES,
   DEFAULT_LLM_REFERENCE_OUTPUT,
   DEFAULT_MAX_MEMORY_ROUNDS,
@@ -481,6 +482,10 @@ export const knowledgeRetrievalCompatibilityProcessor = (shapeData, graph, pageH
           rerankParamValue.push(DEFAULT_KNOWLEDGE_NODE_RERANK_TOP_N);
         }
       }
+
+      if (Array.isArray(optionValue) && !optionValue.some(v => v.name === 'extensions')) {
+        optionValue.push(DEFAULT_KNOWLEDGE_RETRIEVAL_NODE_EXTENSIONS);
+      }
     };
 
     const userIdParamProcess = () => {