lmnr 0.3.3__tar.gz → 0.3.4__tar.gz

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

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

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

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

{lmnr-0.3.3 → 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,

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

@@ -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.3 → 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.3 → 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(