lmnr 0.4.33__py3-none-any.whl → 0.4.35__py3-none-any.whl

lmnr/sdk/laminar.py

@@ -1,18 +1,19 @@
 from contextlib import contextmanager
 from contextvars import Context
-from opentelemetry import context, trace
-from opentelemetry.util.types import AttributeValue
-from opentelemetry.context import set_value, attach, detach
 from lmnr.traceloop_sdk import Traceloop
 from lmnr.traceloop_sdk.instruments import Instruments
 from lmnr.traceloop_sdk.tracing import get_tracer
 from lmnr.traceloop_sdk.tracing.attributes import (
+    ASSOCIATION_PROPERTIES,
     Attributes,
     SPAN_TYPE,
     OVERRIDE_PARENT_SPAN,
 )
 from lmnr.traceloop_sdk.decorators.base import json_dumps
+from opentelemetry import context, trace
+from opentelemetry.context import attach, detach, set_value
 from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
+from opentelemetry.util.types import AttributeValue
 
 from pydantic.alias_generators import to_snake
 from typing import Any, Literal, Optional, Set, Union
@@ -39,6 +40,7 @@ from lmnr.traceloop_sdk.tracing.attributes import (
 )
 from lmnr.traceloop_sdk.tracing.tracing import (
     get_span_path,
+    remove_association_properties,
     set_association_properties,
     update_association_properties,
 )
@@ -298,6 +300,7 @@ class Laminar:
         span_type: Union[Literal["DEFAULT"], Literal["LLM"]] = "DEFAULT",
         context: Optional[Context] = None,
         trace_id: Optional[uuid.UUID] = None,
+        labels: Optional[dict[str, str]] = None,
     ):
         """Start a new span as the current span. Useful for manual
         instrumentation. If `span_type` is set to `"LLM"`, you should report
@@ -323,6 +326,8 @@ class Laminar:
             trace_id (Optional[uuid.UUID], optional): [EXPERIMENTAL] override\
                 the trace id for the span. If not provided, use the current\
                 trace id. Defaults to None.
+            labels (Optional[dict[str, str]], optional): labels to set for the\
+                span. Defaults to None.
         """
 
         with get_tracer() as tracer:
@@ -345,10 +350,26 @@ class Laminar:
                         " is not a valid UUID"
                     )
             ctx_token = attach(ctx)
+            label_props = {}
+            try:
+                if labels:
+                    label_props = dict(
+                        (f"{ASSOCIATION_PROPERTIES}.label.{k}", json_dumps(v))
+                        for k, v in labels.items()  # noqa: F821
+                    )
+            except Exception:
+                cls.__logger.warning(
+                    f"`start_as_current_span` Could not set labels: {labels}. "
+                    "They will be propagated to the next span."
+                )
             with tracer.start_as_current_span(
                 name,
                 context=ctx,
-                attributes={SPAN_PATH: span_path, SPAN_TYPE: span_type},
+                attributes={
+                    SPAN_PATH: span_path,
+                    SPAN_TYPE: span_type,
+                    **(label_props),
+                },
             ) as span:
                 if trace_id is not None and isinstance(trace_id, uuid.UUID):
                     span.set_attribute(OVERRIDE_PARENT_SPAN, True)
@@ -365,6 +386,41 @@ class Laminar:
             except Exception:
                 pass
 
+    @classmethod
+    @contextmanager
+    def with_labels(cls, labels: dict[str, str], context: Optional[Context] = None):
+        """Set labels for spans within this `with` context. This is useful for
+        adding labels to the spans created in the auto-instrumentations.
+
+        Requirements:
+        - Labels must be created in your project in advance.
+        - Keys must be strings from your label names.
+        - Values must be strings matching the label's allowed values.
+
+        Usage example:
+        ```python
+        with Laminar.with_labels({"sentiment": "positive"}):
+            openai_client.chat.completions.create()
+        ```
+        """
+        with get_tracer():
+            label_props = labels.copy()
+            label_props = dict(
+                (f"label.{k}", json_dumps(v)) for k, v in label_props.items()
+            )
+            update_association_properties(
+                label_props, set_on_current_span=False, context=context
+            )
+            yield
+            try:
+                remove_association_properties(label_props)
+            except Exception:
+                cls.__logger.warning(
+                    f"`with_labels` Could not remove labels: {labels}. They will be "
+                    "propagated to the next span."
+                )
+                pass
+
     @classmethod
     def start_span(
         cls,
@@ -373,6 +429,7 @@ class Laminar:
         span_type: Union[Literal["DEFAULT"], Literal["LLM"]] = "DEFAULT",
         context: Optional[Context] = None,
         trace_id: Optional[uuid.UUID] = None,
+        labels: Optional[dict[str, str]] = None,
     ):
         """Start a new span. Useful for manual instrumentation.
         If `span_type` is set to `"LLM"`, you should report usage and response
@@ -417,6 +474,8 @@ class Laminar:
             trace_id (Optional[uuid.UUID], optional): [EXPERIMENTAL] override\
                 the trace id for the span. If not provided, use the current\
                 trace id. Defaults to None.
+            labels (Optional[dict[str, str]], optional): labels to set for the\
+                span. Defaults to None.
         """
         with get_tracer() as tracer:
             span_path = get_span_path(name)
