mirror of
https://github.com/langgenius/dify.git
synced 2025-12-19 17:27:16 -05:00
Fix UpdatedVariable truncation crash (#27359)
Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
@@ -283,7 +283,7 @@ class VariableTruncator:
|
|||||||
break
|
break
|
||||||
|
|
||||||
remaining_budget = target_size - used_size
|
remaining_budget = target_size - used_size
|
||||||
if item is None or isinstance(item, (str, list, dict, bool, int, float)):
|
if item is None or isinstance(item, (str, list, dict, bool, int, float, UpdatedVariable)):
|
||||||
part_result = self._truncate_json_primitives(item, remaining_budget)
|
part_result = self._truncate_json_primitives(item, remaining_budget)
|
||||||
else:
|
else:
|
||||||
raise UnknownTypeError(f"got unknown type {type(item)} in array truncation")
|
raise UnknownTypeError(f"got unknown type {type(item)} in array truncation")
|
||||||
@@ -373,6 +373,11 @@ class VariableTruncator:
|
|||||||
|
|
||||||
return _PartResult(truncated_obj, used_size, truncated)
|
return _PartResult(truncated_obj, used_size, truncated)
|
||||||
|
|
||||||
|
@overload
|
||||||
|
def _truncate_json_primitives(
|
||||||
|
self, val: UpdatedVariable, target_size: int
|
||||||
|
) -> _PartResult[Mapping[str, object]]: ...
|
||||||
|
|
||||||
@overload
|
@overload
|
||||||
def _truncate_json_primitives(self, val: str, target_size: int) -> _PartResult[str]: ...
|
def _truncate_json_primitives(self, val: str, target_size: int) -> _PartResult[str]: ...
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user