lmnr 0.4.43__tar.gz → 0.4.45__tar.gz

{lmnr-0.4.43 → lmnr-0.4.45}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lmnr
-Version: 0.4.43
+Version: 0.4.45
 Summary: Python SDK for Laminar AI
 License: Apache-2.0
 Author: lmnr.ai
@@ -38,11 +38,9 @@ Provides-Extra: transformers
 Provides-Extra: vertexai
 Provides-Extra: watsonx
 Provides-Extra: weaviate
-Requires-Dist: aiohttp (>=3.0,<4.0)
-Requires-Dist: argparse (>=1.0,<2.0)
-Requires-Dist: backoff (>=2.0,<3.0)
-Requires-Dist: deprecated (>=1.0,<2.0)
-Requires-Dist: jinja2 (>=3.0,<4.0)
+Requires-Dist: aiohttp (>=3.0)
+Requires-Dist: argparse (>=1.0)
+Requires-Dist: deprecated (>=1.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)
@@ -77,11 +75,10 @@ Requires-Dist: opentelemetry-instrumentation-watsonx (>=0.33.12) ; extra == "all
 Requires-Dist: opentelemetry-instrumentation-weaviate (>=0.33.12) ; extra == "all" or extra == "weaviate"
 Requires-Dist: opentelemetry-sdk (>=1.28.0)
 Requires-Dist: opentelemetry-semantic-conventions-ai (==0.4.2)
-Requires-Dist: pydantic (>=2.7,<3.0)
-Requires-Dist: python-dotenv (>=1.0,<2.0)
-Requires-Dist: requests (>=2.0,<3.0)
-Requires-Dist: tenacity (>=8.0)
-Requires-Dist: tqdm (>=4.0,<5.0)
+Requires-Dist: pydantic (>=2.7)
+Requires-Dist: python-dotenv (>=1.0)
+Requires-Dist: requests (>=2.0)
+Requires-Dist: tqdm (>=4.0)
 Description-Content-Type: text/markdown
 
 # Laminar Python

{lmnr-0.4.43 → lmnr-0.4.45}/pyproject.toml

@@ -6,7 +6,7 @@
 
 [project]
 name = "lmnr"
-version = "0.4.43"
+version = "0.4.45"
 description = "Python SDK for Laminar AI"
 authors = [
   { name = "lmnr.ai", email = "founders@lmnr.ai" }
@@ -17,7 +17,7 @@ license = "Apache-2.0"
 
 [tool.poetry]
 name = "lmnr"
-version = "0.4.43"
+version = "0.4.45"
 description = "Python SDK for Laminar AI"
 authors = ["lmnr.ai"]
 readme = "README.md"
@@ -25,10 +25,9 @@ license = "Apache-2.0"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<4"
-pydantic = "~=2.7"
-requests = "~=2.0"
-python-dotenv = "~=1.0"
-backoff = "~=2.0"
+pydantic = ">=2.7"
+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"
@@ -38,12 +37,10 @@ opentelemetry-instrumentation-sqlalchemy = "^0.49b0"
 opentelemetry-instrumentation-urllib3 = "^0.49b0"
 opentelemetry-instrumentation-threading = "^0.49b0"
 opentelemetry-semantic-conventions-ai = "0.4.2"
-tenacity = ">=8.0"
-jinja2 = "~=3.0"
-deprecated = "~=1.0"
-tqdm = "~=4.0"
-argparse = "~=1.0"
-aiohttp = "~=3.0"
+deprecated = ">=1.0"
+tqdm = ">=4.0"
+argparse = ">=1.0"
+aiohttp = ">=3.0"
 opentelemetry-instrumentation-alephalpha = {version = ">=0.33.12", optional = true}
 opentelemetry-instrumentation-anthropic = {version = ">=0.33.12", optional = true}
 opentelemetry-instrumentation-bedrock = {version = ">=0.33.12", optional = true}

{lmnr-0.4.43 → lmnr-0.4.45}/src/lmnr/__init__.py

@@ -7,6 +7,7 @@ from .sdk.types import (
     NodeInput,
     PipelineRunError,
     PipelineRunResponse,
+    TracingLevel,
 )
 from .sdk.decorators import observe
 from .openllmetry_sdk import Instruments

{lmnr-0.4.43 → lmnr-0.4.45}/src/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-0.4.43 → lmnr-0.4.45}/src/lmnr/openllmetry_sdk/tracing/attributes.py

