payi 0.1.0a26__py3-none-any.whl → 0.1.0a28__py3-none-any.whl

payi/resources/ingest.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Union, Iterable, Optional
+from typing import Dict, Union, Iterable, Optional
 from datetime import datetime
 
 import httpx
@@ -89,14 +89,14 @@ class IngestResource(SyncAPIResource):
         self,
         *,
         category: str,
-        input: int,
-        output: int,
         resource: str,
+        units: Dict[str, ingest_units_params.Units],
         event_timestamp: Union[str, datetime, None] | NotGiven = NOT_GIVEN,
         provisioned_resource_name: Optional[str] | NotGiven = NOT_GIVEN,
         budget_ids: Union[list[str], None] | NotGiven = NOT_GIVEN,
         request_tags: Union[list[str], None] | NotGiven = NOT_GIVEN,
         experience_id: Union[str, None] | NotGiven = NOT_GIVEN,
+        experience_name: Union[str, None] | NotGiven = NOT_GIVEN,
         user_id: Union[str, None] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -123,6 +123,8 @@ class IngestResource(SyncAPIResource):
           
           request_tags (list[str], optional): The request tags to associate with the request. Defaults to None.
 
+          experience_name (str, optional): The experience name
+          
           experience_id (str, optional): The experience instance id
           
           user_id (str, optional): The user id
@@ -156,6 +158,9 @@ class IngestResource(SyncAPIResource):
             valid_tags = [tag.strip() for tag in request_tags if tag.strip()]
             valid_tags_str = ",".join(valid_tags) if valid_tags else NOT_GIVEN
 
+        if experience_name is None or isinstance(experience_name, NotGiven):
+            experience_name = NOT_GIVEN
+
         if experience_id is None or isinstance(experience_id, NotGiven):
             experience_id = NOT_GIVEN
         
@@ -168,6 +173,7 @@ class IngestResource(SyncAPIResource):
                     "xProxy-Budget-IDs": valid_ids_str,
                     "xProxy-Request-Tags": valid_tags_str,
                     "xProxy-Experience-Id": experience_id,
+                    "xProxy-Experience-Name": experience_name,
                     "xProxy-User-ID": user_id,
                 }
             ),
