# 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] + "..."