prompty 0.1.36__tar.gz → 0.1.38__tar.gz

{prompty-0.1.36 → prompty-0.1.38}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompty
-Version: 0.1.36
+Version: 0.1.38
 Summary: Prompty is a new asset class and format for LLM prompts that aims to provide observability, understandability, and portability for developers. It includes spec, tooling, and a runtime. This Prompty runtime supports Python
 Author-Email: Seth Juarez <seth.juarez@microsoft.com>
 License: MIT

{prompty-0.1.36 → prompty-0.1.38}/prompty/tracer.py

@@ -28,6 +28,10 @@ def sanitize(key: str, value: Any) -> Any:
 class Tracer:
     _tracers: Dict[str, Callable[[str], Iterator[Callable[[str, Any], None]]]] = {}
 
+    SIGNATURE = "signature"
+    INPUTS = "inputs"
+    RESULT = "result"
+
     @classmethod
     def add(
         cls, name: str, tracer: Callable[[str], Iterator[Callable[[str, Any], None]]]
@@ -40,11 +44,17 @@ class Tracer:
 
     @classmethod
     @contextlib.contextmanager
-    def start(cls, name: str) -> Iterator[Callable[[str, Any], None]]:
+    def start(cls, name: str, attributes: Dict[str, Any] = None) -> Iterator[Callable[[str, Any], None]]:
         with contextlib.ExitStack() as stack:
             traces = [
                 stack.enter_context(tracer(name)) for tracer in cls._tracers.values()
             ]
+
+            if attributes:
+                for trace in traces:
+                    for key, value in attributes.items():
+                        trace(key, value)
+
             yield lambda key, value: [
                 # normalize and sanitize any trace values
                 trace(key, sanitize(key, to_dict(value)))
@@ -124,7 +134,17 @@ def _trace_sync(
     @wraps(func)
     def wrapper(*args, **kwargs):
         name, signature = _name(func, args)
+        altname: str = None
+        # special case
+        if "name" in okwargs:
+            altname = name
+            name = okwargs["name"]
+            del okwargs["name"]
+
         with Tracer.start(name) as trace:
+            if altname != None:
+                trace("function", altname)
+
             trace("signature", signature)
 
             # support arbitrary keyword
@@ -168,10 +188,20 @@ def _trace_async(
     @wraps(func)
     async def wrapper(*args, **kwargs):
         name, signature = _name(func, args)
+        altname: str = None
+        # special case
+        if "name" in okwargs:
+            altname = name
+            name = okwargs["name"]
+            del okwargs["name"]
+
         with Tracer.start(name) as trace:
+            if altname != None:
+                trace("function", altname)
+                
             trace("signature", signature)
 
-            # support arbitrary keyword 
+            # support arbitrary keyword
             # arguments for trace decorator
             for k, v in okwargs.items():
                 trace(k, to_dict(v))

{prompty-0.1.36 → prompty-0.1.38}/pyproject.toml

@@ -15,7 +15,7 @@ dependencies = [
     "click>=8.1.7",
     "aiofiles>=24.1.0",
 ]
-version = "0.1.36"
+version = "0.1.38"
 
 [project.license]
 text = "MIT"

{prompty-0.1.36 → prompty-0.1.38}/tests/test_tracing.py

@@ -296,3 +296,30 @@ async def test_streaming_async():
     if isinstance(result, AsyncIterator):
         async for item in result:
             print(item)
+
+@trace
+def test_tracing_attributes():
+    with Tracer.start("Test1", {Tracer.SIGNATURE: "test1", "two": 2}) as trace:
+        trace(Tracer.INPUTS, 3)
+        trace(Tracer.INPUTS, 4)
+        with Tracer.start("Test2", {"signature": "5", "six": 6}) as trace:
+            trace("inputs", 7)
+            trace(Tracer.RESULT, 8)
+            with Tracer.start("Test3", {"signature": "9", "ten": 10}) as trace:
+                trace("inputs", 11)
+                trace(Tracer.RESULT, 12)
+
+
+@trace(name="OTHER")
+def test_named_tracer():
+    with Tracer.start("Test1", {Tracer.SIGNATURE: "test1", "two": 2}) as trace:
+        trace(Tracer.INPUTS, 3)
+        trace(Tracer.RESULT, 4)
+
+
+@pytest.mark.asyncio
+@trace(name="OTHER")
+async def test_named_tracer_async():
+    with Tracer.start("Test1", {Tracer.SIGNATURE: "test1", "two": 2}) as trace:
+        trace(Tracer.INPUTS, 3)
+        trace(Tracer.RESULT, 4)