aeri-python 4.0.0__py3-none-any.whl

aeri/api/commons/types/score_source.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ...core import enum
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class ScoreSource(enum.StrEnum):
+    ANNOTATION = "ANNOTATION"
+    API = "API"
+    EVAL = "EVAL"
+
+    def visit(
+        self,
+        annotation: typing.Callable[[], T_Result],
+        api: typing.Callable[[], T_Result],
+        eval: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is ScoreSource.ANNOTATION:
+            return annotation()
+        if self is ScoreSource.API:
+            return api()
+        if self is ScoreSource.EVAL:
+            return eval()

aeri/api/commons/types/score_v1.py

@@ -0,0 +1,131 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import datetime as dt
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import UniversalBaseModel
+from ...core.serialization import FieldMetadata
+from .score_source import ScoreSource
+
+
+class ScoreV1_Numeric(UniversalBaseModel):
+    data_type: typing_extensions.Annotated[
+        typing.Literal["NUMERIC"], FieldMetadata(alias="dataType")
+    ] = "NUMERIC"
+    value: float
+    id: str
+    trace_id: typing_extensions.Annotated[str, FieldMetadata(alias="traceId")]
+    name: str
+    source: ScoreSource
+    observation_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="observationId")
+    ] = None
+    timestamp: dt.datetime
+    created_at: typing_extensions.Annotated[
+        dt.datetime, FieldMetadata(alias="createdAt")
+    ]
+    updated_at: typing_extensions.Annotated[
+        dt.datetime, FieldMetadata(alias="updatedAt")
+    ]
+    author_user_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="authorUserId")
+    ] = None
+    comment: typing.Optional[str] = None
+    metadata: typing.Any
+    config_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="configId")
+    ] = None
+    queue_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="queueId")
+    ] = None
+    environment: str
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )
+
+
+class ScoreV1_Categorical(UniversalBaseModel):
+    data_type: typing_extensions.Annotated[
+        typing.Literal["CATEGORICAL"], FieldMetadata(alias="dataType")
+    ] = "CATEGORICAL"
+    value: float
+    string_value: typing_extensions.Annotated[str, FieldMetadata(alias="stringValue")]
+    id: str
+    trace_id: typing_extensions.Annotated[str, FieldMetadata(alias="traceId")]
+    name: str
+    source: ScoreSource
+    observation_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="observationId")
+    ] = None
+    timestamp: dt.datetime
+    created_at: typing_extensions.Annotated[
+        dt.datetime, FieldMetadata(alias="createdAt")
+    ]
+    updated_at: typing_extensions.Annotated[
+        dt.datetime, FieldMetadata(alias="updatedAt")
+    ]
+    author_user_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="authorUserId")
+    ] = None
+    comment: typing.Optional[str] = None
+    metadata: typing.Any
+    config_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="configId")
+    ] = None
+    queue_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="queueId")
+    ] = None
+    environment: str
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )
+
+
+class ScoreV1_Boolean(UniversalBaseModel):
+    data_type: typing_extensions.Annotated[
+        typing.Literal["BOOLEAN"], FieldMetadata(alias="dataType")
+    ] = "BOOLEAN"
+    value: float
+    string_value: typing_extensions.Annotated[str, FieldMetadata(alias="stringValue")]
+    id: str
+    trace_id: typing_extensions.Annotated[str, FieldMetadata(alias="traceId")]
+    name: str
+    source: ScoreSource
+    observation_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="observationId")
+    ] = None
+    timestamp: dt.datetime
+    created_at: typing_extensions.Annotated[
+        dt.datetime, FieldMetadata(alias="createdAt")
+    ]
+    updated_at: typing_extensions.Annotated[
+        dt.datetime, FieldMetadata(alias="updatedAt")
+    ]
+    author_user_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="authorUserId")
+    ] = None
+    comment: typing.Optional[str] = None
+    metadata: typing.Any
+    config_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="configId")
+    ] = None
+    queue_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="queueId")
+    ] = None
+    environment: str
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )
+
+
+ScoreV1 = typing_extensions.Annotated[
+    typing.Union[ScoreV1_Numeric, ScoreV1_Categorical, ScoreV1_Boolean],
+    pydantic.Field(discriminator="data_type"),
+]

aeri/api/commons/types/session.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import UniversalBaseModel
+from ...core.serialization import FieldMetadata
+
+
+class Session(UniversalBaseModel):
+    id: str
+    created_at: typing_extensions.Annotated[
+        dt.datetime, FieldMetadata(alias="createdAt")
+    ]
+    project_id: typing_extensions.Annotated[str, FieldMetadata(alias="projectId")]
+    environment: str = pydantic.Field()
+    """
+    The environment from which this session originated.
+    """
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/commons/types/session_with_traces.py

