promptlayer 1.0.17__tar.gz → 1.0.19__tar.gz

{promptlayer-1.0.17 → promptlayer-1.0.19}/PKG-INFO

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

{promptlayer-1.0.17 → promptlayer-1.0.19}/promptlayer/__init__.py

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

{promptlayer-1.0.17 → promptlayer-1.0.19}/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__(
@@ -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(
@@ -355,12 +356,13 @@ class PromptLayer:
         else:
             return self._run_internal(**_run_internal_kwargs)
 
-    def traceable(self, attributes=None):
+    def traceable(self, attributes=None, name=None):
         def decorator(func):
             @wraps(func)
             def sync_wrapper(*args, **kwargs):
                 if self.tracer:
-                    with self.tracer.start_as_current_span(func.__name__) as span:
+                    span_name = name or func.__name__
+                    with self.tracer.start_as_current_span(span_name) as span:
                         if attributes:
                             for key, value in attributes.items():
                                 span.set_attribute(key, value)
@@ -378,7 +380,8 @@ class PromptLayer:
             @wraps(func)
             async def async_wrapper(*args, **kwargs):
                 if self.tracer:
-                    with self.tracer.start_as_current_span(func.__name__) as span:
+                    span_name = name or func.__name__
+                    with self.tracer.start_as_current_span(span_name) as span:
                         if attributes:
                             for key, value in attributes.items():
                                 span.set_attribute(key, value)

{promptlayer-1.0.17 → promptlayer-1.0.19}/pyproject.toml

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