@@ -178,9 +184,8 @@ class IngestResource(SyncAPIResource):
             body=maybe_transform(
                 {
                     "category": category,
-                    "input": input,
-                    "output": output,
                     "resource": resource,
+                    "units": units,
                     "event_timestamp": event_timestamp,
                     "provisioned_resource_name": provisioned_resource_name,
                 },
@@ -251,16 +256,15 @@ class AsyncIngestResource(AsyncAPIResource):
         self,
         *,
         category: str,
-        input: int,
-        output: int,
         resource: str,
+        units: Dict[str, ingest_units_params.Units],
         event_timestamp: Union[str, datetime, None] | NotGiven = NOT_GIVEN,
         provisioned_resource_name: Optional[str] | NotGiven = NOT_GIVEN,
         budget_ids: Union[list[str], None] | NotGiven = NOT_GIVEN,
         request_tags: Union[list[str], None] | NotGiven = NOT_GIVEN,
+        experience_name: Union[str, None] | NotGiven = NOT_GIVEN,
         experience_id: Union[str, None] | NotGiven = NOT_GIVEN,
         user_id: Union[str, None] | NotGiven = NOT_GIVEN,
-
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
@@ -285,6 +289,8 @@ class AsyncIngestResource(AsyncAPIResource):
           
           request_tags (list[str], optional): The request tags to associate with the request. Defaults to None.
           
+          experience_name (str, optional): The experience name
+          
           experience_id (str, optional): The experience instance id
           
           user_id (str, optional): The user id
@@ -318,6 +324,9 @@ class AsyncIngestResource(AsyncAPIResource):
             valid_tags = [tag.strip() for tag in request_tags if tag.strip()]
             valid_tags_str = ",".join(valid_tags) if valid_tags else NOT_GIVEN
 
+        if experience_name is None or isinstance(experience_name, NotGiven):
+            experience_name = NOT_GIVEN
+
         if experience_id is None or isinstance(experience_id, NotGiven):
             experience_id = NOT_GIVEN
         
@@ -329,6 +338,7 @@ class AsyncIngestResource(AsyncAPIResource):
                 {
                     "xProxy-Budget-IDs": valid_ids_str,
                     "xProxy-Request-Tags": valid_tags_str,
+                    "xProxy-Experience-Name": experience_name,
                     "xProxy-Experience-Id": experience_id,
                     "xProxy-User-ID": user_id,
                 }
@@ -340,9 +350,8 @@ class AsyncIngestResource(AsyncAPIResource):
             body=await async_maybe_transform(
                 {
                     "category": category,
-                    "input": input,
-                    "output": output,
                     "resource": resource,
+                    "units": units,
                     "event_timestamp": event_timestamp,
                     "provisioned_resource_name": provisioned_resource_name,
                 },

payi/resources/price_modifiers.py

@@ -0,0 +1,353 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from ..types import price_modifier_create_params, price_modifier_update_params
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._utils import (
+    maybe_transform,
+    async_maybe_transform,
+)
+from .._compat import cached_property
+from .._resource import SyncAPIResource, AsyncAPIResource
+from .._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from .._base_client import make_request_options
+from ..types.price_modifier import PriceModifier
+from ..types.price_modifier_retrieve_response import PriceModifierRetrieveResponse
+
+__all__ = ["PriceModifiersResource", "AsyncPriceModifiersResource"]
+
+
+class PriceModifiersResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> PriceModifiersResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return the
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/Pay-i/pay-i-python#accessing-raw-response-data-eg-headers
+        """
+        return PriceModifiersResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> PriceModifiersResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/Pay-i/pay-i-python#with_streaming_response
+        """
+        return PriceModifiersResourceWithStreamingResponse(self)
+
+    def create(
+        self,
+        *,
+        billing_model_id: str,
+        category: str,
+        price_modifier: float,
+        resource: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> PriceModifier:
+        """
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/api/v1/price-modifier",
+            body=maybe_transform(
+                {
+                    "billing_model_id": billing_model_id,
+                    "category": category,
+                    "price_modifier": price_modifier,
+                    "resource": resource,
+                },
+                price_modifier_create_params.PriceModifierCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=PriceModifier,
+        )
+
+    def retrieve(
+        self,
+        billing_model_id: str,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> PriceModifierRetrieveResponse:
+        """
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not billing_model_id:
+            raise ValueError(f"Expected a non-empty value for `billing_model_id` but received {billing_model_id!r}")
+        return self._get(
+            f"/api/v1/price-modifier/{billing_model_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=PriceModifierRetrieveResponse,
+        )
+
+    def update(
+        self,
+        *,
+        billing_model_id: str,
+        category: str,
+        price_modifier: float,
+        resource: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> PriceModifier:
+        """
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._put(
+            "/api/v1/price-modifier",
+            body=maybe_transform(
+                {
+                    "billing_model_id": billing_model_id,
+                    "category": category,
+                    "price_modifier": price_modifier,
+                    "resource": resource,
+                },
+                price_modifier_update_params.PriceModifierUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=PriceModifier,
+        )
+
+
+class AsyncPriceModifiersResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncPriceModifiersResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return the
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/Pay-i/pay-i-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncPriceModifiersResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncPriceModifiersResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/Pay-i/pay-i-python#with_streaming_response
+        """
+        return AsyncPriceModifiersResourceWithStreamingResponse(self)
+
+    async def create(
+        self,
+        *,
+        billing_model_id: str,
+        category: str,
+        price_modifier: float,
+        resource: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> PriceModifier:
+        """
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/api/v1/price-modifier",
+            body=await async_maybe_transform(
+                {
+                    "billing_model_id": billing_model_id,
+                    "category": category,
+                    "price_modifier": price_modifier,
+                    "resource": resource,
+                },
+                price_modifier_create_params.PriceModifierCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=PriceModifier,
+        )
+
+    async def retrieve(
+        self,
+        billing_model_id: str,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> PriceModifierRetrieveResponse:
+        """
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not billing_model_id:
+            raise ValueError(f"Expected a non-empty value for `billing_model_id` but received {billing_model_id!r}")
+        return await self._get(
+            f"/api/v1/price-modifier/{billing_model_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=PriceModifierRetrieveResponse,
+        )
+
+    async def update(
+        self,
+        *,
+        billing_model_id: str,
+        category: str,
+        price_modifier: float,
+        resource: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> PriceModifier:
+        """
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._put(
+            "/api/v1/price-modifier",
+            body=await async_maybe_transform(
+                {
+                    "billing_model_id": billing_model_id,
+                    "category": category,
+                    "price_modifier": price_modifier,
+                    "resource": resource,
+                },
+                price_modifier_update_params.PriceModifierUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=PriceModifier,
+        )
+
+
+class PriceModifiersResourceWithRawResponse:
+    def __init__(self, price_modifiers: PriceModifiersResource) -> None:
+        self._price_modifiers = price_modifiers
+
+        self.create = to_raw_response_wrapper(
+            price_modifiers.create,
+        )
+        self.retrieve = to_raw_response_wrapper(
+            price_modifiers.retrieve,
+        )
+        self.update = to_raw_response_wrapper(
+            price_modifiers.update,
+        )
+
+
+class AsyncPriceModifiersResourceWithRawResponse:
+    def __init__(self, price_modifiers: AsyncPriceModifiersResource) -> None:
+        self._price_modifiers = price_modifiers
+
+        self.create = async_to_raw_response_wrapper(
+            price_modifiers.create,
+        )
+        self.retrieve = async_to_raw_response_wrapper(
+            price_modifiers.retrieve,
+        )
+        self.update = async_to_raw_response_wrapper(
+            price_modifiers.update,
+        )
+
+
+class PriceModifiersResourceWithStreamingResponse:
+    def __init__(self, price_modifiers: PriceModifiersResource) -> None:
+        self._price_modifiers = price_modifiers
+
+        self.create = to_streamed_response_wrapper(
+            price_modifiers.create,
+        )
+        self.retrieve = to_streamed_response_wrapper(
+            price_modifiers.retrieve,
+        )
+        self.update = to_streamed_response_wrapper(
+            price_modifiers.update,
+        )
+
+
+class AsyncPriceModifiersResourceWithStreamingResponse:
+    def __init__(self, price_modifiers: AsyncPriceModifiersResource) -> None:
+        self._price_modifiers = price_modifiers
+
+        self.create = async_to_streamed_response_wrapper(
+            price_modifiers.create,
+        )
+        self.retrieve = async_to_streamed_response_wrapper(
+            price_modifiers.retrieve,
+        )
+        self.update = async_to_streamed_response_wrapper(
+            price_modifiers.update,
+        )

payi/types/__init__.py

@@ -5,7 +5,9 @@ from __future__ import annotations
 from .csat import Csat as Csat
 from .cost_data import CostData as CostData
 from .cost_details import CostDetails as CostDetails
+from .billing_model import BillingModel as BillingModel
 from .requests_data import RequestsData as RequestsData
+from .price_modifier import PriceModifier as PriceModifier
 from .budget_response import BudgetResponse as BudgetResponse
 from .ingest_response import IngestResponse as IngestResponse
 from .total_cost_data import TotalCostData as TotalCostData
@@ -25,5 +27,11 @@ from .category_list_response import CategoryListResponse as CategoryListResponse
 from .budget_history_response import BudgetHistoryResponse as BudgetHistoryResponse
 from .category_delete_response import CategoryDeleteResponse as CategoryDeleteResponse
 from .category_resource_response import CategoryResourceResponse as CategoryResourceResponse
+from .billing_model_create_params import BillingModelCreateParams as BillingModelCreateParams
+from .billing_model_list_response import BillingModelListResponse as BillingModelListResponse
+from .billing_model_update_params import BillingModelUpdateParams as BillingModelUpdateParams
+from .price_modifier_create_params import PriceModifierCreateParams as PriceModifierCreateParams
+from .price_modifier_update_params import PriceModifierUpdateParams as PriceModifierUpdateParams
 from .category_list_resources_response import CategoryListResourcesResponse as CategoryListResourcesResponse
+from .price_modifier_retrieve_response import PriceModifierRetrieveResponse as PriceModifierRetrieveResponse
 from .category_delete_resource_response import CategoryDeleteResourceResponse as CategoryDeleteResourceResponse

payi/types/billing_model.py

@@ -0,0 +1,29 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from datetime import datetime
+from typing_extensions import Literal
+
+from .._models import BaseModel
+
+__all__ = ["BillingModel"]
+
+
+class BillingModel(BaseModel):
+    billing_model_id: Optional[str] = None
+
+    created_on: datetime
+
+    default: bool
+
+    name: str
+
+    type: Literal["costplus"]
+
+    updated_on: datetime
+
+    prepaid_amount: Optional[float] = None
+
+    prepaid_max: Optional[float] = None
+
+    threshold: Optional[float] = None

payi/types/billing_model_create_params.py

@@ -0,0 +1,20 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Literal, Required, TypedDict
+
+__all__ = ["BillingModelCreateParams"]
+
+
+class BillingModelCreateParams(TypedDict, total=False):
+    name: Required[str]
+
+    type: Required[Literal["costplus"]]
+
+    prepaid_amount: Optional[float]
+
+    prepaid_max: Optional[float]
+
+    threshold: Optional[float]

payi/types/billing_model_list_response.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+from typing_extensions import TypeAlias
+
+from .billing_model import BillingModel
+
+__all__ = ["BillingModelListResponse"]
+
+BillingModelListResponse: TypeAlias = List[BillingModel]

payi/types/billing_model_update_params.py

@@ -0,0 +1,20 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Literal, Required, TypedDict
+
+__all__ = ["BillingModelUpdateParams"]
+
+
+class BillingModelUpdateParams(TypedDict, total=False):
+    type: Required[Literal["costplus"]]
+
+    name: Optional[str]
+
+    prepaid_amount: Optional[float]
+
+    prepaid_max: Optional[float]
+
+    threshold: Optional[float]

payi/types/budget_create_params.py

@@ -21,8 +21,8 @@ class BudgetCreateParams(TypedDict, total=False):
 
     budget_tags: Optional[List[str]]
 
-    budget_type: Literal["conservative", "liberal"]
-
     cost_basis: Literal["base", "billed"]
 
     currency: Literal["usd"]
+
+    threshold: Optional[float]

payi/types/budget_response.py

@@ -21,8 +21,6 @@ class Budget(BaseModel):
 
     budget_response_type: Literal["block", "allow"]
 
-    budget_type: Literal["conservative", "liberal"]
-
     budget_update_timestamp: datetime
 
     currency: Literal["usd"]
@@ -33,6 +31,8 @@ class Budget(BaseModel):
 
     budget_tags: Optional[List[str]] = None
 
+    threshold: Optional[float] = None
+
 
 class BudgetResponse(BaseModel):
     budget: Budget

payi/types/categories/resource_create_params.py

@@ -2,19 +2,19 @@
 
 from __future__ import annotations
 
-from typing import Union
+from typing import Dict, Union
 from datetime import datetime
 from typing_extensions import Required, Annotated, TypedDict
 
 from ..._utils import PropertyInfo
 
-__all__ = ["ResourceCreateParams"]
+__all__ = ["ResourceCreateParams", "Units"]
 
 
 class ResourceCreateParams(TypedDict, total=False):
     category: Required[str]
 
-    input_price: float
+    units: Required[Dict[str, Units]]
 
     max_input_units: int
 
@@ -22,6 +22,10 @@ class ResourceCreateParams(TypedDict, total=False):
 
     max_total_units: int
 
-    output_price: float
-
     start_timestamp: Annotated[Union[str, datetime, None], PropertyInfo(format="iso8601")]
+
+
+class Units(TypedDict, total=False):
+    input_price: Required[float]
+
+    output_price: Required[float]

payi/types/category_resource_response.py

@@ -1,11 +1,17 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import Optional
+from typing import Dict, Optional
 from datetime import datetime
 
 from .._models import BaseModel
 
-__all__ = ["CategoryResourceResponse"]
+__all__ = ["CategoryResourceResponse", "Units"]
+
+
+class Units(BaseModel):
+    input_price: float
+
+    output_price: float
 
 
 class CategoryResourceResponse(BaseModel):
@@ -17,12 +23,10 @@ class CategoryResourceResponse(BaseModel):
 
     start_timestamp: datetime
 
-    input_price: Optional[float] = None
+    units: Dict[str, Units]
 
     max_input_units: Optional[int] = None
 
     max_output_units: Optional[int] = None
 
     max_total_units: Optional[int] = None
-
-    output_price: Optional[float] = None

payi/types/cost_data.py

@@ -1,7 +1,6 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 
-
 from .._models import BaseModel
 from .cost_details import CostDetails