lmnr 0.6.2__tar.gz → 0.6.4__tar.gz

{lmnr-0.6.2 → lmnr-0.6.4}/PKG-INFO

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

{lmnr-0.6.2 → lmnr-0.6.4}/pyproject.toml

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

{lmnr-0.6.2 → lmnr-0.6.4}/src/lmnr/cli.py

@@ -4,11 +4,10 @@ import importlib.util
 import os
 import re
 import sys
-from typing import Optional
 
 from lmnr.sdk.evaluations import Evaluation
 
-from .sdk.eval_control import PREPARE_ONLY, EVALUATION_INSTANCE
+from .sdk.eval_control import PREPARE_ONLY, EVALUATION_INSTANCES
 from .sdk.log import get_default_logger
 
 LOG = get_default_logger(__name__)
@@ -38,10 +37,10 @@ async def run_evaluation(args):
     else:
         files = [args.file]
 
-    for file in files:
-        prep_token = PREPARE_ONLY.set(True)
-        LOG.info(f"Running evaluation from {file}")
-        try:
+    prep_token = PREPARE_ONLY.set(True)
+    try:
+        for file in files:
+            LOG.info(f"Running evaluation from {file}")
             file = os.path.abspath(file)
             name = "user_module" + file
 
@@ -55,16 +54,24 @@ async def run_evaluation(args):
             sys.modules[name] = mod
 
             spec.loader.exec_module(mod)
-            evaluation: Optional[Evaluation] = EVALUATION_INSTANCE.get()
-            if evaluation is None:
+            evaluations: list[Evaluation] | None = EVALUATION_INSTANCES.get()
+            if evaluations is None:
                 LOG.warning("Evaluation instance not found")
                 if args.fail_on_error:
                     return
                 continue
 
-            await evaluation.run()
-        finally:
-            PREPARE_ONLY.reset(prep_token)
+            LOG.info(f"Loaded {len(evaluations)} evaluations from {file}")
+
+            for evaluation in evaluations:
+                try:
+                    await evaluation.run()
+                except Exception as e:
+                    LOG.error(f"Error running evaluation: {e}")
+                    if args.fail_on_error:
+                        raise
+    finally:
+        PREPARE_ONLY.reset(prep_token)
 
 
 def cli():

{lmnr-0.6.2 → lmnr-0.6.4}/src/lmnr/opentelemetry_lib/__init__.py

@@ -1,7 +1,6 @@
 import logging
 import sys
 
-from typing import Optional, Set
 from opentelemetry.sdk.trace.export import SpanExporter
 from opentelemetry.sdk.resources import SERVICE_NAME
 
