From 11ec8b9584b5dfe54863ff8284133b44be4de62f Mon Sep 17 00:00:00 2001 From: kenwoodjw Date: Thu, 5 Jun 2025 22:28:14 +0800 Subject: [PATCH] fix MyPy check Signed-off-by: kenwoodjw --- .../nodes/knowledge_retrieval/knowledge_retrieval_node.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py index c964c252ce..8381bfd8f4 100644 --- a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py +++ b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py @@ -375,17 +375,17 @@ class KnowledgeRetrievalNode(LLMNode): expected_value = self.graph_runtime_state.variable_pool.convert_template( expected_value ).value[0] - if hasattr(expected_value, "value_type"): + if hasattr(expected_value, "value_type") and expected_value is not None: 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 ( + elif expected_value.value_type in ( # type: ignore "array[number]", "array[string]", "array[object]", "array", - ): # type: ignore + ): expected_value = expected_value.value # type: ignore else: raise ValueError("Invalid expected metadata value type")