promptlayer 1.0.37__tar.gz → 1.0.38__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: promptlayer
-Version: 1.0.37
+Version: 1.0.38
 Summary: PromptLayer is a platform for prompt engineering and tracks your LLM requests.
 License: Apache-2.0
 Author: Magniv
@@ -32,11 +32,11 @@ Description-Content-Type: text/markdown
 <a href="https://docs.promptlayer.com"><img alt="Docs" src="https://custom-icon-badges.herokuapp.com/badge/docs-PL-green.svg?logo=cake&style=for-the-badge"></a>
 <a href="https://www.loom.com/share/196c42e43acd4a369d75e9a7374a0850"><img alt="Demo with Loom" src="https://img.shields.io/badge/Demo-loom-552586.svg?logo=loom&style=for-the-badge&labelColor=gray"></a>
 
----  
+---
 
 <div align="left">
 
-[PromptLayer](https://promptlayer.com/) is the first platform that allows you to track, manage, and share your GPT prompt engineering. PromptLayer acts a middleware between your code and OpenAI’s python library. 
+[PromptLayer](https://promptlayer.com/) is the first platform that allows you to track, manage, and share your GPT prompt engineering. PromptLayer acts a middleware between your code and OpenAI’s python library.
 
 PromptLayer records all your OpenAI API requests, allowing you to search and explore request history in the PromptLayer dashboard.
 
@@ -77,14 +77,14 @@ openai = promptlayer.openai
 
 ### Adding PromptLayer tags: `pl_tags`
 
-PromptLayer allows you to add tags through the `pl_tags` argument. This allows you to track and group requests in the dashboard. 
+PromptLayer allows you to add tags through the `pl_tags` argument. This allows you to track and group requests in the dashboard.
 
 *Tags are not required but we recommend them!*
 
 ```python
 openai.Completion.create(
-  engine="text-ada-001", 
-  prompt="My name is", 
+  engine="text-ada-001",
+  prompt="My name is",
   pl_tags=["name-guessing", "pipeline-2"]
 )
 ```
@@ -93,11 +93,11 @@ After making your first few requests, you should be able to see them in the Prom
 
 ## Using the REST API
 
-This Python library is a wrapper over PromptLayer's REST API. If you use another language, like Javascript, just interact directly with the API. 
+This Python library is a wrapper over PromptLayer's REST API. If you use another language, like Javascript, just interact directly with the API.
 
 Here is an example request below:
 
-```jsx
+```python
 import requests
 request_response = requests.post(
   "https://api.promptlayer.com/track-request",

{promptlayer-1.0.37 → promptlayer-1.0.38}/README.md

@@ -8,11 +8,11 @@
 <a href="https://docs.promptlayer.com"><img alt="Docs" src="https://custom-icon-badges.herokuapp.com/badge/docs-PL-green.svg?logo=cake&style=for-the-badge"></a>
 <a href="https://www.loom.com/share/196c42e43acd4a369d75e9a7374a0850"><img alt="Demo with Loom" src="https://img.shields.io/badge/Demo-loom-552586.svg?logo=loom&style=for-the-badge&labelColor=gray"></a>
 
----  
+---
 
 <div align="left">
 
-[PromptLayer](https://promptlayer.com/) is the first platform that allows you to track, manage, and share your GPT prompt engineering. PromptLayer acts a middleware between your code and OpenAI’s python library. 
+[PromptLayer](https://promptlayer.com/) is the first platform that allows you to track, manage, and share your GPT prompt engineering. PromptLayer acts a middleware between your code and OpenAI’s python library.
 
 PromptLayer records all your OpenAI API requests, allowing you to search and explore request history in the PromptLayer dashboard.
 
@@ -53,14 +53,14 @@ openai = promptlayer.openai
 
 ### Adding PromptLayer tags: `pl_tags`
 
-PromptLayer allows you to add tags through the `pl_tags` argument. This allows you to track and group requests in the dashboard. 
+PromptLayer allows you to add tags through the `pl_tags` argument. This allows you to track and group requests in the dashboard.
 
 *Tags are not required but we recommend them!*
 
 ```python
 openai.Completion.create(
-  engine="text-ada-001", 
-  prompt="My name is", 
+  engine="text-ada-001",
+  prompt="My name is",
   pl_tags=["name-guessing", "pipeline-2"]
 )
 ```
@@ -69,11 +69,11 @@ After making your first few requests, you should be able to see them in the Prom
 
 ## Using the REST API
 
-This Python library is a wrapper over PromptLayer's REST API. If you use another language, like Javascript, just interact directly with the API. 
+This Python library is a wrapper over PromptLayer's REST API. If you use another language, like Javascript, just interact directly with the API.
 
 Here is an example request below:
 
-```jsx
+```python
 import requests
 request_response = requests.post(
   "https://api.promptlayer.com/track-request",

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

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

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

@@ -61,9 +61,7 @@ class PromptLayer(PromptLayerMixin):
         self.api_key = api_key
         self.templates = TemplateManager(api_key)
         self.group = GroupManager(api_key)
-        self.tracer_provider, self.tracer = self._initialize_tracer(
-            api_key, enable_tracing
-        )
+        self.tracer_provider, self.tracer = self._initialize_tracer(api_key, enable_tracing)
         self.track = TrackManager(api_key)
 
     def __getattr__(
@@ -233,9 +231,7 @@ class PromptLayer(PromptLayerMixin):
                 span.set_attribute("prompt_name", prompt_name)
                 span.set_attribute("function_input", str(_run_internal_kwargs))
                 pl_run_span_id = hex(span.context.span_id)[2:].zfill(16)
-                result = self._run_internal(
-                    **_run_internal_kwargs, pl_run_span_id=pl_run_span_id
-                )
+                result = self._run_internal(**_run_internal_kwargs, pl_run_span_id=pl_run_span_id)
                 span.set_attribute("function_output", str(result))
                 return result
         else:
@@ -285,18 +281,12 @@ class PromptLayer(PromptLayerMixin):
 
             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")
-                    ]
+                    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
-                    ):
+                    if not any(node.get("status") == "SUCCESS" for node in output_nodes):
                         raise Exception(json.dumps(results, indent=4))
 
             return results
@@ -364,14 +354,10 @@ class AsyncPromptLayer(PromptLayerMixin):
         self.api_key = api_key
         self.templates = AsyncTemplateManager(api_key)
         self.group = AsyncGroupManager(api_key)
-        self.tracer_provider, self.tracer = self._initialize_tracer(
-            api_key, enable_tracing
-        )
+        self.tracer_provider, self.tracer = self._initialize_tracer(api_key, enable_tracing)
         self.track = AsyncTrackManager(api_key)
 
-    def __getattr__(
-        self, name: Union[Literal["openai"], Literal["anthropic"], Literal["prompts"]]
-    ):
+    def __getattr__(self, name: Union[Literal["openai"], Literal["anthropic"], Literal["prompts"]]):
         if name == "openai":
             import openai as openai_module
 
@@ -400,9 +386,7 @@ class AsyncPromptLayer(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,  # This is the version number, not the version ID
         return_all_outputs: Optional[bool] = False,
     ) -> Dict[str, Any]:
         try:
@@ -448,9 +432,7 @@ class AsyncPromptLayer(PromptLayerMixin):
                 span.set_attribute("prompt_name", prompt_name)
                 span.set_attribute("function_input", str(_run_internal_kwargs))
                 pl_run_span_id = hex(span.context.span_id)[2:].zfill(16)
-                result = await self._run_internal(
-                    **_run_internal_kwargs, pl_run_span_id=pl_run_span_id
-                )
+                result = await self._run_internal(**_run_internal_kwargs, pl_run_span_id=pl_run_span_id)
                 span.set_attribute("function_output", str(result))
                 return result
         else:
@@ -563,9 +545,7 @@ class AsyncPromptLayer(PromptLayerMixin):
             input_variables=input_variables,
             metadata=metadata,
         )
-        prompt_blueprint = await self.templates.get(
-            prompt_name, get_prompt_template_params
-        )
+        prompt_blueprint = await self.templates.get(prompt_name, get_prompt_template_params)
         prompt_blueprint_model = self._validate_and_extract_model_from_prompt_blueprint(
             prompt_blueprint=prompt_blueprint, prompt_name=prompt_name
         )

{promptlayer-1.0.37 → promptlayer-1.0.38}/promptlayer/promptlayer_base.py

@@ -15,9 +15,7 @@ class PromptLayerBase(object):
         "_tracer",
     ]
 
-    def __init__(
-        self, obj, function_name="", provider_type="openai", api_key=None, tracer=None
-    ):
+    def __init__(self, obj, function_name="", provider_type="openai", api_key=None, tracer=None):
         object.__setattr__(self, "_obj", obj)
         object.__setattr__(self, "_function_name", function_name)
         object.__setattr__(self, "_provider_type", provider_type)
@@ -29,29 +27,22 @@ class PromptLayerBase(object):
 
         if (
             name != "count_tokens"  # fix for anthropic count_tokens
-            and not re.match(
-                r"<class 'anthropic\..*Error'>", str(attr)
-            )  # fix for anthropic errors
-            and not re.match(
-                r"<class 'openai\..*Error'>", str(attr)
-            )  # fix for openai errors
+            and not re.match(r"<class 'anthropic\..*Error'>", str(attr))  # fix for anthropic errors
+            and not re.match(r"<class 'openai\..*Error'>", str(attr))  # fix for openai errors
             and (
                 inspect.isclass(attr)
                 or inspect.isfunction(attr)
                 or inspect.ismethod(attr)
-                or str(type(attr))
-                == "<class 'anthropic.resources.completions.Completions'>"
-                or str(type(attr))
-                == "<class 'anthropic.resources.completions.AsyncCompletions'>"
+                or str(type(attr)) == "<class 'anthropic.resources.completions.Completions'>"
+                or str(type(attr)) == "<class 'anthropic.resources.completions.AsyncCompletions'>"
                 or str(type(attr)) == "<class 'anthropic.resources.messages.Messages'>"
-                or str(type(attr))
-                == "<class 'anthropic.resources.messages.AsyncMessages'>"
+                or str(type(attr)) == "<class 'anthropic.resources.messages.AsyncMessages'>"
                 or re.match(r"<class 'openai\.resources.*'>", str(type(attr)))
             )
         ):
             return PromptLayerBase(
                 attr,
-                function_name=f'{object.__getattribute__(self, "_function_name")}.{name}',
+                function_name=f"{object.__getattribute__(self, '_function_name')}.{name}",
                 provider_type=object.__getattribute__(self, "_provider_type"),
                 api_key=object.__getattribute__(self, "_api_key"),
                 tracer=object.__getattribute__(self, "_tracer"),
@@ -77,16 +68,10 @@ class PromptLayerBase(object):
 
         if tracer:
             with tracer.start_as_current_span(function_name) as llm_request_span:
-                llm_request_span_id = hex(llm_request_span.context.span_id)[2:].zfill(
-                    16
-                )
-                llm_request_span.set_attribute(
-                    "provider", object.__getattribute__(self, "_provider_type")
-                )
+                llm_request_span_id = hex(llm_request_span.context.span_id)[2:].zfill(16)
+                llm_request_span.set_attribute("provider", object.__getattribute__(self, "_provider_type"))
                 llm_request_span.set_attribute("function_name", function_name)
-                llm_request_span.set_attribute(
-                    "function_input", str({"args": args, "kwargs": kwargs})
-                )
+                llm_request_span.set_attribute("function_input", str({"args": args, "kwargs": kwargs}))
 
                 if inspect.isclass(function_object):
                     result = PromptLayerBase(
@@ -101,9 +86,7 @@ class PromptLayerBase(object):
 
                 function_response = function_object(*args, **kwargs)
 
-                if inspect.iscoroutinefunction(function_object) or inspect.iscoroutine(
-                    function_response
-                ):
+                if inspect.iscoroutinefunction(function_object) or inspect.iscoroutine(function_response):
                     return async_wrapper(
                         function_response,
                         return_pl_id,
@@ -146,9 +129,7 @@ class PromptLayerBase(object):
 
             function_response = function_object(*args, **kwargs)
 
-            if inspect.iscoroutinefunction(function_object) or inspect.iscoroutine(
-                function_response
-            ):
+            if inspect.iscoroutinefunction(function_object) or inspect.iscoroutine(function_response):
                 return async_wrapper(
                     function_response,
                     return_pl_id,

{promptlayer-1.0.37 → promptlayer-1.0.38}/promptlayer/promptlayer_mixins.py

@@ -138,9 +138,7 @@ class PromptLayerMixin:
     @staticmethod
     def _initialize_tracer(api_key: str = None, enable_tracing: bool = False):
         if enable_tracing:
-            resource = Resource(
-                attributes={ResourceAttributes.SERVICE_NAME: "prompt-layer-library"}
-            )
+            resource = Resource(attributes={ResourceAttributes.SERVICE_NAME: "prompt-layer-library"})
             tracer_provider = TracerProvider(resource=resource)
             promptlayer_exporter = PromptLayerSpanExporter(api_key=api_key)
             span_processor = BatchSpanProcessor(promptlayer_exporter)
@@ -151,9 +149,7 @@ class PromptLayerMixin:
             return None, None
 
     @staticmethod
-    def _prepare_get_prompt_template_params(
-        *, prompt_version, prompt_release_label, input_variables, metadata
-    ):
+    def _prepare_get_prompt_template_params(*, prompt_version, prompt_release_label, input_variables, metadata):
         params = {}
 
         if prompt_version:
@@ -206,27 +202,19 @@ class PromptLayerMixin:
         }
 
     @staticmethod
-    def _validate_and_extract_model_from_prompt_blueprint(
-        *, prompt_blueprint, prompt_name
-    ):
+    def _validate_and_extract_model_from_prompt_blueprint(*, prompt_blueprint, prompt_name):
         if not prompt_blueprint["llm_kwargs"]:
-            raise ValueError(
-                f"Prompt '{prompt_name}' does not have any LLM kwargs associated with it."
-            )
+            raise ValueError(f"Prompt '{prompt_name}' does not have any LLM kwargs associated with it.")
 
         prompt_blueprint_metadata = prompt_blueprint.get("metadata")
 
         if not prompt_blueprint_metadata:
-            raise ValueError(
-                f"Prompt '{prompt_name}' does not have any metadata associated with it."
-            )
+            raise ValueError(f"Prompt '{prompt_name}' does not have any metadata associated with it.")
 
         prompt_blueprint_model = prompt_blueprint_metadata.get("model")
 
         if not prompt_blueprint_model:
-            raise ValueError(
-                f"Prompt '{prompt_name}' does not have a model parameters associated with it."
-            )
+            raise ValueError(f"Prompt '{prompt_name}' does not have a model parameters associated with it.")
 
         return prompt_blueprint_model
 
@@ -247,12 +235,8 @@ class PromptLayerMixin:
             "args": [],
             "kwargs": request_params["kwargs"],
             "tags": tags,
-            "request_start_time": datetime.datetime.now(
-                datetime.timezone.utc
-            ).timestamp(),
-            "request_end_time": datetime.datetime.now(
-                datetime.timezone.utc
-            ).timestamp(),
+            "request_start_time": datetime.datetime.now(datetime.timezone.utc).timestamp(),
+            "request_end_time": datetime.datetime.now(datetime.timezone.utc).timestamp(),
             "api_key": api_key,
             "metadata": metadata,
             "prompt_id": request_params["prompt_blueprint"]["id"],
@@ -275,9 +259,7 @@ class PromptLayerMixin:
                             for key, value in attributes.items():
                                 span.set_attribute(key, value)
 
-                        span.set_attribute(
-                            "function_input", str({"args": args, "kwargs": kwargs})
-                        )
+                        span.set_attribute("function_input", str({"args": args, "kwargs": kwargs}))
                         result = func(*args, **kwargs)
                         span.set_attribute("function_output", str(result))
 
@@ -294,9 +276,7 @@ class PromptLayerMixin:
                             for key, value in attributes.items():
                                 span.set_attribute(key, value)
 
-                        span.set_attribute(
-                            "function_input", str({"args": args, "kwargs": kwargs})
-                        )
+                        span.set_attribute("function_input", str({"args": args, "kwargs": kwargs}))
                         result = await func(*args, **kwargs)
                         span.set_attribute("function_output", str(result))
 

{promptlayer-1.0.37 → promptlayer-1.0.38}/promptlayer/span_exporter.py

@@ -19,12 +19,8 @@ class PromptLayerSpanExporter(SpanExporter):
             span_info = {
                 "name": span.name,
                 "context": {
-                    "trace_id": hex(span.context.trace_id)[2:].zfill(
-                        32
-                    ),  # Ensure 32 characters
-                    "span_id": hex(span.context.span_id)[2:].zfill(
-                        16
-                    ),  # Ensure 16 characters
+                    "trace_id": hex(span.context.trace_id)[2:].zfill(32),  # Ensure 32 characters
+                    "span_id": hex(span.context.span_id)[2:].zfill(16),  # Ensure 16 characters
                     "trace_state": str(span.context.trace_state),
                 },
                 "kind": str(span.kind),
@@ -44,10 +40,7 @@ class PromptLayerSpanExporter(SpanExporter):
                     }
                     for event in span.events
                 ],
-                "links": [
-                    {"context": link.context, "attributes": dict(link.attributes)}
-                    for link in span.links
-                ],
+                "links": [{"context": link.context, "attributes": dict(link.attributes)} for link in span.links],
                 "resource": {
                     "attributes": dict(span.resource.attributes),
                     "schema_url": span.resource.schema_url,

{promptlayer-1.0.37 → promptlayer-1.0.38}/promptlayer/templates.py

@@ -28,9 +28,7 @@ class AsyncTemplateManager:
     def __init__(self, api_key: str):
         self.api_key = api_key
 
-    async def get(
-        self, prompt_name: str, params: Union[GetPromptTemplate, None] = None
-    ):
+    async def get(self, prompt_name: str, params: Union[GetPromptTemplate, None] = None):
         return await aget_prompt_template(prompt_name, params, self.api_key)
 
     async def all(self, page: int = 1, per_page: int = 30):

{promptlayer-1.0.37 → promptlayer-1.0.38}/promptlayer/track/__init__.py

@@ -1,7 +1,15 @@
-from promptlayer.track.track import agroup, ametadata, aprompt, ascore, group
-from promptlayer.track.track import metadata as metadata_
-from promptlayer.track.track import prompt
-from promptlayer.track.track import score as score_
+from promptlayer.track.track import (
+    agroup,
+    ametadata,
+    aprompt,
+    ascore,
+    group,
+    metadata as metadata_,
+    prompt,
+    score as score_,
+)
+
+# TODO(dmu) LOW: Move this code to another file
 
 
 class TrackManager:
@@ -14,9 +22,7 @@ class TrackManager:
     def metadata(self, request_id, metadata):
         return metadata_(request_id, metadata, self.api_key)
 
-    def prompt(
-        self, request_id, prompt_name, prompt_input_variables, version=None, label=None
-    ):
+    def prompt(self, request_id, prompt_name, prompt_input_variables, version=None, label=None):
         return prompt(
             request_id,
             prompt_name,
@@ -40,9 +46,7 @@ class AsyncTrackManager:
     async def metadata(self, request_id, metadata):
         return await ametadata(request_id, metadata, self.api_key)
 
-    async def prompt(
-        self, request_id, prompt_name, prompt_input_variables, version=None, label=None
-    ):
+    async def prompt(self, request_id, prompt_name, prompt_input_variables, version=None, label=None):
         return await aprompt(
             request_id,
             prompt_name,

{promptlayer-1.0.37 → promptlayer-1.0.38}/promptlayer/track/track.py

@@ -20,9 +20,7 @@ def prompt(
 ):
     if not isinstance(prompt_input_variables, dict):
         raise Exception("Please provide a dictionary of input variables.")
-    return promptlayer_track_prompt(
-        request_id, prompt_name, prompt_input_variables, api_key, version, label
-    )
+    return promptlayer_track_prompt(request_id, prompt_name, prompt_input_variables, api_key, version, label)
 
 
 def metadata(request_id, metadata, api_key: str = None):
@@ -30,9 +28,7 @@ def metadata(request_id, metadata, api_key: str = None):
         raise Exception("Please provide a dictionary of metadata.")
     for key, value in metadata.items():
         if not isinstance(key, str) or not isinstance(value, str):
-            raise Exception(
-                "Please provide a dictionary of metadata with key value pair of strings."
-            )
+            raise Exception("Please provide a dictionary of metadata with key value pair of strings.")
     return promptlayer_track_metadata(request_id, metadata, api_key)
 
 
@@ -60,9 +56,7 @@ async def aprompt(
 ):
     if not isinstance(prompt_input_variables, dict):
         raise Exception("Please provide a dictionary of input variables.")
-    return await apromptlayer_track_prompt(
-        request_id, prompt_name, prompt_input_variables, api_key, version, label
-    )
+    return await apromptlayer_track_prompt(request_id, prompt_name, prompt_input_variables, api_key, version, label)
 
 
 async def ametadata(request_id, metadata, api_key: str = None):
@@ -70,9 +64,7 @@ async def ametadata(request_id, metadata, api_key: str = None):
         raise Exception("Please provide a dictionary of metadata.")
     for key, value in metadata.items():
         if not isinstance(key, str) or not isinstance(value, str):
-            raise Exception(
-                "Please provide a dictionary of metadata with key-value pairs of strings."
-            )
+            raise Exception("Please provide a dictionary of metadata with key-value pairs of strings.")
     return await apromptlayer_track_metadata(request_id, metadata, api_key)
 
 

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

@@ -35,9 +35,7 @@ from promptlayer.types.prompt_template import (
     PublishPromptTemplateResponse,
 )
 
-URL_API_PROMPTLAYER = os.environ.setdefault(
-    "URL_API_PROMPTLAYER", "https://api.promptlayer.com"
-)
+URL_API_PROMPTLAYER = os.environ.setdefault("URL_API_PROMPTLAYER", "https://api.promptlayer.com")
 
 
 async def arun_workflow_request(
@@ -237,10 +235,7 @@ def convert_native_object_to_dict(native_object):
     if isinstance(native_object, Enum):
         return native_object.value
     if hasattr(native_object, "__dict__"):
-        return {
-            k: convert_native_object_to_dict(v)
-            for k, v in native_object.__dict__.items()
-        }
+        return {k: convert_native_object_to_dict(v) for k, v in native_object.__dict__.items()}
     return native_object
 
 
@@ -262,9 +257,7 @@ def promptlayer_api_request(
     if isinstance(response, dict) and hasattr(response, "to_dict_recursive"):
         response = response.to_dict_recursive()
     request_response = None
-    if hasattr(
-        response, "dict"
-    ):  # added this for anthropic 3.0 changes, they return a completion object
+    if hasattr(response, "dict"):  # added this for anthropic 3.0 changes, they return a completion object
         response = response.dict()
     try:
         request_response = requests.post(
@@ -371,9 +364,7 @@ def promptlayer_api_request_async(
     )
 
 
-def promptlayer_get_prompt(
-    prompt_name, api_key, version: int = None, label: str = None
-):
+def promptlayer_get_prompt(prompt_name, api_key, version: int = None, label: str = None):
     """
     Get a prompt from the PromptLayer library
     version: version of the prompt to get, None for latest
@@ -386,9 +377,7 @@ def promptlayer_get_prompt(
             params={"prompt_name": prompt_name, "version": version, "label": label},
         )
     except Exception as e:
-        raise Exception(
-            f"PromptLayer had the following error while getting your prompt: {e}"
-        )
+        raise Exception(f"PromptLayer had the following error while getting your prompt: {e}")
     if request_response.status_code != 200:
         raise_on_bad_response(
             request_response,
@@ -398,9 +387,7 @@ def promptlayer_get_prompt(
     return request_response.json()
 
 
-def promptlayer_publish_prompt(
-    prompt_name, prompt_template, commit_message, tags, api_key, metadata=None
-):
+def promptlayer_publish_prompt(prompt_name, prompt_template, commit_message, tags, api_key, metadata=None):
     try:
         request_response = requests.post(
             f"{URL_API_PROMPTLAYER}/library-publish-prompt-template",
@@ -414,9 +401,7 @@ def promptlayer_publish_prompt(
             },
         )
     except Exception as e:
-        raise Exception(
-            f"PromptLayer had the following error while publishing your prompt: {e}"
-        )
+        raise Exception(f"PromptLayer had the following error while publishing your prompt: {e}")
     if request_response.status_code != 200:
         raise_on_bad_response(
             request_response,
@@ -425,9 +410,7 @@ def promptlayer_publish_prompt(
     return True
 
 
-def promptlayer_track_prompt(
-    request_id, prompt_name, input_variables, api_key, version, label
-):
+def promptlayer_track_prompt(request_id, prompt_name, input_variables, api_key, version, label):
     try:
         request_response = requests.post(
             f"{URL_API_PROMPTLAYER}/library-track-prompt",
@@ -516,9 +499,7 @@ def promptlayer_track_metadata(request_id, metadata, api_key):
     return True
 
 
-async def apromptlayer_track_metadata(
-    request_id: str, metadata: Dict[str, Any], api_key: Optional[str] = None
-) -> bool:
+async def apromptlayer_track_metadata(request_id: str, metadata: Dict[str, Any], api_key: Optional[str] = None) -> bool:
     url = f"{URL_API_PROMPTLAYER}/library-track-metadata"
     payload = {
         "request_id": request_id,
@@ -649,9 +630,7 @@ class GeneratorProxy:
 
     def __getattr__(self, name):
         if name == "text_stream":  # anthropic async stream
-            return GeneratorProxy(
-                self.generator.text_stream, self.api_request_arugments, self.api_key
-            )
+            return GeneratorProxy(self.generator.text_stream, self.api_request_arugments, self.api_key)
         return getattr(self.generator, name)
 
     def _abstracted_next(self, result):
@@ -668,8 +647,7 @@ class GeneratorProxy:
                 end_anthropic = True
 
         end_openai = provider_type == "openai" and (
-            result.choices[0].finish_reason == "stop"
-            or result.choices[0].finish_reason == "length"
+            result.choices[0].finish_reason == "stop" or result.choices[0].finish_reason == "length"
         )
 
         if end_anthropic or end_openai:
@@ -684,9 +662,7 @@ class GeneratorProxy:
                 request_end_time=self.api_request_arugments["request_end_time"],
                 api_key=self.api_key,
                 return_pl_id=self.api_request_arugments["return_pl_id"],
-                llm_request_span_id=self.api_request_arugments.get(
-                    "llm_request_span_id"
-                ),
+                llm_request_span_id=self.api_request_arugments.get("llm_request_span_id"),
             )
 
             if self.api_request_arugments["return_pl_id"]:
@@ -716,8 +692,7 @@ class GeneratorProxy:
                 elif hasattr(result, "delta") and hasattr(result.delta, "text"):
                     response = f"{response}{result.delta.text}"
             if (
-                hasattr(self.results[-1], "type")
-                and self.results[-1].type == "message_stop"
+                hasattr(self.results[-1], "type") and self.results[-1].type == "message_stop"
             ):  # this is a message stream and not the correct event
                 final_result = deepcopy(self.results[0].message)
                 final_result.usage = None
@@ -735,23 +710,15 @@ class GeneratorProxy:
             final_result = deepcopy(self.results[-1])
             final_result.choices[0].text = response
             return final_result
-        elif hasattr(
-            self.results[0].choices[0], "delta"
-        ):  # this is completion with delta
+        elif hasattr(self.results[0].choices[0], "delta"):  # this is completion with delta
             response = {"role": "", "content": ""}
             for result in self.results:
-                if (
-                    hasattr(result.choices[0].delta, "role")
-                    and result.choices[0].delta.role is not None
-                ):
+                if hasattr(result.choices[0].delta, "role") and result.choices[0].delta.role is not None:
                     response["role"] = result.choices[0].delta.role
-                if (
-                    hasattr(result.choices[0].delta, "content")
-                    and result.choices[0].delta.content is not None
-                ):
-                    response["content"] = response[
-                        "content"
-                    ] = f"{response['content']}{result.choices[0].delta.content}"
+                if hasattr(result.choices[0].delta, "content") and result.choices[0].delta.content is not None:
+                    response["content"] = response["content"] = (
+                        f"{response['content']}{result.choices[0].delta.content}"
+                    )
             final_result = deepcopy(self.results[-1])
             final_result.choices[0] = response
             return final_result
@@ -854,9 +821,7 @@ def promptlayer_create_group(api_key: str = None):
             return False
     except requests.exceptions.RequestException as e:
         # I'm aiming for a more specific exception catch here
-        raise Exception(
-            f"PromptLayer had the following error while creating your group: {e}"
-        )
+        raise Exception(f"PromptLayer had the following error while creating your group: {e}")
     return request_response.json()["id"]
 
 
@@ -877,9 +842,7 @@ async def apromptlayer_create_group(api_key: Optional[str] = None) -> str:
             return False
         return response.json()["id"]
     except httpx.RequestError as e:
-        raise Exception(
-            f"PromptLayer had the following error while creating your group: {str(e)}"
-        ) from e
+        raise Exception(f"PromptLayer had the following error while creating your group: {str(e)}") from e
 
 
 def promptlayer_track_group(request_id, group_id, api_key: str = None):
@@ -900,9 +863,7 @@ def promptlayer_track_group(request_id, group_id, api_key: str = None):
             return False
     except requests.exceptions.RequestException as e:
         # I'm aiming for a more specific exception catch here
-        raise Exception(
-            f"PromptLayer had the following error while tracking your group: {e}"
-        )
+        raise Exception(f"PromptLayer had the following error while tracking your group: {e}")
     return True
 
 
@@ -948,9 +909,7 @@ def get_prompt_template(
             json=json_body,
         )
         if response.status_code != 200:
-            raise Exception(
-                f"PromptLayer had the following error while getting your prompt template: {response.text}"
-            )
+            raise Exception(f"PromptLayer had the following error while getting your prompt template: {response.text}")
 
         warning = response.json().get("warning", None)
         if warning is not None:
@@ -960,9 +919,7 @@ def get_prompt_template(
             )
         return response.json()
     except requests.exceptions.RequestException as e:
-        raise Exception(
-            f"PromptLayer had the following error while getting your prompt template: {e}"
-        )
+        raise Exception(f"PromptLayer had the following error while getting your prompt template: {e}")
 
 
 async def aget_prompt_template(
@@ -993,9 +950,7 @@ async def aget_prompt_template(
             )
         return response.json()
     except httpx.RequestError as e:
-        raise Exception(
-            f"PromptLayer had the following error while getting your prompt template: {str(e)}"
-        ) from e
+        raise Exception(f"PromptLayer had the following error while getting your prompt template: {str(e)}") from e
 
 
 def publish_prompt_template(
@@ -1018,9 +973,7 @@ def publish_prompt_template(
             )
         return response.json()
     except requests.exceptions.RequestException as e:
-        raise Exception(
-            f"PromptLayer had the following error while publishing your prompt template: {e}"
-        )
+        raise Exception(f"PromptLayer had the following error while publishing your prompt template: {e}")
 
 
 async def apublish_prompt_template(
@@ -1049,9 +1002,7 @@ async def apublish_prompt_template(
             )
         return response.json()
     except httpx.RequestError as e:
-        raise Exception(
-            f"PromptLayer had the following error while publishing your prompt template: {str(e)}"
-        ) from e
+        raise Exception(f"PromptLayer had the following error while publishing your prompt template: {str(e)}") from e
 
 
 def get_all_prompt_templates(
@@ -1070,9 +1021,7 @@ def get_all_prompt_templates(
         items = response.json().get("items", [])
         return items
     except requests.exceptions.RequestException as e:
-        raise Exception(
-            f"PromptLayer had the following error while getting all your prompt templates: {e}"
-        )
+        raise Exception(f"PromptLayer had the following error while getting all your prompt templates: {e}")
 
 
 async def aget_all_prompt_templates(
@@ -1093,9 +1042,7 @@ async def aget_all_prompt_templates(
         items = response.json().get("items", [])
         return items
     except httpx.RequestError as e:
-        raise Exception(
-            f"PromptLayer had the following error while getting all your prompt templates: {str(e)}"
-        ) from e
+        raise Exception(f"PromptLayer had the following error while getting all your prompt templates: {str(e)}") from e
 
 
 def openai_stream_chat(results: list):
@@ -1158,10 +1105,10 @@ def openai_stream_chat(results: list):
                     )
                 )
                 continue
-            last_tool_call.function.name = (
-                f"{last_tool_call.function.name}{tool_call.function.name or ''}"
+            last_tool_call.function.name = f"{last_tool_call.function.name}{tool_call.function.name or ''}"
+            last_tool_call.function.arguments = (
+                f"{last_tool_call.function.arguments}{tool_call.function.arguments or ''}"
             )
-            last_tool_call.function.arguments = f"{last_tool_call.function.arguments}{tool_call.function.arguments or ''}"
 
     response.choices[0].message.content = content
     response.choices[0].message.tool_calls = tool_calls
@@ -1224,10 +1171,10 @@ async def aopenai_stream_chat(generator: AsyncIterable[Any]) -> Any:
                     )
                 )
                 continue
-            last_tool_call.function.name = (
-                f"{last_tool_call.function.name}{tool_call.function.name or ''}"
+            last_tool_call.function.name = f"{last_tool_call.function.name}{tool_call.function.name or ''}"
+            last_tool_call.function.arguments = (
+                f"{last_tool_call.function.arguments}{tool_call.function.arguments or ''}"
             )
-            last_tool_call.function.arguments = f"{last_tool_call.function.arguments}{tool_call.function.arguments or ''}"
 
     # After collecting all chunks, set the response attributes
     if chat_completion_chunks:
@@ -1414,9 +1361,7 @@ async def aanthropic_stream_completion(generator: AsyncIterable[Any]) -> Any:
     return response
 
 
-def stream_response(
-    generator: Generator, after_stream: Callable, map_results: Callable
-):
+def stream_response(generator: Generator, after_stream: Callable, map_results: Callable):
     data = {
         "request_id": None,
         "raw_response": None,
@@ -1455,9 +1400,7 @@ async def astream_response(
             yield item
 
     request_response = await map_results(async_generator_from_list(results))
-    after_stream_response = await after_stream(
-        request_response=request_response.model_dump()
-    )
+    after_stream_response = await after_stream(request_response=request_response.model_dump())
     data["request_id"] = after_stream_response.get("request_id")
     data["prompt_blueprint"] = after_stream_response.get("prompt_blueprint")
     yield data
@@ -1481,9 +1424,7 @@ def openai_request(prompt_blueprint: GetPromptTemplateResponse, **kwargs):
     from openai import OpenAI
 
     client = OpenAI(base_url=kwargs.pop("base_url", None))
-    request_to_make = MAP_TYPE_TO_OPENAI_FUNCTION[
-        prompt_blueprint["prompt_template"]["type"]
-    ]
+    request_to_make = MAP_TYPE_TO_OPENAI_FUNCTION[prompt_blueprint["prompt_template"]["type"]]
     return request_to_make(client, **kwargs)
 
 
@@ -1505,9 +1446,7 @@ async def aopenai_request(prompt_blueprint: GetPromptTemplateResponse, **kwargs)
     from openai import AsyncOpenAI
 
     client = AsyncOpenAI(base_url=kwargs.pop("base_url", None))
-    request_to_make = AMAP_TYPE_TO_OPENAI_FUNCTION[
-        prompt_blueprint["prompt_template"]["type"]
-    ]
+    request_to_make = AMAP_TYPE_TO_OPENAI_FUNCTION[prompt_blueprint["prompt_template"]["type"]]
     return await request_to_make(client, **kwargs)
 
 
@@ -1515,9 +1454,7 @@ def azure_openai_request(prompt_blueprint: GetPromptTemplateResponse, **kwargs):
     from openai import AzureOpenAI
 
     client = AzureOpenAI(azure_endpoint=kwargs.pop("base_url", None))
-    request_to_make = MAP_TYPE_TO_OPENAI_FUNCTION[
-        prompt_blueprint["prompt_template"]["type"]
-    ]
+    request_to_make = MAP_TYPE_TO_OPENAI_FUNCTION[prompt_blueprint["prompt_template"]["type"]]
     return request_to_make(client, **kwargs)
 
 
@@ -1525,9 +1462,7 @@ async def aazure_openai_request(prompt_blueprint: GetPromptTemplateResponse, **k
     from openai import AsyncAzureOpenAI
 
     client = AsyncAzureOpenAI(azure_endpoint=kwargs.pop("base_url", None))
-    request_to_make = AMAP_TYPE_TO_OPENAI_FUNCTION[
-        prompt_blueprint["prompt_template"]["type"]
-    ]
+    request_to_make = AMAP_TYPE_TO_OPENAI_FUNCTION[prompt_blueprint["prompt_template"]["type"]]
     return await request_to_make(client, **kwargs)
 
 
@@ -1549,9 +1484,7 @@ def anthropic_request(prompt_blueprint: GetPromptTemplateResponse, **kwargs):
     from anthropic import Anthropic
 
     client = Anthropic(base_url=kwargs.pop("base_url", None))
-    request_to_make = MAP_TYPE_TO_ANTHROPIC_FUNCTION[
-        prompt_blueprint["prompt_template"]["type"]
-    ]
+    request_to_make = MAP_TYPE_TO_ANTHROPIC_FUNCTION[prompt_blueprint["prompt_template"]["type"]]
     return request_to_make(client, **kwargs)
 
 
@@ -1573,9 +1506,7 @@ async def aanthropic_request(prompt_blueprint: GetPromptTemplateResponse, **kwar
     from anthropic import AsyncAnthropic
 
     client = AsyncAnthropic(base_url=kwargs.pop("base_url", None))
-    request_to_make = AMAP_TYPE_TO_ANTHROPIC_FUNCTION[
-        prompt_blueprint["prompt_template"]["type"]
-    ]
+    request_to_make = AMAP_TYPE_TO_ANTHROPIC_FUNCTION[prompt_blueprint["prompt_template"]["type"]]
     return await request_to_make(client, **kwargs)
 
 
@@ -1715,11 +1646,11 @@ def mistral_stream_chat(results: list):
                 else:
                     last_tool_call = tool_calls[-1]
                     if tool_call.function.name:
-                        last_tool_call.function.name = (
-                            f"{last_tool_call.function.name}{tool_call.function.name}"
-                        )
+                        last_tool_call.function.name = f"{last_tool_call.function.name}{tool_call.function.name}"
                     if tool_call.function.arguments:
-                        last_tool_call.function.arguments = f"{last_tool_call.function.arguments}{tool_call.function.arguments}"
+                        last_tool_call.function.arguments = (
+                            f"{last_tool_call.function.arguments}{tool_call.function.arguments}"
+                        )
 
     response.choices[0].message.content = content
     response.choices[0].message.tool_calls = tool_calls
@@ -1779,11 +1710,11 @@ async def amistral_stream_chat(generator: AsyncIterable[Any]) -> Any:
                 else:
                     last_tool_call = tool_calls[-1]
                     if tool_call.function.name:
-                        last_tool_call.function.name = (
-                            f"{last_tool_call.function.name}{tool_call.function.name}"
-                        )
+                        last_tool_call.function.name = f"{last_tool_call.function.name}{tool_call.function.name}"
                     if tool_call.function.arguments:
-                        last_tool_call.function.arguments = f"{last_tool_call.function.arguments}{tool_call.function.arguments}"
+                        last_tool_call.function.arguments = (
+                            f"{last_tool_call.function.arguments}{tool_call.function.arguments}"
+                        )
 
     if completion_chunks:
         last_result = completion_chunks[-1]

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "promptlayer"
-version = "1.0.37"
+version = "1.0.38"
 description = "PromptLayer is a platform for prompt engineering and tracks your LLM requests."
 authors = ["Magniv <hello@magniv.io>"]
 license = "Apache-2.0"
@@ -28,3 +28,22 @@ openai = "^1.60.1"
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
+
+[tool.ruff]
+line-length = 120
+indent-width = 4  # mimic Black
+target-version = "py38"
+
+[tool.ruff.lint]
+ignore = ["E501", "E711", "E712"]
+
+[tool.ruff.lint.isort]
+combine-as-imports = true
+relative-imports-order = "closest-to-furthest"
+known-first-party = ["promptlayer", "tests"]
+
+[tool.ruff.format]
+quote-style = "double"  # mimic Black
+indent-style = "space"  # also mimic Black
+skip-magic-trailing-comma = false  # also mimic Black
+line-ending = "auto"  # mimic Black