@@ -437,10 +496,26 @@ class Laminar:
                         "trace_id provided to `Laminar.start_span`"
                         " is not a valid UUID"
                     )
+            label_props = {}
+            try:
+                if labels:
+                    label_props = dict(
+                        (f"{ASSOCIATION_PROPERTIES}.label.{k}", json_dumps(v))
+                        for k, v in labels.items()  # noqa: F821
+                    )
+            except Exception:
+                cls.__logger.warning(
+                    f"`start_span` Could not set labels: {labels}. They will be "
+                    "propagated to the next span."
+                )
             span = tracer.start_span(
                 name,
                 context=ctx,
-                attributes={SPAN_PATH: span_path, SPAN_TYPE: span_type},
+                attributes={
+                    SPAN_PATH: span_path,
+                    SPAN_TYPE: span_type,
+                    **(label_props),
+                },
             )
             if trace_id is not None and isinstance(trace_id, uuid.UUID):
                 span.set_attribute(OVERRIDE_PARENT_SPAN, True)

lmnr/traceloop_sdk/tracing/tracing.py

@@ -1,8 +1,18 @@
 import atexit
+import copy
 import logging
-import os
 
 
+from contextvars import Context
+from lmnr.traceloop_sdk.instruments import Instruments
+from lmnr.traceloop_sdk.tracing.attributes import (
+    ASSOCIATION_PROPERTIES,
+    SPAN_INSTRUMENTATION_SOURCE,
+    SPAN_PATH,
+)
+from lmnr.traceloop_sdk.tracing.content_allow_list import ContentAllowList
+from lmnr.traceloop_sdk.utils import is_notebook
+from lmnr.traceloop_sdk.utils.package_check import is_package_installed
 from opentelemetry import trace
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
     OTLPSpanExporter as HTTPExporter,
@@ -10,28 +20,19 @@ from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
 from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
     OTLPSpanExporter as GRPCExporter,
 )
+from opentelemetry.instrumentation.threading import ThreadingInstrumentor
+from opentelemetry.context import get_value, attach, set_value
+from opentelemetry.propagate import set_global_textmap
+from opentelemetry.propagators.textmap import TextMapPropagator
 from opentelemetry.sdk.resources import Resource
 from opentelemetry.sdk.trace import TracerProvider, SpanProcessor
-from opentelemetry.propagators.textmap import TextMapPropagator
-from opentelemetry.propagate import set_global_textmap
 from opentelemetry.sdk.trace.export import (
     SpanExporter,
     SimpleSpanProcessor,
     BatchSpanProcessor,
 )
 from opentelemetry.trace import get_tracer_provider, ProxyTracerProvider
-from opentelemetry.context import get_value, attach, set_value
-from opentelemetry.instrumentation.threading import ThreadingInstrumentor
 
-from lmnr.traceloop_sdk.instruments import Instruments
-from lmnr.traceloop_sdk.tracing.attributes import (
-    ASSOCIATION_PROPERTIES,
-    SPAN_INSTRUMENTATION_SOURCE,
-    SPAN_PATH,
-)
-from lmnr.traceloop_sdk.tracing.content_allow_list import ContentAllowList
-from lmnr.traceloop_sdk.utils import is_notebook
-from lmnr.traceloop_sdk.utils.package_check import is_package_installed
 from typing import Dict, Optional, Set
 
 
@@ -188,15 +189,27 @@ def set_association_properties(properties: dict) -> None:
     _set_association_properties_attributes(span, properties)
 
 
-def update_association_properties(properties: dict) -> None:
+def update_association_properties(
+    properties: dict,
+    set_on_current_span: bool = True,
+    context: Optional[Context] = None,
+) -> None:
     """Only adds or updates properties that are not already present"""
-    association_properties = get_value("association_properties") or {}
+    association_properties = get_value("association_properties", context) or {}
     association_properties.update(properties)
 
-    attach(set_value("association_properties", association_properties))
+    attach(set_value("association_properties", association_properties, context))
 
-    span = trace.get_current_span()
-    _set_association_properties_attributes(span, properties)
+    if set_on_current_span:
+        span = trace.get_current_span()
+        _set_association_properties_attributes(span, properties)
+
+
+def remove_association_properties(properties: dict) -> None:
+    props: dict = copy.copy(get_value("association_properties") or {})
+    for k in properties.keys():
+        props.pop(k, None)
+    set_association_properties(props)
 
 
 def _set_association_properties_attributes(span, properties: dict) -> None:

lmnr/traceloop_sdk/utils/package_check.py

@@ -1,8 +1,6 @@
-import pkg_resources
-
-installed_packages = {p.key for p in pkg_resources.working_set}
+from importlib.metadata import distributions
 
+installed_packages = {dist.metadata["Name"].lower() for dist in distributions()}
 
 def is_package_installed(package_name: str) -> bool:
