promptlayer 1.0.16__py3-none-any.whl → 1.0.18__py3-none-any.whl

promptlayer/__init__.py

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

promptlayer/promptlayer.py

@@ -5,7 +5,6 @@ from copy import deepcopy
 from functools import wraps
 from typing import Any, Dict, List, Literal, Union
 
-from opentelemetry import trace
 from opentelemetry.sdk.resources import Resource
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor
@@ -74,7 +73,9 @@ class PromptLayer:
         self.api_key = api_key
         self.templates = TemplateManager(api_key)
         self.group = GroupManager(api_key)
-        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__(
@@ -112,7 +113,7 @@ class PromptLayer:
         tags,
         input_variables,
         group_id,
-        pl_run_span_id: str | None = None,
+        pl_run_span_id: Union[str, None] = None,
     ):
         def _track_request(**body):
             track_request_kwargs = self._prepare_track_request_kwargs(
@@ -132,10 +133,10 @@ class PromptLayer:
             promptlayer_exporter = PromptLayerSpanExporter(api_key=api_key)
             span_processor = BatchSpanProcessor(promptlayer_exporter)
             tracer_provider.add_span_processor(span_processor)
-            trace.set_tracer_provider(tracer_provider)
-            return trace.get_tracer(__name__)
+            tracer = tracer_provider.get_tracer(__name__)
+            return tracer_provider, tracer
         else:
-            return None
+            return None, None
 
     @staticmethod
     def _prepare_get_prompt_template_params(
@@ -184,8 +185,8 @@ class PromptLayer:
         tags,
         input_variables,
         group_id,
-        pl_run_span_id: str | None = None,
-        metadata: Dict[str, str] | None = None,
+        pl_run_span_id: Union[str, None] = None,
+        metadata: Union[Dict[str, str], None] = None,
         **body,
     ):
         return {
@@ -222,7 +223,7 @@ class PromptLayer:
         metadata: Union[Dict[str, str], None] = None,
         group_id: Union[int, None] = None,
         stream: bool = False,
-        pl_run_span_id: str | None = None,
+        pl_run_span_id: Union[str, None] = None,
     ) -> Dict[str, Any]:
         get_prompt_template_params = self._prepare_get_prompt_template_params(
             prompt_version=prompt_version,
@@ -280,8 +281,8 @@ class PromptLayer:
         tags,
         input_variables,
         group_id,
-        pl_run_span_id: str | None = None,
-        metadata: Dict[str, str] | None = None,
+        pl_run_span_id: Union[str, None] = None,
+        metadata: Union[Dict[str, str], None] = None,
         **body,
     ):
         track_request_kwargs = self._prepare_track_request_kwargs(

{promptlayer-1.0.16.dist-info → promptlayer-1.0.18.dist-info}/METADATA

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

{promptlayer-1.0.16.dist-info → promptlayer-1.0.18.dist-info}/RECORD

@@ -1,7 +1,7 @@
-promptlayer/__init__.py,sha256=NPhgcyisSfks_AFSRjhlScwwHQRlZSKiPBJS-Lm_grg,102
+promptlayer/__init__.py,sha256=MVQ-wlZkL_mEj_JfMdDtR1lv1x5c_b3saQtpNBQCmEM,102
 promptlayer/groups/__init__.py,sha256=-xs-2cn0nc0D_5YxZr3nC86iTdRVZmBhEpOKDJXE-sQ,224
 promptlayer/groups/groups.py,sha256=yeO6T0TM3qB0ondZRiHhcH8G06YygrpFoM8b9RmoIao,165
-promptlayer/promptlayer.py,sha256=xqnwYO5p2JPA_mQbP-5l9yQlQ1bKrfms803CrJBSiFA,13824
+promptlayer/promptlayer.py,sha256=OzwekB-elWljFTUciOg3KiMWJKGleZkR9G3hxtxyj7A,13878
 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,7 +10,7 @@ promptlayer/track/track.py,sha256=XNEZT9yNiRBPp9vaDZo_f0dP_ldOu8q1qafpVfS5Ze8,16
 promptlayer/types/__init__.py,sha256=ulWSyCrk5hZ_PI-nKGpd6GPcRaK8lqP4wFl0LPNUYWk,61
 promptlayer/types/prompt_template.py,sha256=QbxYSeIubrwp8KmDKdt9syAwzONFPh_So9yr4H73ANQ,4429
 promptlayer/utils.py,sha256=-p0qapUvkZYJd_Dfat3c8LANXWU1JN0bJB91IyjB8iA,29656
-promptlayer-1.0.16.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-promptlayer-1.0.16.dist-info/METADATA,sha256=HpWQPpYbuABRRj4feBBLYpZvviPcIe25q5S0fBKlzR8,4609
-promptlayer-1.0.16.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-promptlayer-1.0.16.dist-info/RECORD,,
+promptlayer-1.0.18.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+promptlayer-1.0.18.dist-info/METADATA,sha256=qU7gZWgtYWWWuJ3fXx2cKbaz9W7bmzFGM6pOodi4_KU,4609
+promptlayer-1.0.18.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+promptlayer-1.0.18.dist-info/RECORD,,