@@ -0,0 +1,15 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from .session import Session
+from .trace import Trace
+
+
+class SessionWithTraces(Session):
+    traces: typing.List[Trace]
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/commons/types/trace.py

@@ -0,0 +1,84 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import UniversalBaseModel
+from ...core.serialization import FieldMetadata
+
+
+class Trace(UniversalBaseModel):
+    id: str = pydantic.Field()
+    """
+    The unique identifier of a trace
+    """
+
+    timestamp: dt.datetime = pydantic.Field()
+    """
+    The timestamp when the trace was created
+    """
+
+    name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The name of the trace
+    """
+
+    input: typing.Optional[typing.Any] = pydantic.Field(default=None)
+    """
+    The input data of the trace. Can be any JSON.
+    """
+
+    output: typing.Optional[typing.Any] = pydantic.Field(default=None)
+    """
+    The output data of the trace. Can be any JSON.
+    """
+
+    session_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="sessionId")
+    ] = pydantic.Field(default=None)
+    """
+    The session identifier associated with the trace
+    """
+
+    release: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The release version of the application when the trace was created
+    """
+
+    version: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The version of the trace
+    """
+
+    user_id: typing_extensions.Annotated[
+        typing.Optional[str], FieldMetadata(alias="userId")
+    ] = pydantic.Field(default=None)
+    """
+    The user identifier associated with the trace
+    """
+
+    metadata: typing.Optional[typing.Any] = pydantic.Field(default=None)
+    """
+    The metadata associated with the trace. Can be any JSON.
+    """
+
+    tags: typing.List[str] = pydantic.Field()
+    """
+    The tags associated with the trace.
+    """
+
+    public: bool = pydantic.Field()
+    """
+    Public traces are accessible via url without login
+    """
+
+    environment: str = pydantic.Field()
+    """
+    The environment from which this trace originated. Can be any lowercase alphanumeric string with hyphens and underscores that does not start with 'aeri'.
+    """
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/commons/types/trace_with_details.py

@@ -0,0 +1,43 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.serialization import FieldMetadata
+from .trace import Trace
+
+
+class TraceWithDetails(Trace):
+    html_path: typing_extensions.Annotated[str, FieldMetadata(alias="htmlPath")] = (
+        pydantic.Field()
+    )
+    """
+    Path of trace in Aeri UI
+    """
+
+    latency: typing.Optional[float] = pydantic.Field(default=None)
+    """
+    Latency of trace in seconds
+    """
+
+    total_cost: typing_extensions.Annotated[
+        typing.Optional[float], FieldMetadata(alias="totalCost")
+    ] = pydantic.Field(default=None)
+    """
+    Cost of trace in USD
+    """
+
+    observations: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    List of observation ids
+    """
+
+    scores: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    List of score ids
+    """
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/commons/types/trace_with_full_details.py

@@ -0,0 +1,45 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.serialization import FieldMetadata
+from .observations_view import ObservationsView
+from .score_v1 import ScoreV1
+from .trace import Trace
+
+
+class TraceWithFullDetails(Trace):
+    html_path: typing_extensions.Annotated[str, FieldMetadata(alias="htmlPath")] = (
+        pydantic.Field()
+    )
+    """
+    Path of trace in Aeri UI
+    """
+
+    latency: typing.Optional[float] = pydantic.Field(default=None)
+    """
+    Latency of trace in seconds
+    """
+
+    total_cost: typing_extensions.Annotated[
+        typing.Optional[float], FieldMetadata(alias="totalCost")
+    ] = pydantic.Field(default=None)
+    """
+    Cost of trace in USD
+    """
+
+    observations: typing.List[ObservationsView] = pydantic.Field()
+    """
+    List of observations
+    """
+
+    scores: typing.List[ScoreV1] = pydantic.Field()
+    """
+    List of scores
+    """
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/commons/types/usage.py

@@ -0,0 +1,59 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import UniversalBaseModel
+from ...core.serialization import FieldMetadata
+
+
+class Usage(UniversalBaseModel):
+    """
+    (Deprecated. Use usageDetails and costDetails instead.) Standard interface for usage and cost
+    """
+
+    input: int = pydantic.Field()
+    """
+    Number of input units (e.g. tokens)
+    """
+
+    output: int = pydantic.Field()
+    """
+    Number of output units (e.g. tokens)
+    """
+
+    total: int = pydantic.Field()
+    """
+    Defaults to input+output if not set
+    """
+
+    unit: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Unit of measurement
+    """
+
+    input_cost: typing_extensions.Annotated[
+        typing.Optional[float], FieldMetadata(alias="inputCost")
+    ] = pydantic.Field(default=None)
+    """
+    USD input cost
+    """
+
+    output_cost: typing_extensions.Annotated[
+        typing.Optional[float], FieldMetadata(alias="outputCost")
+    ] = pydantic.Field(default=None)
+    """
+    USD output cost
+    """
+
+    total_cost: typing_extensions.Annotated[
+        typing.Optional[float], FieldMetadata(alias="totalCost")
+    ] = pydantic.Field(default=None)
+    """
+    USD total cost, defaults to input+output
+    """
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/core/__init__.py

