lmnr 0.4.64__py3-none-any.whl → 0.4.66__py3-none-any.whl

{lmnr-0.4.64.dist-info → lmnr-0.4.66.dist-info}/RECORD

@@ -5,32 +5,35 @@ lmnr/openllmetry_sdk/__init__.py,sha256=G_sNpfg0rnA6n4AaKM88g_gwaCNFcg1Ew8dB8el4
 lmnr/openllmetry_sdk/config/__init__.py,sha256=5aGdIdo1LffBkNwIBUbqzN6OUCMCrURU4b0rf5LBSI0,300
 lmnr/openllmetry_sdk/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lmnr/openllmetry_sdk/decorators/base.py,sha256=RNyvSreGBwwh8EWaa3O9samrnmKoZHK7eBzGOPT3lHc,7132
-lmnr/openllmetry_sdk/instruments.py,sha256=X1S3XbFF_RDlbxxbNxqKKJ9JNUStdTI6gLYCPWnoOTs,1066
+lmnr/openllmetry_sdk/instruments.py,sha256=9KoJ19Qar1dBrmO1wikNEkKxRfus8znQTj-g_maRTTM,1098
 lmnr/openllmetry_sdk/tracing/__init__.py,sha256=xT73L1t2si2CM6QmMiTZ7zn-dKKYBLNrpBBWq6WfVBw,68
 lmnr/openllmetry_sdk/tracing/attributes.py,sha256=cLBmSp4AMv9E91Ck3yD5Z1Qx1L5ZRV-80VJxFA-sO0Q,1426
 lmnr/openllmetry_sdk/tracing/content_allow_list.py,sha256=3feztm6PBWNelc8pAZUcQyEGyeSpNiVKjOaDk65l2ps,846
 lmnr/openllmetry_sdk/tracing/context_manager.py,sha256=rdSus-p-TaevQ8hIAhfbnZr5dTqRvACDkzXGDpflncY,306
-lmnr/openllmetry_sdk/tracing/tracing.py,sha256=W6Aps6CfkecscSqvcmMPL-C9Ee3J0jMKIMfK27ce5NI,34081
+lmnr/openllmetry_sdk/tracing/tracing.py,sha256=gn-3jIFHwPFzmzhcfxnqs0Y2YkBbIDtoHCTHFpzhCEQ,35194
 lmnr/openllmetry_sdk/utils/__init__.py,sha256=pNhf0G3vTd5ccoc03i1MXDbricSaiqCbi1DLWhSekK8,604
 lmnr/openllmetry_sdk/utils/in_memory_span_exporter.py,sha256=H_4TRaThMO1H6vUQ0OpQvzJk_fZH0OOsRAM1iZQXsR8,2112
 lmnr/openllmetry_sdk/utils/json_encoder.py,sha256=dK6b_axr70IYL7Vv-bu4wntvDDuyntoqsHaddqX7P58,463
 lmnr/openllmetry_sdk/utils/package_check.py,sha256=_-Fu9Zbp9tOyy27_-Rul7tDc8JaXYR2FmqF8SWOXSCc,244
 lmnr/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lmnr/sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lmnr/sdk/browser/__init__.py,sha256=NSP5sB-dm-f0FP70_GMvVrNFwc5rHf7SW0_Oisyo3cE,343
-lmnr/sdk/browser/playwright_patch.py,sha256=wfbkFXYMl6VmsxzLfaArmuCKtASnd3hUtlL7Shj6lss,13071
+lmnr/sdk/browser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lmnr/sdk/browser/browser_use_otel.py,sha256=lsqc8cpCvpoWUSrodrOptdQSKDqvJ1HUWZ-UI3H-dVg,3917
+lmnr/sdk/browser/playwright_otel.py,sha256=uj9-P8f0e1i17n5fp1V9GE9CVFjfzq-TiqBgw9RYK2A,10027
 lmnr/sdk/browser/rrweb/rrweb.min.js,sha256=X5pgaoX1j_OjKTqGQgKB-83xUSuydNLQa-Kkh1AAZYM,140485
