lmnr 0.6.0__py3-none-any.whl → 0.6.1__py3-none-any.whl

lmnr/opentelemetry_lib/__init__.py

@@ -1,3 +1,4 @@
+import logging
 import sys
 
 from typing import Optional, Set
@@ -20,6 +21,7 @@ class TracerManager:
         exporter: Optional[SpanExporter] = None,
         resource_attributes: dict = {},
         instruments: Optional[Set[Instruments]] = None,
+        block_instruments: Optional[Set[Instruments]] = None,
         base_url: str = "https://api.lmnr.ai",
         port: int = 8443,
         http_port: int = 443,
@@ -27,6 +29,8 @@ class TracerManager:
         max_export_batch_size: Optional[int] = None,
         force_http: bool = False,
         timeout_seconds: int = 30,
+        set_global_tracer_provider: bool = True,
+        otel_logger_level: int = logging.ERROR,
     ) -> None:
         enable_content_tracing = True
 
@@ -37,6 +41,7 @@ class TracerManager:
             disable_batch=disable_batch,
             exporter=exporter,
             instruments=instruments,
+            block_instruments=block_instruments,
             base_url=base_url,
             port=port,
             http_port=http_port,
@@ -44,6 +49,8 @@ class TracerManager:
             max_export_batch_size=max_export_batch_size,
             force_http=force_http,
             timeout_seconds=timeout_seconds,
+            set_global_tracer_provider=set_global_tracer_provider,
+            otel_logger_level=otel_logger_level,
         )
 
     @staticmethod

lmnr/opentelemetry_lib/tracing/__init__.py

@@ -11,13 +11,12 @@ from lmnr.opentelemetry_lib.tracing.instruments import (
 )
 
 from opentelemetry import trace
+from opentelemetry.instrumentation.threading import ThreadingInstrumentor
 from opentelemetry.sdk.resources import Resource
 from opentelemetry.sdk.trace import TracerProvider, SpanProcessor
 from opentelemetry.sdk.trace.export import SpanExporter
-
 from typing import Optional, Set
 
-
 module_logger = logging.getLogger(__name__)
 console_log_handler = logging.StreamHandler()
 console_log_handler.setFormatter(VerboseColorfulFormatter())
@@ -52,7 +51,12 @@ class TracerWrapper(object):
         max_export_batch_size: Optional[int] = None,
         force_http: bool = False,
         timeout_seconds: int = 10,
+        set_global_tracer_provider: bool = True,
+        otel_logger_level: int = logging.ERROR,
     ) -> "TracerWrapper":
+        # Silence some opentelemetry warnings
+        logging.getLogger("opentelemetry.trace").setLevel(otel_logger_level)
+
         base_http_url = f"{base_url}:{http_port}"
         cls._initialize_logger(cls)
         if not hasattr(cls, "instance"):
@@ -68,7 +72,6 @@ class TracerWrapper(object):
             )
 
             obj.__resource = Resource(attributes=TracerWrapper.resource_attributes)
-            obj.__tracer_provider = TracerProvider(resource=obj.__resource)
 
             obj.__span_processor = LaminarSpanProcessor(
                 base_url=base_url,
@@ -81,8 +84,24 @@ class TracerWrapper(object):
                 disable_batch=disable_batch,
             )
 
+            lmnr_provider = TracerProvider(resource=obj.__resource)
+            global_provider = trace.get_tracer_provider()
+            if set_global_tracer_provider and isinstance(
+                global_provider, trace.ProxyTracerProvider
+            ):
+                trace.set_tracer_provider(lmnr_provider)
+
+            obj.__tracer_provider = lmnr_provider
+
             obj.__tracer_provider.add_span_processor(obj.__span_processor)
 
