fix mypy check

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

@ -375,6 +375,7 @@ class KnowledgeRetrievalNode(LLMNode):
expected_value = self.graph_runtime_state.variable_pool.convert_template(
expected_value
).value[0]
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

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