-    # return importlib.util.find_spec(package_name) is not None
-    return package_name in installed_packages
+    return package_name.lower() in installed_packages

{lmnr-0.4.33.dist-info → lmnr-0.4.35.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lmnr
-Version: 0.4.33
+Version: 0.4.35
 Summary: Python SDK for Laminar AI
 License: Apache-2.0
 Author: lmnr.ai

{lmnr-0.4.33.dist-info → lmnr-0.4.35.dist-info}/RECORD

@@ -4,7 +4,7 @@ lmnr/sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lmnr/sdk/datasets.py,sha256=w8U9E6fvetAo65Cb2CbYzlfhY8CfXAR-VysrakG6-4I,1591
 lmnr/sdk/decorators.py,sha256=ZSDaEZyjo-RUzRCltsNbe6x0t9SKl2xRQ2q4uaKvXtk,2250
 lmnr/sdk/evaluations.py,sha256=5Vfyp0aIjuGpqfuM3cqsaaLpcoO7z6lcOOKxnyHCNHk,16264
-lmnr/sdk/laminar.py,sha256=2EMMeppaR8GuB29YGS3WqGssMQvneUxXeOV7w5Ahv4Q,23462
+lmnr/sdk/laminar.py,sha256=DkZQjAwkRzHCQ9V9XUF3fSD76zAaRQBNKQ48GdCscqQ,26382
 lmnr/sdk/log.py,sha256=cZBeUoSK39LMEV-X4-eEhTWOciULRfHaKfRK8YqIM8I,1532
 lmnr/sdk/types.py,sha256=qGD1tkGszd-_sZJaZ_Zx9U_CdUYzoDkUeN2g-o48Gls,5588
 lmnr/sdk/utils.py,sha256=Uk8y15x-sd5tP2ERONahElLDJVEy_3dA_1_5g9A6auY,3358
@@ -39,14 +39,14 @@ lmnr/traceloop_sdk/tracing/__init__.py,sha256=Ckq7zCM26VdJVB5tIZv0GTPyMZKyfso_KW
 lmnr/traceloop_sdk/tracing/attributes.py,sha256=h970zmb7yTszzf2oHBfOY3cDYhE6O7LhkiHLqa_7x1k,1261
 lmnr/traceloop_sdk/tracing/content_allow_list.py,sha256=3feztm6PBWNelc8pAZUcQyEGyeSpNiVKjOaDk65l2ps,846
 lmnr/traceloop_sdk/tracing/context_manager.py,sha256=csVlB6kDmbgSPsROHwnddvGGblx55v6lJMRj0wsSMQM,304
-lmnr/traceloop_sdk/tracing/tracing.py,sha256=wGFU51FHg2S2hxFOTYvXBQL6Z1oeqNpYQnREkh_J6MY,28742
+lmnr/traceloop_sdk/tracing/tracing.py,sha256=enZ6uRH9H9ZShPl4LK9B6ThzOZrBBK1Bap4l3NoqhQ8,29145
 lmnr/traceloop_sdk/utils/__init__.py,sha256=pNhf0G3vTd5ccoc03i1MXDbricSaiqCbi1DLWhSekK8,604
 lmnr/traceloop_sdk/utils/in_memory_span_exporter.py,sha256=H_4TRaThMO1H6vUQ0OpQvzJk_fZH0OOsRAM1iZQXsR8,2112
 lmnr/traceloop_sdk/utils/json_encoder.py,sha256=dK6b_axr70IYL7Vv-bu4wntvDDuyntoqsHaddqX7P58,463
-lmnr/traceloop_sdk/utils/package_check.py,sha256=TZSngzJOpFhfUZLXIs38cpMxQiZSmp0D-sCrIyhz7BA,251
+lmnr/traceloop_sdk/utils/package_check.py,sha256=Ki74WZME-ASF0fx7RXSIMsULGYUzI86sOINi1EGrc_Y,235
 lmnr/traceloop_sdk/version.py,sha256=OlatFEFA4ttqSSIiV8jdE-sq3KG5zu2hnC4B4mzWF3s,23
-lmnr-0.4.33.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
-lmnr-0.4.33.dist-info/METADATA,sha256=Ds9Fe8QmvPfSQsugBVGXr2GT4zJLUR4TzfXFVZB34qI,10322
-lmnr-0.4.33.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-lmnr-0.4.33.dist-info/entry_points.txt,sha256=K1jE20ww4jzHNZLnsfWBvU3YKDGBgbOiYG5Y7ivQcq4,37
-lmnr-0.4.33.dist-info/RECORD,,
+lmnr-0.4.35.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
+lmnr-0.4.35.dist-info/METADATA,sha256=WRv37EYH3Zoe1ccA9wxGx8Ote0A3mVSKArfi98yb7P4,10322
+lmnr-0.4.35.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+lmnr-0.4.35.dist-info/entry_points.txt,sha256=K1jE20ww4jzHNZLnsfWBvU3YKDGBgbOiYG5Y7ivQcq4,37
+lmnr-0.4.35.dist-info/RECORD,,