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

promptlayer/__init__.py

@@ -1,7 +1,7 @@
 import datetime
 import os
 from copy import deepcopy
-from typing import Dict, List, Literal, Union
+from typing import Any, Dict, List, Literal, Union
 
 from promptlayer.groups import GroupManager
 from promptlayer.promptlayer import PromptLayerBase
@@ -90,7 +90,7 @@ class PromptLayer:
         prompt_name: str,
         prompt_version: Union[int, None] = None,
         prompt_release_label: Union[str, None] = None,
-        input_variables: Union[Dict[str, str], None] = None,
+        input_variables: Union[Dict[str, Any], None] = None,
         tags: Union[List[str], None] = None,
         metadata: Union[Dict[str, str], None] = None,
         group_id: Union[int, None] = None,

promptlayer/types/prompt_template.py

@@ -1,4 +1,4 @@
-from typing import Dict, List, Literal, Optional, Sequence, TypedDict, Union
+from typing import Any, Dict, List, Literal, Optional, Sequence, TypedDict, Union
 
 from typing_extensions import Required
 
@@ -7,7 +7,7 @@ class GetPromptTemplate(TypedDict, total=False):
     version: int
     label: str
     provider: str
-    input_variables: Dict[str, str]
+    input_variables: Dict[str, Any]
     metadata_filters: Dict[str, str]
 
 

promptlayer/utils.py

@@ -436,8 +436,11 @@ class GeneratorProxy:
                     response = f"{response}{result.completion}"
                 elif hasattr(result, "message") and isinstance(result.message, str):
                     response = f"{response}{result.message}"
-                elif hasattr(result, "content_block") and hasattr(
-                    result.content_block, "text"
+                elif (
+                    hasattr(result, "content_block")
+                    and hasattr(result.content_block, "text")
+                    and "type" in result
+                    and result.type != "message_stop"
                 ):
                     response = f"{response}{result.content_block.text}"
                 elif hasattr(result, "delta") and hasattr(result.delta, "text"):

{promptlayer-1.0.6.dist-info → promptlayer-1.0.8.dist-info}/METADATA

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

{promptlayer-1.0.6.dist-info → promptlayer-1.0.8.dist-info}/RECORD

@@ -1,4 +1,4 @@
-promptlayer/__init__.py,sha256=7DdTZdYCs3XRkX18DJNZ946J66lDBlALrWe6t8HcOzU,6353
+promptlayer/__init__.py,sha256=Soo7M8Pc7RxVQHFg81MIjsDCsumzebSeD87BtP54JzY,6358
 promptlayer/groups/__init__.py,sha256=-xs-2cn0nc0D_5YxZr3nC86iTdRVZmBhEpOKDJXE-sQ,224
 promptlayer/groups/groups.py,sha256=yeO6T0TM3qB0ondZRiHhcH8G06YygrpFoM8b9RmoIao,165
 promptlayer/promptlayer.py,sha256=1q1cZOBt27Luzu3aRWcYUjQiKbmwrF9R62Uw95Ryqqc,4035
@@ -6,9 +6,9 @@ promptlayer/templates.py,sha256=aY_-BCrL0AgIdYEUE28pi0AP_avTVAgwv5hgzrh75vo,717
 promptlayer/track/__init__.py,sha256=VheO_Au0lffGlPKYYPQwkv8ci16wSXABCVSNRoFWu_w,945
 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=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,,
+promptlayer/types/prompt_template.py,sha256=yMtQ1kMe_Ub12htwu7EC_BBqOsZosrjckc73tJgokmo,4289
+promptlayer/utils.py,sha256=jkto4JeHdEK7eoPN50H0AjlWZHk5twzFpMmvMP-rNZU,27941
+promptlayer-1.0.8.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+promptlayer-1.0.8.dist-info/METADATA,sha256=FsZTOzYib7xxM9OMJU0tmNwB-YX7atgrDaRT7PgFrJA,4506
+promptlayer-1.0.8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+promptlayer-1.0.8.dist-info/RECORD,,