chore: Filter the `finish_reason` for all LLM-based nodes in draft editor of the variable inspect tool

pull/21766/head
Davide Delbianco 11 months ago
parent 4a917daf16
commit e83e8715e0
No known key found for this signature in database
GPG Key ID: 3C00412F2A31305E

@ -520,6 +520,7 @@ class DraftVariableSaver:
_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"]), NodeType.LOOP: frozenset(["loop_round"]),
NodeType.QUESTION_CLASSIFIER: frozenset(["finish_reason"]),
} }
# Database session used for persisting draft variables. # Database session used for persisting draft variables.

Loading…
Cancel
Save