ngskcloud
/
gcgj-dify-1.7.0
17
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix MyPy check

Browse Source 
Signed-off-by: kenwoodjw <blackxin55+@gmail.com>
pull/20589/head
kenwoodjw 12 months ago
parent 5e00ee96ba
commit 11ec8b9584
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File

6
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
Unescape Escape View File

@ -375,17 +375,17 @@ 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 hasattr(expected_value, "value_type") and expected_value is not None:
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
expected_value = re.sub(r"[\r\n\t]+", " ", expected_value.text).strip() # 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[number]",
"array[string]", "array[string]",
"array[object]", "array[object]",
"array", "array",
): # type: ignore ):
expected_value = expected_value.value # type: ignore expected_value = expected_value.value # type: ignore
else: else:
raise ValueError("Invalid expected metadata value type") raise ValueError("Invalid expected metadata value type")

Write Preview
Loading…
Cancel
Save