fix mypy check

Signed-off-by: kenwoodjw <blackxin55+@gmail.com>
pull/20589/head
kenwoodjw 12 months ago
parent 5ab4553793
commit 5e00ee96ba

@ -375,19 +375,20 @@ class KnowledgeRetrievalNode(LLMNode):
expected_value = self.graph_runtime_state.variable_pool.convert_template(
expected_value
).value[0]
if expected_value.value_type == "number": # type: ignore
expected_value = expected_value.value # type: ignore
elif expected_value.value_type == "string": # type: ignore
expected_value = re.sub(r"[\r\n\t]+", " ", expected_value.text).strip() # type: ignore
elif expected_value.value_type in (
"array[number]",
"array[string]",
"array[object]",
"array",
): # type: ignore
expected_value = expected_value.value # type: ignore
else:
raise ValueError("Invalid expected metadata value type")
if hasattr(expected_value, "value_type"):
if expected_value.value_type == "number": # type: ignore
expected_value = expected_value.value # type: ignore
elif expected_value.value_type == "string": # type: ignore
expected_value = re.sub(r"[\r\n\t]+", " ", expected_value.text).strip() # type: ignore
elif expected_value.value_type in (
"array[number]",
"array[string]",
"array[object]",
"array",
): # type: ignore
expected_value = expected_value.value # type: ignore
else:
raise ValueError("Invalid expected metadata value type")
elif isinstance(expected_value, list):
# For constant array values
pass

@ -77,7 +77,7 @@ def _evaluate_condition(
*,
operator: SupportedComparisonOperator,
value: Any,
expected: str | Sequence[str] | None,
expected: str | Sequence[str] | Sequence[int] | Sequence[float] | int | float | None,
) -> bool:
match operator:
case "contains":

Loading…
Cancel
Save