lmnr 0.4.43__py3-none-any.whl → 0.4.44__py3-none-any.whl

lmnr/openllmetry_sdk/decorators/base.py

@@ -3,11 +3,13 @@ from functools import wraps
 import logging
 import os
 import pydantic
+import traceback
 import types
 from typing import Any, Optional
 
 from opentelemetry import trace
 from opentelemetry import context as context_api
+from opentelemetry.trace import Span
 
 from lmnr.sdk.utils import get_input_from_func_args, is_method
 from lmnr.openllmetry_sdk.tracing import get_tracer
@@ -69,7 +71,12 @@ def entity_method(
                 except TypeError:
                     pass
 
-                res = fn(*args, **kwargs)
+                try:
+                    res = fn(*args, **kwargs)
+                except Exception as e:
+                    _process_exception(span, e)
+                    span.end()
+                    raise e
 
                 # span will be ended in the generator
                 if isinstance(res, types.GeneratorType):
@@ -131,7 +138,12 @@ def aentity_method(
                 except TypeError:
                     pass
 
-                res = await fn(*args, **kwargs)
+                try:
+                    res = await fn(*args, **kwargs)
+                except Exception as e:
+                    _process_exception(span, e)
+                    span.end()
+                    raise e
 
                 # span will be ended in the generator
                 if isinstance(res, types.AsyncGeneratorType):
@@ -177,3 +189,17 @@ def _should_send_prompts():
     return (
         os.getenv("TRACELOOP_TRACE_CONTENT") or "true"
     ).lower() == "true" or context_api.get_value("override_enable_content_tracing")
+
+
+def _process_exception(span: Span, e: Exception):
+    exception_path = [type(e).__module__] if type(e).__module__ != "builtins" else []
+    exception_path.append(type(e).__qualname__)
+    span.add_event(
+        "exception",
+        {
+            "exception.message": str(e),
+            "exception.type": ".".join(exception_path),
+            "exception.stacktrace": traceback.format_exc(),
+            "exception.escaped": True,
+        },
+    )

lmnr/sdk/types.py

@@ -124,7 +124,7 @@ ExecutorFunctionReturnType = Any
 EvaluatorFunctionReturnType = Union[Numeric, dict[str, Numeric]]
 
 ExecutorFunction = Callable[
-    [EvaluationDatapointData, Any, dict[str, Any]],
+    [EvaluationDatapointData, Any],
     Union[ExecutorFunctionReturnType, Awaitable[ExecutorFunctionReturnType]],
 ]
 
@@ -133,7 +133,7 @@ ExecutorFunction = Callable[
 # record of string keys and number values. The latter is useful for evaluating
 # multiple criteria in one go instead of running multiple evaluators.
 EvaluatorFunction = Callable[
-    [ExecutorFunctionReturnType, Any, dict[str, Any]],
+    [ExecutorFunctionReturnType, Any],
     Union[EvaluatorFunctionReturnType, Awaitable[EvaluatorFunctionReturnType]],
 ]
 

{lmnr-0.4.43.dist-info → lmnr-0.4.44.dist-info}/METADATA

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

{lmnr-0.4.43.dist-info → lmnr-0.4.44.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=UpOFkiEWEXbfDhJ-wZNzD4GJ9IyIsb13w-mgyUJYPeU,5567
+lmnr/openllmetry_sdk/decorators/base.py,sha256=xwLsUFd5husf3p-ZllVTYEe9gkwl9T3Bf1WpW1-Ua5I,6413
 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=h970zmb7yTszzf2oHBfOY3cDYhE6O7LhkiHLqa_7x1k,1261
@@ -24,10 +24,10 @@ lmnr/sdk/eval_control.py,sha256=G6Fg3Xx_KWv72iBaWlNMdyRTF2bZFQnwJ68sJNSpIcY,177
 lmnr/sdk/evaluations.py,sha256=gLImD_uB9uXgw07QiJ_OYRTFDGxiPtFCO1c8HyOq2s0,15935
 lmnr/sdk/laminar.py,sha256=wCQ-GyJkH58kbazfYfYzZj-oSh-TOZycRIyf2f01v6w,30024
 lmnr/sdk/log.py,sha256=cZBeUoSK39LMEV-X4-eEhTWOciULRfHaKfRK8YqIM8I,1532
-lmnr/sdk/types.py,sha256=fxRslW0IptnAuWCeFyF_jEZUToyi79-du4kyOcqeSfQ,6352
+lmnr/sdk/types.py,sha256=hqaZ8sg97PIrZzmCyEc2uH_g4GKRPIIz4c_LuGlNyHM,6320
 lmnr/sdk/utils.py,sha256=Uk8y15x-sd5tP2ERONahElLDJVEy_3dA_1_5g9A6auY,3358
-lmnr-0.4.43.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
-lmnr-0.4.43.dist-info/METADATA,sha256=_zZTrZtCVj4kbBphWiNvTC0YSRR1AEA3bX_ewbdSnv0,12350
-lmnr-0.4.43.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-lmnr-0.4.43.dist-info/entry_points.txt,sha256=K1jE20ww4jzHNZLnsfWBvU3YKDGBgbOiYG5Y7ivQcq4,37
-lmnr-0.4.43.dist-info/RECORD,,
+lmnr-0.4.44.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
+lmnr-0.4.44.dist-info/METADATA,sha256=gj3Eu8hQmEXB3azu404wIGU6VUFetHDIemd5f3S2EEE,12350
+lmnr-0.4.44.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+lmnr-0.4.44.dist-info/entry_points.txt,sha256=K1jE20ww4jzHNZLnsfWBvU3YKDGBgbOiYG5Y7ivQcq4,37
+lmnr-0.4.44.dist-info/RECORD,,