lmnr 0.5.1a0__py3-none-any.whl → 0.5.3__py3-none-any.whl

lmnr/sdk/eval_control.py

@@ -1,4 +1,5 @@
 from contextvars import ContextVar
 
-PREPARE_ONLY = ContextVar("__lmnr_prepare_only", default=False)
-EVALUATION_INSTANCE = ContextVar("__lmnr_evaluation_instance", default=None)
+
+PREPARE_ONLY: ContextVar[bool] = ContextVar("__lmnr_prepare_only", default=False)
+EVALUATION_INSTANCE = ContextVar("__lmnr_evaluation_instance")

lmnr/sdk/evaluations.py

@@ -5,8 +5,8 @@ import dotenv
 from tqdm import tqdm
 from typing import Any, Awaitable, Optional, Set, Union
 
-from lmnr.openllmetry_sdk.instruments import Instruments
-from lmnr.openllmetry_sdk.tracing.attributes import SPAN_TYPE
+from lmnr.opentelemetry_lib.instruments import Instruments
+from lmnr.opentelemetry_lib.tracing.attributes import SPAN_TYPE
 
 from lmnr.sdk.client.asynchronous.async_client import AsyncLaminarClient
 from lmnr.sdk.client.synchronous.sync_client import LaminarClient
@@ -176,7 +176,6 @@ class Evaluation:
 
         base_url = base_url or from_env("LMNR_BASE_URL") or "https://api.lmnr.ai"
 
