lmnr 0.4.66__tar.gz → 0.5.0__tar.gz

{lmnr-0.4.66 → lmnr-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lmnr
-Version: 0.4.66
+Version: 0.5.0
 Summary: Python SDK for Laminar
 License: Apache-2.0
 Author: lmnr.ai
@@ -39,12 +39,12 @@ Provides-Extra: transformers
 Provides-Extra: vertexai
 Provides-Extra: watsonx
 Provides-Extra: weaviate
-Requires-Dist: aiohttp (>=3.0)
 Requires-Dist: argparse (>=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: httpx (>=0.28.1)
+Requires-Dist: opentelemetry-api (>=1.31.1)
+Requires-Dist: opentelemetry-exporter-otlp-proto-grpc (>=1.31.1)
+Requires-Dist: opentelemetry-exporter-otlp-proto-http (>=1.31.1)
 Requires-Dist: opentelemetry-instrumentation-alephalpha (>=0.38.12) ; extra == "alephalpha"
 Requires-Dist: opentelemetry-instrumentation-alephalpha (>=0.38.12) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-anthropic (>=0.38.12) ; extra == "all"
@@ -83,27 +83,26 @@ Requires-Dist: opentelemetry-instrumentation-qdrant (>=0.38.12) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-qdrant (>=0.38.12) ; extra == "qdrant"
 Requires-Dist: opentelemetry-instrumentation-replicate (>=0.38.12) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-replicate (>=0.38.12) ; extra == "replicate"
-Requires-Dist: opentelemetry-instrumentation-requests (>=0.50b0)
+Requires-Dist: opentelemetry-instrumentation-requests (>=0.52b0)
 Requires-Dist: opentelemetry-instrumentation-sagemaker (>=0.38.12) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-sagemaker (>=0.38.12) ; extra == "sagemaker"
-Requires-Dist: opentelemetry-instrumentation-sqlalchemy (>=0.50b0)
-Requires-Dist: opentelemetry-instrumentation-threading (>=0.50b0)
+Requires-Dist: opentelemetry-instrumentation-sqlalchemy (>=0.52b0)
+Requires-Dist: opentelemetry-instrumentation-threading (>=0.52b0)
 Requires-Dist: opentelemetry-instrumentation-together (>=0.38.12) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-together (>=0.38.12) ; extra == "together"
 Requires-Dist: opentelemetry-instrumentation-transformers (>=0.38.12) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-transformers (>=0.38.12) ; extra == "transformers"
-Requires-Dist: opentelemetry-instrumentation-urllib3 (>=0.50b0)
+Requires-Dist: opentelemetry-instrumentation-urllib3 (>=0.52b0)
 Requires-Dist: opentelemetry-instrumentation-vertexai (>=0.38.12) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-vertexai (>=0.38.12) ; extra == "vertexai"
 Requires-Dist: opentelemetry-instrumentation-watsonx (>=0.38.12) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-watsonx (>=0.38.12) ; extra == "watsonx"
 Requires-Dist: opentelemetry-instrumentation-weaviate (>=0.38.12) ; extra == "all"
 Requires-Dist: opentelemetry-instrumentation-weaviate (>=0.38.12) ; extra == "weaviate"
-Requires-Dist: opentelemetry-sdk (>=1.28.0)
+Requires-Dist: opentelemetry-sdk (>=1.31.1)
 Requires-Dist: opentelemetry-semantic-conventions-ai (>=0.4.2)
 Requires-Dist: pydantic (>=2.0.3)
 Requires-Dist: python-dotenv (>=1.0)
-Requires-Dist: requests (>=2.0)
 Requires-Dist: tenacity (>=8.0)
 Requires-Dist: tqdm (>=4.0)
 Description-Content-Type: text/markdown
@@ -145,7 +144,28 @@ from lmnr import Laminar
 Laminar.initialize(project_api_key="<PROJECT_API_KEY>")
 ```
 
-Note that you need to only initialize Laminar once in your application.
+You can also skip passing the `project_api_key`, in which case it will be looked
+in the environment (or local .env file) by the key `LMNR_PROJECT_API_KEY`.
+
+Note that you need to only initialize Laminar once in your application. You should
+try to do that as early as possible in your application, e.g. at server startup.
+
+## Set-up for self-hosting
+
+If you self-host a Laminar instance, the default connection settings to it are
+`http://localhost:8000` for HTTP and `http://localhost:8001` for gRPC. Initialize
+the SDK accordingly:
+
+```python
+from lmnr import Laminar
+
+Laminar.initialize(
+    project_api_key="<PROJECT_API_KEY>",
+    base_url="http://localhost",
+    http_port=8000,
+    grpc_port=8001,
+)
+```
 
 ## Instrumentation
 
@@ -281,49 +301,79 @@ You can run evaluations locally by providing executor (part of the logic used in
 
 Read the [docs](https://docs.lmnr.ai/evaluations/introduction) to learn more about evaluations.
 
-## Laminar pipelines as prompt chain managers
+## Client for HTTP operations
 
-You can create Laminar pipelines in the UI and manage chains of LLM calls there.
+Various interactions with Laminar [API](https://docs.lmnr.ai/api-reference/) are available in `LaminarClient`
+and its asynchronous version `AsyncLaminarClient`.
 
-After you are ready to use your pipeline in your code, deploy it in Laminar by selecting the target version for the pipeline.
+### Agent
 
-Once your pipeline target is set, you can call it from Python in just a few lines.
-
-Example use:
+To run Laminar agent, you can invoke `client.agent.run`
 
 ```python
-from lmnr import Laminar
+from lmnr import LaminarClient
 
-Laminar.initialize('<YOUR_PROJECT_API_KEY>', instruments=set())
+client = LaminarClient(project_api_key="<YOUR_PROJECT_API_KEY>")
 
-result = Laminar.run(
-    pipeline = 'my_pipeline_name',
-    inputs = {'input_node_name': 'some_value'},
-    # all environment variables
-    env = {'OPENAI_API_KEY': 'sk-some-key'},
+response = client.agent.run(
+    prompt="What is the weather in London today?"
 )
+
+print(response.result.content)
 ```
 
-Resulting in:
+#### Streaming
+
+Agent run supports streaming as well.
 
 ```python
->>> result
-PipelineRunResponse(
-    outputs={'output': {'value': [ChatMessage(role='user', content='hello')]}},
-    # useful to locate your trace
-    run_id='53b012d5-5759-48a6-a9c5-0011610e3669'
-)
+from lmnr import LaminarClient
+
+client = LaminarClient(project_api_key="<YOUR_PROJECT_API_KEY>")
+
+for chunk in client.agent.run(
+    prompt="What is the weather in London today?",
+    stream=True
+):
+    if chunk.chunkType == 'step':
+        print(chunk.summary)
+    elif chunk.chunkType == 'finalOutput':
+        print(chunk.content.result.content)
 ```
 
-## Semantic search
+#### Async mode
+
+```python
+from lmnr import AsyncLaminarClient
+
+client = AsyncLaminarClient(project_api_key="<YOUR_PROJECT_API_KEY>")
 
-You can perform a semantic search on a dataset in Laminar by calling `Laminar.semantic_search`.
+response = await client.agent.run(
+    prompt="What is the weather in London today?"
+)
+
+print(response.result.content)
+```
+
+#### Async mode with streaming
 
 ```python
-response = Laminar.semantic_search(
-    query="Greatest Chinese architectural wonders",
-    dataset_id=uuid.UUID("413f8404-724c-4aa4-af16-714d84fd7958"),
+from lmnr import AsyncLaminarClient
+
+client = AsyncLaminarClient(project_api_key="<YOUR_PROJECT_API_KEY>")
+
+# Note that you need to await the operation even though we use `async for` below
+response = await client.agent.run(
+    prompt="What is the weather in London today?",
+    stream=True
 )
+async for chunk in client.agent.run(
+    prompt="What is the weather in London today?",
+    stream=True
+):
+    if chunk.chunkType == 'step':
+        print(chunk.summary)
+    else if chunk.chunkType == 'finalOutput':
+        print(chunk.content.result.content)
 ```
 
-[Read more](https://docs.lmnr.ai/datasets/indexing) about indexing and semantic search.

{lmnr-0.4.66 → lmnr-0.5.0}/README.md

@@ -35,7 +35,28 @@ from lmnr import Laminar
 Laminar.initialize(project_api_key="<PROJECT_API_KEY>")
 ```
 
-Note that you need to only initialize Laminar once in your application.
+You can also skip passing the `project_api_key`, in which case it will be looked
+in the environment (or local .env file) by the key `LMNR_PROJECT_API_KEY`.
+
+Note that you need to only initialize Laminar once in your application. You should
+try to do that as early as possible in your application, e.g. at server startup.
+
+## Set-up for self-hosting
+
+If you self-host a Laminar instance, the default connection settings to it are
+`http://localhost:8000` for HTTP and `http://localhost:8001` for gRPC. Initialize
+the SDK accordingly:
+
+```python
+from lmnr import Laminar
+
+Laminar.initialize(
+    project_api_key="<PROJECT_API_KEY>",
+    base_url="http://localhost",
+    http_port=8000,
+    grpc_port=8001,
+)
+```
 
 ## Instrumentation
 
@@ -171,49 +192,78 @@ You can run evaluations locally by providing executor (part of the logic used in
 
 Read the [docs](https://docs.lmnr.ai/evaluations/introduction) to learn more about evaluations.
 
-## Laminar pipelines as prompt chain managers
+## Client for HTTP operations
 
-You can create Laminar pipelines in the UI and manage chains of LLM calls there.
+Various interactions with Laminar [API](https://docs.lmnr.ai/api-reference/) are available in `LaminarClient`
+and its asynchronous version `AsyncLaminarClient`.
 
-After you are ready to use your pipeline in your code, deploy it in Laminar by selecting the target version for the pipeline.
+### Agent
 
-Once your pipeline target is set, you can call it from Python in just a few lines.
-
-Example use:
+To run Laminar agent, you can invoke `client.agent.run`
 
 ```python
-from lmnr import Laminar
+from lmnr import LaminarClient
 
-Laminar.initialize('<YOUR_PROJECT_API_KEY>', instruments=set())
+client = LaminarClient(project_api_key="<YOUR_PROJECT_API_KEY>")
 
-result = Laminar.run(
-    pipeline = 'my_pipeline_name',
-    inputs = {'input_node_name': 'some_value'},
-    # all environment variables
-    env = {'OPENAI_API_KEY': 'sk-some-key'},
+response = client.agent.run(
+    prompt="What is the weather in London today?"
 )
+
+print(response.result.content)
 ```
 
-Resulting in:
+#### Streaming
+
+Agent run supports streaming as well.
 
 ```python
->>> result
-PipelineRunResponse(
-    outputs={'output': {'value': [ChatMessage(role='user', content='hello')]}},
-    # useful to locate your trace
-    run_id='53b012d5-5759-48a6-a9c5-0011610e3669'
-)
+from lmnr import LaminarClient
+
+client = LaminarClient(project_api_key="<YOUR_PROJECT_API_KEY>")
+
+for chunk in client.agent.run(
+    prompt="What is the weather in London today?",
+    stream=True
+):
+    if chunk.chunkType == 'step':
+        print(chunk.summary)
+    elif chunk.chunkType == 'finalOutput':
+        print(chunk.content.result.content)
 ```
 
-## Semantic search
-
-You can perform a semantic search on a dataset in Laminar by calling `Laminar.semantic_search`.
+#### Async mode
 
 ```python
-response = Laminar.semantic_search(
-    query="Greatest Chinese architectural wonders",
-    dataset_id=uuid.UUID("413f8404-724c-4aa4-af16-714d84fd7958"),
+from lmnr import AsyncLaminarClient
+
+client = AsyncLaminarClient(project_api_key="<YOUR_PROJECT_API_KEY>")
+
+response = await client.agent.run(
+    prompt="What is the weather in London today?"
 )
+
+print(response.result.content)
 ```
 
-[Read more](https://docs.lmnr.ai/datasets/indexing) about indexing and semantic search.
+#### Async mode with streaming
+
+```python
+from lmnr import AsyncLaminarClient
+
+client = AsyncLaminarClient(project_api_key="<YOUR_PROJECT_API_KEY>")
+
+# Note that you need to await the operation even though we use `async for` below
+response = await client.agent.run(
+    prompt="What is the weather in London today?",
+    stream=True
+)
+async for chunk in client.agent.run(
+    prompt="What is the weather in London today?",
+    stream=True
+):
+    if chunk.chunkType == 'step':
+        print(chunk.summary)
+    else if chunk.chunkType == 'finalOutput':
+        print(chunk.content.result.content)
+```

{lmnr-0.4.66 → lmnr-0.5.0}/pyproject.toml

@@ -6,7 +6,7 @@
 
 [project]
 name = "lmnr"
-version = "0.4.66"
+version = "0.5.0"
 description = "Python SDK for Laminar"
 authors = [
   { name = "lmnr.ai", email = "founders@lmnr.ai" }
@@ -16,20 +16,18 @@ 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-api (>=1.31.1)",
+  "opentelemetry-sdk (>=1.31.1)",
+  "opentelemetry-exporter-otlp-proto-http (>=1.31.1)",
+  "opentelemetry-exporter-otlp-proto-grpc (>=1.31.1)",
+  "opentelemetry-instrumentation-requests (>=0.52b0)",
+  "opentelemetry-instrumentation-sqlalchemy (>=0.52b0)",
+  "opentelemetry-instrumentation-urllib3 (>=0.52b0)",
+  "opentelemetry-instrumentation-threading (>=0.52b0)",
   "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.
@@ -40,6 +38,7 @@ dependencies = [
   # 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",
+  "httpx>=0.28.1",
 ]
 
 [project.scripts]
@@ -114,7 +113,8 @@ dev = [
   "flake8",
   "pytest>=8.3.4",
   "pytest-sugar",
-  "pytest-asyncio>=0.25.2"
+  "pytest-asyncio>=0.25.2",
+  "playwright>=1.51.0"
 ]
 
 [build-system]

lmnr-0.5.0/src/lmnr/__init__.py

@@ -0,0 +1,46 @@
+from .sdk.client.synchronous.sync_client import LaminarClient
+from .sdk.client.asynchronous.async_client import AsyncLaminarClient
+from .sdk.datasets import EvaluationDataset, LaminarDataset
+from .sdk.evaluations import evaluate
+from .sdk.laminar import Laminar
+from .sdk.types import (
+    AgentOutput,
+    FinalOutputChunkContent,
+    ChatMessage,
+    HumanEvaluator,
+    NodeInput,
+    PipelineRunError,
+    PipelineRunResponse,
+    RunAgentResponseChunk,
+    StepChunkContent,
+    TracingLevel,
+)
+from .sdk.decorators import observe
+from .sdk.types import LaminarSpanContext
+from .openllmetry_sdk import Instruments
+from .openllmetry_sdk.tracing.attributes import Attributes
+from opentelemetry.trace import use_span
+
+__all__ = [
+    "AgentOutput",
+    "AsyncLaminarClient",
+    "Attributes",
+    "ChatMessage",
+    "EvaluationDataset",
+    "FinalOutputChunkContent",
+    "HumanEvaluator",
+    "Instruments",
+    "Laminar",
+    "LaminarClient",
+    "LaminarDataset",
+    "LaminarSpanContext",
+    "NodeInput",
+    "PipelineRunError",
+    "PipelineRunResponse",
+    "RunAgentResponseChunk",
+    "StepChunkContent",
+    "TracingLevel",
+    "evaluate",
+    "observe",
+    "use_span",
+]

{lmnr-0.4.66 → lmnr-0.5.0}/src/lmnr/openllmetry_sdk/__init__.py

@@ -16,7 +16,7 @@ from lmnr.openllmetry_sdk.tracing.tracing import TracerWrapper
 from typing import Dict
 
 
-class Traceloop:
+class TracerManager:
     __tracer_wrapper: TracerWrapper
 
     @staticmethod
@@ -44,17 +44,6 @@ class Traceloop:
         if isinstance(headers, str):
             headers = parse_env_headers(headers)
 
-        if (
-            not exporter
-            and not processor
-            and api_endpoint == "https://api.lmnr.ai"
-            and not api_key
-        ):
-            print(
-                "Set the LMNR_PROJECT_API_KEY environment variable to your project API key"
-            )
-            return
-
         if api_key and not exporter and not processor and not headers:
             headers = {
                 "Authorization": f"Bearer {api_key}",
@@ -65,7 +54,7 @@ class Traceloop:
         TracerWrapper.set_static_params(
             resource_attributes, enable_content_tracing, api_endpoint, headers
         )
-        Traceloop.__tracer_wrapper = TracerWrapper(
+        TracerManager.__tracer_wrapper = TracerWrapper(
             disable_batch=disable_batch,
             processor=processor,
             propagator=propagator,
@@ -79,5 +68,5 @@ class Traceloop:
 
     @staticmethod
     def flush():
-        if Traceloop.__tracer_wrapper:
-            Traceloop.__tracer_wrapper.flush()
+        if getattr(TracerManager, "__tracer_wrapper", None):
+            TracerManager.__tracer_wrapper.flush()

{lmnr-0.4.66 → lmnr-0.5.0}/src/lmnr/openllmetry_sdk/tracing/attributes.py

@@ -9,7 +9,6 @@ SPAN_IDS_PATH = "lmnr.span.ids_path"
 SPAN_INSTRUMENTATION_SOURCE = "lmnr.span.instrumentation_source"
 SPAN_SDK_VERSION = "lmnr.span.sdk_version"
 SPAN_LANGUAGE_VERSION = "lmnr.span.language_version"
-OVERRIDE_PARENT_SPAN = "lmnr.internal.override_parent_span"
 
 ASSOCIATION_PROPERTIES = "lmnr.association.properties"
 SESSION_ID = "session_id"

{lmnr-0.4.66 → lmnr-0.5.0}/src/lmnr/openllmetry_sdk/tracing/tracing.py

@@ -4,6 +4,8 @@ import logging
 import uuid
 
 from contextvars import Context
+from lmnr.sdk.client.asynchronous.async_client import AsyncLaminarClient
+from lmnr.sdk.client.synchronous.sync_client import LaminarClient
 from lmnr.sdk.log import VerboseColorfulFormatter
 from lmnr.openllmetry_sdk.instruments import Instruments
 from lmnr.openllmetry_sdk.tracing.attributes import (
@@ -41,7 +43,7 @@ from opentelemetry.trace import get_tracer_provider, ProxyTracerProvider
 
 from typing import Dict, Optional, Set
 
-from lmnr.version import SDK_VERSION, PYTHON_VERSION
+from lmnr.version import __version__, PYTHON_VERSION
 
 module_logger = logging.getLogger(__name__)
 console_log_handler = logging.StreamHandler()
@@ -80,6 +82,8 @@ class TracerWrapper(object):
     __logger: logging.Logger = None
     __span_id_to_path: dict[int, list[str]] = {}
     __span_id_lists: dict[int, list[str]] = {}
+    __client: LaminarClient = None
+    __async_client: AsyncLaminarClient = None
 
     def __new__(
         cls,
@@ -99,6 +103,15 @@ class TracerWrapper(object):
             if not TracerWrapper.endpoint:
                 return obj
 
+            obj.__client = LaminarClient(
+                base_url=base_http_url,
+                project_api_key=project_api_key,
+            )
+            obj.__async_client = AsyncLaminarClient(
+                base_url=base_http_url,
+                project_api_key=project_api_key,
+            )
+
             obj.__resource = Resource(attributes=TracerWrapper.resource_attributes)
             obj.__tracer_provider = init_tracer_provider(resource=obj.__resource)
             if processor:
@@ -135,8 +148,8 @@ class TracerWrapper(object):
             instrument_set = init_instrumentations(
                 should_enrich_metrics,
                 instruments,
-                base_http_url=base_http_url,
-                project_api_key=project_api_key,
+                client=obj.__client,
+                async_client=obj.__async_client,
             )
 
             if not instrument_set:
@@ -184,7 +197,7 @@ class TracerWrapper(object):
         self.__span_id_lists[span.get_span_context().span_id] = span_ids_path
 
         span.set_attribute(SPAN_INSTRUMENTATION_SOURCE, "python")
-        span.set_attribute(SPAN_SDK_VERSION, SDK_VERSION)
+        span.set_attribute(SPAN_SDK_VERSION, __version__)
         span.set_attribute(SPAN_LANGUAGE_VERSION, f"python@{PYTHON_VERSION}")
 
         association_properties = get_value("association_properties")
@@ -320,8 +333,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,
+    client: Optional[LaminarClient] = None,
+    async_client: Optional[AsyncLaminarClient] = None,
 ):
     block_instruments = block_instruments or set()
     # These libraries are not instrumented by default,
@@ -434,7 +447,7 @@ def init_instrumentations(
             if init_weaviate_instrumentor():
                 instrument_set = True
         elif instrument == Instruments.PLAYWRIGHT:
-            if init_playwright_instrumentor():
+            if init_playwright_instrumentor(client, async_client):
                 instrument_set = True
         elif instrument == Instruments.BROWSER_USE:
             if init_browser_use_instrumentor():
@@ -465,12 +478,14 @@ def init_browser_use_instrumentor():
         return False
 
 
-def init_playwright_instrumentor():
+def init_playwright_instrumentor(
+    client: LaminarClient, async_client: AsyncLaminarClient
+):
     try:
         if is_package_installed("playwright"):
             from lmnr.sdk.browser.playwright_otel import PlaywrightInstrumentor
 
-            instrumentor = PlaywrightInstrumentor()
+            instrumentor = PlaywrightInstrumentor(client, async_client)
             instrumentor.instrument()
         return True
     except Exception as e:

{lmnr-0.4.66 → lmnr-0.5.0}/src/lmnr/sdk/browser/browser_use_otel.py

@@ -1,7 +1,7 @@
 from lmnr.openllmetry_sdk.decorators.base import json_dumps
-from lmnr.sdk.browser.utils import _with_tracer_wrapper
+from lmnr.sdk.browser.utils import with_tracer_wrapper
 from lmnr.sdk.utils import get_input_from_func_args
-from lmnr.version import SDK_VERSION
+from lmnr.version import __version__
 
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.instrumentation.utils import unwrap
@@ -50,7 +50,7 @@ WRAPPED_METHODS = [
 ]
 
 
-@_with_tracer_wrapper
+@with_tracer_wrapper
 async def _wrap(tracer: Tracer, to_wrap, wrapped, instance, args, kwargs):
     span_name = to_wrap.get("span_name")
     attributes = {
@@ -86,7 +86,7 @@ class BrowserUseInstrumentor(BaseInstrumentor):
 
     def _instrument(self, **kwargs):
         tracer_provider = kwargs.get("tracer_provider")
-        tracer = get_tracer(__name__, SDK_VERSION, tracer_provider)
+        tracer = get_tracer(__name__, __version__, tracer_provider)
 
         for wrapped_method in WRAPPED_METHODS:
             wrap_package = wrapped_method.get("package")