@@ -16,17 +15,17 @@ class TracerManager:
 
     @staticmethod
     def init(
-        app_name: Optional[str] = sys.argv[0],
+        app_name: str | None = sys.argv[0],
         disable_batch=False,
-        exporter: Optional[SpanExporter] = None,
+        exporter: SpanExporter | None = None,
         resource_attributes: dict = {},
-        instruments: Optional[Set[Instruments]] = None,
-        block_instruments: Optional[Set[Instruments]] = 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: Optional[str] = None,
-        max_export_batch_size: Optional[int] = None,
+        project_api_key: str | None = None,
+        max_export_batch_size: int | None = None,
         force_http: bool = False,
         timeout_seconds: int = 30,
         set_global_tracer_provider: bool = True,

{lmnr-0.6.2 → lmnr-0.6.4}/src/lmnr/opentelemetry_lib/decorators/__init__.py

@@ -3,12 +3,16 @@ import json
 import logging
 import pydantic
 import types
-from typing import Any, Literal, Optional, Union
+from typing import Any, Literal
 
 from opentelemetry import trace
 from opentelemetry import context as context_api
 from opentelemetry.trace import Span
 
+from lmnr.opentelemetry_lib.tracing.context_properties import (
+    remove_association_properties,
+    update_association_properties,
+)
 from lmnr.sdk.utils import get_input_from_func_args, is_method
 from lmnr.opentelemetry_lib import MAX_MANUAL_SPAN_PAYLOAD_SIZE
 from lmnr.opentelemetry_lib.tracing.tracer import get_tracer
@@ -37,11 +41,12 @@ def json_dumps(data: dict) -> str:
 
 
 def entity_method(
-    name: Optional[str] = None,
+    name: str | None = None,
     ignore_input: bool = False,
-    ignore_inputs: Optional[list[str]] = None,
+    ignore_inputs: list[str] | None = None,
     ignore_output: bool = False,
-    span_type: Union[Literal["DEFAULT"], Literal["LLM"], Literal["TOOL"]] = "DEFAULT",
+    span_type: Literal["DEFAULT", "LLM", "TOOL"] = "DEFAULT",
+    association_properties: dict[str, Any] | None = None,
 ):
     def decorate(fn):
         @wraps(fn)
@@ -51,6 +56,9 @@ def entity_method(
 
             span_name = name or fn.__name__
 
+            if association_properties is not None:
+                update_association_properties(association_properties)
+
             with get_tracer() as tracer:
                 span = tracer.start_span(span_name, attributes={SPAN_TYPE: span_type})
 
@@ -111,7 +119,8 @@ def entity_method(
 
                 span.end()
                 context_api.detach(ctx_token)
-
+                if association_properties is not None:
+                    remove_association_properties(association_properties)
                 return res
 
         return wrap
@@ -121,11 +130,12 @@ def entity_method(
 
 # Async Decorators
 def aentity_method(
-    name: Optional[str] = None,
+    name: str | None = None,
     ignore_input: bool = False,
-    ignore_inputs: Optional[list[str]] = None,
+    ignore_inputs: list[str] | None = None,
     ignore_output: bool = False,
-    span_type: Union[Literal["DEFAULT"], Literal["LLM"], Literal["TOOL"]] = "DEFAULT",
+    span_type: Literal["DEFAULT", "LLM", "TOOL"] = "DEFAULT",
+    association_properties: dict[str, Any] | None = None,
 ):
     def decorate(fn):
         @wraps(fn)
@@ -135,6 +145,9 @@ def aentity_method(
 
             span_name = name or fn.__name__
 
+            if association_properties is not None:
+                update_association_properties(association_properties)
+
             with get_tracer() as tracer:
                 span = tracer.start_span(span_name, attributes={SPAN_TYPE: span_type})
 
@@ -187,6 +200,8 @@ def aentity_method(
                     pass
 
                 span.end()
+                if association_properties is not None:
+                    remove_association_properties(association_properties)
                 context_api.detach(ctx_token)
 
                 return res

{lmnr-0.6.2 → lmnr-0.6.4}/src/lmnr/opentelemetry_lib/opentelemetry/instrumentation/google_genai/__init__.py

@@ -3,7 +3,7 @@
 from collections import defaultdict
 import logging
 import os
-from typing import AsyncGenerator, Callable, Collection, Generator, Optional
+from typing import AsyncGenerator, Callable, Collection, Generator
 
 from google.genai import types
 
@@ -152,7 +152,7 @@ def _set_request_attributes(span, args, kwargs):
 
     if should_send_prompts():
         i = 0
-        system_instruction: Optional[types.ContentUnion] = config_dict.get(
+        system_instruction: types.ContentUnion | None = config_dict.get(
             "system_instruction"
         )
         if system_instruction:

{lmnr-0.6.2 → lmnr-0.6.4}/src/lmnr/opentelemetry_lib/opentelemetry/instrumentation/google_genai/config.py

@@ -1,9 +1,9 @@
-from typing import Callable, Coroutine, Optional
+from typing import Callable, Coroutine
 
 
 class Config:
     exception_logger = None
-    upload_base64_image: Optional[
-        Callable[[str, str, str, str], Coroutine[None, None, str]]
-    ] = None
+    upload_base64_image: (
+        Callable[[str, str, str, str], Coroutine[None, None, str]] | None
+    ) = None
     convert_image_to_openai_format: bool = True

{lmnr-0.6.2 → lmnr-0.6.4}/src/lmnr/opentelemetry_lib/opentelemetry/instrumentation/google_genai/utils.py

@@ -9,7 +9,7 @@ from google.genai import types
 from google.genai._common import BaseModel
 import pydantic
 from opentelemetry.trace import Span
-from typing import Any, Optional, Union
+from typing import Any
 
 
 def set_span_attribute(span: Span, name: str, value: str):
@@ -44,7 +44,7 @@ def dont_throw(func):
     return wrapper
 
 
-def to_dict(obj: Union[BaseModel, pydantic.BaseModel, dict]) -> dict[str, Any]:
+def to_dict(obj: BaseModel | pydantic.BaseModel | dict) -> dict[str, Any]:
     try:
         if isinstance(obj, BaseModel):
             return obj.model_dump()
@@ -59,11 +59,11 @@ def to_dict(obj: Union[BaseModel, pydantic.BaseModel, dict]) -> dict[str, Any]:
 
 
 def process_content_union(
-    content: Union[types.ContentUnion, types.ContentUnionDict],
-    trace_id: Optional[str] = None,
-    span_id: Optional[str] = None,
+    content: types.ContentUnion | types.ContentUnionDict,
+    trace_id: str | None = None,
+    span_id: str | None = None,
     message_index: int = 0,
-) -> Optional[str]:
+) -> str | None:
     parts = _process_content_union(content, trace_id, span_id, message_index)
     if parts is None:
         return None
@@ -83,11 +83,11 @@ def process_content_union(
 
 
 def _process_content_union(
-    content: Union[types.ContentUnion, types.ContentUnionDict],
-    trace_id: Optional[str] = None,
-    span_id: Optional[str] = None,
+    content: types.ContentUnion | types.ContentUnionDict,
+    trace_id: str | None = None,
+    span_id: str | None = None,
     message_index: int = 0,
-) -> Union[str, list[str], None]:
+) -> str | list[str] | None:
     if isinstance(content, types.Content):
         parts = to_dict(content).get("parts", [])
         return [_process_part(part) for part in parts]
@@ -111,12 +111,12 @@ def _process_content_union(
 
 
 def _process_part_union(
-    content: Union[types.PartDict, types.File, types.Part, str],
-    trace_id: Optional[str] = None,
-    span_id: Optional[str] = None,
+    content: types.PartDict | types.File | types.Part | str,
+    trace_id: str | None = None,
+    span_id: str | None = None,
     message_index: int = 0,
     content_index: int = 0,
-) -> Optional[str]:
+) -> str | None:
     if isinstance(content, str):
         return content
     elif isinstance(content, types.File):
@@ -135,11 +135,11 @@ def _process_part_union(
 
 def _process_part(
     content: types.Part,
-    trace_id: Optional[str] = None,
-    span_id: Optional[str] = None,
+    trace_id: str | None = None,
+    span_id: str | None = None,
     message_index: int = 0,
     content_index: int = 0,
-) -> Optional[str]:
+) -> str | None:
     part_dict = to_dict(content)
     if part_dict.get("text") is not None:
         return part_dict.get("text")
@@ -157,8 +157,8 @@ def _process_part(
 
 
 def role_from_content_union(
-    content: Union[types.ContentUnion, types.ContentUnionDict],
-) -> Optional[str]:
+    content: types.ContentUnion | types.ContentUnionDict,
+) -> str | None:
     if isinstance(content, types.Content):
         return to_dict(content).get("role")
     elif isinstance(content, list) and len(content) > 0:

{lmnr-0.6.2 → lmnr-0.6.4}/src/lmnr/opentelemetry_lib/tracing/__init__.py

@@ -15,7 +15,6 @@ 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
-from typing import Optional, Set
 
 module_logger = logging.getLogger(__name__)
 console_log_handler = logging.StreamHandler()
@@ -31,7 +30,7 @@ MAX_EVENTS_OR_ATTRIBUTES_PER_SPAN = 5000
 class TracerWrapper(object):
     resource_attributes: dict = {}
     enable_content_tracing: bool = True
-    __tracer_provider: Optional[TracerProvider] = None
+    __tracer_provider: TracerProvider | None = None
     __logger: logging.Logger
     __client: LaminarClient
     __async_client: AsyncLaminarClient
@@ -41,14 +40,14 @@ class TracerWrapper(object):
     def __new__(
         cls,
         disable_batch=False,
-        exporter: Optional[SpanExporter] = None,
-        instruments: Optional[Set[Instruments]] = None,
-        block_instruments: Optional[Set[Instruments]] = None,
+        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: Optional[str] = None,
-        max_export_batch_size: Optional[int] = None,
+        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,

{lmnr-0.6.2 → lmnr-0.6.4}/src/lmnr/opentelemetry_lib/tracing/_instrument_initializers.py

@@ -1,6 +1,5 @@
 import abc
 
-from typing import Optional
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 
 from lmnr.opentelemetry_lib.utils.package_check import is_package_installed
@@ -8,12 +7,12 @@ from lmnr.opentelemetry_lib.utils.package_check import is_package_installed
 
 class InstrumentorInitializer(abc.ABC):
     @abc.abstractmethod
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         pass
 
 
 class AlephAlphaInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("aleph_alpha_client"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-alephalpha"):
@@ -25,7 +24,7 @@ class AlephAlphaInstrumentorInitializer(InstrumentorInitializer):
 
 
 class AnthropicInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("anthropic"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-anthropic"):
@@ -39,7 +38,7 @@ class AnthropicInstrumentorInitializer(InstrumentorInitializer):
 
 
 class BedrockInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("boto3"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-bedrock"):
@@ -51,7 +50,7 @@ class BedrockInstrumentorInitializer(InstrumentorInitializer):
 
 
 class BrowserUseInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("browser-use"):
             return None
 
@@ -61,7 +60,7 @@ class BrowserUseInstrumentorInitializer(InstrumentorInitializer):
 
 
 class ChromaInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("chromadb"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-chromadb"):
@@ -73,7 +72,7 @@ class ChromaInstrumentorInitializer(InstrumentorInitializer):
 
 
 class CohereInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("cohere"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-cohere"):
@@ -85,7 +84,7 @@ class CohereInstrumentorInitializer(InstrumentorInitializer):
 
 
 class CrewAIInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("crewai"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-crewai"):
@@ -97,7 +96,7 @@ class CrewAIInstrumentorInitializer(InstrumentorInitializer):
 
 
 class GoogleGenerativeAIInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("google-generativeai"):
             return None
         if not is_package_installed(
@@ -113,7 +112,7 @@ class GoogleGenerativeAIInstrumentorInitializer(InstrumentorInitializer):
 
 
 class GoogleGenAIInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("google-genai"):
             return None
 
@@ -125,7 +124,7 @@ class GoogleGenAIInstrumentorInitializer(InstrumentorInitializer):
 
 
 class GroqInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("groq"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-groq"):
@@ -137,7 +136,7 @@ class GroqInstrumentorInitializer(InstrumentorInitializer):
 
 
 class HaystackInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("haystack"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-haystack"):
@@ -149,7 +148,7 @@ class HaystackInstrumentorInitializer(InstrumentorInitializer):
 
 
 class LanceDBInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("lancedb"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-lancedb"):
@@ -161,7 +160,7 @@ class LanceDBInstrumentorInitializer(InstrumentorInitializer):
 
 
 class LangchainInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("langchain"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-langchain"):
@@ -173,7 +172,7 @@ class LangchainInstrumentorInitializer(InstrumentorInitializer):
 
 
 class LlamaIndexInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not (
             is_package_installed("llama-index") or is_package_installed("llama_index")
         ):
@@ -187,7 +186,7 @@ class LlamaIndexInstrumentorInitializer(InstrumentorInitializer):
 
 
 class MarqoInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("marqo"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-marqo"):
@@ -199,7 +198,7 @@ class MarqoInstrumentorInitializer(InstrumentorInitializer):
 
 
 class MCPInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("mcp"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-mcp"):
@@ -211,7 +210,7 @@ class MCPInstrumentorInitializer(InstrumentorInitializer):
 
 
 class MilvusInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("pymilvus"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-milvus"):
@@ -223,7 +222,7 @@ class MilvusInstrumentorInitializer(InstrumentorInitializer):
 
 
 class MistralInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("mistralai"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-mistralai"):
@@ -235,7 +234,7 @@ class MistralInstrumentorInitializer(InstrumentorInitializer):
 
 
 class OllamaInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("ollama"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-ollama"):
@@ -247,7 +246,7 @@ class OllamaInstrumentorInitializer(InstrumentorInitializer):
 
 
 class OpenAIInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("openai"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-openai"):
@@ -266,7 +265,7 @@ class OpenAIInstrumentorInitializer(InstrumentorInitializer):
 class PatchrightInstrumentorInitializer(InstrumentorInitializer):
     def init_instrumentor(
         self, client, async_client, *args, **kwargs
-    ) -> Optional[BaseInstrumentor]:
+    ) -> BaseInstrumentor | None:
         if not is_package_installed("patchright"):
             return None
 
@@ -276,7 +275,7 @@ class PatchrightInstrumentorInitializer(InstrumentorInitializer):
 
 
 class PineconeInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("pinecone"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-pinecone"):
@@ -290,7 +289,7 @@ class PineconeInstrumentorInitializer(InstrumentorInitializer):
 class PlaywrightInstrumentorInitializer(InstrumentorInitializer):
     def init_instrumentor(
         self, client, async_client, *args, **kwargs
-    ) -> Optional[BaseInstrumentor]:
+    ) -> BaseInstrumentor | None:
         if not is_package_installed("playwright"):
             return None
 
@@ -300,7 +299,7 @@ class PlaywrightInstrumentorInitializer(InstrumentorInitializer):
 
 
 class QdrantInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("qdrant_client"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-qdrant"):
@@ -312,7 +311,7 @@ class QdrantInstrumentorInitializer(InstrumentorInitializer):
 
 
 class ReplicateInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("replicate"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-replicate"):
@@ -324,7 +323,7 @@ class ReplicateInstrumentorInitializer(InstrumentorInitializer):
 
 
 class SageMakerInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("boto3"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-sagemaker"):
@@ -336,7 +335,7 @@ class SageMakerInstrumentorInitializer(InstrumentorInitializer):
 
 
 class TogetherInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("together"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-together"):
@@ -348,7 +347,7 @@ class TogetherInstrumentorInitializer(InstrumentorInitializer):
 
 
 class TransformersInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("transformers"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-transformers"):
@@ -360,7 +359,7 @@ class TransformersInstrumentorInitializer(InstrumentorInitializer):
 
 
 class VertexAIInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("vertexai"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-vertexai"):
@@ -372,7 +371,7 @@ class VertexAIInstrumentorInitializer(InstrumentorInitializer):
 
 
 class WatsonxInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not (
             is_package_installed("ibm-watsonx-ai")
             or is_package_installed("ibm-watson-machine-learning")
@@ -387,7 +386,7 @@ class WatsonxInstrumentorInitializer(InstrumentorInitializer):
 
 
 class WeaviateInstrumentorInitializer(InstrumentorInitializer):
-    def init_instrumentor(self, *args, **kwargs) -> Optional[BaseInstrumentor]:
+    def init_instrumentor(self, *args, **kwargs) -> BaseInstrumentor | None:
         if not is_package_installed("weaviate"):
             return None
         if not is_package_installed("opentelemetry-instrumentation-weaviate"):

{lmnr-0.6.2 → lmnr-0.6.4}/src/lmnr/opentelemetry_lib/tracing/context_properties.py

@@ -1,5 +1,4 @@
 import copy
-from typing import Optional
 
 from lmnr.opentelemetry_lib.tracing.attributes import (
     ASSOCIATION_PROPERTIES,
@@ -18,14 +17,14 @@ def set_association_properties(properties: dict) -> None:
     _set_association_properties_attributes(span, properties)
 
 
-def get_association_properties(context: Optional[Context] = None) -> dict:
+def get_association_properties(context: Context | None = None) -> dict:
     return get_value("association_properties", context) or {}
 
 
 def update_association_properties(
     properties: dict,
     set_on_current_span: bool = True,
-    context: Optional[Context] = None,
+    context: Context | None = None,
 ) -> None:
     """Only adds or updates properties that are not already present"""
     association_properties = get_value("association_properties", context) or {}