prompty 0.1.23__py2.py3-none-any.whl → 0.1.24__py2.py3-none-any.whl

prompty/tracer.py

@@ -1,6 +1,7 @@
 import os
 import json
 import inspect
+import numbers
 import traceback
 import importlib
 import contextlib
@@ -139,7 +140,11 @@ def _trace_sync(
                     {
                         "exception": {
                             "type": type(e),
-                            "traceback": traceback.format_tb(tb=e.__traceback__) if e.__traceback__ else None,
+                            "traceback": (
+                                traceback.format_tb(tb=e.__traceback__)
+                                if e.__traceback__
+                                else None
+                            ),
                             "message": str(e),
                             "args": to_dict(e.args),
                         }
@@ -179,7 +184,11 @@ def _trace_async(
                     {
                         "exception": {
                             "type": type(e),
-                            "traceback": traceback.format_tb(tb=e.__traceback__) if e.__traceback__ else None,
+                            "traceback": (
+                                traceback.format_tb(tb=e.__traceback__)
+                                if e.__traceback__
+                                else None
+                            ),
                             "message": str(e),
                             "args": to_dict(e.args),
                         }
@@ -250,11 +259,10 @@ class PromptyTracer:
             # hoist usage to parent frame
             if "result" in frame and isinstance(frame["result"], dict):
                 if "usage" in frame["result"]:
-                    if "__usage" in frame:
-                        for key, value in frame["result"]["usage"].items():
-                            frame["__usage"][key] += value
-                    else:
-                        frame["__usage"] = frame["result"]["usage"]
+                    frame["__usage"] = self.hoist_item(
+                        frame["result"]["usage"],
+                        frame["__usage"] if "__usage" in frame else {},
+                    )
 
             # streamed results may have usage as well
             if "result" in frame and isinstance(frame["result"], list):
@@ -264,48 +272,59 @@ class PromptyTracer:
                         and "usage" in result
                         and isinstance(result["usage"], dict)
                     ):
-                        if "__usage" not in frame:
-                            frame["__usage"] = {}
-                        for key, value in result["usage"].items():
-                            if key not in frame["__usage"]:
-                                frame["__usage"][key] = value
-                            else:
-                                frame["__usage"][key] += value
+                        frame["__usage"] = self.hoist_item(
+                            result["usage"],
+                            frame["__usage"] if "__usage" in frame else {},
+                        )
 
             # add any usage frames from below
             if "__frames" in frame:
                 for child in frame["__frames"]:
                     if "__usage" in child:
-                        if "__usage" not in frame:
-                            frame["__usage"] = {}
-                        for key, value in child["__usage"].items():
-                            if key not in frame["__usage"]:
-                                frame["__usage"][key] = value
-                            else:
-                                frame["__usage"][key] += value
+                        frame["__usage"] = self.hoist_item(
+                            child["__usage"],
+                            frame["__usage"] if "__usage" in frame else {},
+                        )
 
             # if stack is empty, dump the frame
             if len(self.stack) == 0:
-                trace_file = (
-                    self.output
-                    / f"{frame['name']}.{datetime.now().strftime('%Y%m%d.%H%M%S')}.tracy"
-                )
-
-                v = importlib.metadata.version("prompty")
-                enriched_frame = {
-                    "runtime": "python",
-                    "version": v,
-                    "trace": frame,
-                }
-
-                with open(trace_file, "w") as f:
-                    json.dump(enriched_frame, f, indent=4)
+                self.write_trace(frame)
             # otherwise, append the frame to the parent
             else:
                 if "__frames" not in self.stack[-1]:
                     self.stack[-1]["__frames"] = []
                 self.stack[-1]["__frames"].append(frame)
 
+    def hoist_item(self, src: Dict[str, Any], cur: Dict[str, Any]) -> None:
+        for key, value in src.items():
+            if value is None or isinstance(value, list) or isinstance(value, dict):
+                continue
+            try:
+                if key not in cur:
+                    cur[key] = value
+                else:
+                    cur[key] += value
+            except:
+                continue
+
+        return cur
+
+    def write_trace(self, frame: Dict[str, Any]) -> None:
+        trace_file = (
+            self.output
+            / f"{frame['name']}.{datetime.now().strftime('%Y%m%d.%H%M%S')}.tracy"
+        )
+
+        v = importlib.metadata.version("prompty")
+        enriched_frame = {
+            "runtime": "python",
+            "version": v,
+            "trace": frame,
+        }
+
+        with open(trace_file, "w") as f:
+            json.dump(enriched_frame, f, indent=4)
+
 
 @contextlib.contextmanager
 def console_tracer(name: str) -> Iterator[Callable[[str, Any], None]]:

{prompty-0.1.23.dist-info → prompty-0.1.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompty
-Version: 0.1.23
+Version: 0.1.24
 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>
 Requires-Dist: pyyaml>=6.0.1
@@ -22,7 +22,7 @@ Prompty is an asset class and format for LLM prompts designed to enhance observa
 
 The file format has a supporting toolchain with a VS Code extension and runtimes in multiple programming languages to simplify and accelerate your AI application development.
 
-The tooling comes together in three ways: the *prompty file asset*, the *VS Code extension tool*, and *runtimes* in multiple programming languges.
+The tooling comes together in three ways: the *prompty file asset*, the *VS Code extension tool*, and *runtimes* in multiple programming languages.
 
 ## The Prompty File Format
 Prompty is a language agnostic prompt asset for creating prompts and engineering the responses. Learn more about the format [here](https://prompty.ai/docs/prompty-file-spec).
@@ -95,7 +95,7 @@ The Prompty runtime comes with a set of built-in invokers that can be used to ex
 
 
 ## Using Tracing in Prompty
-Prompty supports tracing to help you understand the execution of your prompts. This functionality is customizeable and can be used to trace the execution of your prompts in a way that makes sense to you. Prompty has two default traces built in: `console_tracer` and `PromptyTracer`. The `console_tracer` writes the trace to the console, and the `PromptyTracer` writes the trace to a JSON file. You can also create your own tracer by creating your own hook.
+Prompty supports tracing to help you understand the execution of your prompts. This functionality is customizable and can be used to trace the execution of your prompts in a way that makes sense to you. Prompty has two default traces built in: `console_tracer` and `PromptyTracer`. The `console_tracer` writes the trace to the console, and the `PromptyTracer` writes the trace to a JSON file. You can also create your own tracer by creating your own hook.
 
 ```python
 import prompty
@@ -117,7 +117,7 @@ print(response)
 ```
 
 You can also bring your own tracer by your own tracing hook. The `console_tracer` is the simplest example of a tracer. It writes the trace to the console.
-This is what it loks like:
+This is what it looks like:
 
 ```python
 @contextlib.contextmanager
@@ -212,4 +212,4 @@ prompty -s path/to/prompty/file -e .env
 This will execute the prompt and print the response to the console. If there are any environment variables the CLI should take into account, you can pass those in via the `-e` flag. It also has default tracing enabled.
 
 ## Contributing
-We welcome contributions to the Prompty project! This community led project is open to all contributors. The project cvan be found on [GitHub](https://github.com/Microsoft/prompty).
+We welcome contributions to the Prompty project! This community led project is open to all contributors. The project can be found on [GitHub](https://github.com/Microsoft/prompty).

{prompty-0.1.23.dist-info → prompty-0.1.24.dist-info}/RECORD

@@ -1,7 +1,7 @@
-prompty-0.1.23.dist-info/METADATA,sha256=jxIN-iLH2JzTWPDha6eHMX61oKf-Do0raKMpC4NAXrQ,9037
-prompty-0.1.23.dist-info/WHEEL,sha256=CuZGaXTwoRLAOVv0AcE3bCTxO5ejVuBEJkUBe9C-kvk,94
-prompty-0.1.23.dist-info/entry_points.txt,sha256=9y1lKPWUpPWRJzUslcVH-gMwbNoa2PzjyoZsKYLQqyw,45
-prompty-0.1.23.dist-info/licenses/LICENSE,sha256=KWSC4z9cfML_t0xThoQYjzTdcZQj86Y_mhXdatzU-KM,1052
+prompty-0.1.24.dist-info/METADATA,sha256=Za-iNjmfbvfVGUVDesnmA55OTFhsRMEO86UU0Ry_jMw,9037
+prompty-0.1.24.dist-info/WHEEL,sha256=CuZGaXTwoRLAOVv0AcE3bCTxO5ejVuBEJkUBe9C-kvk,94
+prompty-0.1.24.dist-info/entry_points.txt,sha256=9y1lKPWUpPWRJzUslcVH-gMwbNoa2PzjyoZsKYLQqyw,45
+prompty-0.1.24.dist-info/licenses/LICENSE,sha256=KWSC4z9cfML_t0xThoQYjzTdcZQj86Y_mhXdatzU-KM,1052
 prompty/__init__.py,sha256=XTUgJ3xT7HYJieuWW5PBItey0BWneg3G7iBBjIeNJZU,11628
 prompty/azure/__init__.py,sha256=ptGajCh68s_tugPv45Y4GJCyBToNFCExUzUh9yIBIfo,292
 prompty/azure/executor.py,sha256=4BiVgDcoDaJ2QcSd3O1LIm_3Fh-ln6wjqO2ZoVYjslc,4700
@@ -16,5 +16,5 @@ prompty/renderers.py,sha256=RSHFQFx7AtKLUfsMLCXR0a56Mb7DL1NJNgjUqgg3IqU,776
 prompty/serverless/__init__.py,sha256=NPqoFATEMQ96G8OQkVcGxUWU4llIQCwxfJePPo8YFY8,279
 prompty/serverless/executor.py,sha256=jbTnYE2aq8oS5PWcZ96NhhjL-gU1a_tlUoB449QqHaY,4648
 prompty/serverless/processor.py,sha256=pft1XGbPzo0MzQMbAt1VxsLsvRrjQO3B8MXEE2PfSA0,1982
-prompty/tracer.py,sha256=WNI7V9ySGrOywY_StGt3swM7BuT-XtaPTSVSs4DteSs,10709
-prompty-0.1.23.dist-info/RECORD,,
+prompty/tracer.py,sha256=qYivi1z-xkXiIikM45uG8yFlEoXwaxoP2Xog_7yjz5Y,10990
+prompty-0.1.24.dist-info/RECORD,,