promptlayer 1.0.24__py3-none-any.whl → 1.0.25__py3-none-any.whl

promptlayer/__init__.py

@@ -1,4 +1,4 @@
 from .promptlayer import PromptLayer
 
-__version__ = "1.0.24"
+__version__ = "1.0.25"
 __all__ = ["PromptLayer", "__version__"]

promptlayer/promptlayer.py

@@ -24,7 +24,7 @@ from promptlayer.utils import (
     openai_request,
     openai_stream_chat,
     openai_stream_completion,
-    run_workflow_request,
+    run_workflow_async,
     stream_response,
     track_request,
     util_log_request,
@@ -380,15 +380,17 @@ class PromptLayer:
         workflow_version: Optional[
             int
         ] = None,  # This is the version number, not the version ID
+        return_all_outputs: Optional[bool] = False,
     ) -> Dict[str, Any]:
         try:
-            result = run_workflow_request(
+            result = run_workflow_async(
                 workflow_name=workflow_name,
                 input_variables=input_variables or {},
                 metadata=metadata,
                 workflow_label_name=workflow_label_name,
                 workflow_version_number=workflow_version,
                 api_key=self.api_key,
+                return_all_outputs=return_all_outputs,
             )
             return result
         except Exception as e:

promptlayer/utils.py

@@ -10,7 +10,9 @@ from copy import deepcopy
 from enum import Enum
 from typing import Any, Callable, Dict, Generator, List, Optional, Union
 
+import aiohttp
 import requests
+from ably import AblyRealtime
 from opentelemetry import context, trace
 
 from promptlayer.types import RequestLog
@@ -27,7 +29,7 @@ URL_API_PROMPTLAYER = os.environ.setdefault(
 )
 
 
-def run_workflow_request(
+def run_workflow_async(
     *,
     workflow_name: str,
     input_variables: Dict[str, Any],
@@ -35,37 +37,103 @@ def run_workflow_request(
     workflow_label_name: Optional[str] = None,
     workflow_version_number: Optional[int] = None,
     api_key: str,
+    return_all_outputs: Optional[bool] = False,
+    timeout: Optional[int] = 120,
+) -> Dict[str, Any]:
+    return asyncio.run(
+        run_workflow_request(
+            workflow_name=workflow_name,
+            input_variables=input_variables,
+            metadata=metadata,
+            workflow_label_name=workflow_label_name,
+            workflow_version_number=workflow_version_number,
+            api_key=api_key,
+            return_all_outputs=return_all_outputs,
+            timeout=timeout,
+        )
+    )
+
+
+async def run_workflow_request(
+    *,
+    workflow_name: str,
+    input_variables: Dict[str, Any],
+    metadata: Optional[Dict[str, str]] = None,
+    workflow_label_name: Optional[str] = None,
+    workflow_version_number: Optional[int] = None,
+    api_key: str,
+    return_all_outputs: Optional[bool] = None,
+    timeout: Optional[int] = 120,
 ) -> Dict[str, Any]:
     payload = {
         "input_variables": input_variables,
         "metadata": metadata,
         "workflow_label_name": workflow_label_name,
         "workflow_version_number": workflow_version_number,
+        "return_all_outputs": return_all_outputs,
     }
 
     url = f"{URL_API_PROMPTLAYER}/workflows/{workflow_name}/run"
     headers = {"X-API-KEY": api_key}
 
     try:
-        response = requests.post(url, json=payload, headers=headers)
-    except requests.exceptions.RequestException as e:
+        async with aiohttp.ClientSession() as session:
+            async with session.post(url, json=payload, headers=headers) as response:
+                if response.status != 201:
+                    error_message = f"Failed to run workflow: {response.status} {await response.text()}"
+                    print(error_message)
+                    raise Exception(error_message)
+                result = await response.json()
+    except Exception as e:
         error_message = f"Failed to run workflow: {e}"
-        print(error_message, file=sys.stderr)
+        print(error_message)
         raise Exception(error_message)
 
-    if response.status_code != 201:
-        try:
-            error_details = response.json().get("error", "Unknown error")
-        except ValueError:
-            error_details = response.text or "Unknown error"
+    execution_id = result.get("workflow_version_execution_id")
+    if not execution_id:
+        raise Exception("No execution ID returned from workflow run")
 
-        error_message = f"Failed to run workflow: {error_details}"
-        print(error_message, file=sys.stderr)
-        raise Exception(error_message)
+    channel_name = f"workflow_updates:{execution_id}"
+
+    ws_response = requests.post(
+        f"{URL_API_PROMPTLAYER}/ws-token-request-library",
+        headers=headers,
+        params={"capability": channel_name},
+    )
+    token_details = ws_response.json()["token_details"]
+
+    # Initialize Ably client
+    ably_client = AblyRealtime(token=token_details["token"])
+
+    # Subscribe to the channel named after the execution ID
+    channel = ably_client.channels.get(channel_name)
+
+    final_output = {}
+    message_received_event = asyncio.Event()
+
+    async def message_listener(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()
+
+    # Subscribe to the channel
+    await channel.subscribe("set_workflow_node_output", 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)
+        await ably_client.close()
+        raise Exception("Workflow execution did not complete properly")
 
-    result = response.json()
+    # Unsubscribe from the channel
+    channel.unsubscribe("set_workflow_node_output", message_listener)
+    await ably_client.close()
 
-    return result
+    return final_output
 
 
 def promptlayer_api_handler(

{promptlayer-1.0.24.dist-info → promptlayer-1.0.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: promptlayer
-Version: 1.0.24
+Version: 1.0.25
 Summary: PromptLayer is a platform for prompt engineering and tracks your LLM requests.
 License: Apache-2.0
 Author: Magniv
@@ -13,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: ably (>=2.0.6,<3.0.0)
 Requires-Dist: opentelemetry-api (>=1.26.0,<2.0.0)
 Requires-Dist: opentelemetry-sdk (>=1.26.0,<2.0.0)
 Requires-Dist: requests (>=2.31.0,<3.0.0)

{promptlayer-1.0.24.dist-info → promptlayer-1.0.25.dist-info}/RECORD

@@ -1,7 +1,7 @@
-promptlayer/__init__.py,sha256=JTVK-6FTroG1GYTAY06uIChUCDWGll51745efZYKqlw,102
+promptlayer/__init__.py,sha256=RBD1elQNn3QMEckznmpU0Sw1FOhFNU6EgPXEM5yc5oE,102
 promptlayer/groups/__init__.py,sha256=-xs-2cn0nc0D_5YxZr3nC86iTdRVZmBhEpOKDJXE-sQ,224
 promptlayer/groups/groups.py,sha256=yeO6T0TM3qB0ondZRiHhcH8G06YygrpFoM8b9RmoIao,165
-promptlayer/promptlayer.py,sha256=Qyju-mDJcf0NiGjV6SQHsyf1syXHekvKhzRzExOn31k,16719
+promptlayer/promptlayer.py,sha256=Zl4nGbCzukfadiLWCPjGyXTN5Zmrjzu_u2jHn9nK6NY,16822
 promptlayer/promptlayer_base.py,sha256=sev-EZehRXJSZSmJtMkqmAUK1345pqbDY_lNjPP5MYA,7158
 promptlayer/span_exporter.py,sha256=zIJNsb3Fe6yb5wKLDmkoPF2wqFjk1p39E0jWHD2plzI,2658
 promptlayer/templates.py,sha256=aY_-BCrL0AgIdYEUE28pi0AP_avTVAgwv5hgzrh75vo,717
@@ -10,8 +10,8 @@ promptlayer/track/track.py,sha256=XNEZT9yNiRBPp9vaDZo_f0dP_ldOu8q1qafpVfS5Ze8,16
 promptlayer/types/__init__.py,sha256=xJcvQuOk91ZBBePb40-1FDNDKYrZoH5lPE2q6_UhprM,111
 promptlayer/types/prompt_template.py,sha256=TUXLXvuvew0EBLfTMBa2LhFeQoF7R-tcFKg7_UUtHMQ,4433
 promptlayer/types/request_log.py,sha256=xU6bcxQar6GaBOJlgZTavXUV3FjE8sF_nSjPu4Ya_00,174
-promptlayer/utils.py,sha256=Oyk_n0hIbYaCFK98QaUzHQeEdGxfQL3HRfXY-AlbPm0,33662
-promptlayer-1.0.24.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-promptlayer-1.0.24.dist-info/METADATA,sha256=iGmVrzJEanaYQsBLYY7t2hF4ly4jCAHQnJJD01QHXdc,4660
-promptlayer-1.0.24.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-promptlayer-1.0.24.dist-info/RECORD,,
+promptlayer/utils.py,sha256=yvlZDT9bunypNnIE4Xa8f2O5219F347HLKO3TaVmLyM,36135
+promptlayer-1.0.25.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+promptlayer-1.0.25.dist-info/METADATA,sha256=1IJZrzSuS_440xIzR2Kavo7QPFVbL9-rpK9-ieft4OI,4697
+promptlayer-1.0.25.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+promptlayer-1.0.25.dist-info/RECORD,,