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 = self.graph_runtime_state.variable_pool.convert_template(
expected_value expected_value
).value[0] ).value[0]
if hasattr(expected_value, "value_type"):
if expected_value.value_type == "number": # type: ignore if expected_value.value_type == "number": # type: ignore
expected_value = expected_value.value # type: ignore expected_value = expected_value.value # type: ignore
elif expected_value.value_type == "string": # type: ignore elif expected_value.value_type == "string": # type: ignore

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

Loading…
Cancel
Save