lmnr 0.6.3__py3-none-any.whl → 0.6.5__py3-none-any.whl

lmnr/cli.py

@@ -4,7 +4,6 @@ import importlib.util
 import os
 import re
 import sys
-from typing import Optional
 
 from lmnr.sdk.evaluations import Evaluation
 
@@ -55,7 +54,7 @@ async def run_evaluation(args):
             sys.modules[name] = mod
 
             spec.loader.exec_module(mod)
-            evaluations: Optional[list[Evaluation]] = EVALUATION_INSTANCES.get()
+            evaluations: list[Evaluation] | None = EVALUATION_INSTANCES.get()
             if evaluations is None:
                 LOG.warning("Evaluation instance not found")
                 if args.fail_on_error:

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/opentelemetry_lib/decorators/__init__.py

@@ -3,7 +3,7 @@ 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
@@ -41,12 +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",
-    association_properties: Optional[dict[str, Any]] = None,
+    span_type: Literal["DEFAULT", "LLM", "TOOL"] = "DEFAULT",
+    association_properties: dict[str, Any] | None = None,
 ):
     def decorate(fn):
         @wraps(fn)
@@ -130,12 +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",
-    association_properties: Optional[dict[str, Any]] = None,
+    span_type: Literal["DEFAULT", "LLM", "TOOL"] = "DEFAULT",
+    association_properties: dict[str, Any] | None = None,
 ):
     def decorate(fn):
         @wraps(fn)

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/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/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/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/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,19 +198,19 @@ 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"):
             return None
 
-        from opentelemetry.instrumentation.mcp import MCPInstrumentor
+        from opentelemetry.instrumentation.mcp import McpInstrumentor
 
-        return MCPInstrumentor()
+        return McpInstrumentor()
 
 
 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/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 {}

lmnr/opentelemetry_lib/tracing/exporter.py

@@ -9,19 +9,18 @@ from opentelemetry.exporter.otlp.proto.http import Compression
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
     OTLPSpanExporter as HTTPOTLPSpanExporter,
 )
-from typing import Optional, Union
 
 from lmnr.sdk.utils import from_env
 
 
 class LaminarSpanExporter(SpanExporter):
-    instance: Union[OTLPSpanExporter, HTTPOTLPSpanExporter]
+    instance: OTLPSpanExporter | HTTPOTLPSpanExporter
 
     def __init__(
         self,
-        base_url: Optional[str] = None,
-        port: Optional[int] = None,
-        api_key: Optional[str] = None,
+        base_url: str | None = None,
+        port: int | None = None,
+        api_key: str | None = None,
         timeout_seconds: int = 30,
         force_http: bool = False,
     ):

lmnr/opentelemetry_lib/tracing/instruments.py

@@ -1,7 +1,6 @@
 import logging
 
 from enum import Enum
-from typing import Optional, Set, Dict
 
 from opentelemetry.trace import TracerProvider
 import lmnr.opentelemetry_lib.tracing._instrument_initializers as initializers
@@ -47,7 +46,7 @@ class Instruments(Enum):
     WEAVIATE = "weaviate"
 
 
-INSTRUMENTATION_INITIALIZERS: Dict[
+INSTRUMENTATION_INITIALIZERS: dict[
     Instruments, initializers.InstrumentorInitializer
 ] = {
     Instruments.ALEPHALPHA: initializers.AlephAlphaInstrumentorInitializer(),
@@ -86,10 +85,10 @@ INSTRUMENTATION_INITIALIZERS: Dict[
 
 def init_instrumentations(
     tracer_provider: TracerProvider,
-    instruments: Optional[Set[Instruments]] = None,
-    block_instruments: Optional[Set[Instruments]] = None,
-    client: Optional[LaminarClient] = None,
-    async_client: Optional[AsyncLaminarClient] = None,
+    instruments: set[Instruments] | None = None,
+    block_instruments: set[Instruments] | None = None,
+    client: LaminarClient | None = None,
+    async_client: AsyncLaminarClient | None = None,
 ):
     block_instruments = block_instruments or set()
     if instruments is None:

lmnr/opentelemetry_lib/tracing/processor.py

@@ -1,6 +1,5 @@
 import uuid
 
-from typing import Optional, Union
 from opentelemetry.sdk.trace.export import (
     SpanProcessor,
     SpanExporter,
@@ -25,20 +24,20 @@ from lmnr.version import PYTHON_VERSION, __version__
 
 
 class LaminarSpanProcessor(SpanProcessor):
-    instance: Union[BatchSpanProcessor, SimpleSpanProcessor]
+    instance: BatchSpanProcessor | SimpleSpanProcessor
     __span_id_to_path: dict[int, list[str]] = {}
     __span_id_lists: dict[int, list[str]] = {}
 
     def __init__(
         self,
-        base_url: Optional[str] = None,
-        port: Optional[int] = None,
-        api_key: Optional[str] = None,
+        base_url: str | None = None,
+        port: int | None = None,
+        api_key: str | None = None,
         timeout_seconds: int = 30,
         force_http: bool = False,
         max_export_batch_size: int = 512,
         disable_batch: bool = False,
-        exporter: Optional[SpanExporter] = None,
+        exporter: SpanExporter | None = None,
     ):
         self.exporter = exporter or LaminarSpanExporter(
             base_url=base_url,
@@ -55,7 +54,7 @@ class LaminarSpanProcessor(SpanProcessor):
             )
         )
 
-    def on_start(self, span: Span, parent_context: Optional[Context] = None):
+    def on_start(self, span: Span, parent_context: Context | None = None):
         span_path_in_context = get_value("span_path", parent_context or get_current())
         parent_span_path = span_path_in_context or (
             self.__span_id_to_path.get(span.parent.span_id) if span.parent else None