ifEsle node add regex match (#8007)

This commit is contained in:
Charlie.Wei 2024-09-06 17:44:09 +08:00 committed by GitHub
parent 2060db8e11
commit 01858e1caf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
7 changed files with 23 additions and 2 deletions

View File

@ -12,7 +12,7 @@ class Condition(BaseModel):
variable_selector: list[str]
comparison_operator: Literal[
# for string or array
"contains", "not contains", "start with", "end with", "is", "is not", "empty", "not empty",
"contains", "not contains", "start with", "end with", "is", "is not", "empty", "not empty", "regex match",
# for number
"=", "", ">", "<", "", "", "null", "not null"
]

View File

@ -1,3 +1,4 @@
import re
from collections.abc import Sequence
from typing import Optional, cast
@ -136,6 +137,8 @@ class IfElseNode(BaseNode):
return self._assert_null(actual_value)
elif comparison_operator == "not null":
return self._assert_not_null(actual_value)
elif comparison_operator == "regex match":
return self._assert_regex_match(actual_value, expected_value)
else:
raise ValueError(f"Invalid comparison operator: {comparison_operator}")
@ -285,6 +288,21 @@ class IfElseNode(BaseNode):
return True
return False
def _assert_regex_match(self, actual_value: Optional[str], expected_value: str) -> bool:
"""
Assert empty
:param actual_value: actual value
:return:
"""
if actual_value is None:
return False
pattern = re.compile(expected_value)
regex_result = pattern.findall(actual_value)
if len(regex_result) > 0:
return True
return False
def _assert_not_empty(self, actual_value: Optional[str]) -> bool:
"""
Assert not empty

View File

@ -88,7 +88,6 @@ const ConfigVar: FC<IConfigVarProps> = ({ promptVariables, readonly, onPromptVar
} as InputVar
})()
const updatePromptVariableItem = (payload: InputVar) => {
console.log(payload)
const newPromptVariables = produce(promptVariables, (draft) => {
const { variable, label, type, ...rest } = payload
draft[currIndex] = {

View File

@ -28,6 +28,7 @@ export enum ComparisonOperator {
lessThanOrEqual = '≤',
isNull = 'is null',
isNotNull = 'is not null',
regexMatch = 'regex match',
}
export type Condition = {

View File

@ -30,6 +30,7 @@ export const getOperators = (type?: VarType) => {
ComparisonOperator.isNot,
ComparisonOperator.empty,
ComparisonOperator.notEmpty,
ComparisonOperator.regexMatch,
]
case VarType.number:
return [

View File

@ -412,6 +412,7 @@ const translation = {
'not empty': 'is not empty',
'null': 'is null',
'not null': 'is not null',
'regex match': 'regex match',
},
enterValue: 'Enter value',
addCondition: 'Add Condition',

View File

@ -412,6 +412,7 @@ const translation = {
'not empty': '不为空',
'null': '空',
'not null': '不为空',
'regex match': '正则匹配',
},
enterValue: '输入值',
addCondition: '添加条件',