payi 0.1.0a1__py3-none-any.whl

payi/resources/ingest_requests.py

@@ -0,0 +1,187 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from ..types import ingest_request_create_params
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._utils import (
+    maybe_transform,
+    strip_not_given,
+    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.successful_proxy_result import SuccessfulProxyResult
+
+__all__ = ["IngestRequestsResource", "AsyncIngestRequestsResource"]
+
+
+class IngestRequestsResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> IngestRequestsResourceWithRawResponse:
+        return IngestRequestsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> IngestRequestsResourceWithStreamingResponse:
+        return IngestRequestsResourceWithStreamingResponse(self)
+
+    def create(
+        self,
+        *,
+        category: str,
+        resource: str,
+        units: ingest_request_create_params.Units,
+        x_proxy_budget_ids: str | NotGiven = NOT_GIVEN,
+        x_proxy_request_tags: str | 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,
+    ) -> SuccessfulProxyResult:
+        """
+        Ingest a request
+
+        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
+        """
+        extra_headers = {
+            **strip_not_given(
+                {
+                    "xProxy-Budget-IDs": x_proxy_budget_ids,
+                    "xProxy-Request-Tags": x_proxy_request_tags,
+                }
+            ),
+            **(extra_headers or {}),
+        }
+        return self._post(
+            "/api/v1/ingest",
+            body=maybe_transform(
+                {
+                    "category": category,
+                    "resource": resource,
+                    "units": units,
+                },
+                ingest_request_create_params.IngestRequestCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SuccessfulProxyResult,
+        )
+
+
+class AsyncIngestRequestsResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncIngestRequestsResourceWithRawResponse:
+        return AsyncIngestRequestsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncIngestRequestsResourceWithStreamingResponse:
+        return AsyncIngestRequestsResourceWithStreamingResponse(self)
+
+    async def create(
+        self,
+        *,
+        category: str,
+        resource: str,
+        units: ingest_request_create_params.Units,
+        x_proxy_budget_ids: str | NotGiven = NOT_GIVEN,
+        x_proxy_request_tags: str | 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,
+    ) -> SuccessfulProxyResult:
+        """
+        Ingest a request
+
+        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
+        """
+        extra_headers = {
+            **strip_not_given(
+                {
+                    "xProxy-Budget-IDs": x_proxy_budget_ids,
+                    "xProxy-Request-Tags": x_proxy_request_tags,
+                }
+            ),
+            **(extra_headers or {}),
+        }
+        return await self._post(
+            "/api/v1/ingest",
+            body=await async_maybe_transform(
+                {
+                    "category": category,
+                    "resource": resource,
+                    "units": units,
+                },
+                ingest_request_create_params.IngestRequestCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SuccessfulProxyResult,
+        )
+
+
+class IngestRequestsResourceWithRawResponse:
+    def __init__(self, ingest_requests: IngestRequestsResource) -> None:
+        self._ingest_requests = ingest_requests
+
+        self.create = to_raw_response_wrapper(
+            ingest_requests.create,
+        )
+
+
+class AsyncIngestRequestsResourceWithRawResponse:
+    def __init__(self, ingest_requests: AsyncIngestRequestsResource) -> None:
+        self._ingest_requests = ingest_requests
+
+        self.create = async_to_raw_response_wrapper(
+            ingest_requests.create,
+        )
+
+
+class IngestRequestsResourceWithStreamingResponse:
+    def __init__(self, ingest_requests: IngestRequestsResource) -> None:
+        self._ingest_requests = ingest_requests
+
+        self.create = to_streamed_response_wrapper(
+            ingest_requests.create,
+        )
+
+
+class AsyncIngestRequestsResourceWithStreamingResponse:
+    def __init__(self, ingest_requests: AsyncIngestRequestsResource) -> None:
+        self._ingest_requests = ingest_requests
+
+        self.create = async_to_streamed_response_wrapper(
+            ingest_requests.create,
+        )

payi/types/__init__.py

@@ -0,0 +1,13 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .budget_response import BudgetResponse as BudgetResponse
+from .default_response import DefaultResponse as DefaultResponse
+from .paged_budget_list import PagedBudgetList as PagedBudgetList
+from .budget_list_params import BudgetListParams as BudgetListParams
+from .budget_create_params import BudgetCreateParams as BudgetCreateParams
+from .budget_update_params import BudgetUpdateParams as BudgetUpdateParams
+from .budget_history_response import BudgetHistoryResponse as BudgetHistoryResponse
+from .successful_proxy_result import SuccessfulProxyResult as SuccessfulProxyResult
+from .ingest_request_create_params import IngestRequestCreateParams as IngestRequestCreateParams

payi/types/budget_create_params.py

@@ -0,0 +1,26 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List, Optional
+from typing_extensions import Literal, Required, Annotated, TypedDict
+
+from .._utils import PropertyInfo
+
+__all__ = ["BudgetCreateParams"]
+
+
+class BudgetCreateParams(TypedDict, total=False):
+    budget_name: Required[str]
+
+    max: Required[float]
+
+    base_cost_estimate: Literal["Max"]
+
+    budget_response_type: Literal["Block", "Allow"]
+
+    budget_tags: Optional[List[str]]
+
+    budget_type: Literal["Conservative", "Liberal"]
+
+    x_proxy_application_key: Annotated[str, PropertyInfo(alias="xProxy-Application-Key")]

payi/types/budget_history_response.py

@@ -0,0 +1,110 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+from datetime import datetime
+from typing_extensions import Literal
+
+from pydantic import Field as FieldInfo
+
+from .._models import BaseModel
+
+__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: Optional[BudgetHistoryTotalsCostInput] = None
+
+    output: Optional[BudgetHistoryTotalsCostOutput] = None
+
+    total: Optional[BudgetHistoryTotalsCostTotal] = None
+
+
+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: Optional[BudgetHistoryTotalsCost] = None
+
+    requests: Optional[BudgetHistoryTotalsRequests] = None
+
+
+class BudgetHistory(BaseModel):
+    base_cost_estimate: Optional[Literal["Max"]] = None
+
+    budget_id: Optional[str] = None
+
+    budget_name: Optional[str] = None
+
+    budget_response_type: Optional[Literal["Block", "Allow"]] = None
+
+    budget_tags: Optional[List[str]] = None
+
+    budget_type: Optional[Literal["Conservative", "Liberal"]] = None
+
+    max: Optional[float] = None
+
+    reset_date: Optional[datetime] = None
+
+    totals: Optional[BudgetHistoryTotals] = None
+
+
+class BudgetHistoryResponse(BaseModel):
+    budget_history: BudgetHistory
+
+    request_id: str
+
+    message: Optional[str] = None

payi/types/budget_list_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_extensions import Annotated, TypedDict
+
+from .._utils import PropertyInfo
+
+__all__ = ["BudgetListParams"]
+
+
+class BudgetListParams(TypedDict, total=False):
+    budget_name: Annotated[str, PropertyInfo(alias="BudgetName")]
+
+    page_number: Annotated[int, PropertyInfo(alias="PageNumber")]
+
+    page_size: Annotated[int, PropertyInfo(alias="PageSize")]
+
+    sort_ascending: Annotated[bool, PropertyInfo(alias="SortAscending")]
+
+    sort_by: Annotated[str, PropertyInfo(alias="SortBy")]
+
+    tags: Annotated[str, PropertyInfo(alias="Tags")]
+
+    x_proxy_application_key: Annotated[str, PropertyInfo(alias="xProxy-Application-Key")]

payi/types/budget_response.py

@@ -0,0 +1,98 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+from datetime import datetime
+from typing_extensions import Literal
+
+from pydantic import Field as FieldInfo
+
+from .._models import BaseModel
+
+__all__ = [
+    "BudgetResponse",
+    "Budget",
+    "BudgetTotals",
+    "BudgetTotalsCost",
+    "BudgetTotalsCostInputCost",
+    "BudgetTotalsCostOutputCost",
+    "BudgetTotalsCostTotalCost",
+    "BudgetTotalsRequests",
+]
+
+
+class BudgetTotalsCostInputCost(BaseModel):
+    base: Optional[float] = None
+
+    overrun_base: Optional[float] = None
+
+
+class BudgetTotalsCostOutputCost(BaseModel):
+    base: Optional[float] = None
+
+    overrun_base: Optional[float] = None
+
+
+class BudgetTotalsCostTotalCost(BaseModel):
+    base: Optional[float] = None
+
+    overrun_base: Optional[float] = None
+
+
+class BudgetTotalsCost(BaseModel):
+    input_cost: Optional[BudgetTotalsCostInputCost] = FieldInfo(alias="inputCost", default=None)
+
+    output_cost: Optional[BudgetTotalsCostOutputCost] = FieldInfo(alias="outputCost", default=None)
+
+    total_cost: Optional[BudgetTotalsCostTotalCost] = FieldInfo(alias="totalCost", default=None)
+
+
+class BudgetTotalsRequests(BaseModel):
+    blocked: Optional[int] = None
+
+    error: Optional[int] = None
+
+    exceeded: Optional[int] = None
+
+    failed: Optional[int] = None
+
+    successful: Optional[int] = None
+
+    total: Optional[int] = None
+
+
+class BudgetTotals(BaseModel):
+    cost: Optional[BudgetTotalsCost] = None
+
+    requests: Optional[BudgetTotalsRequests] = None
+
+
+class Budget(BaseModel):
+    base_cost_estimate: Literal["Max"]
+
+    budget_creation_timestamp: datetime
+
+    budget_id: str
+
+    budget_name: str
+
+    budget_response_type: Literal["Block", "Allow"]
+
+    budget_type: Literal["Conservative", "Liberal"]
+
+    budget_update_timestamp: datetime
+
+    currency: str
+
+    max: float
+
+    totals: BudgetTotals
+
+    budget_tags: Optional[List[str]] = None
+
+
+class BudgetResponse(BaseModel):
+    budget: Budget
+
+    request_id: str
+
+    message: Optional[str] = None

payi/types/budget_update_params.py

@@ -0,0 +1,17 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, Annotated, TypedDict
+
+from .._utils import PropertyInfo
+
+__all__ = ["BudgetUpdateParams"]
+
+
+class BudgetUpdateParams(TypedDict, total=False):
+    budget_name: Required[str]
+
+    max: float
+
+    x_proxy_application_key: Annotated[str, PropertyInfo(alias="xProxy-Application-Key")]

payi/types/budgets/__init__.py

@@ -0,0 +1,13 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .tag_create_params import TagCreateParams as TagCreateParams
+from .tag_list_response import TagListResponse as TagListResponse
+from .tag_remove_params import TagRemoveParams as TagRemoveParams
+from .tag_update_params import TagUpdateParams as TagUpdateParams
+from .tag_create_response import TagCreateResponse as TagCreateResponse
+from .tag_delete_response import TagDeleteResponse as TagDeleteResponse
+from .tag_remove_response import TagRemoveResponse as TagRemoveResponse
+from .tag_update_response import TagUpdateResponse as TagUpdateResponse
+from .budget_tags_response import BudgetTagsResponse as BudgetTagsResponse

payi/types/budgets/budget_tags_response.py

@@ -0,0 +1,18 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from datetime import datetime
+
+from pydantic import Field as FieldInfo
+
+from ..._models import BaseModel
+
+__all__ = ["BudgetTagsResponse"]
+
+
+class BudgetTagsResponse(BaseModel):
+    created_on: Optional[datetime] = FieldInfo(alias="createdOn", default=None)
+
+    tag_id: Optional[int] = FieldInfo(alias="tagId", default=None)
+
+    tag_name: Optional[str] = FieldInfo(alias="tagName", default=None)

payi/types/budgets/tag_create_params.py

@@ -0,0 +1,16 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List
+from typing_extensions import Required, Annotated, TypedDict
+
+from ..._utils import PropertyInfo
+
+__all__ = ["TagCreateParams"]
+
+
+class TagCreateParams(TypedDict, total=False):
+    budget_tags: Required[List[str]]
+
+    x_proxy_application_key: Annotated[str, PropertyInfo(alias="xProxy-Application-Key")]

payi/types/budgets/tag_create_response.py

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

payi/types/budgets/tag_delete_response.py

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

payi/types/budgets/tag_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 .budget_tags_response import BudgetTagsResponse
+
+__all__ = ["TagListResponse"]
+
+TagListResponse = List[BudgetTagsResponse]

payi/types/budgets/tag_remove_params.py

@@ -0,0 +1,16 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List
+from typing_extensions import Required, Annotated, TypedDict
+
+from ..._utils import PropertyInfo
+
+__all__ = ["TagRemoveParams"]
+
+
+class TagRemoveParams(TypedDict, total=False):
+    budget_tags: Required[List[str]]
+
+    x_proxy_application_key: Annotated[str, PropertyInfo(alias="xProxy-Application-Key")]

payi/types/budgets/tag_remove_response.py

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

payi/types/budgets/tag_update_params.py

@@ -0,0 +1,16 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List
+from typing_extensions import Required, Annotated, TypedDict
+
+from ..._utils import PropertyInfo
+
+__all__ = ["TagUpdateParams"]
+
+
+class TagUpdateParams(TypedDict, total=False):
+    budget_tags: Required[List[str]]
+
+    x_proxy_application_key: Annotated[str, PropertyInfo(alias="xProxy-Application-Key")]

payi/types/budgets/tag_update_response.py

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

payi/types/default_response.py

@@ -0,0 +1,13 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from .._models import BaseModel
+
+__all__ = ["DefaultResponse"]
+
+
+class DefaultResponse(BaseModel):
+    request_id: str
+
+    message: Optional[str] = None

payi/types/ingest_request_create_params.py

@@ -0,0 +1,27 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, Annotated, TypedDict
+
+from .._utils import PropertyInfo
+
+__all__ = ["IngestRequestCreateParams", "Units"]
+
+
+class IngestRequestCreateParams(TypedDict, total=False):
+    category: Required[str]
+
+    resource: Required[str]
+
+    units: Required[Units]
+
+    x_proxy_budget_ids: Annotated[str, PropertyInfo(alias="xProxy-Budget-IDs")]
+
+    x_proxy_request_tags: Annotated[str, PropertyInfo(alias="xProxy-Request-Tags")]
+
+
+class Units(TypedDict, total=False):
+    input: int
+
+    output: int