ifEsle node add regex match (#8007)

pull/8059/head^2
Charlie.Wei 2 years ago committed by GitHub
parent 2060db8e11
commit 01858e1caf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -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"
]

@ -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

@ -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] = {

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

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

@ -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',

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

Loading…
Cancel
Save