prompty 0.1.7__py3-none-any.whl → 0.1.8__py3-none-any.whl

prompty/tracer.py

@@ -89,58 +89,92 @@ class Trace:
             return str(obj)
 
 
-def trace(func: Callable = None, *, description: str = None) -> Callable:
-    if func is None:
-        return partial(trace, description=description)
+def _name(func: Callable, args):
+    if hasattr(func, "__qualname__"):
+        signature = f"{func.__module__}.{func.__qualname__}"
+    else:
+        signature = f"{func.__module__}.{func.__name__}"
+
+    # core invoker gets special treatment
+    core_invoker = signature == "prompty.core.Invoker.__call__"
+    if core_invoker:
+        name = type(args[0]).__name__
+        signature = f"{args[0].__module__}.{args[0].__class__.__name__}.invoke"
+    else:
+        name = func.__name__
+
+    return name, signature
+
+
+def _inputs(func: Callable, args, kwargs) -> dict:
+    ba = inspect.signature(func).bind(*args, **kwargs)
+    ba.apply_defaults()
+
+    inputs = {k: Trace.to_dict(v) for k, v in ba.arguments.items() if k != "self"}
+
+    return inputs
 
+def _results(result: Any) -> dict:
+    return {
+        "result": Trace.to_dict(result) if result is not None else "None",
+    }
+
+def _trace_sync(func: Callable = None, *, description: str = None) -> Callable:
     description = description or ""
 
     @wraps(func)
     def wrapper(*args, **kwargs):
-        if hasattr(func, "__qualname__"):
-            signature = f"{func.__module__}.{func.__qualname__}"
-        else:
-            signature = f"{func.__module__}.{func.__name__}"
+        name, signature = _name(func, args)
+        Trace.start(name)
+        Trace.add("signature", signature)
+        if description and description != "":
+            Trace.add("description", description)
 
-        # core invoker gets special treatment
-        core_invoker = signature == "prompty.core.Invoker.__call__"
-        if core_invoker:
-            name = type(args[0]).__name__
-        else:
-            name = func.__name__
+        inputs = _inputs(func, args, kwargs)
+        Trace.add("inputs", inputs)
 
-        Trace.start(name)
+        result = func(*args, **kwargs)
+        Trace.add("result", _results(result))
 
-        if core_invoker:
-            Trace.add(
-                "signature",
-                f"{args[0].__module__}.{args[0].__class__.__name__}.invoke",
-            )
-        else:
-            Trace.add("signature", signature)
+        Trace.end()
 
-        if len(description) > 0:
-            Trace.add("description", description)
+        return result
+    
+    return wrapper
 
-        ba = inspect.signature(func).bind(*args, **kwargs)
-        ba.apply_defaults()
+def _trace_async(func: Callable = None, *, description: str = None) -> Callable:
+    description = description or ""
 
-        inputs = {k: Trace.to_dict(v) for k, v in ba.arguments.items() if k != "self"}
+    @wraps(func)
+    async def wrapper(*args, **kwargs):
+        name, signature = _name(func, args)
+        Trace.start(name)
+        Trace.add("signature", signature)
+        if description and description != "":
+            Trace.add("description", description)
 
-        Trace.add("input", Trace.to_dict(inputs))
-        result = func(*args, **kwargs)
+        inputs = _inputs(func, args, kwargs)
+        Trace.add("inputs", inputs)
 
-        Trace.add(
-            "result",
-            Trace.to_dict(result) if result is not None else "None",
-        )
+        result = await func(*args, **kwargs)
+        Trace.add("result", _results(result))
 
         Trace.end()
 
         return result
-
+    
     return wrapper
 
+def trace(func: Callable = None, *, description: str = None) -> Callable:
+    if func is None:
+        return partial(trace, description=description)
+    
+    wrapped_method = (
+        _trace_async if inspect.iscoroutinefunction(func) else _trace_sync
+    )
+
+    return wrapped_method(func, description=description)
+
 
 class PromptyTracer(Tracer):
     _stack: List[Dict[str, Any]] = []

{prompty-0.1.7.dist-info → prompty-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompty
-Version: 0.1.7
+Version: 0.1.8
 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.7.dist-info → prompty-0.1.8.dist-info}/RECORD

@@ -1,6 +1,6 @@
-prompty-0.1.7.dist-info/METADATA,sha256=WnpHqfmMVpAyXSIjChwWCNo5lTB19gLTiYVEm0xCvCU,4665
-prompty-0.1.7.dist-info/WHEEL,sha256=rSwsxJWe3vzyR5HCwjWXQruDgschpei4h_giTm0dJVE,90
-prompty-0.1.7.dist-info/licenses/LICENSE,sha256=KWSC4z9cfML_t0xThoQYjzTdcZQj86Y_mhXdatzU-KM,1052
+prompty-0.1.8.dist-info/METADATA,sha256=1sVPpxf3pjHAhCIJoXa-v02zF6P5w4aCdBcQZV3kEm4,4665
+prompty-0.1.8.dist-info/WHEEL,sha256=rSwsxJWe3vzyR5HCwjWXQruDgschpei4h_giTm0dJVE,90
+prompty-0.1.8.dist-info/licenses/LICENSE,sha256=KWSC4z9cfML_t0xThoQYjzTdcZQj86Y_mhXdatzU-KM,1052
 prompty/__init__.py,sha256=Msp8eiKdrDq0wyl6G5DFDH8r5BxM2_E60uzzL7_MJ5w,11183
 prompty/cli.py,sha256=_bx_l5v7OGhtAn4d_73b8tyfEw7OOkjCqGMQPu0YP5A,2489
 prompty/core.py,sha256=WYSvognjMUl08FT0_mkcqZfymb_guKcp3sK8_RO4Kq0,13528
@@ -8,5 +8,5 @@ prompty/executors.py,sha256=TankDTAEBTZkvnPfNUw2KNb1TnNuWhyY8TkWOogUXKs,3185
 prompty/parsers.py,sha256=4mmIn4SVNs8B0R1BufanqUJk8v4r0OEEo8yx6UOxQpA,4670
 prompty/processors.py,sha256=GmReygLx2XW1UuanlX71HG3rTZL86y0yAGyNdbGWkcg,2366
 prompty/renderers.py,sha256=RSHFQFx7AtKLUfsMLCXR0a56Mb7DL1NJNgjUqgg3IqU,776
-prompty/tracer.py,sha256=ac7b0M94RjeTqmMAypTlqq5kSnfxge4Oo8sJFEqVp9c,5497
-prompty-0.1.7.dist-info/RECORD,,
+prompty/tracer.py,sha256=XMS4aJD_Tp76wm2UFB8amtXn7ioGmPBUy11LmklSUFQ,6490
+prompty-0.1.8.dist-info/RECORD,,