+            # This is not a real instrumentation and does not generate telemetry
+            # data, but it is required to ensure that OpenTelemetry context
+            # propagation is enabled.
+            # See the README at:
+            # https://pypi.org/project/opentelemetry-instrumentation-threading/
+            ThreadingInstrumentor().instrument()
+
             init_instrumentations(
                 tracer_provider=obj.__tracer_provider,
                 instruments=instruments,

lmnr/sdk/browser/playwright_otel.py

@@ -1,6 +1,7 @@
 import logging
 import uuid
 
+from lmnr.opentelemetry_lib.utils.package_check import is_package_installed
 from lmnr.sdk.browser.pw_utils import handle_navigation_async, handle_navigation_sync
 from lmnr.sdk.browser.utils import with_tracer_and_client_wrapper
 from lmnr.sdk.client.asynchronous.async_client import AsyncLaminarClient
@@ -22,17 +23,31 @@ from typing import Collection
 from wrapt import wrap_function_wrapper
 
 try:
-    from playwright.async_api import Browser, BrowserContext, Page
-    from playwright.sync_api import (
-        Browser as SyncBrowser,
-        BrowserContext as SyncBrowserContext,
-        Page as SyncPage,
-    )
+    if is_package_installed("playwright"):
+        from playwright.async_api import Browser, BrowserContext, Page
+        from playwright.sync_api import (
+            Browser as SyncBrowser,
+            BrowserContext as SyncBrowserContext,
+            Page as SyncPage,
+        )
+    elif is_package_installed("patchright"):
+        from patchright.async_api import Browser, BrowserContext, Page
+        from patchright.sync_api import (
+            Browser as SyncBrowser,
+            BrowserContext as SyncBrowserContext,
+            Page as SyncPage,
+        )
+    else:
+        raise ImportError(
+            "Attempted to import lmnr.sdk.browser.playwright_otel, but neither "
+            "playwright nor patchright is installed. Use `pip install playwright` "
+            "or `pip install patchright` to install one of the supported browsers."
+        )
 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."
+        "or `pip install patchright` to install Playwright or remove this import."
     ) from e
 
 # all available versions at https://pypi.org/project/playwright/#history

lmnr/sdk/browser/pw_utils.py

@@ -6,19 +6,31 @@ import threading
 
 from opentelemetry import trace
 
+from lmnr.opentelemetry_lib.utils.package_check import is_package_installed
 from lmnr.sdk.decorators import observe
 from lmnr.sdk.browser.utils import retry_sync, retry_async
 from lmnr.sdk.client.synchronous.sync_client import LaminarClient
 from lmnr.sdk.client.asynchronous.async_client import AsyncLaminarClient
 
+
 try:
-    from playwright.async_api import Page
-    from playwright.sync_api import Page as SyncPage
+    if is_package_installed("playwright"):
+        from playwright.async_api import Page
+        from playwright.sync_api import Page as SyncPage
+    elif is_package_installed("patchright"):
+        from patchright.async_api import Page
+        from patchright.sync_api import Page as SyncPage
+    else:
+        raise ImportError(
+            "Attempted to import lmnr.sdk.browser.pw_utils, but neither "
+            "playwright nor patchright is installed. Use `pip install playwright` "
+            "or `pip install patchright` to install one of the supported browsers."
+        )
 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."
+        "Attempted to import lmnr.sdk.browser.pw_utils, but neither "
+        "playwright nor patchright is installed. Use `pip install playwright` "
+        "or `pip install patchright` to install one of the supported browsers."
     ) from e
 
 logger = logging.getLogger(__name__)

lmnr/sdk/laminar.py

@@ -59,9 +59,12 @@ class Laminar:
         http_port: Optional[int] = None,
         grpc_port: Optional[int] = None,
         instruments: Optional[Set[Instruments]] = None,
+        disabled_instruments: Optional[Set[Instruments]] = None,
         disable_batch: bool = False,
         max_export_batch_size: Optional[int] = None,
         export_timeout_seconds: Optional[int] = None,
