langwatch 0.3.1__py3-none-any.whl → 0.4.1__py3-none-any.whl

langwatch/__version__.py

@@ -1,3 +1,3 @@
 """Version information for LangWatch."""
 
-__version__ = "0.3.1"
+__version__ = "0.4.1"

langwatch/prompts/decorators/prompt_tracing.py

@@ -1,6 +1,7 @@
 from functools import wraps
 from opentelemetry import trace
-from typing import TYPE_CHECKING, TypeVar, Callable, Any
+from opentelemetry.trace import Span
+from typing import TYPE_CHECKING, TypeVar, Callable, Any, Optional, Union, Dict
 import json
 
 from langwatch.attributes import AttributeKey
@@ -11,9 +12,37 @@ if TYPE_CHECKING:
 T = TypeVar("T", bound="Prompt")
 
 
+def _set_attribute_if_not_none(
+    span: Span, key: str, value: Optional[Union[str, int, float, bool]]
+) -> None:
+    """Set span attribute only if value is not None."""
+    if value is not None:
+        span.set_attribute(key, value)
+
+
 class PromptTracing:
     """Namespace for Prompt method tracing decorators"""
 
+    @staticmethod
+    def _set_prompt_attributes(span: Span, prompt: "Prompt") -> None:
+        """Set prompt-related attributes on the span."""
+        _set_attribute_if_not_none(
+            span, AttributeKey.LangWatchPromptId, getattr(prompt, "id", None)
+        )
+        _set_attribute_if_not_none(
+            span, AttributeKey.LangWatchPromptHandle, getattr(prompt, "handle", None)
+        )
+        _set_attribute_if_not_none(
+            span,
+            AttributeKey.LangWatchPromptVersionId,
+            getattr(prompt, "version_id", None),
+        )
+        _set_attribute_if_not_none(
+            span,
+            AttributeKey.LangWatchPromptVersionNumber,
+            getattr(prompt, "version", None),
+        )
+
     @staticmethod
     def _create_compile_decorator(
         span_name: str,
@@ -28,26 +57,15 @@ class PromptTracing:
                 with trace.get_tracer(__name__).start_as_current_span(
                     PromptTracing._create_span_name(span_name)
                 ) as span:
-                    # Set base prompt
-                    span.set_attributes(
-                        {
-                            AttributeKey.LangWatchPromptId: getattr(self, "id", None),
-                            AttributeKey.LangWatchPromptHandle: getattr(
-                                self, "handle", None
-                            ),
-                            AttributeKey.LangWatchPromptVersionId: getattr(
-                                self, "version_id", None
-                            ),
-                            AttributeKey.LangWatchPromptVersionNumber: getattr(
-                                self, "version", None
-                            ),
-                        }
-                    )
+                    # Set prompt attributes
+                    PromptTracing._set_prompt_attributes(span, self)
 
                     # Create variables dict from args and kwargs
-                    variables_dict: dict[str, Any] = {}
-                    if args and args[0] is not None:
-                        variables_dict.update(args[0])
+                    variables_dict: Dict[str, Any] = {}
+                    if args:
+                        first_arg = args[0]
+                        if first_arg is not None and hasattr(first_arg, "update"):
+                            variables_dict.update(first_arg)
                     variables_dict.update(kwargs)
 
                     span.set_attribute(
@@ -88,7 +106,7 @@ class PromptTracing:
     @staticmethod
     def _create_span_name(span_name: str) -> str:
         """Create a span name for the prompt"""
-        return "Prompt" + "." + span_name
+        return f"Prompt.{span_name}"
 
 
 prompt_tracing = PromptTracing()

{langwatch-0.3.1.dist-info → langwatch-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langwatch
-Version: 0.3.1
+Version: 0.4.1
 Summary: LangWatch Python SDK, for monitoring your LLMs
 Author-email: Langwatch Engineers <engineering@langwatch.ai>
 License: MIT
@@ -31,6 +31,7 @@ Requires-Dist: opentelemetry-sdk>=1.32.1
 Requires-Dist: pksuid>=1.1.2
 Requires-Dist: pydantic>=2
 Requires-Dist: python-liquid>=2.0.2
+Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: retry>=0.9.2
 Requires-Dist: termcolor>=3.0.1
 Provides-Extra: dev

{langwatch-0.3.1.dist-info → langwatch-0.4.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 langwatch/__init__.py,sha256=TzPHzqCFGZJByI3sAIKrNB33Qi4PqVmgYDZuBwPnhPc,4222
-langwatch/__version__.py,sha256=76Y19D8flgXvQE1gQJQMULzN9YyGslJSW_cUX8J6s6I,64
+langwatch/__version__.py,sha256=9l3owj3_pNV7TsTQT4fmHpK6UvxlyyebOVlDH3-Xcgw,64
 langwatch/attributes.py,sha256=nXdI_G85wQQCAdAcwjCiLYdEYj3wATmfgCmhlf6dVIk,3910
 langwatch/batch_evaluation.py,sha256=piez7TYqUZPb9NlIShTuTPmSzrZqX-vm2Grz_NGXe04,16078
 langwatch/client.py,sha256=WTNcYSik7kZ2kH-qGDnhbMTosc8e_Xhab_lZlfh5TC8,25559
@@ -390,7 +390,7 @@ langwatch/prompts/prompt_api_service.py,sha256=tHhwIRjUBSM43_jwDAoGCHJjvvqVeSCrU
 langwatch/prompts/prompt_facade.py,sha256=47matSK4G2Ce3HWUnO13-k7jlDuZQePGCck4gkbTmXM,5052
 langwatch/prompts/types.py,sha256=p1bRMvfCCpGGiVwzFtQijVtWl5GWugL_vBOFc4B2348,269
 langwatch/prompts/decorators/prompt_service_tracing.py,sha256=uSYw0vExo7AuxbcCRnxbYl6UOfOQSC0IsisSqYy153Y,2395
-langwatch/prompts/decorators/prompt_tracing.py,sha256=GD1iA7j7_BzqO7iwkRUt4-RQrursuRg1gvtu89SsiY8,3443
+langwatch/prompts/decorators/prompt_tracing.py,sha256=x_PQvJlGbGF1h2HtGNiqaZ8K1qNd1jRf5pTOBTQx-7M,3963
 langwatch/prompts/types/__init__.py,sha256=uEnjOQC4LkLMWQ0fXfKe573xKOvoMdPgC6uY-yo9B_g,506
 langwatch/prompts/types/prompt_data.py,sha256=g_EQ94-PGfa4Ptwd3e2rMqoIZiX052MEEZKyF77m9D0,3137
 langwatch/prompts/types/structures.py,sha256=cB94bn-qhFgHHYXcrmJV6Bk9idk5ZmyfXhFNQAaXw-M,951
@@ -407,6 +407,6 @@ langwatch/utils/initialization.py,sha256=1KoZmkHOvGEVF0j-4t4xRQdA_2C_SPiF7qFXqEG
 langwatch/utils/module.py,sha256=KLBNOK3mA9gCSifCcQX_lOtU48BJQDWvFKtF6NMvwVA,688
 langwatch/utils/transformation.py,sha256=76MGXyrYTxM0Yri36NJqLK-XxL4BBYdmKWAXXlw3D4Q,7690
 langwatch/utils/utils.py,sha256=ZCOSie4o9LdJ7odshNfCNjmgwgQ27ojc5ENqt1rXuSs,596
-langwatch-0.3.1.dist-info/METADATA,sha256=RhkQSUrkCjU-O4v2J-MBLavOEYaML6VtK1MmE1CJBPE,13123
-langwatch-0.3.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langwatch-0.3.1.dist-info/RECORD,,
+langwatch-0.4.1.dist-info/METADATA,sha256=OoFLwcDgnCsFrigOQCb3r6oPYa2db0pAjhjiOn3Aa8E,13152
+langwatch-0.4.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langwatch-0.4.1.dist-info/RECORD,,