promptlayer 1.0.13__tar.gz → 1.0.15__tar.gz

{promptlayer-1.0.13 → promptlayer-1.0.15}/PKG-INFO

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

{promptlayer-1.0.13 → promptlayer-1.0.15}/promptlayer/__init__.py

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

{promptlayer-1.0.13 → promptlayer-1.0.15}/promptlayer/promptlayer.py

@@ -61,7 +61,6 @@ class PromptLayer:
         self,
         api_key: str = None,
         enable_tracing: bool = False,
-        workspace_id: int = None,
     ):
         if api_key is None:
             api_key = os.environ.get("PROMPTLAYER_API_KEY")
@@ -72,15 +71,11 @@ class PromptLayer:
                 "Please set the PROMPTLAYER_API_KEY environment variable or pass the api_key parameter."
             )
 
-        if enable_tracing and not workspace_id:
-            raise ValueError("Please set a workspace_id to enable tracing.")
-
         self.api_key = api_key
         self.templates = TemplateManager(api_key)
         self.group = GroupManager(api_key)
-        self.tracer = self._initialize_tracer(api_key, enable_tracing, workspace_id)
+        self.tracer = self._initialize_tracer(api_key, enable_tracing)
         self.track = TrackManager(api_key)
-        self.workspace_id = workspace_id
 
     def __getattr__(
         self,
@@ -140,17 +135,13 @@ class PromptLayer:
         return self.templates.get(prompt_name, template_params)
 
     @staticmethod
-    def _initialize_tracer(
-        api_key: str = None, enable_tracing: bool = False, workspace_id: int = None
-    ):
+    def _initialize_tracer(api_key: str = None, enable_tracing: bool = False):
         if enable_tracing:
             resource = Resource(
                 attributes={ResourceAttributes.SERVICE_NAME: "prompt-layer-library"}
             )
             tracer_provider = TracerProvider(resource=resource)
-            promptlayer_exporter = PromptLayerSpanExporter(
-                api_key=api_key, workspace_id=workspace_id
-            )
+            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)
@@ -374,14 +365,14 @@ class PromptLayer:
         else:
             return self._run_internal(**_run_internal_kwargs)
 
-    def traceable(self, metadata=None):
+    def traceable(self, attributes=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:
-                        if metadata:
-                            for key, value in metadata.items():
+                        if attributes:
+                            for key, value in attributes.items():
                                 span.set_attribute(key, value)
 
                         span.set_attribute(
@@ -398,8 +389,8 @@ class PromptLayer:
             async def async_wrapper(*args, **kwargs):
                 if self.tracer:
                     with self.tracer.start_as_current_span(func.__name__) as span:
-                        if metadata:
-                            for key, value in metadata.items():
+                        if attributes:
+                            for key, value in attributes.items():
                                 span.set_attribute(key, value)
 
                         span.set_attribute(

{promptlayer-1.0.13 → promptlayer-1.0.15}/promptlayer/span_exporter.py

@@ -8,10 +8,9 @@ from promptlayer.utils import URL_API_PROMPTLAYER
 
 
 class PromptLayerSpanExporter(SpanExporter):
-    def __init__(self, api_key: str = None, workspace_id: int = None):
+    def __init__(self, api_key: str = None):
         self.api_key = api_key
         self.url = f"{URL_API_PROMPTLAYER}/spans-bulk"
-        self.workspace_id = workspace_id
 
     def export(self, spans: Sequence[ReadableSpan]) -> SpanExportResult:
         request_data = []
@@ -65,7 +64,6 @@ class PromptLayerSpanExporter(SpanExporter):
                 },
                 json={
                     "spans": request_data,
-                    "workspace_id": self.workspace_id,
                 },
             )
             response.raise_for_status()

{promptlayer-1.0.13 → promptlayer-1.0.15}/pyproject.toml

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