promptlayer 1.0.12__tar.gz → 1.0.13__tar.gz

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

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

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

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

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

@@ -57,18 +57,30 @@ MAP_PROVIDER_TO_FUNCTION = {
 
 
 class PromptLayer:
-    def __init__(self, api_key: str = None, enable_tracing: bool = False):
+    def __init__(
+        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")
+
         if api_key is None:
             raise ValueError(
-                "PromptLayer API key not provided. Please set the PROMPTLAYER_API_KEY environment variable or pass the api_key parameter."
+                "PromptLayer API key not provided. "
+                "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.track = TrackManager(api_key)
-        self.tracer = self._initialize_tracer(enable_tracing)
+        self.workspace_id = workspace_id
 
     def __getattr__(
         self,
@@ -127,13 +139,18 @@ class PromptLayer:
                 return result
         return self.templates.get(prompt_name, template_params)
 
-    def _initialize_tracer(self, enable_tracing: bool):
+    @staticmethod
+    def _initialize_tracer(
+        api_key: str = None, enable_tracing: bool = False, workspace_id: int = None
+    ):
         if enable_tracing:
             resource = Resource(
                 attributes={ResourceAttributes.SERVICE_NAME: "prompt-layer-library"}
             )
             tracer_provider = TracerProvider(resource=resource)
-            promptlayer_exporter = PromptLayerSpanExporter(api_key=self.api_key)
+            promptlayer_exporter = PromptLayerSpanExporter(
+                api_key=api_key, workspace_id=workspace_id
+            )
             span_processor = BatchSpanProcessor(promptlayer_exporter)
             tracer_provider.add_span_processor(span_processor)
             trace.set_tracer_provider(tracer_provider)

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

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

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

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