promptlayer 1.0.5__py3-none-any.whl → 1.0.6__py3-none-any.whl

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.5.dist-info → promptlayer-1.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: promptlayer
-Version: 1.0.5
+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.5.dist-info → promptlayer-1.0.6.dist-info}/RECORD

@@ -7,8 +7,8 @@ promptlayer/track/__init__.py,sha256=VheO_Au0lffGlPKYYPQwkv8ci16wSXABCVSNRoFWu_w
 promptlayer/track/track.py,sha256=XNEZT9yNiRBPp9vaDZo_f0dP_ldOu8q1qafpVfS5Ze8,1610
 promptlayer/types/__init__.py,sha256=ulWSyCrk5hZ_PI-nKGpd6GPcRaK8lqP4wFl0LPNUYWk,61
 promptlayer/types/prompt_template.py,sha256=1iqwH0f7MIITjUEH0BMnSKzMh2BbcB7W8rCmQEaUhQ0,4284
-promptlayer/utils.py,sha256=pVRLCAIikoCn1p95otxZYn0xMFN-cnDBJyN1fbT58tE,27305
-promptlayer-1.0.5.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-promptlayer-1.0.5.dist-info/METADATA,sha256=Yh49bWt30Wo_kzVVy-VGlt-jsN2_wO9zoor0d-x9V9c,4506
-promptlayer-1.0.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-promptlayer-1.0.5.dist-info/RECORD,,
+promptlayer/utils.py,sha256=vourOtK8UnTAMKw9jjXhXR4zR3XYJsdLBBF31A3yW2w,27824
+promptlayer-1.0.6.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+promptlayer-1.0.6.dist-info/METADATA,sha256=0nb2ghDvKr2sm9GR-o0EE-_ZQGpDKliiEg9051J6Rog,4506
+promptlayer-1.0.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+promptlayer-1.0.6.dist-info/RECORD,,