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 5ab4553793
commit 5e00ee96ba
2 changed files with 15 additions and 14 deletions
Show Stats Download Patch File Download Diff File

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

@ -375,19 +375,20 @@ 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 expected_value.value_type == "number": # type: ignore if hasattr(expected_value, "value_type"):
expected_value = expected_value.value # type: ignore if expected_value.value_type == "number": # type: ignore
elif expected_value.value_type == "string": # type: ignore expected_value = expected_value.value # type: ignore
expected_value = re.sub(r"[\r\n\t]+", " ", expected_value.text).strip() # type: ignore elif expected_value.value_type == "string": # type: ignore
elif expected_value.value_type in ( expected_value = re.sub(r"[\r\n\t]+", " ", expected_value.text).strip() # type: ignore
"array[number]", elif expected_value.value_type in (
"array[string]", "array[number]",
"array[object]", "array[string]",
"array", "array[object]",
): # type: ignore "array",
expected_value = expected_value.value # type: ignore ): # type: ignore
else: expected_value = expected_value.value # type: ignore
raise ValueError("Invalid expected metadata value type") else:
raise ValueError("Invalid expected metadata value type")
elif isinstance(expected_value, list): elif isinstance(expected_value, list):
# For constant array values # For constant array values
pass pass

2
api/core/workflow/utils/condition/processor.py
Unescape Escape View File

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

Write Preview
Loading…
Cancel
Save