lmnr 0.4.56__tar.gz → 0.4.58__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lmnr
-Version: 0.4.56
+Version: 0.4.58
 Summary: Python SDK for Laminar
 License: Apache-2.0
 Author: lmnr.ai
@@ -41,65 +41,64 @@ Provides-Extra: watsonx
 Provides-Extra: weaviate
 Requires-Dist: aiohttp (>=3.0)
 Requires-Dist: argparse (>=1.0)
-Requires-Dist: deprecated (>=1.0)
 Requires-Dist: grpcio (<1.68.0)
 Requires-Dist: opentelemetry-api (>=1.28.0)
 Requires-Dist: opentelemetry-exporter-otlp-proto-grpc (>=1.28.0)
 Requires-Dist: opentelemetry-exporter-otlp-proto-http (>=1.28.0)
-Requires-Dist: opentelemetry-instrumentation-alephalpha (>=0.36.1) ; extra == "alephalpha"
-Requires-Dist: opentelemetry-instrumentation-alephalpha (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-anthropic (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-anthropic (>=0.36.1) ; extra == "anthropic"
-Requires-Dist: opentelemetry-instrumentation-bedrock (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-bedrock (>=0.36.1) ; extra == "bedrock"
-Requires-Dist: opentelemetry-instrumentation-chromadb (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-chromadb (>=0.36.1) ; extra == "chromadb"
-Requires-Dist: opentelemetry-instrumentation-cohere (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-cohere (>=0.36.1) ; extra == "cohere"
-Requires-Dist: opentelemetry-instrumentation-google-generativeai (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-google-generativeai (>=0.36.1) ; extra == "google-generativeai"
-Requires-Dist: opentelemetry-instrumentation-groq (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-groq (>=0.36.1) ; extra == "groq"
-Requires-Dist: opentelemetry-instrumentation-haystack (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-haystack (>=0.36.1) ; extra == "haystack"
-Requires-Dist: opentelemetry-instrumentation-lancedb (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-lancedb (>=0.36.1) ; extra == "lancedb"
-Requires-Dist: opentelemetry-instrumentation-langchain (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-langchain (>=0.36.1) ; extra == "langchain"
-Requires-Dist: opentelemetry-instrumentation-llamaindex (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-llamaindex (>=0.36.1) ; extra == "llamaindex"
-Requires-Dist: opentelemetry-instrumentation-marqo (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-marqo (>=0.36.1) ; extra == "marqo"
-Requires-Dist: opentelemetry-instrumentation-milvus (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-milvus (>=0.36.1) ; extra == "milvus"
-Requires-Dist: opentelemetry-instrumentation-mistralai (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-mistralai (>=0.36.1) ; extra == "mistralai"
-Requires-Dist: opentelemetry-instrumentation-ollama (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-ollama (>=0.36.1) ; extra == "ollama"
-Requires-Dist: opentelemetry-instrumentation-openai (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-openai (>=0.36.1) ; extra == "openai"
-Requires-Dist: opentelemetry-instrumentation-pinecone (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-pinecone (>=0.36.1) ; extra == "pinecone"
-Requires-Dist: opentelemetry-instrumentation-qdrant (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-qdrant (>=0.36.1) ; extra == "qdrant"
-Requires-Dist: opentelemetry-instrumentation-replicate (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-replicate (>=0.36.1) ; extra == "replicate"
+Requires-Dist: opentelemetry-instrumentation-alephalpha (>=0.37.1) ; extra == "alephalpha"
+Requires-Dist: opentelemetry-instrumentation-alephalpha (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-anthropic (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-anthropic (>=0.37.1) ; extra == "anthropic"
+Requires-Dist: opentelemetry-instrumentation-bedrock (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-bedrock (>=0.37.1) ; extra == "bedrock"
+Requires-Dist: opentelemetry-instrumentation-chromadb (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-chromadb (>=0.37.1) ; extra == "chromadb"
+Requires-Dist: opentelemetry-instrumentation-cohere (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-cohere (>=0.37.1) ; extra == "cohere"
+Requires-Dist: opentelemetry-instrumentation-google-generativeai (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-google-generativeai (>=0.37.1) ; extra == "google-generativeai"
+Requires-Dist: opentelemetry-instrumentation-groq (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-groq (>=0.37.1) ; extra == "groq"
+Requires-Dist: opentelemetry-instrumentation-haystack (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-haystack (>=0.37.1) ; extra == "haystack"
+Requires-Dist: opentelemetry-instrumentation-lancedb (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-lancedb (>=0.37.1) ; extra == "lancedb"
+Requires-Dist: opentelemetry-instrumentation-langchain (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-langchain (>=0.37.1) ; extra == "langchain"
+Requires-Dist: opentelemetry-instrumentation-llamaindex (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-llamaindex (>=0.37.1) ; extra == "llamaindex"
+Requires-Dist: opentelemetry-instrumentation-marqo (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-marqo (>=0.37.1) ; extra == "marqo"
+Requires-Dist: opentelemetry-instrumentation-milvus (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-milvus (>=0.37.1) ; extra == "milvus"
+Requires-Dist: opentelemetry-instrumentation-mistralai (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-mistralai (>=0.37.1) ; extra == "mistralai"
+Requires-Dist: opentelemetry-instrumentation-ollama (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-ollama (>=0.37.1) ; extra == "ollama"
+Requires-Dist: opentelemetry-instrumentation-openai (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-openai (>=0.37.1) ; extra == "openai"
+Requires-Dist: opentelemetry-instrumentation-pinecone (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-pinecone (>=0.37.1) ; extra == "pinecone"
+Requires-Dist: opentelemetry-instrumentation-qdrant (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-qdrant (>=0.37.1) ; extra == "qdrant"
+Requires-Dist: opentelemetry-instrumentation-replicate (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-replicate (>=0.37.1) ; extra == "replicate"
 Requires-Dist: opentelemetry-instrumentation-requests (>=0.50b0)
-Requires-Dist: opentelemetry-instrumentation-sagemaker (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-sagemaker (>=0.36.1) ; extra == "sagemaker"
+Requires-Dist: opentelemetry-instrumentation-sagemaker (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-sagemaker (>=0.37.1) ; extra == "sagemaker"
 Requires-Dist: opentelemetry-instrumentation-sqlalchemy (>=0.50b0)
 Requires-Dist: opentelemetry-instrumentation-threading (>=0.50b0)
-Requires-Dist: opentelemetry-instrumentation-together (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-together (>=0.36.1) ; extra == "together"
-Requires-Dist: opentelemetry-instrumentation-transformers (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-transformers (>=0.36.1) ; extra == "transformers"
+Requires-Dist: opentelemetry-instrumentation-together (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-together (>=0.37.1) ; extra == "together"
+Requires-Dist: opentelemetry-instrumentation-transformers (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-transformers (>=0.37.1) ; extra == "transformers"
 Requires-Dist: opentelemetry-instrumentation-urllib3 (>=0.50b0)
-Requires-Dist: opentelemetry-instrumentation-vertexai (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-vertexai (>=0.36.1) ; extra == "vertexai"
-Requires-Dist: opentelemetry-instrumentation-watsonx (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-watsonx (>=0.36.1) ; extra == "watsonx"
-Requires-Dist: opentelemetry-instrumentation-weaviate (>=0.36.1) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-weaviate (>=0.36.1) ; extra == "weaviate"
+Requires-Dist: opentelemetry-instrumentation-vertexai (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-vertexai (>=0.37.1) ; extra == "vertexai"
+Requires-Dist: opentelemetry-instrumentation-watsonx (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-watsonx (>=0.37.1) ; extra == "watsonx"
+Requires-Dist: opentelemetry-instrumentation-weaviate (>=0.37.1) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-weaviate (>=0.37.1) ; extra == "weaviate"
 Requires-Dist: opentelemetry-sdk (>=1.28.0)
 Requires-Dist: opentelemetry-semantic-conventions-ai (>=0.4.2)
 Requires-Dist: pydantic (>=2.0.3)

lmnr-0.4.58/pyproject.toml

@@ -0,0 +1,122 @@
+# Laminar Python
+
+# If you are looking for information about possible extras installations,
+# i.e. what you can pass into `pip install 'lmnr[extra1,extra2]'`, please see the
+# `[project.optional-dependencies]` section below.
+
+[project]
+name = "lmnr"
+version = "0.4.58"
+description = "Python SDK for Laminar"
+authors = [
+  { name = "lmnr.ai", email = "founders@lmnr.ai" }
+]
+readme = "README.md"
+requires-python = ">=3.9,<4"
+license = "Apache-2.0"
+dependencies = [
+  "pydantic (>=2.0.3)",
+  "requests (>=2.0)",
+  "python-dotenv (>=1.0)",
+  "opentelemetry-api (>=1.28.0)",
+  "opentelemetry-sdk (>=1.28.0)",
+  "opentelemetry-exporter-otlp-proto-http (>=1.28.0)",
+  "opentelemetry-exporter-otlp-proto-grpc (>=1.28.0)",
+  "opentelemetry-instrumentation-requests (>=0.50b0)",
+  "opentelemetry-instrumentation-sqlalchemy (>=0.50b0)",
+  "opentelemetry-instrumentation-urllib3 (>=0.50b0)",
+  "opentelemetry-instrumentation-threading (>=0.50b0)",
+  "opentelemetry-semantic-conventions-ai (>=0.4.2)",
+  "tqdm (>=4.0)",
+  "argparse (>=1.0)",
+  "aiohttp (>=3.0)",
+  "tenacity (>=8.0)",
+  # explicitly freeze grpcio. Since 1.68.0, grpcio writes a warning message
+  # that looks scary, but is harmless.
+  # WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
+  # E0000 00:00:1737439981.199902 9456033 init.cc:229] grpc_wait_for_shutdown_with_timeout() timed out.
+  #
+  # Related issue:
+  # https://discuss.ai.google.dev/t/warning-all-log-messages-before-absl-initializelog-is-called-are-written-to-stderr-e0000-001731955515-629532-17124-init-cc-229-grpc-wait-for-shutdown-with-timeout-timed-out/50020
+  # https://github.com/grpc/grpc/issues/38490
+  "grpcio<1.68.0",
+]
+
+[project.scripts]
+lmnr = "lmnr.cli:cli"
+
+[project.optional-dependencies]
+# List of all possible extras. You can specify one or more of these extras
+# when installing the package, using any of the following examples:
+# `pip install 'lmnr[anthropic,openai]'`
+# `uv pip install 'lmnr[anthropic,openai]'`
+# `uv add lmnr --extra anthropic --extra openai`
+# `poetry add 'lmnr[anthropic,openai]'`
+
+alephalpha=["opentelemetry-instrumentation-alephalpha>=0.37.1"]
+anthropic=["opentelemetry-instrumentation-anthropic>=0.37.1"]
+bedrock=["opentelemetry-instrumentation-bedrock>=0.37.1"]
+chromadb=["opentelemetry-instrumentation-chromadb>=0.37.1"]
+cohere=["opentelemetry-instrumentation-cohere>=0.37.1"]
+google-generativeai=["opentelemetry-instrumentation-google-generativeai>=0.37.1"]
+groq=["opentelemetry-instrumentation-groq>=0.37.1"]
+haystack=["opentelemetry-instrumentation-haystack>=0.37.1"]
+lancedb=["opentelemetry-instrumentation-lancedb>=0.37.1"]
+langchain=["opentelemetry-instrumentation-langchain>=0.37.1"]
+llamaindex=["opentelemetry-instrumentation-llamaindex>=0.37.1"]
+marqo=["opentelemetry-instrumentation-marqo>=0.37.1"]
+milvus=["opentelemetry-instrumentation-milvus>=0.37.1"]
+mistralai=["opentelemetry-instrumentation-mistralai>=0.37.1"]
+ollama=["opentelemetry-instrumentation-ollama>=0.37.1"]
+openai=["opentelemetry-instrumentation-openai>=0.37.1"]
+pinecone=["opentelemetry-instrumentation-pinecone>=0.37.1"]
+qdrant=["opentelemetry-instrumentation-qdrant>=0.37.1"]
+replicate=["opentelemetry-instrumentation-replicate>=0.37.1"]
+sagemaker=["opentelemetry-instrumentation-sagemaker>=0.37.1"]
+together=["opentelemetry-instrumentation-together>=0.37.1"]
+transformers=["opentelemetry-instrumentation-transformers>=0.37.1"]
+vertexai=["opentelemetry-instrumentation-vertexai>=0.37.1"]
+watsonx=["opentelemetry-instrumentation-watsonx>=0.37.1"]
+weaviate=["opentelemetry-instrumentation-weaviate>=0.37.1"]
+# `all` is the group added for convenience, if you want to install all
+# the instrumentations.
+all = [
+  "opentelemetry-instrumentation-alephalpha>=0.37.1",
+  "opentelemetry-instrumentation-anthropic>=0.37.1",
+  "opentelemetry-instrumentation-bedrock>=0.37.1",
+  "opentelemetry-instrumentation-chromadb>=0.37.1",
+  "opentelemetry-instrumentation-cohere>=0.37.1",
+  "opentelemetry-instrumentation-google-generativeai>=0.37.1",
+  "opentelemetry-instrumentation-groq>=0.37.1",
+  "opentelemetry-instrumentation-haystack>=0.37.1",
+  "opentelemetry-instrumentation-lancedb>=0.37.1",
+  "opentelemetry-instrumentation-langchain>=0.37.1",
+  "opentelemetry-instrumentation-llamaindex>=0.37.1",
+  "opentelemetry-instrumentation-marqo>=0.37.1",
+  "opentelemetry-instrumentation-milvus>=0.37.1",
+  "opentelemetry-instrumentation-mistralai>=0.37.1",
+  "opentelemetry-instrumentation-ollama>=0.37.1",
+  "opentelemetry-instrumentation-openai>=0.37.1",
+  "opentelemetry-instrumentation-pinecone>=0.37.1",
+  "opentelemetry-instrumentation-qdrant>=0.37.1",
+  "opentelemetry-instrumentation-replicate>=0.37.1",
+  "opentelemetry-instrumentation-sagemaker>=0.37.1",
+  "opentelemetry-instrumentation-together>=0.37.1",
+  "opentelemetry-instrumentation-transformers>=0.37.1",
+  "opentelemetry-instrumentation-vertexai>=0.37.1",
+  "opentelemetry-instrumentation-watsonx>=0.37.1",
+  "opentelemetry-instrumentation-weaviate>=0.37.1"
+]
+
+[dependency-groups]
+dev = [
+  "autopep8",
+  "flake8",
+  "pytest>=8.3.4",
+  "pytest-sugar",
+  "pytest-asyncio>=0.25.2"
+]
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

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

@@ -34,6 +34,7 @@ class Traceloop:
         instruments: Optional[Set[Instruments]] = None,
         base_http_url: Optional[str] = None,
         project_api_key: Optional[str] = None,
+        max_export_batch_size: Optional[int] = None,
     ) -> None:
         if not is_tracing_enabled():
             return
@@ -73,4 +74,5 @@ class Traceloop:
             instruments=instruments,
             base_http_url=base_http_url,
             project_api_key=project_api_key,
+            max_export_batch_size=max_export_batch_size,
         )

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

@@ -83,6 +83,7 @@ class TracerWrapper(object):
         instruments: Optional[Set[Instruments]] = None,
         base_http_url: Optional[str] = None,
         project_api_key: Optional[str] = None,
+        max_export_batch_size: Optional[int] = None,
     ) -> "TracerWrapper":
         cls._initialize_logger(cls)
         if not hasattr(cls, "instance"):
@@ -109,7 +110,8 @@ class TracerWrapper(object):
                     )
                 else:
                     obj.__spans_processor: SpanProcessor = BatchSpanProcessor(
-                        obj.__spans_exporter
+                        obj.__spans_exporter,
+                        max_export_batch_size=max_export_batch_size,
                     )
                 obj.__spans_processor_original_on_start = None
 
@@ -430,6 +432,10 @@ def init_openai_instrumentor(should_enrich_metrics: bool):
             instrumentor = OpenAIInstrumentor(
                 enrich_assistant=should_enrich_metrics,
                 enrich_token_usage=should_enrich_metrics,
+                # Default in the package provided is an empty function, which
+                # results in dropping the image data if we don't explicitly
+                # set it to None.
+                upload_base64_image=None,
             )
             if not instrumentor.is_instrumented_by_opentelemetry:
                 instrumentor.instrument()

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

@@ -0,0 +1,248 @@
+import opentelemetry
+import uuid
+import asyncio
+
+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
+
+INJECT_PLACEHOLDER = """
+([baseUrl, projectApiKey]) => {
+    const serverUrl = `${baseUrl}/v1/browser-sessions/events`;
+    const FLUSH_INTERVAL = 1000;
+    const HEARTBEAT_INTERVAL = 1000;
+
+    window.rrwebEventsBatch = [];
+    
+    window.sendBatch = async () => {
+        if (window.rrwebEventsBatch.length === 0) return;
+        
+        const eventsPayload = {
+            sessionId: window.rrwebSessionId,
+            traceId: window.traceId,
+            events: window.rrwebEventsBatch
+        };
+        
+        try {
+            const jsonString = JSON.stringify(eventsPayload);
+            const uint8Array = new TextEncoder().encode(jsonString);
+            
+            const cs = new CompressionStream('gzip');
+            const compressedStream = await new Response(
+                new Response(uint8Array).body.pipeThrough(cs)
+            ).arrayBuffer();
+            
+            const compressedArray = new Uint8Array(compressedStream);
+            
+            const blob = new Blob([compressedArray], { type: 'application/octet-stream' });
+            
+            const response = await fetch(serverUrl, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    'Content-Encoding': 'gzip',
+                    'Authorization': `Bearer ${projectApiKey}`
+                },
+                body: blob,
+                credentials: 'omit',
+                mode: 'cors',
+                cache: 'no-cache',
+            });
+            
+            if (!response.ok) {
+                throw new Error(`HTTP error! status: ${response.status}`);
+            }
+            
+            window.rrwebEventsBatch = [];
+        } catch (error) {
+            console.error('Failed to send events:', error);
+        }
+    };
+
+    setInterval(() => window.sendBatch(), FLUSH_INTERVAL);
+
+    setInterval(() => {
+        window.rrwebEventsBatch.push({
+            type: 6,
+            data: { source: 'heartbeat' },
+            timestamp: Date.now()
+        });
+    }, HEARTBEAT_INTERVAL);
+
+    window.rrweb.record({
+        emit(event) {
+            window.rrwebEventsBatch.push(event);            
+        }
+    });
+
+    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 from CDN
+        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):
+        try:
+            # 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)
+
+            # 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 from CDN
+            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],
+            )
+        except Exception as e:
+            print(f"Error injecting rrweb: {e}")
+
+    def handle_navigation(page: SyncPage):
+        def on_load():
+            inject_rrweb(page)
+
+        page.on("load", on_load)
+        inject_rrweb(page)
+
+    async def handle_navigation_async(page: Page):
+        async def on_load():
+            await inject_rrweb_async(page)
+
+        page.on("load", lambda: asyncio.create_task(on_load()))
+        await inject_rrweb_async(page)
+
+    async def patched_new_page_async(self: BrowserContext, *args, **kwargs):
+        # Modify CSP to allow required domains
+        async def handle_route(route):
+            try:
+                response = await route.fetch()
+                headers = dict(response.headers)
+
+                # Find and modify CSP header
+                for header_name in headers:
+                    if header_name.lower() == "content-security-policy":
+                        csp = headers[header_name]
+                        parts = csp.split(";")
+                        for i, part in enumerate(parts):
+                            if "script-src" in part:
+                                parts[i] = f"{part.strip()} cdn.jsdelivr.net"
+                            elif "connect-src" in part:
+                                parts[i] = f"{part.strip()} " + http_url
+                        if not any("connect-src" in part for part in parts):
+                            parts.append(" connect-src 'self' " + http_url)
+                        headers[header_name] = ";".join(parts)
+
+                await route.fulfill(response=response, headers=headers)
+            except Exception:
+                await route.continue_()
+
+        await self.route("**/*", handle_route)
+        page = await _original_new_page_async(self, *args, **kwargs)
+        await handle_navigation_async(page)
+        return page
+
+    def patched_new_page(self: SyncBrowserContext, *args, **kwargs):
+        # Modify CSP to allow required domains
+        def handle_route(route):
+            try:
+                response = route.fetch()
+                headers = dict(response.headers)
+
+                # Find and modify CSP header
+                for header_name in headers:
+                    if header_name.lower() == "content-security-policy":
+                        csp = headers[header_name]
+                        parts = csp.split(";")
+                        for i, part in enumerate(parts):
+                            if "script-src" in part:
+                                parts[i] = f"{part.strip()} cdn.jsdelivr.net"
+                            elif "connect-src" in part:
+                                parts[i] = f"{part.strip()} " + http_url
+                        if not any("connect-src" in part for part in parts):
+                            parts.append(" connect-src 'self' " + http_url)
+                        headers[header_name] = ";".join(parts)
+
+                route.fulfill(response=response, headers=headers)
+            except Exception:
+                # Continue with the original request without modification
+                route.continue_()
+
+        self.route("**/*", handle_route)
+        page = _original_new_page(self, *args, **kwargs)
+        handle_navigation(page)
+        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()