-lmnr/sdk/datasets.py,sha256=hJcQcwTJbtA4COoVG3god4xll9TBSDMfvrhKmMfanjg,1567
+lmnr/sdk/browser/utils.py,sha256=IGiZc9ydRARk_J6PdxtQb4ffkuUYprdZLdsHYtdRrbU,3500
+lmnr/sdk/client.py,sha256=f5MsMadaBlEfw9DOSgb6r2Y-l8O8gISnZ_gzzPcW-xU,10567
+lmnr/sdk/datasets.py,sha256=XMbZtcrm56XPOezKhbyJQWdRZX3_wcxU1Eneld_vKVQ,1579
 lmnr/sdk/decorators.py,sha256=g0VBqUEMCPRbgjgGHauVuKK1wHEd9rkiGzlYUYrcml4,2336
 lmnr/sdk/eval_control.py,sha256=G6Fg3Xx_KWv72iBaWlNMdyRTF2bZFQnwJ68sJNSpIcY,177
-lmnr/sdk/evaluations.py,sha256=WmPAgQVm2C83xsG-qMdPrMuvCMFMoVdO37LqZtLc8xw,18702
-lmnr/sdk/laminar.py,sha256=1KkEq8DDogobtrFwgTOo71hDptX1oMMkDcYn973S-eU,38717
+lmnr/sdk/evaluations.py,sha256=2YccO4yiAxWUhXhFBW1D3OY44g1vdpbtKig5yVDlOrU,19712
+lmnr/sdk/laminar.py,sha256=MKlcLdPBSS2KltSX0xhYIOx4D2rpOvuISCMUPfjYbrc,33884
 lmnr/sdk/log.py,sha256=nt_YMmPw1IRbGy0b7q4rTtP4Yo3pQfNxqJPXK3nDSNQ,2213
-lmnr/sdk/types.py,sha256=NUVAIZAjjyWB6Ze7naa3Vx9g14Em6OtIXsSNSCJWISE,9892
+lmnr/sdk/types.py,sha256=83rXmIGP12kyF0f27Wrw4kzJPBC44uyiaUwK32eusS0,10822
 lmnr/sdk/utils.py,sha256=sD1YEqhdPaHweY2VGmjMF9MC-X7Ikdc49E01D-HF77E,3377
-lmnr/version.py,sha256=v7DBfMNhZpmxPRXlj_blnGYLZeSnUSzmF20-ubdgmzs,1328
-lmnr-0.4.64.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
-lmnr-0.4.64.dist-info/METADATA,sha256=B-pNM9ZEKV7zQcuQMQWVv-HekuIxDyLTbYXbX6Jqorc,13827
-lmnr-0.4.64.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-lmnr-0.4.64.dist-info/entry_points.txt,sha256=K1jE20ww4jzHNZLnsfWBvU3YKDGBgbOiYG5Y7ivQcq4,37
-lmnr-0.4.64.dist-info/RECORD,,
+lmnr/version.py,sha256=1e1FKNz0PvRx7Z8F9UKp3GgWQ-IIiy8dRjfAuxHmGMk,1328
+lmnr-0.4.66.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
+lmnr-0.4.66.dist-info/METADATA,sha256=mNr-SMHD1amELqAE4xgMig5xAFpRJem0JhUm_s7R9Z8,13877
+lmnr-0.4.66.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+lmnr-0.4.66.dist-info/entry_points.txt,sha256=K1jE20ww4jzHNZLnsfWBvU3YKDGBgbOiYG5Y7ivQcq4,37
+lmnr-0.4.66.dist-info/RECORD,,

lmnr/sdk/browser/playwright_patch.py

