lark-billing 0.0.6__py3-none-any.whl → 0.0.7__py3-none-any.whl

lark/pricing_metrics/client.py

@@ -0,0 +1,219 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+from ..types.create_pricing_metric_response import CreatePricingMetricResponse
+from ..types.get_pricing_metric_response import GetPricingMetricResponse
+from .raw_client import AsyncRawPricingMetricsClient, RawPricingMetricsClient
+from .types.pricing_metric_aggregation import PricingMetricAggregation
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class PricingMetricsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawPricingMetricsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawPricingMetricsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawPricingMetricsClient
+        """
+        return self._raw_client
+
+    def create_pricing_metric(
+        self,
+        *,
+        name: str,
+        event_name: str,
+        aggregation: PricingMetricAggregation,
+        unit: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> CreatePricingMetricResponse:
+        """
+        Parameters
+        ----------
+        name : str
+
+        event_name : str
+
+        aggregation : PricingMetricAggregation
+
+        unit : str
+            Unit of measurement for the pricing metric
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CreatePricingMetricResponse
+            Successful Response
+
+        Examples
+        --------
+        from lark import Lark
+        from lark.pricing_metrics import PricingMetricAggregation_Count
+
+        client = Lark(
+            api_key="YOUR_API_KEY",
+        )
+        client.pricing_metrics.create_pricing_metric(
+            name="name",
+            event_name="event_name",
+            aggregation=PricingMetricAggregation_Count(),
+            unit="unit",
+        )
+        """
+        _response = self._raw_client.create_pricing_metric(
+            name=name, event_name=event_name, aggregation=aggregation, unit=unit, request_options=request_options
+        )
+        return _response.data
+
+    def get_pricing_metric(
+        self, pricing_metric_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> GetPricingMetricResponse:
+        """
+        Parameters
+        ----------
+        pricing_metric_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        GetPricingMetricResponse
+            Successful Response
+
+        Examples
+        --------
+        from lark import Lark
+
+        client = Lark(
+            api_key="YOUR_API_KEY",
+        )
+        client.pricing_metrics.get_pricing_metric(
+            pricing_metric_id="pricing_metric_id",
+        )
+        """
+        _response = self._raw_client.get_pricing_metric(pricing_metric_id, request_options=request_options)
+        return _response.data
+
+
+class AsyncPricingMetricsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawPricingMetricsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawPricingMetricsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawPricingMetricsClient
+        """
+        return self._raw_client
+
+    async def create_pricing_metric(
+        self,
+        *,
+        name: str,
+        event_name: str,
+        aggregation: PricingMetricAggregation,
+        unit: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> CreatePricingMetricResponse:
+        """
+        Parameters
+        ----------
+        name : str
+
+        event_name : str
+
+        aggregation : PricingMetricAggregation
+
+        unit : str
+            Unit of measurement for the pricing metric
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CreatePricingMetricResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from lark import AsyncLark
+        from lark.pricing_metrics import PricingMetricAggregation_Count
+
+        client = AsyncLark(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.pricing_metrics.create_pricing_metric(
+                name="name",
+                event_name="event_name",
+                aggregation=PricingMetricAggregation_Count(),
+                unit="unit",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create_pricing_metric(
+            name=name, event_name=event_name, aggregation=aggregation, unit=unit, request_options=request_options
+        )
+        return _response.data
+
+    async def get_pricing_metric(
+        self, pricing_metric_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> GetPricingMetricResponse:
+        """
+        Parameters
+        ----------
+        pricing_metric_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        GetPricingMetricResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from lark import AsyncLark
+
+        client = AsyncLark(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.pricing_metrics.get_pricing_metric(
+                pricing_metric_id="pricing_metric_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_pricing_metric(pricing_metric_id, request_options=request_options)
+        return _response.data

lark/pricing_metrics/raw_client.py

@@ -0,0 +1,268 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.http_response import AsyncHttpResponse, HttpResponse
+from ..core.jsonable_encoder import jsonable_encoder
+from ..core.pydantic_utilities import parse_obj_as
+from ..core.request_options import RequestOptions
+from ..core.serialization import convert_and_respect_annotation_metadata
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..types.create_pricing_metric_response import CreatePricingMetricResponse
+from ..types.get_pricing_metric_response import GetPricingMetricResponse
+from ..types.http_validation_error import HttpValidationError
+from .types.pricing_metric_aggregation import PricingMetricAggregation
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class RawPricingMetricsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def create_pricing_metric(
+        self,
+        *,
+        name: str,
+        event_name: str,
+        aggregation: PricingMetricAggregation,
+        unit: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[CreatePricingMetricResponse]:
+        """
+        Parameters
+        ----------
+        name : str
+
+        event_name : str
+
+        aggregation : PricingMetricAggregation
+
+        unit : str
+            Unit of measurement for the pricing metric
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[CreatePricingMetricResponse]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "pricing-metrics",
+            method="POST",
+            json={
+                "name": name,
+                "event_name": event_name,
+                "aggregation": convert_and_respect_annotation_metadata(
+                    object_=aggregation, annotation=PricingMetricAggregation, direction="write"
+                ),
+                "unit": unit,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    CreatePricingMetricResponse,
+                    parse_obj_as(
+                        type_=CreatePricingMetricResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        HttpValidationError,
+                        parse_obj_as(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def get_pricing_metric(
+        self, pricing_metric_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[GetPricingMetricResponse]:
+        """
+        Parameters
+        ----------
+        pricing_metric_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[GetPricingMetricResponse]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"pricing-metrics/{jsonable_encoder(pricing_metric_id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    GetPricingMetricResponse,
+                    parse_obj_as(
+                        type_=GetPricingMetricResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        HttpValidationError,
+                        parse_obj_as(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+
+class AsyncRawPricingMetricsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def create_pricing_metric(
+        self,
+        *,
+        name: str,
+        event_name: str,
+        aggregation: PricingMetricAggregation,
+        unit: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[CreatePricingMetricResponse]:
+        """
+        Parameters
+        ----------
+        name : str
+
+        event_name : str
+
+        aggregation : PricingMetricAggregation
+
+        unit : str
+            Unit of measurement for the pricing metric
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[CreatePricingMetricResponse]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "pricing-metrics",
+            method="POST",
+            json={
+                "name": name,
+                "event_name": event_name,
+                "aggregation": convert_and_respect_annotation_metadata(
+                    object_=aggregation, annotation=PricingMetricAggregation, direction="write"
+                ),
+                "unit": unit,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    CreatePricingMetricResponse,
+                    parse_obj_as(
+                        type_=CreatePricingMetricResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        HttpValidationError,
+                        parse_obj_as(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def get_pricing_metric(
+        self, pricing_metric_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[GetPricingMetricResponse]:
+        """
+        Parameters
+        ----------
+        pricing_metric_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[GetPricingMetricResponse]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"pricing-metrics/{jsonable_encoder(pricing_metric_id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    GetPricingMetricResponse,
+                    parse_obj_as(
+                        type_=GetPricingMetricResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        HttpValidationError,
+                        parse_obj_as(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

lark/pricing_metrics/types/__init__.py

@@ -0,0 +1,42 @@
+# 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 .pricing_metric_aggregation import (
+        PricingMetricAggregation,
+        PricingMetricAggregation_Count,
+        PricingMetricAggregation_Sum,
+    )
+_dynamic_imports: typing.Dict[str, str] = {
+    "PricingMetricAggregation": ".pricing_metric_aggregation",
+    "PricingMetricAggregation_Count": ".pricing_metric_aggregation",
+    "PricingMetricAggregation_Sum": ".pricing_metric_aggregation",
+}
+
+
+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__ = ["PricingMetricAggregation", "PricingMetricAggregation_Count", "PricingMetricAggregation_Sum"]

lark/{types/credit_grant_resource_input_schedule.py → pricing_metrics/types/pricing_metric_aggregation.py}

@@ -5,11 +5,11 @@ from __future__ import annotations
 import typing
 
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
-class CreditGrantResourceInputSchedule_OneTime(UniversalBaseModel):
-    schedule_type: typing.Literal["one_time"] = "one_time"
+class PricingMetricAggregation_Count(UniversalBaseModel):
+    aggregation_type: typing.Literal["count"] = "count"
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
@@ -21,8 +21,9 @@ class CreditGrantResourceInputSchedule_OneTime(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-class CreditGrantResourceInputSchedule_RateCycleStart(UniversalBaseModel):
-    schedule_type: typing.Literal["rate_cycle_start"] = "rate_cycle_start"
+class PricingMetricAggregation_Sum(UniversalBaseModel):
+    aggregation_type: typing.Literal["sum"] = "sum"
+    value_field: str
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
@@ -34,6 +35,4 @@ class CreditGrantResourceInputSchedule_RateCycleStart(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-CreditGrantResourceInputSchedule = typing.Union[
-    CreditGrantResourceInputSchedule_OneTime, CreditGrantResourceInputSchedule_RateCycleStart
-]
+PricingMetricAggregation = typing.Union[PricingMetricAggregation_Count, PricingMetricAggregation_Sum]

lark/rate_cards/__init__.py

@@ -6,26 +6,8 @@ import typing
 from importlib import import_module
 
 if typing.TYPE_CHECKING:
-    from .types import (
-        CreateRateCardRequestBillingInterval,
-        CreateRateCardRequestUsageBasedRatesItem,
-        CreateRateCardRequestUsageBasedRatesItem_Dimensional,
-        CreateRateCardRequestUsageBasedRatesItem_Simple,
-        CreateRateCardVersionRequestUsageBasedRatesItem,
-        CreateRateCardVersionRequestUsageBasedRatesItem_Dimensional,
-        CreateRateCardVersionRequestUsageBasedRatesItem_Simple,
-        CreateRateCardVersionRequestVersionType,
-    )
-_dynamic_imports: typing.Dict[str, str] = {
-    "CreateRateCardRequestBillingInterval": ".types",
-    "CreateRateCardRequestUsageBasedRatesItem": ".types",
-    "CreateRateCardRequestUsageBasedRatesItem_Dimensional": ".types",
-    "CreateRateCardRequestUsageBasedRatesItem_Simple": ".types",
-    "CreateRateCardVersionRequestUsageBasedRatesItem": ".types",
-    "CreateRateCardVersionRequestUsageBasedRatesItem_Dimensional": ".types",
-    "CreateRateCardVersionRequestUsageBasedRatesItem_Simple": ".types",
-    "CreateRateCardVersionRequestVersionType": ".types",
-}
+    from .types import CreateRateCardRequestBillingInterval
+_dynamic_imports: typing.Dict[str, str] = {"CreateRateCardRequestBillingInterval": ".types"}
 
 
 def __getattr__(attr_name: str) -> typing.Any:
@@ -49,13 +31,4 @@ def __dir__():
     return sorted(lazy_attrs)
 
 
-__all__ = [
-    "CreateRateCardRequestBillingInterval",
-    "CreateRateCardRequestUsageBasedRatesItem",
-    "CreateRateCardRequestUsageBasedRatesItem_Dimensional",
-    "CreateRateCardRequestUsageBasedRatesItem_Simple",
-    "CreateRateCardVersionRequestUsageBasedRatesItem",
-    "CreateRateCardVersionRequestUsageBasedRatesItem_Dimensional",
-    "CreateRateCardVersionRequestUsageBasedRatesItem_Simple",
-    "CreateRateCardVersionRequestVersionType",
-]
+__all__ = ["CreateRateCardRequestBillingInterval"]