Add in and not in filtering to the metadata of the knowledge-retrieval node

api/core/app/app_config/entities.py

@@ -147,6 +147,8 @@ SupportedComparisonOperator = Literal[
     "is not",
     "empty",
     "not empty",
+    "in",
+    "not in",
     # for number
     "=",
     "≠",

api/core/rag/entities/metadata_entities.py

@@ -13,6 +13,8 @@ SupportedComparisonOperator = Literal[
     "is not",
     "empty",
     "not empty",
+    "in",
+    "not in",
     # for number
     "=",
     "≠",

api/core/workflow/nodes/knowledge_retrieval/entities.py

@@ -85,6 +85,8 @@ SupportedComparisonOperator = Literal[
     "is not",
     "empty",
     "not empty",
+    "in",
+    "not in",
     # for number
     "=",
     "≠",

api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py

@@ -509,6 +509,18 @@ class KnowledgeRetrievalNode(LLMNode):
                         **{key: metadata_name, key_value: f"%{value}"}
                     )
                 )
+            case "in":
+                filters.append(
+                    (text(f"documents.doc_metadata ->> :{key} = any(string_to_array(:{key_value},',')")).params(
+                        **{key: metadata_name, key_value: f"{value}"}
+                    )
+                )
+            case "not in":
+                filters.append(
+                    (text(f"documents.doc_metadata ->> :{key} != all(string_to_array(:{key_value},',')")).params(
+                        **{key: metadata_name, key_value: f"{value}"}
+                    )
+                )
             case "=" | "is":
                 if isinstance(value, str):
                     filters.append(Document.doc_metadata[metadata_name] == f'"{value}"')

web/app/components/workflow/nodes/knowledge-retrieval/components/metadata/condition-list/utils.ts

@@ -32,6 +32,8 @@ export const getOperators = (type?: MetadataFilteringVariableType) => {
         ComparisonOperator.endWith,
         ComparisonOperator.empty,
         ComparisonOperator.notEmpty,
+        ComparisonOperator.in,
+        ComparisonOperator.notIn,
       ]
     case MetadataFilteringVariableType.number:
       return [

web/i18n/zh-Hans/workflow.ts

@@ -576,8 +576,8 @@ const translation = {
         'not empty': '不为空',
         'null': '空',
         'not null': '不为空',
-        'in': '是',
+        'in': '在',
-        'not in': '不是',
+        'not in': '不在',
         'all of': '全部是',
         'exists': '存在',
         'not exists': '不存在',