Spaces:
Running
Running
File size: 3,030 Bytes
041d087 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
# import json
# def truncate_result(result, max_length: int = 100000, truncation_indicator: str = "...") -> str:
# """
# Truncate the result to specified length while preserving JSON structure when possible.
# Args:
# result: The result to truncate (can be str, list, dict, or other types)
# max_length: Maximum length of the output string (default: 1000)
# truncation_indicator: String to indicate truncation (default: "...")
# Returns:
# str: Truncated string representation of the result
# """
# if isinstance(result, (dict, list)):
# try:
# result_str = json.dumps(result, ensure_ascii=False)
# except:
# result_str = str(result)
# else:
# result_str = str(result)
# indicator_length = len(truncation_indicator)
# if len(result_str) > max_length:
# # For JSON-like strings, try to find the last complete structure
# if result_str.startswith('{') or result_str.startswith('['):
# # Find last complete element
# pos = max_length - indicator_length
# while pos > 0 and not (
# result_str[pos] in ',]}' and
# result_str[pos:].count('"') % 2 == 0
# ):
# pos -= 1
# if pos > 0:
# return result_str[:pos + 1] + truncation_indicator
# # Default truncation if not JSON or no suitable truncation point found
# return result_str[:max_length - indicator_length] + truncation_indicator
# return result_str
def make_json_serializable(obj):
if isinstance(obj, (str, int, float, bool, type(None))):
return obj
elif isinstance(obj, dict):
return {make_json_serializable(key): make_json_serializable(value) for key, value in obj.items()}
elif isinstance(obj, list):
return [make_json_serializable(element) for element in obj]
elif hasattr(obj, '__dict__'):
return make_json_serializable(obj.__dict__)
else:
return str(obj)
def make_json_serializable_truncated(obj, max_length: int = 100000):
if isinstance(obj, (int, float, bool, type(None))):
if isinstance(obj, (int, float)) and len(str(obj)) > max_length:
return str(obj)[:max_length - 3] + "..."
return obj
elif isinstance(obj, str):
return obj if len(obj) <= max_length else obj[:max_length - 3] + "..."
elif isinstance(obj, dict):
return {make_json_serializable_truncated(key, max_length): make_json_serializable_truncated(value, max_length)
for key, value in obj.items()}
elif isinstance(obj, list):
return [make_json_serializable_truncated(element, max_length) for element in obj]
elif hasattr(obj, '__dict__'):
return make_json_serializable_truncated(obj.__dict__, max_length)
else:
result = str(obj)
return result if len(result) <= max_length else result[:max_length - 3] + "..."
|