lmnr 0.4.55__tar.gz → 0.4.56__tar.gz

{lmnr-0.4.55 → lmnr-0.4.56}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lmnr
-Version: 0.4.55
+Version: 0.4.56
 Summary: Python SDK for Laminar
 License: Apache-2.0
 Author: lmnr.ai

{lmnr-0.4.55 → lmnr-0.4.56}/pyproject.toml

@@ -6,7 +6,7 @@
 
 [project]
 name = "lmnr"
-version = "0.4.55"
+version = "0.4.56"
 description = "Python SDK for Laminar"
 authors = [
   { name = "lmnr.ai", email = "founders@lmnr.ai" }

{lmnr-0.4.55 → lmnr-0.4.56}/src/lmnr/openllmetry_sdk/__init__.py

@@ -32,6 +32,8 @@ class Traceloop:
         should_enrich_metrics: bool = False,
         resource_attributes: dict = {},
         instruments: Optional[Set[Instruments]] = None,
+        base_http_url: Optional[str] = None,
+        project_api_key: Optional[str] = None,
     ) -> None:
         if not is_tracing_enabled():
             return
@@ -69,4 +71,6 @@ class Traceloop:
             exporter=exporter,
             should_enrich_metrics=should_enrich_metrics,
             instruments=instruments,
+            base_http_url=base_http_url,
+            project_api_key=project_api_key,
         )

{lmnr-0.4.55 → lmnr-0.4.56}/src/lmnr/openllmetry_sdk/instruments.py

@@ -21,6 +21,7 @@ class Instruments(Enum):
     OLLAMA = "ollama"
     OPENAI = "openai"
     PINECONE = "pinecone"
+    PLAYWRIGHT = "playwright"
     QDRANT = "qdrant"
     REPLICATE = "replicate"
     SAGEMAKER = "sagemaker"

{lmnr-0.4.55 → lmnr-0.4.56}/src/lmnr/openllmetry_sdk/tracing/tracing.py

@@ -6,6 +6,7 @@ import logging
 from contextvars import Context
 from lmnr.sdk.log import VerboseColorfulFormatter
 from lmnr.openllmetry_sdk.instruments import Instruments
+from lmnr.sdk.browser import init_browser_tracing
 from lmnr.openllmetry_sdk.tracing.attributes import (
     ASSOCIATION_PROPERTIES,
     SPAN_INSTRUMENTATION_SOURCE,
@@ -80,6 +81,8 @@ class TracerWrapper(object):
         exporter: Optional[SpanExporter] = None,
         should_enrich_metrics: bool = False,
         instruments: Optional[Set[Instruments]] = None,
+        base_http_url: Optional[str] = None,
+        project_api_key: Optional[str] = None,
     ) -> "TracerWrapper":
         cls._initialize_logger(cls)
         if not hasattr(cls, "instance"):
@@ -122,6 +125,8 @@ class TracerWrapper(object):
             instrument_set = init_instrumentations(
                 should_enrich_metrics,
                 instruments,
+                base_http_url=base_http_url,
+                project_api_key=project_api_key,
             )
 
             if not instrument_set:
@@ -286,6 +291,8 @@ def init_instrumentations(
     should_enrich_metrics: bool,
     instruments: Optional[Set[Instruments]] = None,
     block_instruments: Optional[Set[Instruments]] = None,
+    base_http_url: Optional[str] = None,
+    project_api_key: Optional[str] = None,
 ):
     block_instruments = block_instruments or set()
     # These libraries are not instrumented by default,
