lmnr 0.4.14__tar.gz → 0.4.15__tar.gz

{lmnr-0.4.14 → lmnr-0.4.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lmnr
-Version: 0.4.14
+Version: 0.4.15
 Summary: Python SDK for Laminar AI
 License: Apache-2.0
 Author: lmnr.ai
@@ -49,11 +49,10 @@ Requires-Dist: opentelemetry-instrumentation-watsonx (>=0.30.0,<0.31.0)
 Requires-Dist: opentelemetry-instrumentation-weaviate (>=0.30.0,<0.31.0)
 Requires-Dist: opentelemetry-sdk (>=1.27.0,<2.0.0)
 Requires-Dist: opentelemetry-semantic-conventions-ai (==0.4.1)
-Requires-Dist: posthog (>=3.0,<4.0)
 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,<9.0)
+Requires-Dist: tenacity (>=8.0)
 Requires-Dist: tqdm (>=4.0,<5.0)
 Description-Content-Type: text/markdown
 

{lmnr-0.4.14 → lmnr-0.4.15}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lmnr"
-version = "0.4.14"
+version = "0.4.15"
 description = "Python SDK for Laminar AI"
 authors = [
   { name = "lmnr.ai", email = "founders@lmnr.ai" }
@@ -11,7 +11,7 @@ license = "Apache-2.0"
 
 [tool.poetry]
 name = "lmnr"
-version = "0.4.14"
+version = "0.4.15"
 description = "Python SDK for Laminar AI"
 authors = ["lmnr.ai"]
 readme = "README.md"
@@ -33,10 +33,9 @@ opentelemetry-instrumentation-sqlalchemy = "^0.48b0"
 opentelemetry-instrumentation-urllib3 = "^0.48b0"
 opentelemetry-instrumentation-threading = "^0.48b0"
 opentelemetry-semantic-conventions-ai = "0.4.1"
-tenacity = "~=8.0"
+tenacity = ">=8.0"
 jinja2 = "~=3.0"
 deprecated = "~=1.0"
-posthog = "~=3.0"
 opentelemetry-instrumentation-mistralai = "^0.30.0"
 opentelemetry-instrumentation-openai = "^0.30.0"
 opentelemetry-instrumentation-ollama = "^0.30.0"

{lmnr-0.4.14 → lmnr-0.4.15}/src/lmnr/sdk/decorators.py

@@ -4,7 +4,8 @@ from lmnr.traceloop_sdk.decorators.base import (
 )
 from opentelemetry.trace import INVALID_SPAN, get_current_span
 
-from typing import Callable, Optional, cast
+from typing import Callable, Optional, TypeVar, cast
+from typing_extensions import ParamSpec
 
 from lmnr.traceloop_sdk.tracing.attributes import SESSION_ID, USER_ID
 from lmnr.traceloop_sdk.tracing.tracing import update_association_properties
@@ -12,12 +13,16 @@ from lmnr.traceloop_sdk.tracing.tracing import update_association_properties
 from .utils import is_async
 
 
+P = ParamSpec("P")
+R = TypeVar("R")
+
+
 def observe(
     *,
     name: Optional[str] = None,
     user_id: Optional[str] = None,
     session_id: Optional[str] = None,
-) -> Callable[[Callable], Callable]:
+) -> Callable[[Callable[P, R]], Callable[P, R]]:
     """The main decorator entrypoint for Laminar. This is used to wrap
     functions and methods to create spans.
 
@@ -43,13 +48,9 @@ def observe(
         current_span = get_current_span()
         if current_span != INVALID_SPAN:
             if session_id is not None:
-                current_span.set_attribute(
-                    SESSION_ID, session_id
-                )
+                current_span.set_attribute(SESSION_ID, session_id)
             if user_id is not None:
-                current_span.set_attribute(
-                    USER_ID, user_id
-                )
+                current_span.set_attribute(USER_ID, user_id)
         association_properties = {}
         if session_id is not None:
             association_properties["session_id"] = session_id