@@ -0,0 +1,111 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .api_error import ApiError
+    from .client_wrapper import AsyncClientWrapper, BaseClientWrapper, SyncClientWrapper
+    from .datetime_utils import serialize_datetime
+    from .file import File, convert_file_dict_to_httpx_tuples, with_content_type
+    from .http_client import AsyncHttpClient, HttpClient
+    from .http_response import AsyncHttpResponse, HttpResponse
+    from .jsonable_encoder import jsonable_encoder
+    from .pydantic_utilities import (
+        IS_PYDANTIC_V2,
+        UniversalBaseModel,
+        UniversalRootModel,
+        parse_obj_as,
+        universal_field_validator,
+        universal_root_validator,
+        update_forward_refs,
+    )
+    from .query_encoder import encode_query
+    from .remove_none_from_dict import remove_none_from_dict
+    from .request_options import RequestOptions
+    from .serialization import FieldMetadata, convert_and_respect_annotation_metadata
+_dynamic_imports: typing.Dict[str, str] = {
+    "ApiError": ".api_error",
+    "AsyncClientWrapper": ".client_wrapper",
+    "AsyncHttpClient": ".http_client",
+    "AsyncHttpResponse": ".http_response",
+    "BaseClientWrapper": ".client_wrapper",
+    "FieldMetadata": ".serialization",
+    "File": ".file",
+    "HttpClient": ".http_client",
+    "HttpResponse": ".http_response",
+    "IS_PYDANTIC_V2": ".pydantic_utilities",
+    "RequestOptions": ".request_options",
+    "SyncClientWrapper": ".client_wrapper",
+    "UniversalBaseModel": ".pydantic_utilities",
+    "UniversalRootModel": ".pydantic_utilities",
+    "convert_and_respect_annotation_metadata": ".serialization",
+    "convert_file_dict_to_httpx_tuples": ".file",
+    "encode_query": ".query_encoder",
+    "jsonable_encoder": ".jsonable_encoder",
+    "parse_obj_as": ".pydantic_utilities",
+    "remove_none_from_dict": ".remove_none_from_dict",
+    "serialize_datetime": ".datetime_utils",
+    "universal_field_validator": ".pydantic_utilities",
+    "universal_root_validator": ".pydantic_utilities",
+    "update_forward_refs": ".pydantic_utilities",
+    "with_content_type": ".file",
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(
+            f"No {attr_name} found in _dynamic_imports for module name -> {__name__}"
+        )
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(
+            f"Failed to import {attr_name} from {module_name}: {e}"
+        ) from e
+    except AttributeError as e:
+        raise AttributeError(
+            f"Failed to get {attr_name} from {module_name}: {e}"
+        ) from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = [
+    "ApiError",
+    "AsyncClientWrapper",
+    "AsyncHttpClient",
+    "AsyncHttpResponse",
+    "BaseClientWrapper",
+    "FieldMetadata",
+    "File",
+    "HttpClient",
+    "HttpResponse",
+    "IS_PYDANTIC_V2",
+    "RequestOptions",
+    "SyncClientWrapper",
+    "UniversalBaseModel",
+    "UniversalRootModel",
+    "convert_and_respect_annotation_metadata",
+    "convert_file_dict_to_httpx_tuples",
+    "encode_query",
+    "jsonable_encoder",
+    "parse_obj_as",
+    "remove_none_from_dict",
+    "serialize_datetime",
+    "universal_field_validator",
+    "universal_root_validator",
+    "update_forward_refs",
+    "with_content_type",
+]

aeri/api/core/api_error.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from typing import Any, Dict, Optional
+
+
+class ApiError(Exception):
+    headers: Optional[Dict[str, str]]
+    status_code: Optional[int]
+    body: Any
+
+    def __init__(
+        self,
+        *,
+        headers: Optional[Dict[str, str]] = None,
+        status_code: Optional[int] = None,
+        body: Any = None,
+    ) -> None:
+        self.headers = headers
+        self.status_code = status_code
+        self.body = body
+
+    def __str__(self) -> str:
+        return f"headers: {self.headers}, status_code: {self.status_code}, body: {self.body}"