feat: AsyncPosthog privacy_mode and enable_full_ai_capture

Author: eli-r-ph

posthog/async_client.py

@@ -113,8 +113,8 @@ class AsyncClient:
     events, on a separate queue that posts to the AI capture endpoint.
     ``flush()``, ``join()``, and ``shutdown()`` are awaitable lifecycle methods.
 
-    The ``capture_ai_*`` options configure the AI queue the same way as on
-    :class:`posthog.Client`.
+    The ``capture_ai_*``, ``privacy_mode`` and ``enable_full_ai_capture``
+    options work the same way as on :class:`posthog.Client`.
     """
 
     log = logging.getLogger("posthog")
@@ -153,6 +153,8 @@ def __init__(
         capture_ai_max_queue_size: int = 1000,
         capture_ai_timeout: float = 30,
         capture_ai_max_event_bytes: int = AI_MAX_MSG_SIZE,
+        privacy_mode: bool = False,
+        enable_full_ai_capture: bool = False,
         capture_trace_context: bool = False,
         secret_key: Optional[str] = None,
         personal_api_key: Optional[str] = None,
@@ -196,6 +198,8 @@ def __init__(
             capture_ai_compression
         )
         self.capture_trace_context = capture_trace_context
+        self.privacy_mode = privacy_mode
+        self.enable_full_ai_capture = enable_full_ai_capture is True
         if personal_api_key is not None and secret_key is None:
             warnings.warn(
                 "`personal_api_key` is deprecated; use `secret_key` instead.",

posthog/test/test_ai_capture_lane.py

@@ -9,6 +9,7 @@
 import posthog
 
 from posthog.ai.utils import _capture_ai_event, finalize_ai_content, with_privacy_mode
+from posthog.async_client import AsyncPosthog
 from posthog.capture_compression import CAPTURE_COMPRESSION_ENV_VAR, CaptureCompression
 from posthog.client import Client
 from posthog.consumer import AI_MAX_MSG_SIZE, AI_MAX_PROPERTIES_SIZE, MAX_MSG_SIZE
@@ -697,14 +698,16 @@ def drop_uuid(event):
 class TestCaptureAiPrivacyMode(unittest.TestCase):
     """Privacy mode always wins over `enable_full_ai_capture`."""
 
-    def test_privacy_mode_strips_content_despite_full_ai_capture(self):
-        client = Client(
+    @parameterized.expand([("sync", Client), ("async", AsyncPosthog)])
+    def test_privacy_mode_strips_content_despite_full_ai_capture(
+        self, _name, client_cls
+    ):
+        client = client_cls(
             TEST_API_KEY,
             send=False,
             enable_full_ai_capture=True,
             privacy_mode=True,
         )
-        self.addCleanup(client.join)
         payload = {"role": "user", "content": "sensitive prompt"}
 
         sanitized = with_privacy_mode(
@@ -713,6 +716,33 @@ def test_privacy_mode_strips_content_despite_full_ai_capture(self):
 
         self.assertIsNone(sanitized)
 
+    @parameterized.expand(
+        [
+            ("sync_default", Client, False),
+            ("sync_full_capture", Client, True),
+            ("async_default", AsyncPosthog, False),
+            ("async_full_capture", AsyncPosthog, True),
+        ]
+    )
+    def test_full_ai_capture_controls_media_redaction(
+        self, _name, client_cls, full_capture
+    ):
+        client = client_cls(
+            TEST_API_KEY, send=False, enable_full_ai_capture=full_capture
+        )
+        image = "data:image/jpeg;base64," + "A" * 64
+        payload = [
+            {
+                "role": "user",
+                "content": [{"type": "image_url", "image_url": {"url": image}}],
+            }
+        ]
+
+        sanitized = finalize_ai_content(payload, ph_client=client)
+
+        sent_url = sanitized[0]["content"][0]["image_url"]["url"]
+        self.assertEqual(sent_url == image, full_capture)
+
 
 if __name__ == "__main__":
     unittest.main()

references/public_api_snapshot.txt

@@ -668,6 +668,7 @@ attribute posthog.async_client.AsyncClient.debug = debug
 attribute posthog.async_client.AsyncClient.disable_geoip = disable_geoip
 attribute posthog.async_client.AsyncClient.disabled = disabled or not self.api_key
 attribute posthog.async_client.AsyncClient.distinct_ids_feature_flags_reported = SizeLimitedDict(_MAX_DICT_SIZE, set)
+attribute posthog.async_client.AsyncClient.enable_full_ai_capture = enable_full_ai_capture is True
 attribute posthog.async_client.AsyncClient.feature_flags_request_max_retries = max(0, feature_flags_request_max_retries)
 attribute posthog.async_client.AsyncClient.feature_flags_request_timeout_seconds = feature_flags_request_timeout_seconds
 attribute posthog.async_client.AsyncClient.historical_migration = historical_migration
@@ -679,6 +680,7 @@ attribute posthog.async_client.AsyncClient.log_captured_exceptions = log_capture
 attribute posthog.async_client.AsyncClient.max_retries = max(0, max_retries)
 attribute posthog.async_client.AsyncClient.on_error = on_error
 attribute posthog.async_client.AsyncClient.personal_api_key = self.secret_key
+attribute posthog.async_client.AsyncClient.privacy_mode = privacy_mode
 attribute posthog.async_client.AsyncClient.project_root = os.getcwd()
 attribute posthog.async_client.AsyncClient.raw_host = normalize_host(host)
 attribute posthog.async_client.AsyncClient.secret_key = (resolved_secret_key.strip() if isinstance(resolved_secret_key, str) else resolved_secret_key) or None
@@ -1177,7 +1179,7 @@ class posthog.ai.types.TokenUsage
 class posthog.ai.types.ToolInProgress 
 class posthog.args.OptionalCaptureArgs 
 class posthog.args.OptionalSetArgs 
-class posthog.async_client.AsyncClient(project_api_key: str, host: Optional[str] = None, *, debug: bool = False, max_queue_size: int = 10000, send: bool = True, on_error=None, flush_at: int = 100, flush_interval: float = 5.0, max_retries: int = 3, timeout: int = 15, thread: int = 1, disabled: bool = False, disable_geoip: bool = True, is_server: bool = True, historical_migration: bool = False, super_properties: Optional[dict[str, Any]] = None, super_options: Optional[dict[str, Any]] = None, before_send=None, log_captured_exceptions: bool = False, project_root: Optional[str] = None, capture_exception_code_variables: bool = False, code_variables_mask_patterns=None, code_variables_ignore_patterns=None, code_variables_mask_url_credentials=None, code_variables_detect_secrets=None, in_app_modules: Optional[list[str]] = None, capture_compression: Optional[Union[CaptureCompression, str]] = None, capture_ai_compression: Optional[Union[CaptureCompression, str]] = None, capture_ai_max_queue_size: int = 1000, capture_ai_timeout: float = 30, capture_ai_max_event_bytes: int = AI_MAX_MSG_SIZE, capture_trace_context: bool = False, secret_key: Optional[str] = None, personal_api_key: Optional[str] = None, feature_flags_request_timeout_seconds: int = 3, feature_flags_request_max_retries: int = 1)
+class posthog.async_client.AsyncClient(project_api_key: str, host: Optional[str] = None, *, debug: bool = False, max_queue_size: int = 10000, send: bool = True, on_error=None, flush_at: int = 100, flush_interval: float = 5.0, max_retries: int = 3, timeout: int = 15, thread: int = 1, disabled: bool = False, disable_geoip: bool = True, is_server: bool = True, historical_migration: bool = False, super_properties: Optional[dict[str, Any]] = None, super_options: Optional[dict[str, Any]] = None, before_send=None, log_captured_exceptions: bool = False, project_root: Optional[str] = None, capture_exception_code_variables: bool = False, code_variables_mask_patterns=None, code_variables_ignore_patterns=None, code_variables_mask_url_credentials=None, code_variables_detect_secrets=None, in_app_modules: Optional[list[str]] = None, capture_compression: Optional[Union[CaptureCompression, str]] = None, capture_ai_compression: Optional[Union[CaptureCompression, str]] = None, capture_ai_max_queue_size: int = 1000, capture_ai_timeout: float = 30, capture_ai_max_event_bytes: int = AI_MAX_MSG_SIZE, privacy_mode: bool = False, enable_full_ai_capture: bool = False, capture_trace_context: bool = False, secret_key: Optional[str] = None, personal_api_key: Optional[str] = None, feature_flags_request_timeout_seconds: int = 3, feature_flags_request_max_retries: int = 1)
 class posthog.async_client.AsyncPosthog 
 class posthog.bucketed_rate_limiter.BucketedRateLimiter(bucket_size: Number, refill_rate: Number, refill_interval_seconds: Number, on_bucket_rate_limited: Optional[Callable[[Hashable], None]] = None, clock: Callable[[], float] = time.monotonic)
 class posthog.capture_compression.CaptureCompression