feat(api): exclude the `loop_round` field while saving LOOP outputs

pull/20699/head
QuantumGhost 11 months ago
parent 6bae477fc0
commit 7aed680953

@ -522,6 +522,7 @@ class DraftVariableSaver:
# that aren't meant to be directly edited or viewed by users. # that aren't meant to be directly edited or viewed by users.
_EXCLUDE_VARIABLE_NAMES_MAPPING: dict[NodeType, frozenset[str]] = { _EXCLUDE_VARIABLE_NAMES_MAPPING: dict[NodeType, frozenset[str]] = {
NodeType.LLM: frozenset(["finish_reason"]), NodeType.LLM: frozenset(["finish_reason"]),
NodeType.LOOP: frozenset(["loop_round"]),
} }
# Database session used for persisting draft variables. # Database session used for persisting draft variables.

Loading…
Cancel
Save