Преглед на файлове

fix: handle non-array segment types in Loop node (#24590)

Co-authored-by: QuantumGhost <obelisk.reg+git@gmail.com>
tags/1.8.0
-LAN- преди 2 месеца
родител
ревизия
726c429772
No account linked to committer's email address
променени са 2 файла, в които са добавени 6 реда и са изтрити 4 реда
  1. 3
    1
      api/core/llm_generator/llm_generator.py
  2. 3
    3
      api/core/workflow/nodes/loop/loop_node.py

+ 3
- 1
api/core/llm_generator/llm_generator.py Целия файл

@@ -572,5 +572,7 @@ class LLMGenerator:
error = str(e)
return {"error": f"Failed to generate code. Error: {error}"}
except Exception as e:
logger.exception("Failed to invoke LLM model, model: " + json.dumps(model_config.get("name")), exc_info=e)
logger.exception(
"Failed to invoke LLM model, model: %s", json.dumps(model_config.get("name")), exc_info=True
)
return {"error": f"An unexpected error occurred: {str(e)}"}

+ 3
- 3
api/core/workflow/nodes/loop/loop_node.py Целия файл

@@ -524,7 +524,9 @@ class LoopNode(BaseNode):
@staticmethod
def _get_segment_for_constant(var_type: SegmentType, original_value: Any) -> Segment:
"""Get the appropriate segment type for a constant value."""
if var_type in [
if not var_type.is_array_type() or var_type == SegmentType.BOOLEAN:
value = original_value
elif var_type in [
SegmentType.ARRAY_NUMBER,
SegmentType.ARRAY_OBJECT,
SegmentType.ARRAY_STRING,
@@ -534,8 +536,6 @@ class LoopNode(BaseNode):
else:
logger.warning("unexpected value for LoopNode, value_type=%s, value=%s", original_value, var_type)
value = []
elif var_type == SegmentType.ARRAY_BOOLEAN:
value = original_value
else:
raise AssertionError("this statement should be unreachable.")
try:

Loading…
Отказ
Запис