@@ -397,6 +404,9 @@ def init_instrumentations(
         elif instrument == Instruments.WEAVIATE:
             if init_weaviate_instrumentor():
                 instrument_set = True
+        elif instrument == Instruments.PLAYWRIGHT:
+            if init_browser_tracing(base_http_url, project_api_key):
+                instrument_set = True
         else:
             module_logger.warning(
                 f"Warning: {instrument} instrumentation does not exist."

lmnr-0.4.56/src/lmnr/sdk/browser/__init__.py

@@ -0,0 +1,9 @@
+from lmnr.openllmetry_sdk.utils.package_check import is_package_installed
+
+
+def init_browser_tracing(http_url: str, project_api_key: str):
+    if is_package_installed("playwright"):
+        from .playwright_patch import init_playwright_tracing
+
+        init_playwright_tracing(http_url, project_api_key)
+    # Other browsers can be added here

lmnr-0.4.56/src/lmnr/sdk/browser/playwright_patch.py

@@ -0,0 +1,192 @@
+import opentelemetry
+import uuid
+
+try:
+    from playwright.async_api import BrowserContext, Page
+    from playwright.sync_api import (
+        BrowserContext as SyncBrowserContext,
+        Page as SyncPage,
+    )
+except ImportError as e:
+    raise ImportError(
+        f"Attempated to import {__file__}, but it is designed "
+        "to patch Playwright, which is not installed. Use `pip install playwright` "
+        "to install Playwright or remove this import."
+    ) from e
+
+_original_new_page = None
+_original_goto = None
+_original_new_page_async = None
+_original_goto_async = None
+
+INJECT_PLACEHOLDER = """
+([baseUrl, projectApiKey]) => {
+    const serverUrl = `${baseUrl}/v1/browser-sessions/events`;
+    const BATCH_SIZE = 16;
+    const FLUSH_INTERVAL = 1000;
+    const HEARTBEAT_INTERVAL = 1000; // 1 second heartbeat
+
+    window.rrwebEventsBatch = [];
+    
+    window.sendBatch = async () => {
+        if (window.rrwebEventsBatch.length === 0) return;
+        
+        const eventsPayload = {
+            sessionId: window.rrwebSessionId,
+            traceId: window.traceId,
+            events: window.rrwebEventsBatch
+        };
+
+        try {
+            await fetch(serverUrl, {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json', 'Authorization': `Bearer ${projectApiKey}` },
+                body: JSON.stringify(eventsPayload),
+            });
+            window.rrwebEventsBatch = [];
+        } catch (error) {
+            console.error('Failed to send events:', error);
+        }
+    };
+
+    setInterval(() => window.sendBatch(), FLUSH_INTERVAL);
+
+    // Add heartbeat event
+    setInterval(() => {
+        window.rrwebEventsBatch.push({
+            type: 6, // Custom event type
+            data: { source: 'heartbeat' },
+            timestamp: Date.now()
+        });
+    }, HEARTBEAT_INTERVAL);
+
+    window.rrweb.record({
+        emit(event) {
+            window.rrwebEventsBatch.push(event);
+            
+            if (window.rrwebEventsBatch.length >= BATCH_SIZE) {
+                window.sendBatch();
+            }
+        }
+    });
+
+    // Simplified beforeunload handler
+    window.addEventListener('beforeunload', () => {
+        window.sendBatch();
+    });
+}
+"""
+
+
+def init_playwright_tracing(http_url: str, project_api_key: str):
+    def inject_rrweb(page: SyncPage):
+        # Get current trace ID from active span
+        current_span = opentelemetry.trace.get_current_span()
+        current_span.set_attribute("lmnr.internal.has_browser_session", True)
+        trace_id = format(current_span.get_span_context().trace_id, "032x")
+        session_id = str(uuid.uuid4().hex)
+
+        # Generate UUID session ID and set trace ID
+        page.evaluate(
+            """([traceId, sessionId]) => {
+            window.rrwebSessionId = sessionId;
+            window.traceId = traceId;
+        }""",
+            [trace_id, session_id],
+        )
+
+        # Load rrweb and set up recording
+        page.add_script_tag(
+            url="https://cdn.jsdelivr.net/npm/rrweb@latest/dist/rrweb.min.js"
+        )
+
+        # Update the recording setup to include trace ID
+        page.evaluate(
+            INJECT_PLACEHOLDER,
+            [http_url, project_api_key],
+        )
+
+    async def inject_rrweb_async(page: Page):
+        # Wait for the page to be in a ready state first
+        await page.wait_for_load_state("domcontentloaded")
+
+        # Get current trace ID from active span
+        current_span = opentelemetry.trace.get_current_span()
+        current_span.set_attribute("lmnr.internal.has_browser_session", True)
+        trace_id = format(current_span.get_span_context().trace_id, "032x")
+        session_id = str(uuid.uuid4().hex)
+
+        # Wait for any existing script load to complete
+        await page.wait_for_load_state("networkidle")
+
+        # Generate UUID session ID and set trace ID
+        await page.evaluate(
+            """([traceId, sessionId]) => {
+            window.rrwebSessionId = sessionId;
+            window.traceId = traceId;
+        }""",
+            [trace_id, session_id],
+        )
+
+        # Load rrweb and set up recording
+        await page.add_script_tag(
+            url="https://cdn.jsdelivr.net/npm/rrweb@latest/dist/rrweb.min.js"
+        )
+
+        await page.wait_for_function("""(() => window.rrweb || 'rrweb' in window)""")
+
+        # Update the recording setup to include trace ID
+        await page.evaluate(
+            INJECT_PLACEHOLDER,
+            [http_url, project_api_key],
+        )
+
+    async def patched_new_page_async(self: BrowserContext, *args, **kwargs):
+        # Call the original new_page (returns a Page object)
+        page = await _original_new_page_async(self, *args, **kwargs)
+        # Inject rrweb automatically after the page is created
+        await inject_rrweb_async(page)
+        return page
+
+    async def patched_goto_async(self: Page, *args, **kwargs):
+        # Call the original goto
+        result = await _original_goto_async(self, *args, **kwargs)
+        # Inject rrweb after navigation
+        await inject_rrweb_async(self)
+        return result
+
+    def patched_new_page(self: SyncBrowserContext, *args, **kwargs):
+        # Call the original new_page (returns a Page object)
+        page = _original_new_page(self, *args, **kwargs)
+        # Inject rrweb automatically after the page is created
+        inject_rrweb(page)
+        return page
+
+    def patched_goto(self: SyncPage, *args, **kwargs):
+        # Call the original goto
+        result = _original_goto(self, *args, **kwargs)
+        # Inject rrweb after navigation
+        inject_rrweb(self)
+        return result
+
+    def patch_browser():
+        """
+        Overrides BrowserContext.new_page with a patched async function
+        that injects rrweb into every new page.
+        """
+        global _original_new_page, _original_goto, _original_new_page_async, _original_goto_async
+        if _original_new_page_async is None or _original_goto_async is None:
+            _original_new_page_async = BrowserContext.new_page
+            BrowserContext.new_page = patched_new_page_async
+
+            _original_goto_async = Page.goto
+            Page.goto = patched_goto_async
+
+        if _original_new_page is None or _original_goto is None:
+            _original_new_page = SyncBrowserContext.new_page
+            SyncBrowserContext.new_page = patched_new_page
+
+            _original_goto = SyncPage.goto
+            SyncPage.goto = patched_goto
+
+    patch_browser()

{lmnr-0.4.55 → lmnr-0.4.56}/src/lmnr/sdk/laminar.py

@@ -142,6 +142,8 @@ class Laminar:
         cls._initialize_logger()
 
         Traceloop.init(
+            base_http_url=cls.__base_http_url,
+            project_api_key=cls.__project_api_key,
             exporter=OTLPSpanExporter(
                 endpoint=cls.__base_grpc_url,
                 headers={"authorization": f"Bearer {cls.__project_api_key}"},