lmnr 0.6.9__tar.gz → 0.6.11__tar.gz

{lmnr-0.6.9 → lmnr-0.6.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lmnr
-Version: 0.6.9
+Version: 0.6.11
 Summary: Python SDK for Laminar
 License: Apache-2.0
 Author: lmnr.ai

{lmnr-0.6.9 → lmnr-0.6.11}/pyproject.toml

@@ -6,7 +6,7 @@
 
 [project]
 name = "lmnr"
-version = "0.6.9"
+version = "0.6.11"
 description = "Python SDK for Laminar"
 authors = [
   { name = "lmnr.ai", email = "founders@lmnr.ai" }

{lmnr-0.6.9 → lmnr-0.6.11}/src/lmnr/cli.py

@@ -100,9 +100,14 @@ async def run_evaluation(args):
             sys.modules[name] = mod
 
             spec.loader.exec_module(mod)
-            evaluations: list[Evaluation] | None = EVALUATION_INSTANCES.get()
-            if evaluations is None:
-                LOG.warning("Evaluation instance not found")
+            evaluations = []
+            try:
+                evaluations: list[Evaluation] | None = EVALUATION_INSTANCES.get()
+                if evaluations is None:
+                    raise LookupError()
+            # may be raised by `get()` or manually by us above
+            except LookupError:
+                log_evaluation_instance_not_found()
                 if args.continue_on_error:
                     continue
                 return
@@ -130,6 +135,14 @@ async def run_evaluation(args):
         PREPARE_ONLY.reset(prep_token)
 
 
+def log_evaluation_instance_not_found():
+    LOG.warning(
+        "Evaluation instance not found. "
+        "`evaluate` must be called at the top level of the file, "
+        "not inside a function when running evaluations from the CLI."
+    )
+
+
 def cli():
     parser = ArgumentParser(
         prog="lmnr",

{lmnr-0.6.9 → lmnr-0.6.11}/src/lmnr/opentelemetry_lib/__init__.py

@@ -54,6 +54,8 @@ class TracerManager:
 
     @staticmethod
     def flush() -> bool:
+        if not hasattr(TracerManager, "_TracerManager__tracer_wrapper"):
+            return False
         return TracerManager.__tracer_wrapper.flush()
 
     @staticmethod

{lmnr-0.6.9 → lmnr-0.6.11}/src/lmnr/opentelemetry_lib/opentelemetry/instrumentation/langgraph/__init__.py

@@ -11,7 +11,7 @@ from .utils import (
 from langchain_core.runnables.graph import Graph
 from opentelemetry.trace import Tracer
 from wrapt import wrap_function_wrapper
-from opentelemetry.trace import get_tracer, get_current_span
+from opentelemetry.trace import get_tracer
 
 from lmnr.opentelemetry_lib.tracing.context_properties import (
     update_association_properties,
@@ -81,7 +81,9 @@ async def async_wrap_pregel_stream(
             "langgraph.nodes": json.dumps(nodes),
         },
     )
-    return await wrapped(*args, **kwargs)
+
+    async for item in wrapped(*args, **kwargs):
+        yield item
 
 
 class LanggraphInstrumentor(BaseInstrumentor):

lmnr-0.6.11/src/lmnr/opentelemetry_lib/tracing/__init__.py

@@ -0,0 +1,162 @@
+import atexit
+import logging
+import threading
+
+from lmnr.opentelemetry_lib.tracing.processor import LaminarSpanProcessor
+from lmnr.sdk.client.asynchronous.async_client import AsyncLaminarClient
+from lmnr.sdk.client.synchronous.sync_client import LaminarClient
+from lmnr.sdk.log import VerboseColorfulFormatter
+from lmnr.opentelemetry_lib.tracing.instruments import (
+    Instruments,
+    init_instrumentations,
+)
+
+from opentelemetry import trace
+from opentelemetry.instrumentation.threading import ThreadingInstrumentor
+from opentelemetry.sdk.resources import Resource
+from opentelemetry.sdk.trace import TracerProvider, SpanProcessor
+from opentelemetry.sdk.trace.export import SpanExporter
+
+TRACER_NAME = "lmnr.tracer"
+
+MAX_EVENTS_OR_ATTRIBUTES_PER_SPAN = 5000
+
+
+class TracerWrapper(object):
+    resource_attributes: dict = {}
+    enable_content_tracing: bool = True
+    _lock = threading.Lock()
+    _tracer_provider: TracerProvider | None = None
+    _logger: logging.Logger
+    _client: LaminarClient
+    _async_client: AsyncLaminarClient
+    _resource: Resource
+    _span_processor: SpanProcessor
+
+    def __new__(
+        cls,
+        disable_batch=False,
+        exporter: SpanExporter | None = None,
+        instruments: set[Instruments] | None = None,
+        block_instruments: set[Instruments] | None = None,
+        base_url: str = "https://api.lmnr.ai",
+        port: int = 8443,
+        http_port: int = 443,
+        project_api_key: str | None = None,
+        max_export_batch_size: int | None = None,
+        force_http: bool = False,
+        timeout_seconds: int = 10,
+        set_global_tracer_provider: bool = True,
+        otel_logger_level: int = logging.ERROR,
+    ) -> "TracerWrapper":
+        # Silence some opentelemetry warnings
+        logging.getLogger("opentelemetry.trace").setLevel(otel_logger_level)
+
+        base_http_url = f"{base_url}:{http_port}"
+        with cls._lock:
+            if not hasattr(cls, "instance"):
+                cls._initialize_logger(cls)
+                obj = super(TracerWrapper, cls).__new__(cls)
+
+                obj._client = LaminarClient(
+                    base_url=base_http_url,
+                    project_api_key=project_api_key,
+                )
+                obj._async_client = AsyncLaminarClient(
+                    base_url=base_http_url,
+                    project_api_key=project_api_key,
+                )
+
+                obj._resource = Resource(attributes=TracerWrapper.resource_attributes)
+
+                obj._span_processor = LaminarSpanProcessor(
+                    base_url=base_url,
+                    api_key=project_api_key,
+                    port=http_port if force_http else port,
+                    exporter=exporter,
+                    max_export_batch_size=max_export_batch_size,
+                    timeout_seconds=timeout_seconds,
+                    force_http=force_http,
+                    disable_batch=disable_batch,
+                )
+
+                lmnr_provider = TracerProvider(resource=obj._resource)
+                global_provider = trace.get_tracer_provider()
+                if set_global_tracer_provider and isinstance(
+                    global_provider, trace.ProxyTracerProvider
+                ):
+                    trace.set_tracer_provider(lmnr_provider)
+
+                obj._tracer_provider = lmnr_provider
+
+                obj._tracer_provider.add_span_processor(obj._span_processor)
+
+                # This is not a real instrumentation and does not generate telemetry
+                # data, but it is required to ensure that OpenTelemetry context
+                # propagation is enabled.
+                # See the README at:
+                # https://pypi.org/project/opentelemetry-instrumentation-threading/
+                ThreadingInstrumentor().instrument()
+
+                init_instrumentations(
+                    tracer_provider=obj._tracer_provider,
+                    instruments=instruments,
+                    block_instruments=block_instruments,
+                    client=obj._client,
+                    async_client=obj._async_client,
+                )
+
+                cls.instance = obj
+
+                # Force flushes for debug environments (e.g. local development)
+                atexit.register(obj.exit_handler)
+
+            return cls.instance
+
+    def exit_handler(self):
+        if isinstance(self._span_processor, LaminarSpanProcessor):
+            self._span_processor.clear()
+        self.flush()
+
+    def _initialize_logger(self):
+        self._logger = logging.getLogger(__name__)
+        console_log_handler = logging.StreamHandler()
+        console_log_handler.setFormatter(VerboseColorfulFormatter())
+        self._logger.addHandler(console_log_handler)
+
+    @staticmethod
+    def set_static_params(
+        resource_attributes: dict,
+        enable_content_tracing: bool,
+    ) -> None:
+        TracerWrapper.resource_attributes = resource_attributes
+        TracerWrapper.enable_content_tracing = enable_content_tracing
+
+    @classmethod
+    def verify_initialized(cls) -> bool:
+        with cls._lock:
+            return hasattr(cls, "instance") and hasattr(cls.instance, "_span_processor")
+
+    @classmethod
+    def clear(cls):
+        if not cls.verify_initialized():
+            return
+        # Any state cleanup. Now used in between tests
+        if isinstance(cls.instance._span_processor, LaminarSpanProcessor):
+            cls.instance._span_processor.clear()
+
+    def shutdown(self):
+        if self._tracer_provider is None:
+            return
+        self._tracer_provider.shutdown()
+
+    def flush(self):
+        if not hasattr(self, "_span_processor"):
+            self._logger.warning("TracerWrapper not fully initialized, cannot flush")
+            return False
+        return self._span_processor.force_flush()
+
+    def get_tracer(self):
+        if self._tracer_provider is None:
+            return trace.get_tracer_provider().get_tracer(TRACER_NAME)
+        return self._tracer_provider.get_tracer(TRACER_NAME)

{lmnr-0.6.9 → lmnr-0.6.11}/src/lmnr/sdk/client/asynchronous/async_client.py

@@ -89,11 +89,11 @@ class AsyncLaminarClient:
         return self.__agent
 
     @property
-    def _evals(self) -> AsyncEvals:
+    def evals(self) -> AsyncEvals:
         """Get the Evals resource.
 
         Returns:
-            Evals: The Evals resource instance.
+            AsyncEvals: The Evals resource instance.
         """
         return self.__evals
 
@@ -144,3 +144,5 @@ class AsyncLaminarClient:
             "Content-Type": "application/json",
             "Accept": "application/json",
         }
+
+

lmnr-0.6.11/src/lmnr/sdk/client/asynchronous/resources/evals.py

@@ -0,0 +1,158 @@
+"""Evals resource for interacting with Laminar evaluations API."""
+
+from typing import Any
+import uuid
+
+from lmnr.sdk.client.asynchronous.resources.base import BaseAsyncResource
+from lmnr.sdk.types import (
+    InitEvaluationResponse,
+    EvaluationResultDatapoint,
+    PartialEvaluationDatapoint,
+)
+
+
+class AsyncEvals(BaseAsyncResource):
+    """Resource for interacting with Laminar evaluations API."""
+
+    async def init(
+        self, name: str | None = None, group_name: str | None = None
+    ) -> InitEvaluationResponse:
+        """Initialize a new evaluation.
+
+        Args:
+            name (str | None, optional): Name of the evaluation. Defaults to None.
+            group_name (str | None, optional): Group name for the evaluation. Defaults to None.
+
+        Returns:
+            InitEvaluationResponse: The response from the initialization request.
+        """
+        response = await self._client.post(
+            self._base_url + "/v1/evals",
+            json={
+                "name": name,
+                "groupName": group_name,
+            },
+            headers=self._headers(),
+        )
+        if response.status_code != 200:
+            if response.status_code == 401:
+                raise ValueError("Unauthorized. Please check your project API key.")
+            raise ValueError(f"Error initializing evaluation: {response.text}")
+        resp_json = response.json()
+        return InitEvaluationResponse.model_validate(resp_json)
+
+    async def create_evaluation(
+        self,
+        name: str | None = None,
+        group_name: str | None = None,
+    ) -> uuid.UUID:
+        """
+        Create a new evaluation and return its ID.
+        
+        Parameters:
+            name (str | None, optional): Optional name of the evaluation.
+            group_name (str | None, optional): An identifier to group evaluations.
+        
+        Returns:
+            uuid.UUID: The evaluation ID.
+        """
+        evaluation = await self.init(name=name, group_name=group_name)
+        return evaluation.id
+
+    async def create_datapoint(
+        self,
+        eval_id: uuid.UUID,
+        data: Any,
+        target: Any = None,
+        metadata: dict[str, Any] | None = None,
+        index: int | None = None,
+        trace_id: uuid.UUID | None = None,
+    ) -> uuid.UUID:
+        """
+        Create a datapoint for an evaluation.
+        
+        Parameters:
+            eval_id (uuid.UUID): The evaluation ID.
+            data: The input data for the executor.
+            target: The target/expected output for evaluators.
+            metadata (dict[str, Any] | None, optional): Optional metadata.
+            index (int | None, optional): Optional index of the datapoint.
+            trace_id (uuid.UUID | None, optional): Optional trace ID.
+        
+        Returns:
+            uuid.UUID: The datapoint ID.
+        """
+        
+        datapoint_id = uuid.uuid4()
+        
+        # Create a minimal datapoint first
+        partial_datapoint = PartialEvaluationDatapoint(
+            id=datapoint_id,
+            data=data,
+            target=target,
+            index=index or 0,
+            trace_id=trace_id or uuid.uuid4(),
+            executor_span_id=uuid.uuid4(),  # Will be updated when executor runs
+            metadata=metadata,
+        )
+        
+        await self.save_datapoints(eval_id, [partial_datapoint])
+        return datapoint_id
+
+    async def save_datapoints(
+        self,
+        eval_id: uuid.UUID,
+        datapoints: list[EvaluationResultDatapoint | PartialEvaluationDatapoint],
+        group_name: str | None = None,
+    ):
+        """Save evaluation datapoints.
+
+        Args:
+            eval_id (uuid.UUID): The evaluation ID.
+            datapoints (list[EvaluationResultDatapoint | PartialEvaluationDatapoint]): The datapoints to save.
+            group_name (str | None, optional): Group name for the datapoints. Defaults to None.
+
+        Raises:
+            ValueError: If there's an error saving the datapoints.
+        """
+        response = await self._client.post(
+            self._base_url + f"/v1/evals/{eval_id}/datapoints",
+            json={
+                "points": [datapoint.to_dict() for datapoint in datapoints],
+                "groupName": group_name,
+            },
+            headers=self._headers(),
+        )
+        if response.status_code != 200:
+            raise ValueError(f"Error saving evaluation datapoints: {response.text}")
+        
+    
+    async def update_datapoint(
+        self,
+        eval_id: uuid.UUID,
+        datapoint_id: uuid.UUID,
+        scores: dict[str, float | int],
+        executor_output: Any | None = None,
+    ) -> None:
+        """Update a datapoint with evaluation results.
+
+        Args:
+            eval_id (uuid.UUID): The evaluation ID.
+            datapoint_id (uuid.UUID): The datapoint ID.
+            executor_output (Any): The executor output.
+            scores (dict[str, float | int] | None, optional): The scores. Defaults to None.
+        """
+        
+        response = await self._client.post(
+            self._base_url + f"/v1/evals/{eval_id}/datapoints/{datapoint_id}",
+            json={
+                "executorOutput": executor_output,
+                "scores": scores,
+            },
+            headers=self._headers(),
+        )
+
+        if response.status_code != 200:
+            raise ValueError(f"Error updating evaluation datapoint: {response.text}")
+        
+    

{lmnr-0.6.9 → lmnr-0.6.11}/src/lmnr/sdk/client/synchronous/resources/evals.py

@@ -2,6 +2,7 @@
 
 import uuid
 import urllib.parse
+from typing import Any
 
 from lmnr.sdk.client.synchronous.resources.base import BaseResource
 from lmnr.sdk.types import (
@@ -35,9 +36,71 @@ class Evals(BaseResource):
             },
             headers=self._headers(),
         )
+        if response.status_code != 200:
+            if response.status_code == 401:
+                raise ValueError("Unauthorized. Please check your project API key.")
+            raise ValueError(f"Error initializing evaluation: {response.text}")
         resp_json = response.json()
         return InitEvaluationResponse.model_validate(resp_json)
 
+    def create_evaluation(
+        self,
+        name: str | None = None,
+        group_name: str | None = None,
+    ) -> uuid.UUID:
+        """
+        Create a new evaluation and return its ID.
+        
+        Parameters:
+            name (str | None, optional): Optional name of the evaluation.
+            group_name (str | None, optional): An identifier to group evaluations.
+        
+        Returns:
+            uuid.UUID: The evaluation ID.
+        """
+        evaluation = self.init(name=name, group_name=group_name)
+        return evaluation.id
+
+    def create_datapoint(
+        self,
+        eval_id: uuid.UUID,
+        data: Any,
+        target: Any = None,
+        metadata: dict[str, Any] | None = None,
+        index: int | None = None,
+        trace_id: uuid.UUID | None = None,
+    ) -> uuid.UUID:
+        """
+        Create a datapoint for an evaluation.
+        
+        Parameters:
+            eval_id (uuid.UUID): The evaluation ID.
+            data: The input data for the executor.
+            target: The target/expected output for evaluators.
+            metadata (dict[str, Any] | None, optional): Optional metadata.
+            index (int | None, optional): Optional index of the datapoint.
+            trace_id (uuid.UUID | None, optional): Optional trace ID.
+        
+        Returns:
+            uuid.UUID: The datapoint ID.
+        """
+        
+        datapoint_id = uuid.uuid4()
+        
+        # Create a minimal datapoint first
+        partial_datapoint = PartialEvaluationDatapoint(
+            id=datapoint_id,
+            data=data,
+            target=target,
+            index=index or 0,
+            trace_id=trace_id or uuid.uuid4(),
+            executor_span_id=uuid.uuid4(),  # Will be updated when executor runs
+            metadata=metadata,
+        )
+        
+        self.save_datapoints(eval_id, [partial_datapoint])
+        return datapoint_id
+
     def save_datapoints(
         self,
         eval_id: uuid.UUID,
@@ -65,6 +128,34 @@ class Evals(BaseResource):
         if response.status_code != 200:
             raise ValueError(f"Error saving evaluation datapoints: {response.text}")
 
+    def update_datapoint(
+        self,
+        eval_id: uuid.UUID,
+        datapoint_id: uuid.UUID,
+        scores: dict[str, float | int],
+        executor_output: Any | None = None,
+    ) -> None:
+        """Update a datapoint with evaluation results.
+
+        Args:
+            eval_id (uuid.UUID): The evaluation ID.
+            datapoint_id (uuid.UUID): The datapoint ID.
+            executor_output (Any): The executor output.
+            scores (dict[str, float | int] | None, optional): The scores. Defaults to None.
+        """
+        
+        response = self._client.post(
+            self._base_url + f"/v1/evals/{eval_id}/datapoints/{datapoint_id}",
+            json={
+                "executorOutput": executor_output,
+                "scores": scores,
+            },
+            headers=self._headers(),
+        )
+
+        if response.status_code != 200:
+            raise ValueError(f"Error updating evaluation datapoint: {response.text}")
+
     def get_datapoints(
         self,
         dataset_name: str,

{lmnr-0.6.9 → lmnr-0.6.11}/src/lmnr/sdk/client/synchronous/sync_client.py

@@ -89,7 +89,7 @@ class LaminarClient:
         return self.__agent
 
     @property
-    def _evals(self) -> Evals:
+    def evals(self) -> Evals:
         """Get the Evals resource.
 
         Returns:
@@ -155,3 +155,5 @@ class LaminarClient:
             "Content-Type": "application/json",
             "Accept": "application/json",
         }
+
+

{lmnr-0.6.9 → lmnr-0.6.11}/src/lmnr/sdk/datasets.py

@@ -38,7 +38,7 @@ class LaminarDataset(EvaluationDataset):
             f"dataset {self.name}. Fetching batch from {self._offset} to "
             + f"{self._offset + self._fetch_size}"
         )
-        resp = self.client._evals.get_datapoints(
+        resp = self.client.evals.get_datapoints(
             self.name, self._offset, self._fetch_size
         )
         self._fetched_items += resp.items

{lmnr-0.6.9 → lmnr-0.6.11}/src/lmnr/sdk/evaluations.py

@@ -199,11 +199,11 @@ class Evaluation:
         self.base_http_url = f"{base_url}:{http_port or 443}"
 
         api_key = project_api_key or from_env("LMNR_PROJECT_API_KEY")
-        if not api_key:
+        if not api_key and not L.is_initialized():
             raise ValueError(
-                "Please initialize the Laminar object with"
-                " your project API key or set the LMNR_PROJECT_API_KEY"
-                " environment variable in your environment or .env file"
+                "Please pass the project API key to `evaluate`"
+                " or set the LMNR_PROJECT_API_KEY environment variable"
+                " in your environment or .env file"
             )
         self.project_api_key = api_key
 
@@ -212,17 +212,12 @@ class Evaluation:
                 base_url=L.get_base_http_url(),
                 project_api_key=L.get_project_api_key(),
             )
-            if project_api_key and project_api_key != L.get_project_api_key():
-                self._logger.warning(
-                    "Project API key is different from the one used to initialize"
-                    " Laminar. Ignoring the project API key passed to the evaluation."
-                )
-            return
+        else:
+            self.client = AsyncLaminarClient(
+                base_url=self.base_http_url,
+                project_api_key=self.project_api_key,
+            )
 
-        self.client = AsyncLaminarClient(
-            base_url=self.base_http_url,
-            project_api_key=self.project_api_key,
-        )
         L.initialize(
             project_api_key=project_api_key,
             base_url=base_url,
@@ -246,7 +241,7 @@ class Evaluation:
             )
         self.reporter.start(len(self.data))
         try:
-            evaluation = await self.client._evals.init(
+            evaluation = await self.client.evals.init(
                 name=self.name, group_name=self.group_name
             )
             result_datapoints = await self._evaluate_in_batches(evaluation.id)
@@ -261,7 +256,7 @@ class Evaluation:
         except Exception as e:
             self.reporter.stopWithError(e)
             await self._shutdown()
-            return {}
+            raise
 
         average_scores = get_average_scores(result_datapoints)
         self.reporter.stop(average_scores, evaluation.projectId, evaluation.id)
@@ -331,7 +326,7 @@ class Evaluation:
                     metadata=datapoint.metadata,
                 )
                 # First, create datapoint with trace_id so that we can show the dp in the UI
-                await self.client._evals.save_datapoints(
+                await self.client.evals.save_datapoints(
                     eval_id, [partial_datapoint], self.group_name
                 )
                 executor_span.set_attribute(SPAN_TYPE, SpanType.EXECUTOR.value)
@@ -389,7 +384,7 @@ class Evaluation:
 
         # Create background upload task without awaiting it
         upload_task = asyncio.create_task(
-            self.client._evals.save_datapoints(eval_id, [datapoint], self.group_name)
+            self.client.evals.save_datapoints(eval_id, [datapoint], self.group_name)
         )
         self.upload_tasks.append(upload_task)
 

{lmnr-0.6.9 → lmnr-0.6.11}/src/lmnr/sdk/laminar.py

@@ -114,6 +114,12 @@ class Laminar:
         Raises:
             ValueError: If project API key is not set
         """
+        if cls.is_initialized():
+            cls.__logger.info(
+                "Laminar is already initialized. Skipping initialization."
+            )
+            return
+
         cls.__project_api_key = project_api_key or from_env("LMNR_PROJECT_API_KEY")
         if not cls.__project_api_key:
             raise ValueError(
@@ -691,6 +697,7 @@ class Laminar:
     def shutdown(cls):
         if cls.is_initialized():
             TracerManager.shutdown()
+            cls.__initialized = False
 
     @classmethod
     def set_span_tags(cls, tags: list[str]):

{lmnr-0.6.9 → lmnr-0.6.11}/src/lmnr/version.py

@@ -3,7 +3,7 @@ import httpx
 from packaging import version
 
 
-__version__ = "0.6.9"
+__version__ = "0.6.11"
 PYTHON_VERSION = f"{sys.version_info.major}.{sys.version_info.minor}"
 
 

lmnr-0.6.9/src/lmnr/opentelemetry_lib/tracing/__init__.py

@@ -1,157 +0,0 @@
-import atexit
-import logging
-
-from lmnr.opentelemetry_lib.tracing.processor import LaminarSpanProcessor
-from lmnr.sdk.client.asynchronous.async_client import AsyncLaminarClient
-from lmnr.sdk.client.synchronous.sync_client import LaminarClient
-from lmnr.sdk.log import VerboseColorfulFormatter
-from lmnr.opentelemetry_lib.tracing.instruments import (
-    Instruments,
-    init_instrumentations,
-)
-
-from opentelemetry import trace
-from opentelemetry.instrumentation.threading import ThreadingInstrumentor
-from opentelemetry.sdk.resources import Resource
-from opentelemetry.sdk.trace import TracerProvider, SpanProcessor
-from opentelemetry.sdk.trace.export import SpanExporter
-
-module_logger = logging.getLogger(__name__)
-console_log_handler = logging.StreamHandler()
-console_log_handler.setFormatter(VerboseColorfulFormatter())
-module_logger.addHandler(console_log_handler)
-
-
-TRACER_NAME = "lmnr.tracer"
-
-MAX_EVENTS_OR_ATTRIBUTES_PER_SPAN = 5000
-
-
-class TracerWrapper(object):
-    resource_attributes: dict = {}
-    enable_content_tracing: bool = True
-    __tracer_provider: TracerProvider | None = None
-    __logger: logging.Logger
-    __client: LaminarClient
-    __async_client: AsyncLaminarClient
-    __resource: Resource
-    __span_processor: SpanProcessor
-
-    def __new__(
-        cls,
-        disable_batch=False,
-        exporter: SpanExporter | None = None,
-        instruments: set[Instruments] | None = None,
-        block_instruments: set[Instruments] | None = None,
-        base_url: str = "https://api.lmnr.ai",
-        port: int = 8443,
-        http_port: int = 443,
-        project_api_key: str | None = None,
-        max_export_batch_size: int | None = None,
-        force_http: bool = False,
-        timeout_seconds: int = 10,
-        set_global_tracer_provider: bool = True,
-        otel_logger_level: int = logging.ERROR,
-    ) -> "TracerWrapper":
-        # Silence some opentelemetry warnings
-        logging.getLogger("opentelemetry.trace").setLevel(otel_logger_level)
-
-        base_http_url = f"{base_url}:{http_port}"
-        cls._initialize_logger(cls)
-        if not hasattr(cls, "instance"):
-            obj = cls.instance = super(TracerWrapper, cls).__new__(cls)
-
-            obj.__client = LaminarClient(
-                base_url=base_http_url,
-                project_api_key=project_api_key,
-            )
-            obj.__async_client = AsyncLaminarClient(
-                base_url=base_http_url,
-                project_api_key=project_api_key,
-            )
-
-            obj.__resource = Resource(attributes=TracerWrapper.resource_attributes)
-
-            obj.__span_processor = LaminarSpanProcessor(
-                base_url=base_url,
-                api_key=project_api_key,
-                port=http_port if force_http else port,
-                exporter=exporter,
-                max_export_batch_size=max_export_batch_size,
-                timeout_seconds=timeout_seconds,
-                force_http=force_http,
-                disable_batch=disable_batch,
-            )
-
-            lmnr_provider = TracerProvider(resource=obj.__resource)
-            global_provider = trace.get_tracer_provider()
-            if set_global_tracer_provider and isinstance(
-                global_provider, trace.ProxyTracerProvider
-            ):
-                trace.set_tracer_provider(lmnr_provider)
-
-            obj.__tracer_provider = lmnr_provider
-
-            obj.__tracer_provider.add_span_processor(obj.__span_processor)
-
-            # This is not a real instrumentation and does not generate telemetry
-            # data, but it is required to ensure that OpenTelemetry context
-            # propagation is enabled.
-            # See the README at:
-            # https://pypi.org/project/opentelemetry-instrumentation-threading/
-            ThreadingInstrumentor().instrument()
-
-            init_instrumentations(
-                tracer_provider=obj.__tracer_provider,
-                instruments=instruments,
-                block_instruments=block_instruments,
-                client=obj.__client,
-                async_client=obj.__async_client,
-            )
-
-            # Force flushes for debug environments (e.g. local development)
-            atexit.register(obj.exit_handler)
-
-        return cls.instance
-
-    def exit_handler(self):
-        if isinstance(self.__span_processor, LaminarSpanProcessor):
-            self.__span_processor.clear()
-        self.flush()
-
-    def _initialize_logger(self):
-        self.__logger = logging.getLogger(__name__)
-        console_log_handler = logging.StreamHandler()
-        console_log_handler.setFormatter(VerboseColorfulFormatter())
-        self.__logger.addHandler(console_log_handler)
-
-    @staticmethod
-    def set_static_params(
-        resource_attributes: dict,
-        enable_content_tracing: bool,
-    ) -> None:
-        TracerWrapper.resource_attributes = resource_attributes
-        TracerWrapper.enable_content_tracing = enable_content_tracing
-
-    @classmethod
-    def verify_initialized(cls) -> bool:
-        return hasattr(cls, "instance")
-
-    @classmethod
-    def clear(cls):
-        # Any state cleanup. Now used in between tests
-        if isinstance(cls.instance.__span_processor, LaminarSpanProcessor):
-            cls.instance.__span_processor.clear()
-
-    def shutdown(self):
-        if self.__tracer_provider is None:
-            return
-        self.__tracer_provider.shutdown()
-
-    def flush(self):
-        return self.__span_processor.force_flush()
-
-    def get_tracer(self):
-        if self.__tracer_provider is None:
-            return trace.get_tracer_provider().get_tracer(TRACER_NAME)
-        return self.__tracer_provider.get_tracer(TRACER_NAME)

lmnr-0.6.9/src/lmnr/sdk/client/asynchronous/resources/evals.py

@@ -1,64 +0,0 @@
-"""Evals resource for interacting with Laminar evaluations API."""
-
-import uuid
-
-from lmnr.sdk.client.asynchronous.resources.base import BaseAsyncResource
-from lmnr.sdk.types import (
-    InitEvaluationResponse,
-    EvaluationResultDatapoint,
-    PartialEvaluationDatapoint,
-)
-
-
-class AsyncEvals(BaseAsyncResource):
-    """Resource for interacting with Laminar evaluations API."""
-
-    async def init(
-        self, name: str | None = None, group_name: str | None = None
-    ) -> InitEvaluationResponse:
-        """Initialize a new evaluation.
-
-        Args:
-            name (str | None, optional): Name of the evaluation. Defaults to None.
-            group_name (str | None, optional): Group name for the evaluation. Defaults to None.
-
-        Returns:
-            InitEvaluationResponse: The response from the initialization request.
-        """
-        response = await self._client.post(
-            self._base_url + "/v1/evals",
-            json={
-                "name": name,
-                "groupName": group_name,
-            },
-            headers=self._headers(),
-        )
-        resp_json = response.json()
-        return InitEvaluationResponse.model_validate(resp_json)
-
-    async def save_datapoints(
-        self,
-        eval_id: uuid.UUID,
-        datapoints: list[EvaluationResultDatapoint | PartialEvaluationDatapoint],
-        group_name: str | None = None,
-    ):
-        """Save evaluation datapoints.
-
-        Args:
-            eval_id (uuid.UUID): The evaluation ID.
-            datapoints (list[EvaluationResultDatapoint | PartialEvaluationDatapoint]): The datapoints to save.
-            group_name (str | None, optional): Group name for the datapoints. Defaults to None.
-
-        Raises:
-            ValueError: If there's an error saving the datapoints.
-        """
-        response = await self._client.post(
-            self._base_url + f"/v1/evals/{eval_id}/datapoints",
-            json={
-                "points": [datapoint.to_dict() for datapoint in datapoints],
-                "groupName": group_name,
-            },
-            headers=self._headers(),
-        )
-        if response.status_code != 200:
-            raise ValueError(f"Error saving evaluation datapoints: {response.text}")