fix: adjust serialization

Author: ablaszkiewicz

posthog/exception_utils.py

@@ -948,9 +948,13 @@ def _serialize_variable_value(value, limiter, max_length=1024):
         elif isinstance(value, bool):
             result = str(value)
         elif isinstance(value, (int, float)):
-            result = str(value)
+            result_size = len(str(value))
+            if not limiter.can_add(result_size):
+                return None
+            limiter.add(result_size)
+            return value
         elif isinstance(value, str):
-            result = repr(value)
+            result = value
         else:
             result = json.dumps(value)
 

posthog/test/test_exception_capture.py

@@ -92,22 +92,22 @@ def process_data():
     assert b"code_variables" in output
 
     # Variables from trigger_error frame
-    assert b"'my_string': \"'hello world'\"" in output
-    assert b"'my_number': '42'" in output
+    assert b"'my_string': 'hello world'" in output
+    assert b"'my_number': 42" in output
     assert b"'my_bool': 'True'" in output
     assert b'"my_dict": "{\\"name\\": \\"test\\", \\"value\\": 123}"' in output
     assert b'"my_obj": "<UnserializableObject>"' in output
     assert b"'my_password': '$$_posthog_redacted_based_on_masking_rules_$$'" in output
     assert b"'__should_be_ignored':" not in output
 
     # Variables from intermediate_function frame
-    assert b"'request_id': \"'abc-123'\"" in output
-    assert b"'user_count': '100'" in output
+    assert b"'request_id': 'abc-123'" in output
+    assert b"'user_count': 100" in output
     assert b"'is_active': 'True'" in output
 
     # Variables from process_data frame
-    assert b"'batch_size': '50'" in output
-    assert b"'retry_count': '3'" in output
+    assert b"'batch_size': 50" in output
+    assert b"'retry_count': 3" in output
 
 
 def test_code_variables_context_override(tmpdir):
@@ -152,7 +152,7 @@ def process_data():
     assert b"ZeroDivisionError" in output
     assert b"code_variables" in output
     assert b"'bank': '$$_posthog_redacted_based_on_masking_rules_$$'" in output
-    assert b"'__dunder_var': \"'should_be_visible'\"" in output
+    assert b"'__dunder_var': 'should_be_visible'" in output
 
 
 def test_code_variables_size_limiter(tmpdir):