lmnr 0.3.2__tar.gz → 0.3.4__tar.gz

{lmnr-0.3.2 → lmnr-0.3.4}/PKG-INFO

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

{lmnr-0.3.2 → lmnr-0.3.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lmnr"
-version = "0.3.2"
+version = "0.3.4"
 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.3.2"
+version = "0.3.4"
 description = "Python SDK for Laminar AI"
 authors = ["lmnr.ai"]
 readme = "README.md"

{lmnr-0.3.2 → lmnr-0.3.4}/src/lmnr/sdk/context.py

@@ -269,9 +269,9 @@ class LaminarContextManager:
     ) -> Span:
         """Internal method to update a span object. Use `SpanContext.update()` instead."""
         span.update(
-            input=input,
+            input=input or span.input,
+            output=output or span.output,
             end_time=end_time,
-            output=output,
             metadata=metadata,
             attributes=attributes,
             evaluate_events=evaluate_events,
@@ -284,7 +284,7 @@ class LaminarContextManager:
     def event(
         self,
         name: str,
-        value: Optional[Union[str, int]] = None,
+        value: Optional[Union[str, int, float, bool]] = None,
         timestamp: Optional[datetime.datetime] = None,
     ):
         span = _lmnr_stack_context.get()[-1] if _lmnr_stack_context.get() else None

{lmnr-0.3.2 → lmnr-0.3.4}/src/lmnr/sdk/decorators.py

@@ -142,14 +142,14 @@ class LaminarDecorator:
     def event(
         self,
         name: str,
-        value: Optional[Union[str, int]] = None,
+        value: Optional[Union[str, int, float, bool]] = None,
         timestamp: Optional[datetime.datetime] = None,
     ):
         """Associate an event with the current span
 
         Args:
             name (str): name of the event. Must be predefined in the Laminar events page.
-            value (Optional[Union[str, int]], optional): value of the event. Must match range definition in Laminar events page. Defaults to None.
+            value (Optional[Union[str, int, float, bool]], optional): value of the event. Must match range definition in Laminar events page. Defaults to None.
             timestamp (Optional[datetime.datetime], optional): If you need custom timestamp. If not specified, current time is used. Defaults to None.
         """
         laminar = LaminarSingleton().get()
@@ -168,7 +168,7 @@ class LaminarDecorator:
         laminar = LaminarSingleton().get()
         laminar.evaluate_event(name, data)
 
-    def run_pipeline(
+    def run(
         self,
         pipeline: str,
         inputs: dict[str, NodeInput],

{lmnr-0.3.2 → lmnr-0.3.4}/src/lmnr/sdk/interface.py

@@ -156,14 +156,14 @@ class SpanContext(ObservationContext):
     def event(
         self,
         name: str,
-        value: Optional[Union[str, int]] = None,
+        value: Optional[Union[str, int, float, bool]] = None,
         timestamp: Optional[datetime.datetime] = None,
     ) -> "SpanContext":
         """Associate an event with the current span
 
         Args:
             name (str): name of the event. Must be predefined in the Laminar events page.
-            value (Optional[Union[str, int]], optional): value of the event. Must match range definition in Laminar events page. Defaults to None.
+            value (Optional[Union[str, int, float, bool]], optional): value of the event. Must match range definition in Laminar events page. Defaults to None.
             timestamp (Optional[datetime.datetime], optional): If you need custom timestamp. If not specified, current time is used. Defaults to None.
 
         Returns:

{lmnr-0.3.2 → lmnr-0.3.4}/src/lmnr/sdk/tracing_types.py

@@ -160,14 +160,14 @@ class Event(pydantic.BaseModel):
     templateName: str
     timestamp: datetime.datetime
     spanId: uuid.UUID
-    value: Optional[Union[int, str]] = None
+    value: Optional[Union[int, str, float, bool]] = None
 
     def __init__(
         self,
         name: str,
         span_id: uuid.UUID,
         timestamp: Optional[datetime.datetime] = None,
-        value: Optional[Union[int, str]] = None,
+        value: Optional[Union[int, str, float, bool]] = None,
     ):
         super().__init__(
             id=uuid.uuid4(),

{lmnr-0.3.2 → lmnr-0.3.4}/src/lmnr/sdk/types.py

@@ -3,6 +3,8 @@ import pydantic
 import uuid
 from typing import Optional, Union
 
+from .utils import to_dict
+
 
 class ChatMessage(pydantic.BaseModel):
     role: str
@@ -30,7 +32,7 @@ class PipelineRunRequest(pydantic.BaseModel):
     def to_dict(self):
         return {
             "inputs": {
-                k: v.model_dump() if isinstance(v, pydantic.BaseModel) else v
+                k: v.model_dump() if isinstance(v, pydantic.BaseModel) else to_dict(v)
                 for k, v in self.inputs.items()
             },
             "pipeline": self.pipeline,

{lmnr-0.3.2 → lmnr-0.3.4}/src/lmnr/sdk/utils.py

@@ -76,10 +76,7 @@ def to_dict(obj: typing.Any) -> dict[str, typing.Any]:
 
         return str(o)
 
-    for key in obj.keys():
-        obj[key] = to_dict_inner(obj[key])
-
-    return obj
+    return to_dict_inner(obj)
 
 
 def get_input_from_func_args(