+        set_global_tracer_provider: bool = True,
+        otel_logger_level: int = logging.ERROR,
     ):
         """Initialize Laminar context across the application.
         This method must be called before using any other Laminar methods or
@@ -86,6 +89,8 @@ class Laminar:
                         enable. Defaults to all instruments. You can pass\
                         an empty set to disable all instruments. Read more:\
                         https://docs.lmnr.ai/tracing/automatic-instrumentation
+            disabled_instruments (Optional[Set[Instruments]], optional): Instruments to\
+                        disable. Defaults to None.
             disable_batch (bool, optional): If set to True, spans will be sent\
                         immediately to the backend. Useful for debugging, but\
                         may cause performance overhead in production.
@@ -94,6 +99,13 @@ class Laminar:
                         exporter. Defaults to 30 seconds (unlike the\
                         OpenTelemetry default of 10 seconds).
                         Defaults to None.
+            set_global_tracer_provider (bool, optional): If set to True, the\
+                        Laminar tracer provider will be set as the global\
+                        tracer provider. OpenTelemetry allows only one tracer\
+                        provider per app, so set this to False, if you are using\
+                        another tracing library. Setting this to False may break\
+                        some external instrumentations, e.g. LiteLLM.
+                        Defaults to True.
 
         Raises:
             ValueError: If project API key is not set
@@ -133,9 +145,12 @@ class Laminar:
             port=grpc_port or 8443,
             project_api_key=cls.__project_api_key,
             instruments=instruments,
+            block_instruments=disabled_instruments,
             disable_batch=disable_batch,
             max_export_batch_size=max_export_batch_size,
             timeout_seconds=export_timeout_seconds,
+            set_global_tracer_provider=set_global_tracer_provider,
+            otel_logger_level=otel_logger_level,
         )
 
     @classmethod

lmnr/version.py

@@ -3,7 +3,7 @@ import httpx
 from packaging import version
 
 
-__version__ = "0.6.0"
+__version__ = "0.6.1"
 PYTHON_VERSION = f"{sys.version_info.major}.{sys.version_info.minor}"
 
 

{lmnr-0.6.0.dist-info → lmnr-0.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lmnr
-Version: 0.6.0
+Version: 0.6.1
 Summary: Python SDK for Laminar
 License: Apache-2.0
 Author: lmnr.ai
@@ -90,6 +90,7 @@ Requires-Dist: opentelemetry-instrumentation-replicate (>=0.40.5) ; extra == "al
 Requires-Dist: opentelemetry-instrumentation-replicate (>=0.40.5) ; extra == "replicate"
 Requires-Dist: opentelemetry-instrumentation-sagemaker (>=0.40.5) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-sagemaker (>=0.40.5) ; extra == "sagemaker"
+Requires-Dist: opentelemetry-instrumentation-threading (>=0.54b0)
 Requires-Dist: opentelemetry-instrumentation-together (>=0.40.5) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-together (>=0.40.5) ; extra == "together"
 Requires-Dist: opentelemetry-instrumentation-transformers (>=0.40.5) ; extra == "all"

{lmnr-0.6.0.dist-info → lmnr-0.6.1.dist-info}/RECORD

@@ -1,12 +1,12 @@
 lmnr/__init__.py,sha256=eJ-gIHEk8KV-BeaU8c9spQww_T2G5_OMu4F8JEzngvA,1281
 lmnr/cli.py,sha256=e_Jgcwn3Q-hgR6VLLar2ccWLAhJb8yB4cwoIdg5vwDs,3013
 lmnr/opentelemetry_lib/.flake8,sha256=bCxuDlGx3YQ55QHKPiGJkncHanh9qGjQJUujcFa3lAU,150
-lmnr/opentelemetry_lib/__init__.py,sha256=ai_eJynFwWWcrJ7yHUHyDfdjJ3q85GFjjp5pnjROLGI,1768
+lmnr/opentelemetry_lib/__init__.py,sha256=QMPaVYATF_rT50A-lqoJZaH6YlzRu1BQKPgw0L3Rxs0,2107
 lmnr/opentelemetry_lib/decorators/__init__.py,sha256=fTO-p6CaLFw2-8HDVGUS1dvnfCMwUH5vZSYc6_NVAuU,7995
 lmnr/opentelemetry_lib/opentelemetry/instrumentation/google_genai/__init__.py,sha256=jrvHu8aq2EBGZI3ypucUltR4-v8HqTKnqGtWXF5Qbb8,15339
 lmnr/opentelemetry_lib/opentelemetry/instrumentation/google_genai/config.py,sha256=FeQOcC3RNq-yOd8KZ_VoBuZgDl6tnkxx6I2MKVH1GMI,256
 lmnr/opentelemetry_lib/opentelemetry/instrumentation/google_genai/utils.py,sha256=P0fOmGi_0nv3cMFcdWblRuTsDRCZnvdM8GQ3zJT0qbM,6229
-lmnr/opentelemetry_lib/tracing/__init__.py,sha256=dQMccjiEUe66ocb1cn0ZtPXdxgJpxUNAtzzFd_aEGW8,4771
+lmnr/opentelemetry_lib/tracing/__init__.py,sha256=7uGdMNAlUJv2HcNz3XAqpb7iUNsiQwSY9ipv8D2V10I,5713
 lmnr/opentelemetry_lib/tracing/_instrument_initializers.py,sha256=CBgkfVKmvWWvSgN8eZHgEF92ED8K4VkXzz9d0QC4tIs,14269
 lmnr/opentelemetry_lib/tracing/attributes.py,sha256=MvowVluXfCqSIC3Cvx3tWDqB0Cpr9bpSlY91qL4Iy74,1497
 lmnr/opentelemetry_lib/tracing/context_properties.py,sha256=xMLitHrENwSW3U9_PweAFRIdTm8hqwHP4Vi3S8jazXA,1757
@@ -22,8 +22,8 @@ lmnr/sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lmnr/sdk/browser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lmnr/sdk/browser/browser_use_otel.py,sha256=Cx8ZLoEGLAAOIvglmZEAxMkfmdwUfjrvxgiZ5vfY3Ko,4519
 lmnr/sdk/browser/patchright_otel.py,sha256=O7n1dB_Mw-_L70zi0zqpnFFiKuTPEEpO7ry1ia6IXWQ,5454
-lmnr/sdk/browser/playwright_otel.py,sha256=dPmdZo9MeSkPGNG8lUCP9JV1G7pW8MXLWT6xoFVOVUA,12672
-lmnr/sdk/browser/pw_utils.py,sha256=Tw5kmONwPu9Q7_QHGIa-amotImQugGVFeEyMCUxmDhc,10243
+lmnr/sdk/browser/playwright_otel.py,sha256=LFg1iJXbez-BEgEIY9eaO_2T2uR6SxyfFL46fHcS7Mg,13423
+lmnr/sdk/browser/pw_utils.py,sha256=nFqVujQb7owVFzT-31_g09CbfbBwUIXKxs3JxuE47ws,10835
 lmnr/sdk/browser/rrweb/rrweb.umd.min.cjs,sha256=Ly2jiwC7hTEtgiXzBpoJNSE1Vkzu0lZPZS8brjusAW0,260896
 lmnr/sdk/browser/utils.py,sha256=xPpMRP2y9aJIsdIDNg2wN4PSa_4w0LSsra-GIMx9VXc,2366
 lmnr/sdk/client/asynchronous/async_client.py,sha256=NuGla1gGlyYLxyIQ6LSKG-b5eYGNIQA5HXAOzGQR5BU,4036
@@ -42,13 +42,13 @@ lmnr/sdk/datasets.py,sha256=jl5Wj5nEI9pww4Jwn4XKF8h0gXBU4TOIrhqNjTJsHZQ,1709
 lmnr/sdk/decorators.py,sha256=NIen6dUhcryiWnoZTP6C6_llwPZqbGoOr8iWPorKWuY,3346
 lmnr/sdk/eval_control.py,sha256=qMiAI6FnHdIwKX8-W1nDhEcS5Cjm5lYRoIN7x4J-AVI,182
 lmnr/sdk/evaluations.py,sha256=SMCoYMHGwMKEXaGT6tR10dvXXU_OrazlYZCy11hKlQc,20444
-lmnr/sdk/laminar.py,sha256=Zh4KCIDCOe7yTdZBbYNPXG_WxURSkzkJOR5wKFjQPj0,27180
+lmnr/sdk/laminar.py,sha256=QbMbxdAhFzVIWzKQRtoVv9M5qMQj_cdhJ4ph3d22Rxs,28158
 lmnr/sdk/log.py,sha256=nt_YMmPw1IRbGy0b7q4rTtP4Yo3pQfNxqJPXK3nDSNQ,2213
 lmnr/sdk/types.py,sha256=QOIZ18ELzts5AG1Tx-_twHI43_PHoB4X8JYynTH0MCA,12287
 lmnr/sdk/utils.py,sha256=_kUn43AEn7Cdqcsg1CkQvzVQ7PhQT27m-iuLcRrwlZM,3891
-lmnr/version.py,sha256=USfbIJ-o8QurfmN8Y-r1THrXx4dRuPHdtWhHLn46QTU,1321
-lmnr-0.6.0.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
-lmnr-0.6.0.dist-info/METADATA,sha256=-Lu3ClhnpKP7iwOrIe4Fp6vEY98gGpXOClE0D2iEs-E,15069
-lmnr-0.6.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-lmnr-0.6.0.dist-info/entry_points.txt,sha256=K1jE20ww4jzHNZLnsfWBvU3YKDGBgbOiYG5Y7ivQcq4,37
-lmnr-0.6.0.dist-info/RECORD,,
+lmnr/version.py,sha256=gIiYGEDo229n5RGCEUAivOOfgZPjhsZSIzuJPxtIZvQ,1321
+lmnr-0.6.1.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
+lmnr-0.6.1.dist-info/METADATA,sha256=dBBwpBlKLQXHADF1afzpoffOLDIEJmA9ac8diguBYWM,15135
+lmnr-0.6.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+lmnr-0.6.1.dist-info/entry_points.txt,sha256=K1jE20ww4jzHNZLnsfWBvU3YKDGBgbOiYG5Y7ivQcq4,37
+lmnr-0.6.1.dist-info/RECORD,,