lmnr 0.3.3__py3-none-any.whl → 0.3.4__py3-none-any.whl

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/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/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/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(

{lmnr-0.3.3.dist-info → lmnr-0.3.4.dist-info}/METADATA

@@ -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.dist-info → lmnr-0.3.4.dist-info}/RECORD

@@ -3,8 +3,8 @@ lmnr/sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lmnr/sdk/client.py,sha256=e6cIvJq38a6XAU8FGWYtNXVAPlEoZEyKS7hC3M_6EkU,5749
 lmnr/sdk/collector.py,sha256=6LRmPhOcmGplUDWm_sJh0dVrLTHknd_kmq7POGuAvoQ,5338
 lmnr/sdk/constants.py,sha256=USCfwuUqRx6_0xC8WUxqGj766dInqQkWJcf8U5vPK7s,34
-lmnr/sdk/context.py,sha256=y7qepVK0n99RV6XplxDcw6C6svlGyzsBO1HMzTXUl5Q,15369
-lmnr/sdk/decorators.py,sha256=YT93XQZtmngiTVANn5ZDF-2pwDw8AWW60otLgrjnmN0,12037
+lmnr/sdk/context.py,sha256=4ngtUYhix91hUW4Kpo-WplDiNs-SCqz6LmHMl84StoA,15398
+lmnr/sdk/decorators.py,sha256=0UxIcCwlcL8WWzWGjWjbo5pl-HVSL9JkiLwvzrAiGGs,12028
 lmnr/sdk/interface.py,sha256=ugtNIcHZhHC_qSyu_-1Fm62_phOb4vW-Lo5VZ1WaMhc,12255
 lmnr/sdk/providers/__init__.py,sha256=wNCgQnt9-bnTNXLQWdPgyKhqA1ajiaEd1Rr2KPOpazM,54
 lmnr/sdk/providers/base.py,sha256=xc6iA8yY_VK6tbzswt-3naZ53aAXtOLur9j8eimC_ZA,1054
@@ -12,12 +12,12 @@ lmnr/sdk/providers/fallback.py,sha256=9-srLJgDK5CWD8DIdnxo6jbSsnwDvdHC-vA06BfEkq
 lmnr/sdk/providers/openai.py,sha256=9X2VWz4_EqQBEA1UEww2EKdqods6IzOEmaO6dnhY-Lw,5725
 lmnr/sdk/providers/utils.py,sha256=ROt82VrvezExYOxionAynD3dp6oX5JoPW6F1ayTm7q8,946
 lmnr/sdk/tracing_types.py,sha256=O6sfTBRKxz8WtRy4uEr-_wkSqu58mmEa1SXXJXDGwBI,6267
-lmnr/sdk/types.py,sha256=hVxOsa3oCQQ-8aS_WkOtErg4nHJRkBVySfYlTgDlDyk,2084
-lmnr/sdk/utils.py,sha256=1yhXtdGmVXfnc8SOQiTH_zAZGbZrzO8oaFd7q5nE7eY,3349
+lmnr/sdk/types.py,sha256=5-Ft-l35wtmn2xxE8BTqsM3nx1zD799tRv4qiOkED50,2121
+lmnr/sdk/utils.py,sha256=50S-vT2ewiFsdx8-xADnqIM5EAJibw188cX-vo6eJvA,3293
 lmnr/semantic_conventions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lmnr/semantic_conventions/gen_ai_spans.py,sha256=3s-2J5v3t5LcMKwK2DefPn56XpxN5oMEYtb9Mf9D_gA,1541
-lmnr-0.3.3.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
-lmnr-0.3.3.dist-info/METADATA,sha256=D8iU-ZCGZoxRY6GHx0PhHBhBXLZSR6s6y7JflHwilLo,9663
-lmnr-0.3.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-lmnr-0.3.3.dist-info/entry_points.txt,sha256=Qg7ZRax4k-rcQsZ26XRYQ8YFSBiyY2PNxYfq4a6PYXI,41
-lmnr-0.3.3.dist-info/RECORD,,
+lmnr-0.3.4.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
+lmnr-0.3.4.dist-info/METADATA,sha256=AB8bHJTiyL9XKw3d3jSqOheQDZWekoxH7o7vg5cWHGk,9663
+lmnr-0.3.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+lmnr-0.3.4.dist-info/entry_points.txt,sha256=Qg7ZRax4k-rcQsZ26XRYQ8YFSBiyY2PNxYfq4a6PYXI,41
+lmnr-0.3.4.dist-info/RECORD,,