-        self.is_finished = False
         self.reporter = EvaluationReporter(base_url)
         if isinstance(data, list):
             self.data = [
@@ -196,12 +195,7 @@ class Evaluation:
         self.upload_tasks = []
         self.base_http_url = f"{base_url}:{http_port or 443}"
 
-        api_key = project_api_key
-        if not api_key:
-            dotenv_path = dotenv.find_dotenv(usecwd=True)
-            api_key = dotenv.get_key(
-                dotenv_path=dotenv_path, key_to_get="LMNR_PROJECT_API_KEY"
-            )
+        api_key = project_api_key or from_env("LMNR_PROJECT_API_KEY")
         if not api_key:
             raise ValueError(
                 "Please initialize the Laminar object with"
@@ -225,8 +219,6 @@ class Evaluation:
         )
 
     async def run(self) -> Awaitable[None]:
-        if self.is_finished:
-            raise Exception("Evaluation is already finished")
         return await self._run()
 
     async def _run(self) -> None:
@@ -253,17 +245,19 @@ class Evaluation:
                 self._logger.debug("All upload tasks completed")
         except Exception as e:
             self.reporter.stopWithError(e)
-            self.is_finished = True
             await self._shutdown()
             return
 
         average_scores = get_average_scores(result_datapoints)
         self.reporter.stop(average_scores, evaluation.projectId, evaluation.id)
-        self.is_finished = True
         await self._shutdown()
 
     async def _shutdown(self):
-        L.shutdown()
+        # We use flush() instead of shutdown() because multiple evaluations
+        # can be run sequentially in the same process. `shutdown()` would
+        # close the OTLP exporter and we wouldn't be able to export traces in
+        # the next evaluation.
+        L.flush()
         await self.client.close()
         if isinstance(self.data, LaminarDataset) and self.data.client:
             self.data.client.close()

lmnr/sdk/laminar.py

@@ -1,15 +1,15 @@
 from contextlib import contextmanager
 from contextvars import Context
-from lmnr.openllmetry_sdk import TracerManager
-from lmnr.openllmetry_sdk.instruments import Instruments
-from lmnr.openllmetry_sdk import get_tracer
-from lmnr.openllmetry_sdk.tracing.attributes import (
+from lmnr.opentelemetry_lib import TracerManager
+from lmnr.opentelemetry_lib.instruments import Instruments
+from lmnr.opentelemetry_lib.tracing import get_tracer
+from lmnr.opentelemetry_lib.tracing.attributes import (
     ASSOCIATION_PROPERTIES,
     Attributes,
     SPAN_TYPE,
 )
-from lmnr.openllmetry_sdk.config import MAX_MANUAL_SPAN_PAYLOAD_SIZE
-from lmnr.openllmetry_sdk.decorators.base import json_dumps
+from lmnr.opentelemetry_lib.config import MAX_MANUAL_SPAN_PAYLOAD_SIZE
+from lmnr.opentelemetry_lib.decorators.base import json_dumps
 from opentelemetry import context as context_api, trace
 from opentelemetry.context import attach, detach
 from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
@@ -28,13 +28,13 @@ import os
 import re
 import uuid
 
-from lmnr.openllmetry_sdk.tracing.attributes import (
+from lmnr.opentelemetry_lib.tracing.attributes import (
     SESSION_ID,
     SPAN_INPUT,
     SPAN_OUTPUT,
     TRACE_TYPE,
 )
-from lmnr.openllmetry_sdk.tracing.tracing import (
+from lmnr.opentelemetry_lib.tracing.tracing import (
     get_association_properties,
     remove_association_properties,
     set_association_properties,
@@ -652,8 +652,8 @@ class Laminar:
 
     @classmethod
     def shutdown(cls):
-        cls.__initialized = False
-        return TracerManager.shutdown()
+        if cls.is_initialized():
+            TracerManager.shutdown()
 
     @classmethod
     def set_session(

lmnr/sdk/types.py

@@ -1,118 +1,20 @@
 import logging
 import datetime
 from enum import Enum
-import httpx
 import json
 from opentelemetry.trace import SpanContext, TraceFlags
 import pydantic
 from typing import Any, Awaitable, Callable, Literal, Optional, Union
 import uuid
 
-from .utils import serialize
-
-
-class ChatMessage(pydantic.BaseModel):
-    role: str
-    content: str
-
+import pydantic.alias_generators
 
-class ConditionedValue(pydantic.BaseModel):
-    condition: str
-    value: "NodeInput"
+from .utils import serialize
 
 
 Numeric = Union[int, float]
 NumericTypes = (int, float)  # for use with isinstance
 
-NodeInput = Union[str, list[ChatMessage], ConditionedValue, Numeric, bool]
-PipelineOutput = Union[NodeInput]
-
-
-class PipelineRunRequest(pydantic.BaseModel):
-    inputs: dict[str, NodeInput]
-    pipeline: str
-    env: dict[str, str] = pydantic.Field(default_factory=dict)
-    metadata: dict[str, str] = pydantic.Field(default_factory=dict)
-    stream: bool = pydantic.Field(default=False)
-    parent_span_id: Optional[uuid.UUID] = pydantic.Field(default=None)
-    trace_id: Optional[uuid.UUID] = pydantic.Field(default=None)
-
-    # uuid is not serializable by default, so we need to convert it to a string
-    def to_dict(self):
-        return {
-            "inputs": {
-                k: v.model_dump() if isinstance(v, pydantic.BaseModel) else serialize(v)
-                for k, v in self.inputs.items()
-            },
-            "pipeline": self.pipeline,
-            "env": self.env,
-            "metadata": self.metadata,
-            "stream": self.stream,
-            "parentSpanId": str(self.parent_span_id) if self.parent_span_id else None,
-            "traceId": str(self.trace_id) if self.trace_id else None,
-        }
-
-
-class PipelineRunResponse(pydantic.BaseModel):
-    outputs: dict[str, dict[str, PipelineOutput]]
-    run_id: str
-
-
-class SemanticSearchRequest(pydantic.BaseModel):
-    query: str
-    dataset_id: uuid.UUID
-    limit: Optional[int] = pydantic.Field(default=None)
-    threshold: Optional[float] = pydantic.Field(default=None, ge=0.0, le=1.0)
-
-    def to_dict(self):
-        res = {
-            "query": self.query,
-            "datasetId": str(self.dataset_id),
-        }
-        if self.limit is not None:
-            res["limit"] = self.limit
-        if self.threshold is not None:
-            res["threshold"] = self.threshold
-        return res
-
-
-class SemanticSearchResult(pydantic.BaseModel):
-    dataset_id: uuid.UUID
-    score: float
-    data: dict[str, Any]
-    content: str
-
-
-class SemanticSearchResponse(pydantic.BaseModel):
-    results: list[SemanticSearchResult]
-
-
-class PipelineRunError(Exception):
-    error_code: str
-    error_message: str
-
-    def __init__(self, response: httpx.Response):
-        try:
-            resp_json = response.json()
-            try:
-                resp_dict = dict(resp_json)
-            except Exception:
-                resp_dict = {}
-            self.error_code = resp_dict.get("error_code")
-            self.error_message = resp_dict.get("error_message")
-            super().__init__(self.error_message)
-        except Exception:
-            super().__init__(response.text)
-
-    def __str__(self) -> str:
-        try:
-            return str(
-                {"error_code": self.error_code, "error_message": self.error_message}
-            )
-        except Exception:
-            return super().__str__()
-
-
 EvaluationDatapointData = Any  # non-null, must be JSON-serializable
 EvaluationDatapointTarget = Optional[Any]  # must be JSON-serializable
 EvaluationDatapointMetadata = Optional[Any]  # must be JSON-serializable
@@ -322,52 +224,18 @@ class LaminarSpanContext(pydantic.BaseModel):
 class ModelProvider(str, Enum):
     ANTHROPIC = "anthropic"
     BEDROCK = "bedrock"
-
-
-# class AgentChatMessageContentTextBlock(pydantic.BaseModel):
-#     type: Literal["text"]
-#     text: str
-
-
-# class AgentChatMessageImageUrlBlock(pydantic.BaseModel):
-#     type: Literal["image"]
-#     imageUrl: str
-
-
-# class AgentChatMessageImageBase64Block(pydantic.BaseModel):
-#     type: Literal["image"]
-#     imageB64: str
-
-
-# class AgentChatMessageImageBlock(pydantic.RootModel):
-#     root: Union[AgentChatMessageImageUrlBlock, AgentChatMessageImageBase64Block]
-
-
-# class AgentChatMessageContentBlock(pydantic.RootModel):
-#     root: Union[AgentChatMessageContentTextBlock, AgentChatMessageImageBlock]
-
-
-# class AgentChatMessageContent(pydantic.RootModel):
-#     root: Union[str, list[AgentChatMessageContentBlock]]
-
-
-# class AgentChatMessage(pydantic.BaseModel):
-#     role: str
-#     content: AgentChatMessageContent
-#     name: Optional[str] = None
-#     toolCallId: Optional[str] = None
-#     isStateMessage: bool = False
-
-
-# class AgentState(pydantic.BaseModel):
-# messages: str = pydantic.Field(default="")
-# messages: list[AgentChatMessage] = pydantic.Field(default_factory=list)
-# browser_state: Optional[BrowserState] = None
+    OPENAI = "openai"
+    GEMINI = "gemini"
 
 
 class RunAgentRequest(pydantic.BaseModel):
+    model_config = pydantic.ConfigDict(
+        alias_generator=pydantic.alias_generators.to_camel,
+        populate_by_name=True,
+    )
     prompt: str
-    state: Optional[str] = pydantic.Field(default=None)
+    storage_state: Optional[str] = pydantic.Field(default=None)
+    agent_state: Optional[str] = pydantic.Field(default=None)
     parent_span_context: Optional[str] = pydantic.Field(default=None)
     model_provider: Optional[ModelProvider] = pydantic.Field(default=None)
     model: Optional[str] = pydantic.Field(default=None)
@@ -375,50 +243,98 @@ class RunAgentRequest(pydantic.BaseModel):
     enable_thinking: bool = pydantic.Field(default=True)
     cdp_url: Optional[str] = pydantic.Field(default=None)
     return_screenshots: bool = pydantic.Field(default=False)
-
-    def to_dict(self):
-        result = {
-            "prompt": self.prompt,
-            "stream": self.stream,
-            "enableThinking": self.enable_thinking,
-            "returnScreenshots": self.return_screenshots,
-        }
-        if self.state:
-            result["state"] = self.state
-        if self.parent_span_context:
-            result["parentSpanContext"] = self.parent_span_context
-        if self.model_provider:
-            result["modelProvider"] = self.model_provider.value
-        if self.model:
-            result["model"] = self.model
-        if self.cdp_url:
-            result["cdpUrl"] = self.cdp_url
-        return result
+    return_storage_state: bool = pydantic.Field(default=False)
+    return_agent_state: bool = pydantic.Field(default=False)
+    timeout: Optional[int] = pydantic.Field(default=None)
+    max_steps: Optional[int] = pydantic.Field(default=None)
+    thinking_token_budget: Optional[int] = pydantic.Field(default=None)
+    start_url: Optional[str] = pydantic.Field(default=None)
+    disable_give_control: bool = pydantic.Field(default=False)
+    user_agent: Optional[str] = pydantic.Field(default=None)
 
 
 class ActionResult(pydantic.BaseModel):
-    isDone: bool = pydantic.Field(default=False)
+    model_config = pydantic.ConfigDict(
+        alias_generator=pydantic.alias_generators.to_camel
+    )
+    is_done: bool = pydantic.Field(default=False)
     content: Optional[str] = pydantic.Field(default=None)
     error: Optional[str] = pydantic.Field(default=None)
 
 
 class AgentOutput(pydantic.BaseModel):
+    model_config = pydantic.ConfigDict(
+        alias_generator=pydantic.alias_generators.to_camel
+    )
     result: ActionResult = pydantic.Field(default_factory=ActionResult)
+    # Browser state with data related to auth, such as cookies.
+    # A stringified JSON object.
+    # Only returned if return_storage_state is True.
+    # CAUTION: This object may become large. It also may contain sensitive data.
+    storage_state: Optional[str] = pydantic.Field(default=None)
+    # Agent state with data related to the agent's state, such as the chat history.
+    # A stringified JSON object.
+    # Only returned if return_agent_state is True.
+    # CAUTION: This object is large.
+    agent_state: Optional[str] = pydantic.Field(default=None)
 
 
 class StepChunkContent(pydantic.BaseModel):
-    chunkType: Literal["step"]
-    messageId: uuid.UUID
-    actionResult: ActionResult
-    summary: str
+    model_config = pydantic.ConfigDict(
+        alias_generator=pydantic.alias_generators.to_camel
+    )
+    chunk_type: Literal["step"] = pydantic.Field(default="step")
+    message_id: uuid.UUID = pydantic.Field()
+    action_result: ActionResult = pydantic.Field()
+    summary: str = pydantic.Field()
+    screenshot: Optional[str] = pydantic.Field(default=None)
+
+
+class TimeoutChunkContent(pydantic.BaseModel):
+    """Chunk content to indicate that timeout has been hit. The only difference from a regular step
+    is the chunk type. This is the last chunk in the stream.
+    """
+
+    model_config = pydantic.ConfigDict(
+        alias_generator=pydantic.alias_generators.to_camel
+    )
+    chunk_type: Literal["timeout"] = pydantic.Field(default="timeout")
+    message_id: uuid.UUID = pydantic.Field()
+    summary: str = pydantic.Field()
     screenshot: Optional[str] = pydantic.Field(default=None)
 
 
 class FinalOutputChunkContent(pydantic.BaseModel):
-    chunkType: Literal["finalOutput"]
-    messageId: uuid.UUID
-    content: AgentOutput
+    """Chunk content to indicate that the agent has finished executing. This
+    is the last chunk in the stream.
+    """
+
+    model_config = pydantic.ConfigDict(
+        alias_generator=pydantic.alias_generators.to_camel
+    )
+
+    chunk_type: Literal["finalOutput"] = pydantic.Field(default="finalOutput")
+    message_id: uuid.UUID = pydantic.Field()
+    content: AgentOutput = pydantic.Field()
+
+
+class ErrorChunkContent(pydantic.BaseModel):
+    """Chunk content to indicate that an error has occurred. Typically, this
+    is the last chunk in the stream.
+    """
+
+    model_config = pydantic.ConfigDict(
+        alias_generator=pydantic.alias_generators.to_camel
+    )
+    chunk_type: Literal["error"] = pydantic.Field(default="error")
+    message_id: uuid.UUID = pydantic.Field()
+    error: str = pydantic.Field()
 
 
 class RunAgentResponseChunk(pydantic.RootModel):
-    root: Union[StepChunkContent, FinalOutputChunkContent]
+    root: Union[
+        StepChunkContent,
+        FinalOutputChunkContent,
+        ErrorChunkContent,
+        TimeoutChunkContent,
+    ]

lmnr/sdk/utils.py

@@ -88,13 +88,20 @@ def get_input_from_func_args(
     is_method: bool = False,
     func_args: list[typing.Any] = [],
     func_kwargs: dict[str, typing.Any] = {},
+    ignore_inputs: typing.Optional[list[str]] = None,
 ) -> dict[str, typing.Any]:
     # Remove implicitly passed "self" or "cls" argument for
     # instance or class methods
-    res = func_kwargs.copy()
+    res = {
+        k: v
+        for k, v in func_kwargs.items()
+        if not (ignore_inputs and k in ignore_inputs)
+    }
     for i, k in enumerate(inspect.signature(func).parameters.keys()):
         if is_method and k in ["self", "cls"]:
             continue
+        if ignore_inputs and k in ignore_inputs:
+            continue
         # If param has default value, then it's not present in func args
         if i < len(func_args):
             res[k] = func_args[i]

lmnr/version.py

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

{lmnr-0.5.1a0.dist-info → lmnr-0.5.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lmnr
-Version: 0.5.1a0
+Version: 0.5.3
 Summary: Python SDK for Laminar
 License: Apache-2.0
 Author: lmnr.ai
@@ -41,67 +41,67 @@ Provides-Extra: watsonx
 Provides-Extra: weaviate
 Requires-Dist: argparse (>=1.0)
 Requires-Dist: grpcio (<1.68.0)
-Requires-Dist: httpx (>=0.28.1)
+Requires-Dist: httpx (>=0.25.0)
 Requires-Dist: opentelemetry-api (>=1.31.1)
 Requires-Dist: opentelemetry-exporter-otlp-proto-grpc (>=1.31.1)
 Requires-Dist: opentelemetry-exporter-otlp-proto-http (>=1.31.1)
-Requires-Dist: opentelemetry-instrumentation-alephalpha (>=0.38.12) ; extra == "alephalpha"
-Requires-Dist: opentelemetry-instrumentation-alephalpha (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-anthropic (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-anthropic (>=0.38.12) ; extra == "anthropic"
-Requires-Dist: opentelemetry-instrumentation-bedrock (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-bedrock (>=0.38.12) ; extra == "bedrock"
-Requires-Dist: opentelemetry-instrumentation-chromadb (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-chromadb (>=0.38.12) ; extra == "chromadb"
-Requires-Dist: opentelemetry-instrumentation-cohere (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-cohere (>=0.38.12) ; extra == "cohere"
-Requires-Dist: opentelemetry-instrumentation-google-generativeai (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-google-generativeai (>=0.38.12) ; extra == "google-generativeai"
-Requires-Dist: opentelemetry-instrumentation-groq (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-groq (>=0.38.12) ; extra == "groq"
-Requires-Dist: opentelemetry-instrumentation-haystack (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-haystack (>=0.38.12) ; extra == "haystack"
-Requires-Dist: opentelemetry-instrumentation-lancedb (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-lancedb (>=0.38.12) ; extra == "lancedb"
-Requires-Dist: opentelemetry-instrumentation-langchain (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-langchain (>=0.38.12) ; extra == "langchain"
-Requires-Dist: opentelemetry-instrumentation-llamaindex (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-llamaindex (>=0.38.12) ; extra == "llamaindex"
-Requires-Dist: opentelemetry-instrumentation-marqo (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-marqo (>=0.38.12) ; extra == "marqo"
-Requires-Dist: opentelemetry-instrumentation-milvus (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-milvus (>=0.38.12) ; extra == "milvus"
-Requires-Dist: opentelemetry-instrumentation-mistralai (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-mistralai (>=0.38.12) ; extra == "mistralai"
-Requires-Dist: opentelemetry-instrumentation-ollama (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-ollama (>=0.38.12) ; extra == "ollama"
-Requires-Dist: opentelemetry-instrumentation-openai (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-openai (>=0.38.12) ; extra == "openai"
-Requires-Dist: opentelemetry-instrumentation-pinecone (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-pinecone (>=0.38.12) ; extra == "pinecone"
-Requires-Dist: opentelemetry-instrumentation-qdrant (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-qdrant (>=0.38.12) ; extra == "qdrant"
-Requires-Dist: opentelemetry-instrumentation-replicate (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-replicate (>=0.38.12) ; extra == "replicate"
+Requires-Dist: opentelemetry-instrumentation-alephalpha (>=0.39.2) ; extra == "alephalpha"
+Requires-Dist: opentelemetry-instrumentation-alephalpha (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-anthropic (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-anthropic (>=0.39.2) ; extra == "anthropic"
+Requires-Dist: opentelemetry-instrumentation-bedrock (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-bedrock (>=0.39.2) ; extra == "bedrock"
+Requires-Dist: opentelemetry-instrumentation-chromadb (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-chromadb (>=0.39.2) ; extra == "chromadb"
+Requires-Dist: opentelemetry-instrumentation-cohere (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-cohere (>=0.39.2) ; extra == "cohere"
+Requires-Dist: opentelemetry-instrumentation-google-generativeai (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-google-generativeai (>=0.39.2) ; extra == "google-generativeai"
+Requires-Dist: opentelemetry-instrumentation-groq (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-groq (>=0.39.2) ; extra == "groq"
+Requires-Dist: opentelemetry-instrumentation-haystack (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-haystack (>=0.39.2) ; extra == "haystack"
+Requires-Dist: opentelemetry-instrumentation-lancedb (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-lancedb (>=0.39.2) ; extra == "lancedb"
+Requires-Dist: opentelemetry-instrumentation-langchain (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-langchain (>=0.39.2) ; extra == "langchain"
+Requires-Dist: opentelemetry-instrumentation-llamaindex (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-llamaindex (>=0.39.2) ; extra == "llamaindex"
+Requires-Dist: opentelemetry-instrumentation-marqo (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-marqo (>=0.39.2) ; extra == "marqo"
+Requires-Dist: opentelemetry-instrumentation-milvus (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-milvus (>=0.39.2) ; extra == "milvus"
+Requires-Dist: opentelemetry-instrumentation-mistralai (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-mistralai (>=0.39.2) ; extra == "mistralai"
+Requires-Dist: opentelemetry-instrumentation-ollama (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-ollama (>=0.39.2) ; extra == "ollama"
+Requires-Dist: opentelemetry-instrumentation-openai (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-openai (>=0.39.2) ; extra == "openai"
+Requires-Dist: opentelemetry-instrumentation-pinecone (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-pinecone (>=0.39.2) ; extra == "pinecone"
+Requires-Dist: opentelemetry-instrumentation-qdrant (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-qdrant (>=0.39.2) ; extra == "qdrant"
+Requires-Dist: opentelemetry-instrumentation-replicate (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-replicate (>=0.39.2) ; extra == "replicate"
 Requires-Dist: opentelemetry-instrumentation-requests (>=0.52b0)
-Requires-Dist: opentelemetry-instrumentation-sagemaker (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-sagemaker (>=0.38.12) ; extra == "sagemaker"
+Requires-Dist: opentelemetry-instrumentation-sagemaker (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-sagemaker (>=0.39.2) ; extra == "sagemaker"
 Requires-Dist: opentelemetry-instrumentation-sqlalchemy (>=0.52b0)
 Requires-Dist: opentelemetry-instrumentation-threading (>=0.52b0)
-Requires-Dist: opentelemetry-instrumentation-together (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-together (>=0.38.12) ; extra == "together"
-Requires-Dist: opentelemetry-instrumentation-transformers (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-transformers (>=0.38.12) ; extra == "transformers"
+Requires-Dist: opentelemetry-instrumentation-together (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-together (>=0.39.2) ; extra == "together"
+Requires-Dist: opentelemetry-instrumentation-transformers (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-transformers (>=0.39.2) ; extra == "transformers"
 Requires-Dist: opentelemetry-instrumentation-urllib3 (>=0.52b0)
-Requires-Dist: opentelemetry-instrumentation-vertexai (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-vertexai (>=0.38.12) ; extra == "vertexai"
-Requires-Dist: opentelemetry-instrumentation-watsonx (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-watsonx (>=0.38.12) ; extra == "watsonx"
-Requires-Dist: opentelemetry-instrumentation-weaviate (>=0.38.12) ; extra == "all"
-Requires-Dist: opentelemetry-instrumentation-weaviate (>=0.38.12) ; extra == "weaviate"
+Requires-Dist: opentelemetry-instrumentation-vertexai (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-vertexai (>=0.39.2) ; extra == "vertexai"
+Requires-Dist: opentelemetry-instrumentation-watsonx (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-watsonx (>=0.39.2) ; extra == "watsonx"
+Requires-Dist: opentelemetry-instrumentation-weaviate (>=0.39.2) ; extra == "all"
+Requires-Dist: opentelemetry-instrumentation-weaviate (>=0.39.2) ; extra == "weaviate"
 Requires-Dist: opentelemetry-sdk (>=1.31.1)
 Requires-Dist: opentelemetry-semantic-conventions-ai (>=0.4.2)
-Requires-Dist: pydantic (>=2.0.3)
+Requires-Dist: pydantic (>=2.0.3,<3.0.0)
 Requires-Dist: python-dotenv (>=1.0)
 Requires-Dist: tenacity (>=8.0)
 Requires-Dist: tqdm (>=4.0)
@@ -222,7 +222,7 @@ def handle_user_request(topic: str):
 Laminar allows you to automatically instrument majority of the most popular LLM, Vector DB, database, requests, and other libraries.
 
 If you want to automatically instrument a default set of libraries, then simply do NOT pass `instruments` argument to `.initialize()`.
-See the full list of available instrumentations in the [enum](https://github.com/lmnr-ai/lmnr-python/blob/main/src/lmnr/openllmetry_sdk/instruments.py).
+See the full list of available instrumentations in the [enum](https://github.com/lmnr-ai/lmnr-python/blob/main/src/lmnr/opentelemetry_lib/instruments.py).
 
 If you want to automatically instrument only specific LLM, Vector DB, or other
 calls with OpenTelemetry-compatible instrumentation, then pass the appropriate instruments to `.initialize()`.
@@ -335,9 +335,9 @@ for chunk in client.agent.run(
     prompt="What is the weather in London today?",
     stream=True
 ):
-    if chunk.chunkType == 'step':
+    if chunk.chunk_type == 'step':
         print(chunk.summary)
-    elif chunk.chunkType == 'finalOutput':
+    elif chunk.chunk_type == 'finalOutput':
         print(chunk.content.result.content)
 ```
 
@@ -371,9 +371,9 @@ async for chunk in client.agent.run(
     prompt="What is the weather in London today?",
     stream=True
 ):
-    if chunk.chunkType == 'step':
+    if chunk.chunk_type == 'step':
         print(chunk.summary)
-    elif chunk.chunkType == 'finalOutput':
+    elif chunk.chunk_type == 'finalOutput':
         print(chunk.content.result.content)
 ```