promptlayer 1.0.35__tar.gz → 1.0.37__tar.gz

{promptlayer-1.0.35 → promptlayer-1.0.37}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: promptlayer
-Version: 1.0.35
+Version: 1.0.37
 Summary: PromptLayer is a platform for prompt engineering and tracks your LLM requests.
 License: Apache-2.0
 Author: Magniv

{promptlayer-1.0.35 → promptlayer-1.0.37}/promptlayer/__init__.py

@@ -1,4 +1,4 @@
 from .promptlayer import AsyncPromptLayer, PromptLayer
 
-__version__ = "1.0.35"
+__version__ = "1.0.37"
 __all__ = ["PromptLayer", "AsyncPromptLayer", "__version__"]

{promptlayer-1.0.35 → promptlayer-1.0.37}/promptlayer/promptlayer.py

@@ -1,4 +1,5 @@
 import asyncio
+import json
 import os
 from typing import Any, Dict, List, Literal, Optional, Union
 
@@ -21,6 +22,27 @@ from promptlayer.utils import (
 )
 
 
+def is_workflow_results_dict(obj: Any) -> bool:
+    if not isinstance(obj, dict):
+        return False
+
+    required_keys = {
+        "status",
+        "value",
+        "error_message",
+        "raw_error_message",
+        "is_output_node",
+    }
+
+    for val in obj.values():
+        if not isinstance(val, dict):
+            return False
+        if not required_keys.issubset(val.keys()):
+            return False
+
+    return True
+
+
 class PromptLayer(PromptLayerMixin):
     def __init__(
         self,
@@ -225,11 +247,9 @@ class PromptLayer(PromptLayerMixin):
         input_variables: Optional[Dict[str, Any]] = None,
         metadata: Optional[Dict[str, str]] = None,
         workflow_label_name: Optional[str] = None,
-        workflow_version: Optional[
-            int
-        ] = None,  # This is the version number, not the version ID
+        workflow_version: Optional[int] = None,
         return_all_outputs: Optional[bool] = False,
-    ) -> Dict[str, Any]:
+    ) -> Union[Dict[str, Any], Any]:
         try:
             try:
                 # Check if we're inside a running event loop
@@ -239,8 +259,7 @@ class PromptLayer(PromptLayerMixin):
 
             if loop and loop.is_running():
                 nest_asyncio.apply()
-                # If there's an active event loop, use `await` directly
-                return asyncio.run(
+                results = asyncio.run(
                     arun_workflow_request(
                         workflow_name=workflow_name,
                         input_variables=input_variables or {},
@@ -252,8 +271,7 @@ class PromptLayer(PromptLayerMixin):
                     )
                 )
             else:
-                # If there's no active event loop, use `asyncio.run()`
-                return asyncio.run(
+                results = asyncio.run(
                     arun_workflow_request(
                         workflow_name=workflow_name,
                         input_variables=input_variables or {},
@@ -264,6 +282,24 @@ class PromptLayer(PromptLayerMixin):
                         return_all_outputs=return_all_outputs,
                     )
                 )
+
+            if not return_all_outputs:
+                if is_workflow_results_dict(results):
+                    output_nodes = [
+                        node_data
+                        for node_data in results.values()
+                        if node_data.get("is_output_node")
+                    ]
+
+                    if not output_nodes:
+                        raise Exception(json.dumps(results, indent=4))
+
+                    if not any(
+                        node.get("status") == "SUCCESS" for node in output_nodes
+                    ):
+                        raise Exception(json.dumps(results, indent=4))
+
+            return results
         except Exception as e:
             raise Exception(f"Error running workflow: {str(e)}")
 

{promptlayer-1.0.35 → promptlayer-1.0.37}/promptlayer/utils.py

@@ -50,7 +50,7 @@ async def arun_workflow_request(
     api_key: str,
     return_all_outputs: Optional[bool] = False,
     timeout: Optional[int] = 120,
-) -> Dict[str, Any]:
+):
     payload = {
         "input_variables": input_variables,
         "metadata": metadata,
@@ -112,32 +112,33 @@ async def arun_workflow_request(
     # Subscribe to the channel named after the execution ID
     channel = ably_client.channels.get(channel_name)
 
-    final_output = {}
+    results = None
     message_received_event = asyncio.Event()
 
     async def message_listener(message: Message):
-        if message.name == "set_workflow_node_output":
-            data = json.loads(message.data)
-            if data.get("status") == "workflow_complete":
-                final_output.update(data.get("final_output", {}))
-                message_received_event.set()
+        nonlocal results
+
+        if message.name == "SET_WORKFLOW_COMPLETE":
+            message_data = json.loads(message.data)
+            results = message_data["final_output"]
+            message_received_event.set()
 
     # Subscribe to the channel
-    await channel.subscribe("set_workflow_node_output", message_listener)
+    await channel.subscribe("SET_WORKFLOW_COMPLETE", message_listener)
 
     # Wait for the message or timeout
     try:
         await asyncio.wait_for(message_received_event.wait(), timeout)
     except asyncio.TimeoutError:
-        channel.unsubscribe("set_workflow_node_output", message_listener)
+        channel.unsubscribe("SET_WORKFLOW_COMPLETE", message_listener)
         await ably_client.close()
         raise Exception("Workflow execution did not complete properly")
 
     # Unsubscribe from the channel and close the client
-    channel.unsubscribe("set_workflow_node_output", message_listener)
+    channel.unsubscribe("SET_WORKFLOW_COMPLETE", message_listener)
     await ably_client.close()
 
-    return final_output
+    return results
 
 
 def promptlayer_api_handler(

{promptlayer-1.0.35 → promptlayer-1.0.37}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "promptlayer"
-version = "1.0.35"
+version = "1.0.37"
 description = "PromptLayer is a platform for prompt engineering and tracks your LLM requests."
 authors = ["Magniv <hello@magniv.io>"]
 license = "Apache-2.0"
@@ -21,9 +21,9 @@ langchain = "^0.0.260"
 behave = "^1.2.6"
 pytest = "^8.2.0"
 pytest-asyncio = "^0.23.6"
-openai = "^1.26.0"
 anyio = "^4.3.0"
 anthropic = "^0.25.8"
+openai = "^1.60.1"
 
 [build-system]
 requires = ["poetry-core"]