payi 0.1.0a13__py3-none-any.whl → 0.1.0a15__py3-none-any.whl

payi/resources/ingest.py

@@ -2,9 +2,12 @@
 
 from __future__ import annotations
 
+from typing import Dict, Union, Iterable, cast
+from datetime import datetime
+
 import httpx
 
-from ..types import ingest_units_params
+from ..types import ingest_bulk_params, ingest_units_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
 from .._utils import (
     maybe_transform,
@@ -20,7 +23,9 @@ from .._response import (
     async_to_streamed_response_wrapper,
 )
 from .._base_client import make_request_options
-from ..types.proxy_result import ProxyResult
+from ..types.ingest_response import IngestResponse
+from ..types.ingest_units_param import IngestUnitsParam
+from ..types.bulk_ingest_response import BulkIngestResponse
 
 __all__ = ["IngestResource", "AsyncIngestResource"]
 
@@ -34,6 +39,46 @@ class IngestResource(SyncAPIResource):
     def with_streaming_response(self) -> IngestResourceWithStreamingResponse:
         return IngestResourceWithStreamingResponse(self)
 
+    def bulk(
+        self,
+        *,
+        items: Iterable[IngestUnitsParam],
+        # 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,
+    ) -> BulkIngestResponse:
+        """
+        Bulk Ingest
+
+        Args:
+
+          items (Iterable[IngestUnitsParams]): The items to ingest
+
+          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/ingest/bulk",
+            body=cast(Dict[str, object], maybe_transform(
+            { 
+                "items": items, 
+            }, 
+            ingest_bulk_params.IngestBulkParams)
+            )["items"],
+            options=make_request_options(
+                    extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=BulkIngestResponse,
+        )
+
     def units(
         self,
         *,
@@ -41,34 +86,46 @@ class IngestResource(SyncAPIResource):
         input: int,
         output: int,
         resource: str,
-        budget_ids: list[str] | NotGiven = NOT_GIVEN,
-        request_tags: list[str] | NotGiven = NOT_GIVEN,
+        event_timestamp: Union[str, datetime, None] | NotGiven = NOT_GIVEN,
+        budget_ids: Union[list[str], None] | NotGiven = NOT_GIVEN,
+        request_tags: Union[list[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.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ProxyResult:
+    ) -> IngestResponse:
         """
         Ingest a request
 
         Args:
           category (str): The name of the category
-          resource (str): The name of the resource.
+
+          resource (str): The name of the resource
+          
           input (int): The number of input units
+
           output (int): The number of output units
+          
+          event_timestamp: (str, datetime, None): The timestamp of the event. Defaults to None.
+          
           budget_ids (list[str], optional): The budget IDs to associate with the request. Defaults to None.
+          
           request_tags (list[str], optional): The request tags to associate with the request. Defaults to None.
+          
           extra_headers (Dict[str, str], optional): Additional headers for the request. Defaults to None.
+          
           extra_query (Dict[str, str], optional): Additional query parameters. Defaults to None.
+          
           extra_body (Dict[str, Any], optional): Additional body parameters. Defaults to None.
+          
           timeout (Union[float, None], optional): The timeout for the request in seconds. Defaults to None.
         """
         valid_ids_str: str | NotGiven = NOT_GIVEN
         valid_tags_str: str | NotGiven = NOT_GIVEN
 
-        if isinstance(budget_ids, NotGiven):
+        if budget_ids is None or isinstance(budget_ids, NotGiven):
             valid_ids_str = NOT_GIVEN
         elif not isinstance(budget_ids, list): # type: ignore
             raise TypeError("budget_ids must be a list")
@@ -77,7 +134,7 @@ class IngestResource(SyncAPIResource):
             valid_ids = [id.strip() for id in budget_ids if id.strip()]
             valid_ids_str = ",".join(valid_ids) if valid_ids else NOT_GIVEN
 
-        if isinstance(request_tags, NotGiven):
+        if request_tags is None or isinstance(request_tags, NotGiven):
             valid_tags_str = NOT_GIVEN
         elif not isinstance(request_tags, list): # type: ignore
             raise TypeError("request_tags must be a list")
@@ -103,13 +160,14 @@ class IngestResource(SyncAPIResource):
                     "input": input,
                     "output": output,
                     "resource": resource,
+                    "event_timestamp": event_timestamp,
                 },
                 ingest_units_params.IngestUnitsParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ProxyResult,
+            cast_to=IngestResponse,
         )
 
 
@@ -122,26 +180,23 @@ class AsyncIngestResource(AsyncAPIResource):
     def with_streaming_response(self) -> AsyncIngestResourceWithStreamingResponse:
         return AsyncIngestResourceWithStreamingResponse(self)
 
-    async def units(
+    async def bulk(
         self,
         *,
-        category: str,
-        input: int,
-        output: int,
-        resource: str,
-        budget_ids: list[str] | NotGiven = NOT_GIVEN,
-        request_tags: list[str] | NotGiven = NOT_GIVEN,
+        items: Iterable[IngestUnitsParam],
         # 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,
-    ) -> ProxyResult:
+    ) -> BulkIngestResponse:
         """
-        Ingest a request
+        Bulk Ingest
 
         Args:
+          items (Iterable[IngestUnitsParams]): The items to ingest
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -150,17 +205,80 @@ class AsyncIngestResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+        return await self._post(
+            "/api/v1/ingest/bulk",
+            body= cast(Dict[str, object],
+                await async_maybe_transform( 
+            { 
+                "items": items, 
+            }, 
+            ingest_bulk_params.IngestBulkParams)
+            )["items"],
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=BulkIngestResponse,
+        )
+
+    async def units(
+        self,
+        *,
+        category: str,
+        input: int,
+        output: int,
+        resource: str,
+        event_timestamp: Union[str, datetime, None] | NotGiven = NOT_GIVEN,
+        budget_ids: Union[list[str], None] | NotGiven = NOT_GIVEN,
+        request_tags: Union[list[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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> IngestResponse:
+        """
+        Ingest a request
+
+        Args:
+          category (str): The name of the category
+
+          resource (str): The name of the resource
+          
+          input (int): The number of input units
+          
+          output (int): The number of output units
+          
+          event_timestamp: (datetime, None): The timestamp of the event. Defaults to None.
+          
+          budget_ids (list[str], optional): The budget IDs to associate with the request. Defaults to None.
+          
+          request_tags (list[str], optional): The request tags to associate with the request. Defaults to None.
+          
+          extra_headers (Dict[str, str], optional): Additional headers for the request. Defaults to None.
+          
+          extra_query (Dict[str, str], optional): Additional query parameters. Defaults to None.
+          
+          extra_body (Dict[str, Any], optional): Additional body parameters. Defaults to None.
+          
+          timeout (Union[float, None], optional): The timeout for the request in seconds. Defaults to None.
+        """
         valid_ids_str: str | NotGiven = NOT_GIVEN
         valid_tags_str: str | NotGiven = NOT_GIVEN
 
-        if isinstance(budget_ids, NotGiven):
+        if budget_ids is None or isinstance(budget_ids, NotGiven):
             valid_ids_str = NOT_GIVEN
+        elif not isinstance(budget_ids, list): # type: ignore
+            raise TypeError("budget_ids must be a list")
         else:
             # Proceed with the list comprehension if budget_ids is not NotGiven
             valid_ids = [id.strip() for id in budget_ids if id.strip()]
             valid_ids_str = ",".join(valid_ids) if valid_ids else NOT_GIVEN
-        if isinstance(request_tags, NotGiven):
+
+        if request_tags is None or isinstance(request_tags, NotGiven):
             valid_tags_str = NOT_GIVEN
+        elif not isinstance(request_tags, list): # type: ignore
+            raise TypeError("request_tags must be a list")
         else:
             # Proceed with the list comprehension if budget_ids is not NotGiven
             valid_tags = [tag.strip() for tag in request_tags if tag.strip()]
@@ -183,13 +301,14 @@ class AsyncIngestResource(AsyncAPIResource):
                     "input": input,
                     "output": output,
                     "resource": resource,
+                    "event_timestamp": event_timestamp,
                 },
                 ingest_units_params.IngestUnitsParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ProxyResult,
+            cast_to=IngestResponse,
         )
 
 
@@ -197,6 +316,9 @@ class IngestResourceWithRawResponse:
     def __init__(self, ingest: IngestResource) -> None:
         self._ingest = ingest
 
+        self.bulk = to_raw_response_wrapper(
+            ingest.bulk,
+        )
         self.units = to_raw_response_wrapper(
             ingest.units,
         )
@@ -206,6 +328,9 @@ class AsyncIngestResourceWithRawResponse:
     def __init__(self, ingest: AsyncIngestResource) -> None:
         self._ingest = ingest
 
+        self.bulk = async_to_raw_response_wrapper(
+            ingest.bulk,
+        )
         self.units = async_to_raw_response_wrapper(
             ingest.units,
         )
@@ -215,6 +340,9 @@ class IngestResourceWithStreamingResponse:
     def __init__(self, ingest: IngestResource) -> None:
         self._ingest = ingest
 
+        self.bulk = to_streamed_response_wrapper(
+            ingest.bulk,
+        )
         self.units = to_streamed_response_wrapper(
             ingest.units,
         )
@@ -224,6 +352,9 @@ class AsyncIngestResourceWithStreamingResponse:
     def __init__(self, ingest: AsyncIngestResource) -> None:
         self._ingest = ingest
 
+        self.bulk = async_to_streamed_response_wrapper(
+            ingest.bulk,
+        )
         self.units = async_to_streamed_response_wrapper(
             ingest.units,
         )

payi/types/__init__.py

@@ -4,13 +4,23 @@ from __future__ import annotations
 
 from .cost_data import CostData as CostData
 from .cost_details import CostDetails as CostDetails
-from .proxy_result import ProxyResult as ProxyResult
 from .requests_data import RequestsData as RequestsData
 from .budget_response import BudgetResponse as BudgetResponse
+from .ingest_response import IngestResponse as IngestResponse
+from .total_cost_data import TotalCostData as TotalCostData
 from .default_response import DefaultResponse as DefaultResponse
+from .category_response import CategoryResponse as CategoryResponse
 from .paged_budget_list import PagedBudgetList as PagedBudgetList
 from .budget_list_params import BudgetListParams as BudgetListParams
+from .ingest_bulk_params import IngestBulkParams as IngestBulkParams
+from .ingest_units_param import IngestUnitsParam as IngestUnitsParam
 from .ingest_units_params import IngestUnitsParams as IngestUnitsParams
 from .budget_create_params import BudgetCreateParams as BudgetCreateParams
 from .budget_update_params import BudgetUpdateParams as BudgetUpdateParams
+from .bulk_ingest_response import BulkIngestResponse as BulkIngestResponse
+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 .category_list_resources_response import CategoryListResourcesResponse as CategoryListResourcesResponse
+from .category_delete_resource_response import CategoryDeleteResourceResponse as CategoryDeleteResourceResponse

payi/types/budget_create_params.py

@@ -15,10 +15,14 @@ class BudgetCreateParams(TypedDict, total=False):
 
     base_cost_estimate: Literal["max"]
 
+    billing_model_id: Optional[str]
+
     budget_response_type: Literal["block", "allow"]
 
     budget_tags: Optional[List[str]]
 
     budget_type: Literal["conservative", "liberal"]
 
+    cost_basis: Literal["base", "billed"]
+
     currency: Literal["usd"]

payi/types/budget_history_response.py

@@ -4,82 +4,10 @@ from typing import List, Optional
 from datetime import datetime
 from typing_extensions import Literal
 
-from pydantic import Field as FieldInfo
-
 from .._models import BaseModel
+from .total_cost_data import TotalCostData
 
-__all__ = [
-    "BudgetHistoryResponse",
-    "BudgetHistory",
-    "BudgetHistoryTotals",
-    "BudgetHistoryTotalsCost",
-    "BudgetHistoryTotalsCostInput",
-    "BudgetHistoryTotalsCostOutput",
-    "BudgetHistoryTotalsCostTotal",
-    "BudgetHistoryTotalsRequests",
-]
-
-
-class BudgetHistoryTotalsCostInput(BaseModel):
-    base: Optional[float] = None
-
-    billed: Optional[float] = None
-
-    overrun_base: Optional[float] = FieldInfo(alias="overrunBase", default=None)
-
-    overrun_billed: Optional[float] = FieldInfo(alias="overrunBilled", default=None)
-
-    revenue: Optional[float] = None
-
-
-class BudgetHistoryTotalsCostOutput(BaseModel):
-    base: Optional[float] = None
-
-    billed: Optional[float] = None
-
-    overrun_base: Optional[float] = FieldInfo(alias="overrunBase", default=None)
-
-    overrun_billed: Optional[float] = FieldInfo(alias="overrunBilled", default=None)
-
-    revenue: Optional[float] = None
-
-
-class BudgetHistoryTotalsCostTotal(BaseModel):
-    base: Optional[float] = None
-
-    billed: Optional[float] = None
-
-    overrun_base: Optional[float] = FieldInfo(alias="overrunBase", default=None)
-
-    overrun_billed: Optional[float] = FieldInfo(alias="overrunBilled", default=None)
-
-    revenue: Optional[float] = None
-
-
-class BudgetHistoryTotalsCost(BaseModel):
-    input: BudgetHistoryTotalsCostInput
-
-    output: BudgetHistoryTotalsCostOutput
-
-    total: BudgetHistoryTotalsCostTotal
-
-
-class BudgetHistoryTotalsRequests(BaseModel):
-    blocked: Optional[int] = None
-
-    exceeded: Optional[int] = None
-
-    failed: Optional[int] = None
-
-    successful: Optional[int] = None
-
-    total: Optional[int] = None
-
-
-class BudgetHistoryTotals(BaseModel):
-    cost: BudgetHistoryTotalsCost
-
-    requests: BudgetHistoryTotalsRequests
+__all__ = ["BudgetHistoryResponse", "BudgetHistory"]
 
 
 class BudgetHistory(BaseModel):
@@ -99,7 +27,7 @@ class BudgetHistory(BaseModel):
 
     max: Optional[float] = None
 
-    totals: Optional[BudgetHistoryTotals] = None
+    totals: Optional[TotalCostData] = None
 
 
 class BudgetHistoryResponse(BaseModel):

payi/types/budget_response.py

@@ -5,32 +5,9 @@ from datetime import datetime
 from typing_extensions import Literal
 
 from .._models import BaseModel
-from .cost_data import CostData
-from .requests_data import RequestsData
+from .total_cost_data import TotalCostData
 
-__all__ = ["BudgetResponse", "Budget", "BudgetTotals", "BudgetTotalsBudgetTransactions"]
-
-
-class BudgetTotalsBudgetTransactions(BaseModel):
-    blocked: int
-
-    blocked_external: int
-
-    exceeded: int
-
-    successful: int
-
-    error: Optional[int] = None
-
-    total: Optional[int] = None
-
-
-class BudgetTotals(BaseModel):
-    budget_transactions: BudgetTotalsBudgetTransactions
-
-    cost: CostData
-
-    requests: RequestsData
+__all__ = ["BudgetResponse", "Budget"]
 
 
 class Budget(BaseModel):
@@ -52,7 +29,7 @@ class Budget(BaseModel):
 
     max: float
 
-    totals: BudgetTotals
+    totals: TotalCostData
 
     budget_tags: Optional[List[str]] = None
 

payi/types/budget_update_params.py

@@ -2,12 +2,13 @@
 
 from __future__ import annotations
 
-from typing_extensions import Required, TypedDict
+from typing import Optional
+from typing_extensions import TypedDict
 
 __all__ = ["BudgetUpdateParams"]
 
 
 class BudgetUpdateParams(TypedDict, total=False):
-    budget_name: Required[str]
+    budget_name: Optional[str]
 
-    max: float
+    max: Optional[float]

payi/types/bulk_ingest_response.py

@@ -0,0 +1,44 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+from datetime import datetime
+
+from pydantic import Field as FieldInfo
+
+from .._models import BaseModel
+
+__all__ = ["BulkIngestResponse", "Error", "ErrorXproxyResult", "ErrorXproxyResultXproxyError"]
+
+
+class ErrorXproxyResultXproxyError(BaseModel):
+    code: Optional[str] = None
+
+    message: Optional[str] = None
+
+
+class ErrorXproxyResult(BaseModel):
+    message: str
+
+    status_code: int = FieldInfo(alias="statusCode")
+
+    xproxy_error: Optional[ErrorXproxyResultXproxyError] = None
+
+
+class Error(BaseModel):
+    item_index: Optional[int] = None
+
+    xproxy_result: Optional[ErrorXproxyResult] = None
+
+
+class BulkIngestResponse(BaseModel):
+    ingest_count: int
+
+    ingest_timestamp: datetime
+
+    request_id: str
+
+    error_count: Optional[int] = None
+
+    errors: Optional[List[Error]] = None
+
+    total_count: Optional[int] = None

payi/types/categories/__init__.py

@@ -0,0 +1,6 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .resource_create_params import ResourceCreateParams as ResourceCreateParams
+from .resource_list_response import ResourceListResponse as ResourceListResponse

payi/types/categories/resource_create_params.py

@@ -0,0 +1,25 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Union
+from datetime import datetime
+from typing_extensions import Required, Annotated, TypedDict
+
+from ..._utils import PropertyInfo
+
+__all__ = ["ResourceCreateParams"]
+
+
+class ResourceCreateParams(TypedDict, total=False):
+    category: Required[str]
+
+    input_price: float
+
+    max_input_units: int
+
+    max_output_units: int
+
+    output_price: float
+
+    start_timestamp: Annotated[Union[str, datetime, None], PropertyInfo(format="iso8601")]

payi/types/categories/resource_list_response.py

@@ -0,0 +1,9 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+
+from ..category_resource_response import CategoryResourceResponse
+
+__all__ = ["ResourceListResponse"]
+
+ResourceListResponse = List[CategoryResourceResponse]

payi/types/category_delete_resource_response.py

@@ -0,0 +1,9 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+
+from .category_resource_response import CategoryResourceResponse
+
+__all__ = ["CategoryDeleteResourceResponse"]
+
+CategoryDeleteResourceResponse = List[CategoryResourceResponse]

payi/types/category_delete_response.py

@@ -0,0 +1,9 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, List
+
+from .category_resource_response import CategoryResourceResponse
+
+__all__ = ["CategoryDeleteResponse"]
+
+CategoryDeleteResponse = Dict[str, List[CategoryResourceResponse]]

payi/types/category_list_resources_response.py

@@ -0,0 +1,9 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+
+from .category_resource_response import CategoryResourceResponse
+
+__all__ = ["CategoryListResourcesResponse"]
+
+CategoryListResourcesResponse = List[CategoryResourceResponse]

payi/types/category_list_response.py

@@ -0,0 +1,9 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+
+from .category_response import CategoryResponse
+
+__all__ = ["CategoryListResponse"]
+
+CategoryListResponse = List[CategoryResponse]

payi/types/category_resource_response.py

@@ -0,0 +1,24 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from datetime import datetime
+
+from .._models import BaseModel
+
+__all__ = ["CategoryResourceResponse"]
+
+
+class CategoryResourceResponse(BaseModel):
+    category: str
+
+    resource: str
+
+    input_price: Optional[float] = None
+
+    max_input_units: Optional[int] = None
+
+    max_output_units: Optional[int] = None
+
+    output_price: Optional[float] = None
+
+    start_timestamp: Optional[datetime] = None

payi/types/category_response.py

@@ -0,0 +1,15 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from datetime import datetime
+
+from .._models import BaseModel
+
+__all__ = ["CategoryResponse"]
+
+
+class CategoryResponse(BaseModel):
+    category: str
+
+    resource_count: int
+
+    start_timestamp: datetime