promptlayer 1.0.16__tar.gz → 1.0.18__tar.gz

{promptlayer-1.0.16 → promptlayer-1.0.18}/PKG-INFO

@@ -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 → promptlayer-1.0.18}/promptlayer/__init__.py

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

{promptlayer-1.0.16 → promptlayer-1.0.18}/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 → promptlayer-1.0.18}/pyproject.toml

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