@@ -1,377 +0,0 @@
-import uuid
-import asyncio
-import logging
-import time
-import os
-import aiohttp
-import requests
-import threading
-import gzip
-import json
-from lmnr.version import SDK_VERSION, PYTHON_VERSION
-from lmnr import Laminar
-
-logger = logging.getLogger(__name__)
-
-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"Attempted 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_new_page_async = None
-
-current_dir = os.path.dirname(os.path.abspath(__file__))
-with open(os.path.join(current_dir, "rrweb", "rrweb.min.js"), "r") as f:
-    RRWEB_CONTENT = f"() => {{ {f.read()} }}"
-
-INJECT_PLACEHOLDER = """
-() => {
-    const BATCH_SIZE = 1000;  // Maximum events to store in memory
-    
-    window.lmnrRrwebEventsBatch = [];
-    
-    // Utility function to compress individual event data
-    async function compressEventData(data) {
-        const jsonString = JSON.stringify(data);
-        const blob = new Blob([jsonString], { type: 'application/json' });
-        const compressedStream = blob.stream().pipeThrough(new CompressionStream('gzip'));
-        const compressedResponse = new Response(compressedStream);
-        const compressedData = await compressedResponse.arrayBuffer();
-        return Array.from(new Uint8Array(compressedData));
-    }
-    
-    window.lmnrGetAndClearEvents = () => {
-        const events = window.lmnrRrwebEventsBatch;
-        window.lmnrRrwebEventsBatch = [];
-        return events;
-    };
-
-    // Add heartbeat events
-    setInterval(async () => {
-        const heartbeat = {
-            type: 6,
-            data: await compressEventData({ source: 'heartbeat' }),
-            timestamp: Date.now()
-        };
-        
-        window.lmnrRrwebEventsBatch.push(heartbeat);
-        
-        // Prevent memory issues by limiting batch size
-        if (window.lmnrRrwebEventsBatch.length > BATCH_SIZE) {
-            window.lmnrRrwebEventsBatch = window.lmnrRrwebEventsBatch.slice(-BATCH_SIZE);
-        }
-    }, 1000);
-
-    window.lmnrRrweb.record({
-        async emit(event) {
-            // Compress the data field
-            const compressedEvent = {
-                ...event,
-                data: await compressEventData(event.data)
-            };
-            window.lmnrRrwebEventsBatch.push(compressedEvent);
-        }
-    });
-}
-"""
-
-
-def retry_sync(func, retries=5, delay=0.5, error_message="Operation failed"):
-    """Utility function for retry logic in synchronous operations"""
-    for attempt in range(retries):
-        try:
-            result = func()
-            if result:  # If function returns truthy value, consider it successful
-                return result
-            if attempt == retries - 1:  # Last attempt
-                logger.error(f"{error_message} after all retries")
-                return None
-        except Exception as e:
-            if attempt == retries - 1:  # Last attempt
-                logger.error(f"{error_message}: {e}")
-                return None
-        time.sleep(delay)
-    return None
-
-
-async def retry_async(func, retries=5, delay=0.5, error_message="Operation failed"):
-    """Utility function for retry logic in asynchronous operations"""
-    for attempt in range(retries):
-        try:
-            result = await func()
-            if result:  # If function returns truthy value, consider it successful
-                return result
-            if attempt == retries - 1:  # Last attempt
-                logger.error(f"{error_message} after all retries")
-                return None
-        except Exception as e:
-            if attempt == retries - 1:  # Last attempt
-                logger.error(f"{error_message}: {e}")
-                return None
-        await asyncio.sleep(delay)
-    return None
-
-
-async def send_events_async(
-    page: Page, http_url: str, project_api_key: str, session_id: str, trace_id: str
-):
-    """Fetch events from the page and send them to the server"""
-    try:
-        # Check if function exists first
-        has_function = await page.evaluate(
-            """
-            () => typeof window.lmnrGetAndClearEvents === 'function'
-        """
-        )
-        if not has_function:
-            return
-
-        events = await page.evaluate("window.lmnrGetAndClearEvents()")
-        if not events or len(events) == 0:
-            return
-
-        payload = {
-            "sessionId": session_id,
-            "traceId": trace_id,
-            "events": events,
-            "source": f"python@{PYTHON_VERSION}",
-            "sdkVersion": SDK_VERSION,
-        }
-
-        headers = {
-            "Content-Type": "application/json",
-            "Authorization": f"Bearer {project_api_key}",
-            "Accept": "application/json",
-        }
-
-        async with aiohttp.ClientSession() as session:
-            async with session.post(
-                f"{http_url}/v1/browser-sessions/events",
-                json=payload,
-                headers=headers,
-            ) as response:
-                if not response.ok:
-                    logger.error(f"Failed to send events: {response.status}")
-
-    except Exception as e:
-        logger.error(f"Error sending events: {e}")
-
-
-def send_events_sync(
-    page: SyncPage, http_url: str, project_api_key: str, session_id: str, trace_id: str
-):
-    """Synchronous version of send_events"""
-    try:
-        # Check if function exists first
-        has_function = page.evaluate(
-            """
-            () => typeof window.lmnrGetAndClearEvents === 'function'
-        """
-        )
-        if not has_function:
-            return
-
-        events = page.evaluate("window.lmnrGetAndClearEvents()")
-        if not events or len(events) == 0:
-            return
-
-        payload = {
-            "sessionId": session_id,
-            "traceId": trace_id,
-            "events": events,
-            "source": f"python@{PYTHON_VERSION}",
-            "sdkVersion": SDK_VERSION,
-        }
-
-        headers = {
-            "Content-Type": "application/json",
-            "Authorization": f"Bearer {project_api_key}",
-            "Accept": "application/json",
-            "Content-Encoding": "gzip",  # Add Content-Encoding header
-        }
-
-        # Compress the payload
-        compressed_payload = gzip.compress(json.dumps(payload).encode("utf-8"))
-
-        response = requests.post(
-            f"{http_url}/v1/browser-sessions/events",
-            data=compressed_payload,  # Use data instead of json for raw bytes
-            headers=headers,
-        )
-        if not response.ok:
-            logger.error(f"Failed to send events: {response.status_code}")
-
-    except Exception as e:
-        logger.error(f"Error sending events: {e}")
-
-
-def init_playwright_tracing(http_url: str, project_api_key: str):
-
-    def inject_rrweb(page: SyncPage):
-        try:
-            page.wait_for_load_state("domcontentloaded")
-
-            # Wrap the evaluate call in a try-catch
-            try:
-                is_loaded = page.evaluate(
-                    """() => typeof window.lmnrRrweb !== 'undefined'"""
-                )
-            except Exception as e:
-                logger.debug(f"Failed to check if rrweb is loaded: {e}")
-                is_loaded = False
-
-            if not is_loaded:
-                def load_rrweb():
-                    try:
-                        page.evaluate(RRWEB_CONTENT)
-                        page.wait_for_function(
-                            """(() => typeof window.lmnrRrweb !== 'undefined')""",
-                            timeout=5000,
-                        )
-                        return True
-                    except Exception as e:
-                        logger.debug(f"Failed to load rrweb: {e}")
-                        return False
-
-                if not retry_sync(
-                    load_rrweb, delay=1, error_message="Failed to load rrweb"
-                ):
-                    return
-
-            try:
-                page.evaluate(INJECT_PLACEHOLDER)
-            except Exception as e:
-                logger.debug(f"Failed to inject rrweb placeholder: {e}")
-
-        except Exception as e:
-            logger.error(f"Error during rrweb injection: {e}")
-
-    async def inject_rrweb_async(page: Page):
-        try:
-            await page.wait_for_load_state("domcontentloaded")
-
-            # Wrap the evaluate call in a try-catch
-            try:
-                is_loaded = await page.evaluate(
-                    """() => typeof window.lmnrRrweb !== 'undefined'"""
-                )
-            except Exception as e:
-                logger.debug(f"Failed to check if rrweb is loaded: {e}")
-                is_loaded = False
-
-            if not is_loaded:
-                async def load_rrweb():
-                    try:
-                        await page.evaluate(RRWEB_CONTENT)
-                        await page.wait_for_function(
-                            """(() => typeof window.lmnrRrweb !== 'undefined')""",
-                            timeout=5000,
-                        )
-                        return True
-                    except Exception as e:
-                        logger.debug(f"Failed to load rrweb: {e}")
-                        return False
-
-                if not await retry_async(
-                    load_rrweb, delay=1, error_message="Failed to load rrweb"
-                ):
-                    return
-
-            try:
-                await page.evaluate(INJECT_PLACEHOLDER)
-            except Exception as e:
-                logger.debug(f"Failed to inject rrweb placeholder: {e}")
-
-        except Exception as e:
-            logger.error(f"Error during rrweb injection: {e}")
-
-    def handle_navigation(page: SyncPage, session_id: str, trace_id: str):
-        def on_load():
-            try:
-                inject_rrweb(page)
-            except Exception as e:
-                logger.error(f"Error in on_load handler: {e}")
-
-        page.on("load", on_load)
-        inject_rrweb(page)
-
-        def collection_loop():
-            while not page.is_closed():  # Stop when page closes
-                send_events_sync(page, http_url, project_api_key, session_id, trace_id)
-                time.sleep(2)
-
-        thread = threading.Thread(target=collection_loop, daemon=True)
-        thread.start()
-
-    async def handle_navigation_async(page: Page, session_id: str, trace_id: str):
-        async def on_load():
-            try:
-                await inject_rrweb_async(page)
-            except Exception as e:
-                logger.error(f"Error in on_load handler: {e}")
-
-        page.on("load", lambda: asyncio.create_task(on_load()))
-        await inject_rrweb_async(page)
-
-        async def collection_loop():
-            try:
-                while not page.is_closed():  # Stop when page closes
-                    await send_events_async(
-                        page, http_url, project_api_key, session_id, trace_id
-                    )
-                    await asyncio.sleep(2)
-                logger.info("Event collection stopped")
-            except Exception as e:
-                logger.error(f"Event collection stopped: {e}")
-
-        # Create and store task
-        task = asyncio.create_task(collection_loop())
-
-        # Clean up task when page closes
-        page.on("close", lambda: task.cancel())
-
-    def patched_new_page(self: SyncBrowserContext, *args, **kwargs):
-        with Laminar.start_as_current_span(name="browser_context.new_page") as span:
-            page = _original_new_page(self, *args, **kwargs)
-
-            session_id = str(uuid.uuid4().hex)
-            span.set_attribute("lmnr.internal.has_browser_session", True)
-
-            trace_id = format(span.get_span_context().trace_id, "032x")
-            session_id = str(uuid.uuid4().hex)
-
-            handle_navigation(page, session_id, trace_id)
-            return page
-
-    async def patched_new_page_async(self: BrowserContext, *args, **kwargs):
-        with Laminar.start_as_current_span(name="browser_context.new_page") as span:
-            page = await _original_new_page_async(self, *args, **kwargs)
-
-            session_id = str(uuid.uuid4().hex)
-
-            span.set_attribute("lmnr.internal.has_browser_session", True)
-            trace_id = format(span.get_span_context().trace_id, "032x")
-            session_id = str(uuid.uuid4().hex)
-            await handle_navigation_async(page, session_id, trace_id)
-            return page
-
-    def patch_browser():
-        global _original_new_page, _original_new_page_async
-        if _original_new_page_async is None:
-            _original_new_page_async = BrowserContext.new_page
-            BrowserContext.new_page = patched_new_page_async
-
-        if _original_new_page is None:
-            _original_new_page = SyncBrowserContext.new_page
-            SyncBrowserContext.new_page = patched_new_page
-
-    patch_browser()