@@ -12,6 +12,7 @@ ASSOCIATION_PROPERTIES = "lmnr.association.properties"
 SESSION_ID = "session_id"
 USER_ID = "user_id"
 TRACE_TYPE = "trace_type"
+TRACING_LEVEL = "tracing_level"
 
 
 # exposed to the user, configurable

{lmnr-0.4.43 → lmnr-0.4.45}/src/lmnr/openllmetry_sdk/tracing/tracing.py

@@ -10,6 +10,7 @@ from lmnr.openllmetry_sdk.tracing.attributes import (
     ASSOCIATION_PROPERTIES,
     SPAN_INSTRUMENTATION_SOURCE,
     SPAN_PATH,
+    TRACING_LEVEL,
 )
 from lmnr.openllmetry_sdk.tracing.content_allow_list import ContentAllowList
 from lmnr.openllmetry_sdk.utils import is_notebook
@@ -224,6 +225,9 @@ def remove_association_properties(properties: dict) -> None:
 
 def _set_association_properties_attributes(span, properties: dict) -> None:
     for key, value in properties.items():
+        if key == TRACING_LEVEL:
+            span.set_attribute(f"lmnr.internal.{TRACING_LEVEL}", value)
+            continue
         span.set_attribute(f"{ASSOCIATION_PROPERTIES}.{key}", value)
 
 

{lmnr-0.4.43 → lmnr-0.4.45}/src/lmnr/sdk/laminar.py

@@ -59,6 +59,7 @@ from .types import (
     SemanticSearchRequest,
     SemanticSearchResponse,
     TraceType,
+    TracingLevel,
 )
 
 
@@ -558,6 +559,39 @@ class Laminar:
         if output is not None and span != trace.INVALID_SPAN:
             span.set_attribute(SPAN_OUTPUT, json_dumps(output))
 
+    @classmethod
+    @contextmanager
+    def set_tracing_level(self, level: TracingLevel):
+        """Set the tracing level for the current span and the context
+        (i.e. any children spans created from the current span in the current
+        thread).
+
+        Tracing level can be one of:
+        - `TracingLevel.ALL`: Enable tracing for the current span and all
+            children spans.
+        - `TracingLevel.META_ONLY`: Enable tracing for the current span and all
+            children spans, but only record metadata, e.g. tokens, costs.
+        - `TracingLevel.OFF`: Disable recording any spans.
+
+        Example:
+        ```python
+        from lmnr import Laminar, TracingLevel
+
+        with Laminar.set_tracing_level(TracingLevel.META_ONLY):
+            openai_client.chat.completions.create()
+        ```
+        """
+        if level == TracingLevel.ALL:
+            yield
+        else:
+            level = "meta_only" if level == TracingLevel.META_ONLY else "off"
+            update_association_properties({"tracing_level": level})
+            yield
+            try:
+                remove_association_properties({"tracing_level": level})
+            except Exception:
+                pass
+
     @classmethod
     def set_span_attributes(
         cls,
@@ -643,20 +677,6 @@ class Laminar:
                 props.pop(k)
         set_association_properties(props)
 
-    @classmethod
-    def _set_trace_type(
-        cls,
-        trace_type: TraceType,
-    ):
-        """Set the trace_type for the current span and the context
-        Args:
-            trace_type (TraceType): Type of the trace
-        """
-        association_properties = {
-            TRACE_TYPE: trace_type.value,
-        }
-        update_association_properties(association_properties)
-
     @classmethod
     def clear_session(cls):
         """Clear the session and user id from  the context"""
@@ -732,6 +752,20 @@ class Laminar:
             "Content-Type": "application/json",
         }
 
+    @classmethod
+    def _set_trace_type(
+        cls,
+        trace_type: TraceType,
+    ):
+        """Set the trace_type for the current span and the context
+        Args:
+            trace_type (TraceType): Type of the trace
+        """
+        association_properties = {
+            TRACE_TYPE: trace_type.value,
+        }
+        update_association_properties(association_properties)
+
     @classmethod
     async def __run(
         cls,

{lmnr-0.4.43 → lmnr-0.4.45}/src/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]],
 ]
 
@@ -203,3 +203,9 @@ class TraceType(Enum):
 class GetDatapointsResponse(pydantic.BaseModel):
     items: list[Datapoint]
     totalCount: int
+
+
+class TracingLevel(Enum):
+    OFF = 0
+    META_ONLY = 1
+    ALL = 2