feat: add bot classification middleware/integration

Part of AI bot classification feature for Python SDK.

Author: jaredmixpanel

mixpanel/__init__.py

@@ -30,6 +30,9 @@
 from .flags.remote_feature_flags import RemoteFeatureFlagsProvider
 from .flags.types import LocalFlagsConfig, RemoteFlagsConfig
 
+from .ai_bot_classifier import classify_user_agent, create_classifier, get_bot_database
+from .ai_bot_consumer import BotClassifyingConsumer
+
 __version__ = '5.1.0'
 
 logger = logging.getLogger(__name__)

mixpanel/ai_bot_helpers.py

@@ -0,0 +1,79 @@
+# mixpanel/ai_bot_helpers.py
+"""Framework integration helpers for AI bot classification."""
+
+from typing import Any, Dict, Optional
+
+
+def extract_request_context_django(request: Any) -> Dict[str, str]:
+    """
+    Extract user-agent and IP from a Django HttpRequest.
+
+    Usage:
+        from mixpanel.ai_bot_helpers import extract_request_context_django
+
+        mp.track('user_id', 'page_view', {
+            **extract_request_context_django(request),
+            'page_url': request.path,
+        })
+    """
+    ctx = {}
+    ua = request.META.get('HTTP_USER_AGENT')
+    if ua:
+        ctx['$user_agent'] = ua
+
+    # Django's get_host() + REMOTE_ADDR
+    ip = (
+        request.META.get('HTTP_X_FORWARDED_FOR', '').split(',')[0].strip()
+        or request.META.get('REMOTE_ADDR')
+    )
+    if ip:
+        ctx['$ip'] = ip
+
+    return ctx
+
+
+def extract_request_context_flask(request: Any) -> Dict[str, str]:
+    """
+    Extract user-agent and IP from a Flask request.
+
+    Usage:
+        from mixpanel.ai_bot_helpers import extract_request_context_flask
+
+        mp.track('user_id', 'page_view', {
+            **extract_request_context_flask(request),
+            'page_url': request.path,
+        })
+    """
+    ctx = {}
+    ua = request.headers.get('User-Agent')
+    if ua:
+        ctx['$user_agent'] = ua
+
+    ip = request.remote_addr
+    if ip:
+        ctx['$ip'] = ip
+
+    return ctx
+
+
+def extract_request_context_fastapi(request: Any) -> Dict[str, str]:
+    """
+    Extract user-agent and IP from a FastAPI/Starlette Request.
+
+    Usage:
+        from mixpanel.ai_bot_helpers import extract_request_context_fastapi
+
+        mp.track('user_id', 'page_view', {
+            **extract_request_context_fastapi(request),
+            'page_url': str(request.url),
+        })
+    """
+    ctx = {}
+    ua = request.headers.get('user-agent')
+    if ua:
+        ctx['$user_agent'] = ua
+
+    if request.client:
+        ctx['$ip'] = request.client.host
+
+    return ctx