promptlayer 1.0.4__tar.gz → 1.0.6__tar.gz

{promptlayer-1.0.4 → promptlayer-1.0.6}/PKG-INFO

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

{promptlayer-1.0.4 → promptlayer-1.0.6}/promptlayer/__init__.py

@@ -103,6 +103,8 @@ class PromptLayer:
             template_get_params["label"] = prompt_release_label
         if input_variables:
             template_get_params["input_variables"] = input_variables
+        if metadata:
+            template_get_params["metadata_filters"] = metadata
         prompt_blueprint = self.templates.get(prompt_name, template_get_params)
         prompt_template = prompt_blueprint["prompt_template"]
         if not prompt_blueprint["llm_kwargs"]:

{promptlayer-1.0.4 → promptlayer-1.0.6}/promptlayer/types/prompt_template.py

@@ -8,6 +8,7 @@ class GetPromptTemplate(TypedDict, total=False):
     label: str
     provider: str
     input_variables: Dict[str, str]
+    metadata_filters: Dict[str, str]
 
 
 TemplateFormat = Literal["f-string", "jinja2"]

{promptlayer-1.0.4 → promptlayer-1.0.6}/promptlayer/utils.py

@@ -41,7 +41,12 @@ def promptlayer_api_handler(
         isinstance(response, types.GeneratorType)
         or isinstance(response, types.AsyncGeneratorType)
         or type(response).__name__
-        in ["Stream", "AsyncStream", "AsyncMessageStreamManager"]
+        in [
+            "Stream",
+            "AsyncStream",
+            "AsyncMessageStreamManager",
+            "MessageStreamManager",
+        ]
     ):
         return GeneratorProxy(
             response,
@@ -356,6 +361,19 @@ class GeneratorProxy:
                 self.api_key,
             )
 
+    def __enter__(self):
+        api_request_arguments = self.api_request_arugments
+        if hasattr(self.generator, "_MessageStreamManager__api_request"):
+            stream = self.generator.__enter__()
+            return GeneratorProxy(
+                stream,
+                api_request_arguments,
+                self.api_key,
+            )
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        pass
+
     async def __aexit__(self, exc_type, exc_val, exc_tb):
         pass
 
@@ -458,7 +476,7 @@ class GeneratorProxy:
                     hasattr(result.choices[0].delta, "content")
                     and result.choices[0].delta.content is not None
                 ):
-                    response[
+                    response["content"] = response[
                         "content"
                     ] = f"{response['content']}{result.choices[0].delta.content}"
             final_result = deepcopy(self.results[-1])

{promptlayer-1.0.4 → promptlayer-1.0.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "promptlayer"
-version = "1.0.4"
+version = "1.0.6"
 description = "PromptLayer is a platform for prompt engineering and tracks your LLM requests."
 authors = ["Magniv <hello@magniv.io>"]
 license = "Apache-2.0"