lmnr 0.4.46__py3-none-any.whl → 0.4.48__py3-none-any.whl

lmnr/openllmetry_sdk/decorators/base.py

@@ -191,4 +191,5 @@ def _should_send_prompts():
 
 
 def _process_exception(span: Span, e: Exception):
-    span.record_exception(e)
+    # Note that this `escaped` is sent as a StringValue("True"), not a boolean.
+    span.record_exception(e, escaped=True)

lmnr/sdk/laminar.py

@@ -12,6 +12,7 @@ from lmnr.openllmetry_sdk.tracing.attributes import (
 from lmnr.openllmetry_sdk.decorators.base import json_dumps
 from opentelemetry import context, trace
 from opentelemetry.context import attach, detach, set_value
+from opentelemetry.sdk.trace import SpanProcessor
 from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
 from opentelemetry.util.types import AttributeValue
 
@@ -79,6 +80,7 @@ class Laminar:
         http_port: Optional[int] = None,
         grpc_port: Optional[int] = None,
         instruments: Optional[Set[Instruments]] = None,
+        _processor: Optional[SpanProcessor] = None,
     ):
         """Initialize Laminar context across the application.
         This method must be called before using any other Laminar methods or
@@ -133,7 +135,16 @@ class Laminar:
         cls.__env = env
         cls.__initialized = True
         cls._initialize_logger()
+
+        if _processor is not None:
+            cls.__logger.warning(
+                "Using a custom span processor. This feature is added for tests only. "
+                "Any use of this feature outside of tests is not supported and "
+                "advised against."
+            )
+
         Traceloop.init(
+            processor=_processor,
             exporter=OTLPSpanExporter(
                 endpoint=cls.__base_grpc_url,
                 headers={"authorization": f"Bearer {cls.__project_api_key}"},
@@ -268,13 +279,9 @@ class Laminar:
         value: Optional[AttributeValue] = None,
         timestamp: Optional[Union[datetime.datetime, int]] = None,
     ):
-        """Associate an event with the current span. If event with such
-        name never existed, Laminar will create a new event and infer its type
-        from the value. If the event already exists, Laminar will append the
-        value to the event if and only if the value is of a matching type.
-        Otherwise, the event won't be recorded.
-        Supported types are string, numeric, and boolean. If the value
-        is `None`, event is considered a boolean tag with the value of `True`.
+        """Associate an event with the current span. If using manual\
+        instrumentation, use raw OpenTelemetry `span.add_event()` instead.\
+       `value` will be saved as a `lmnr.event.value` attribute.
 
         Args:
             name (str): event name
@@ -672,9 +679,9 @@ class Laminar:
     def clear_metadata(cls):
         """Clear the metadata from the context"""
         props: dict = copy.copy(context.get_value("association_properties"))
-        for k in props.keys():
-            if k.startswith("metadata."):
-                props.pop(k)
+        metadata_keys = [k for k in props.keys() if k.startswith("metadata.")]
+        for k in metadata_keys:
+            props.pop(k)
         set_association_properties(props)
 
     @classmethod

{lmnr-0.4.46.dist-info → lmnr-0.4.48.dist-info}/METADATA

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

{lmnr-0.4.46.dist-info → lmnr-0.4.48.dist-info}/RECORD

@@ -5,7 +5,7 @@ lmnr/openllmetry_sdk/.python-version,sha256=9OLQBQVbD4zE4cJsPePhnAfV_snrPSoqEQw-
 lmnr/openllmetry_sdk/__init__.py,sha256=vVSGTAwUnJvdulHtslkGAd8QCBuv78WUK3bgfBpH6Do,2390
 lmnr/openllmetry_sdk/config/__init__.py,sha256=DliMGp2NjYAqRFLKpWQPUKjGMHRO8QsVfazBA1qENQ8,248
 lmnr/openllmetry_sdk/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lmnr/openllmetry_sdk/decorators/base.py,sha256=f8kh7TG3Q99RGXLMrLNlq12s1FWRBftWjzFBpc813k4,6027
+lmnr/openllmetry_sdk/decorators/base.py,sha256=rZC5BgLULll0fZFRZ_Oe1XN-BhctKCpTo4TDsKvSfL4,6121
 lmnr/openllmetry_sdk/instruments.py,sha256=CGGUEELldrXkQwAzAkDeAtDq07_pjhz7i14a92P7C_E,1036
 lmnr/openllmetry_sdk/tracing/__init__.py,sha256=xT73L1t2si2CM6QmMiTZ7zn-dKKYBLNrpBBWq6WfVBw,68
 lmnr/openllmetry_sdk/tracing/attributes.py,sha256=B_4KVYWAUu-6DQmsm2eCJQcTxm8pG1EByCBK3uOPkuI,1293
@@ -22,12 +22,12 @@ lmnr/sdk/datasets.py,sha256=KNMp_v3z1ocIltIw7kTgj8o-l9R8N8Tgj0sw1ajQ9C8,1582
 lmnr/sdk/decorators.py,sha256=ja2EUWUWvFOp28ER0k78PRuxNahwCVyH0TdM3U-xY7U,1856
 lmnr/sdk/eval_control.py,sha256=G6Fg3Xx_KWv72iBaWlNMdyRTF2bZFQnwJ68sJNSpIcY,177
 lmnr/sdk/evaluations.py,sha256=gLImD_uB9uXgw07QiJ_OYRTFDGxiPtFCO1c8HyOq2s0,15935
-lmnr/sdk/laminar.py,sha256=4Saelm9m6pB9GWquCdHWY-1VhAB8Q2tWlq6hnmorzHU,31250
+lmnr/sdk/laminar.py,sha256=pgvkCswo9csOIWD3DWr6EW2UqMbUWc-AomVVOGD2U2E,31418
 lmnr/sdk/log.py,sha256=cZBeUoSK39LMEV-X4-eEhTWOciULRfHaKfRK8YqIM8I,1532
 lmnr/sdk/types.py,sha256=FCNoFoa0ingOvpXGfbiETVsakYyq9Zpoc56MXJ1YDzQ,6390
 lmnr/sdk/utils.py,sha256=Uk8y15x-sd5tP2ERONahElLDJVEy_3dA_1_5g9A6auY,3358
-lmnr-0.4.46.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
-lmnr-0.4.46.dist-info/METADATA,sha256=cL1X2RUUQ9jI6m8iybxhBOE1LsoKVxXUYYHp_WmzqNA,12244
-lmnr-0.4.46.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-lmnr-0.4.46.dist-info/entry_points.txt,sha256=K1jE20ww4jzHNZLnsfWBvU3YKDGBgbOiYG5Y7ivQcq4,37
-lmnr-0.4.46.dist-info/RECORD,,
+lmnr-0.4.48.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
+lmnr-0.4.48.dist-info/METADATA,sha256=KCZg8NS72mThs3tAJ4SL9OwDSTcvsosx5JxzpDTtVTw,12244
+lmnr-0.4.48.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+lmnr-0.4.48.dist-info/entry_points.txt,sha256=K1jE20ww4jzHNZLnsfWBvU3YKDGBgbOiYG5Y7ivQcq4,37
+lmnr-0.4.48.dist-info/RECORD,,