payi 0.1.0a33__py3-none-any.whl → 0.1.0a35__py3-none-any.whl

payi/_base_client.py

@@ -767,6 +767,9 @@ else:
 
 class SyncHttpxClientWrapper(DefaultHttpxClient):
     def __del__(self) -> None:
+        if self.is_closed:
+            return
+
         try:
             self.close()
         except Exception:
@@ -1334,6 +1337,9 @@ else:
 
 class AsyncHttpxClientWrapper(DefaultAsyncHttpxClient):
     def __del__(self) -> None:
+        if self.is_closed:
+            return
+
         try:
             # TODO(someday): support non asyncio runtimes here
             asyncio.get_running_loop().create_task(self.aclose())

payi/_client.py

@@ -32,7 +32,7 @@ from ._base_client import (
     SyncAPIClient,
     AsyncAPIClient,
 )
-from .resources.budgets import budgets
+from .resources.limits import limits
 from .resources.requests import requests
 from .resources.categories import categories
 from .resources.experiences import experiences
@@ -41,7 +41,7 @@ __all__ = ["Timeout", "Transport", "ProxiesTypes", "RequestOptions", "Payi", "As
 
 
 class Payi(SyncAPIClient):
-    budgets: budgets.BudgetsResource
+    limits: limits.LimitsResource
     ingest: ingest.IngestResource
     categories: categories.CategoriesResource
     experiences: experiences.ExperiencesResource
@@ -105,7 +105,7 @@ class Payi(SyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
-        self.budgets = budgets.BudgetsResource(self)
+        self.limits = limits.LimitsResource(self)
         self.ingest = ingest.IngestResource(self)
         self.categories = categories.CategoriesResource(self)
         self.experiences = experiences.ExperiencesResource(self)
@@ -221,7 +221,7 @@ class Payi(SyncAPIClient):
 
 
 class AsyncPayi(AsyncAPIClient):
-    budgets: budgets.AsyncBudgetsResource
+    limits: limits.AsyncLimitsResource
     ingest: ingest.AsyncIngestResource
     categories: categories.AsyncCategoriesResource
     experiences: experiences.AsyncExperiencesResource
@@ -285,7 +285,7 @@ class AsyncPayi(AsyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
-        self.budgets = budgets.AsyncBudgetsResource(self)
+        self.limits = limits.AsyncLimitsResource(self)
         self.ingest = ingest.AsyncIngestResource(self)
         self.categories = categories.AsyncCategoriesResource(self)
         self.experiences = experiences.AsyncExperiencesResource(self)
@@ -402,7 +402,7 @@ class AsyncPayi(AsyncAPIClient):
 
 class PayiWithRawResponse:
     def __init__(self, client: Payi) -> None:
-        self.budgets = budgets.BudgetsResourceWithRawResponse(client.budgets)
+        self.limits = limits.LimitsResourceWithRawResponse(client.limits)
         self.ingest = ingest.IngestResourceWithRawResponse(client.ingest)
         self.categories = categories.CategoriesResourceWithRawResponse(client.categories)
         self.experiences = experiences.ExperiencesResourceWithRawResponse(client.experiences)
@@ -413,7 +413,7 @@ class PayiWithRawResponse:
 
 class AsyncPayiWithRawResponse:
     def __init__(self, client: AsyncPayi) -> None:
-        self.budgets = budgets.AsyncBudgetsResourceWithRawResponse(client.budgets)
+        self.limits = limits.AsyncLimitsResourceWithRawResponse(client.limits)
         self.ingest = ingest.AsyncIngestResourceWithRawResponse(client.ingest)
         self.categories = categories.AsyncCategoriesResourceWithRawResponse(client.categories)
         self.experiences = experiences.AsyncExperiencesResourceWithRawResponse(client.experiences)
@@ -424,7 +424,7 @@ class AsyncPayiWithRawResponse:
 
 class PayiWithStreamedResponse:
     def __init__(self, client: Payi) -> None:
-        self.budgets = budgets.BudgetsResourceWithStreamingResponse(client.budgets)
+        self.limits = limits.LimitsResourceWithStreamingResponse(client.limits)
         self.ingest = ingest.IngestResourceWithStreamingResponse(client.ingest)
         self.categories = categories.CategoriesResourceWithStreamingResponse(client.categories)
         self.experiences = experiences.ExperiencesResourceWithStreamingResponse(client.experiences)
@@ -435,7 +435,7 @@ class PayiWithStreamedResponse:
 
 class AsyncPayiWithStreamedResponse:
     def __init__(self, client: AsyncPayi) -> None:
-        self.budgets = budgets.AsyncBudgetsResourceWithStreamingResponse(client.budgets)
+        self.limits = limits.AsyncLimitsResourceWithStreamingResponse(client.limits)
         self.ingest = ingest.AsyncIngestResourceWithStreamingResponse(client.ingest)
         self.categories = categories.AsyncCategoriesResourceWithStreamingResponse(client.categories)
         self.experiences = experiences.AsyncExperiencesResourceWithStreamingResponse(client.experiences)

payi/_models.py

@@ -488,7 +488,11 @@ def construct_type(*, value: object, type_: object) -> object:
         _, items_type = get_args(type_)  # Dict[_, items_type]
         return {key: construct_type(value=item, type_=items_type) for key, item in value.items()}
 
-    if not is_literal_type(type_) and (issubclass(origin, BaseModel) or issubclass(origin, GenericModel)):
+    if (
+        not is_literal_type(type_)
+        and inspect.isclass(origin)
+        and (issubclass(origin, BaseModel) or issubclass(origin, GenericModel))
+    ):
         if is_list(value):
             return [cast(Any, type_).construct(**entry) if is_mapping(entry) else entry for entry in value]
 

payi/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "payi"
-__version__ = "0.1.0-alpha.33"  # x-release-please-version
+__version__ = "0.1.0-alpha.35"  # x-release-please-version

payi/lib/helpers.py

@@ -2,13 +2,13 @@
 from typing import Dict, List, Union
 
 
-def create_budget_header_from_ids(budget_ids: List[str]) -> Dict[str, str]:
-    if not isinstance(budget_ids, list): # type: ignore
-        raise TypeError("budget_ids must be a list")
+def create_limit_header_from_ids(limit_ids: List[str]) -> Dict[str, str]:
+    if not isinstance(limit_ids, list): # type: ignore
+        raise TypeError("limit_ids must be a list")
 
-    valid_ids = [id.strip() for id in budget_ids if isinstance(id, str) and id.strip()] # type: ignore
+    valid_ids = [id.strip() for id in limit_ids if isinstance(id, str) and id.strip()] # type: ignore
 
-    return {"xProxy-Budget-IDs": ",".join(valid_ids)} if valid_ids else {}
+    return {"xProxy-Limit-IDs": ",".join(valid_ids)} if valid_ids else {}
 
 def create_request_header_from_tags(request_tags: List[str]) -> Dict[str, str]:
     if not isinstance(request_tags, list): # type: ignore
@@ -19,20 +19,23 @@ def create_request_header_from_tags(request_tags: List[str]) -> Dict[str, str]:
     return {"xProxy-Request-Tags": ",".join(valid_tags)} if valid_tags else {}
 
 def create_headers(
-    budget_ids: Union[List[str], None] = None, 
+    limit_ids: Union[List[str], None] = None, 
     request_tags: Union[List[str], None] = None,
     user_id: Union[str, None] = None,
     experience_id: Union[str, None] = None, 
+    experience_name: Union[str, None] = None, 
     ) -> Dict[str, str]:
     headers: Dict[str, str] = {}
 
-    if budget_ids:
-        headers.update(create_budget_header_from_ids(budget_ids))
+    if limit_ids:
+        headers.update(create_limit_header_from_ids(limit_ids))
     if request_tags:
         headers.update(create_request_header_from_tags(request_tags))
     if user_id:
         headers.update({"xProxy-User-ID": user_id})
     if experience_id:
         headers.update({"xProxy-Experience-Id": experience_id})
+    if experience_name:
+        headers.update({"xProxy-Experience-Name": experience_name})
 
     return headers

payi/resources/__init__.py

@@ -8,13 +8,13 @@ from .ingest import (
     IngestResourceWithStreamingResponse,
     AsyncIngestResourceWithStreamingResponse,
 )
-from .budgets import (
-    BudgetsResource,
-    AsyncBudgetsResource,
-    BudgetsResourceWithRawResponse,
-    AsyncBudgetsResourceWithRawResponse,
-    BudgetsResourceWithStreamingResponse,
-    AsyncBudgetsResourceWithStreamingResponse,
+from .limits import (
+    LimitsResource,
+    AsyncLimitsResource,
+    LimitsResourceWithRawResponse,
+    AsyncLimitsResourceWithRawResponse,
+    LimitsResourceWithStreamingResponse,
+    AsyncLimitsResourceWithStreamingResponse,
 )
 from .requests import (
     RequestsResource,
@@ -58,12 +58,12 @@ from .price_modifiers import (
 )
 
 __all__ = [
-    "BudgetsResource",
-    "AsyncBudgetsResource",
-    "BudgetsResourceWithRawResponse",
-    "AsyncBudgetsResourceWithRawResponse",
-    "BudgetsResourceWithStreamingResponse",
-    "AsyncBudgetsResourceWithStreamingResponse",
+    "LimitsResource",
+    "AsyncLimitsResource",
+    "LimitsResourceWithRawResponse",
+    "AsyncLimitsResourceWithRawResponse",
+    "LimitsResourceWithStreamingResponse",
+    "AsyncLimitsResourceWithStreamingResponse",
     "IngestResource",
     "AsyncIngestResource",
     "IngestResourceWithRawResponse",

payi/resources/billing_models.py

@@ -52,7 +52,7 @@ class BillingModelsResource(SyncAPIResource):
         self,
         *,
         name: str,
-        type: Literal["invalid", "costplus", "subscription", "hybrid"],
+        type: Literal["costplus", "subscription", "hybrid"],
         default_price_modifier: Optional[float] | NotGiven = NOT_GIVEN,
         prepaid_amount: Optional[float] | NotGiven = NOT_GIVEN,
         prepaid_max: Optional[float] | NotGiven = NOT_GIVEN,
@@ -128,7 +128,7 @@ class BillingModelsResource(SyncAPIResource):
         self,
         billing_model_id: str,
         *,
-        type: Literal["invalid", "costplus", "subscription", "hybrid"],
+        type: Literal["costplus", "subscription", "hybrid"],
         default_price_modifier: Optional[float] | NotGiven = NOT_GIVEN,
         name: Optional[str] | NotGiven = NOT_GIVEN,
         prepaid_amount: Optional[float] | NotGiven = NOT_GIVEN,
@@ -246,7 +246,7 @@ class AsyncBillingModelsResource(AsyncAPIResource):
         self,
         *,
         name: str,
-        type: Literal["invalid", "costplus", "subscription", "hybrid"],
+        type: Literal["costplus", "subscription", "hybrid"],
         default_price_modifier: Optional[float] | NotGiven = NOT_GIVEN,
         prepaid_amount: Optional[float] | NotGiven = NOT_GIVEN,
         prepaid_max: Optional[float] | NotGiven = NOT_GIVEN,
@@ -322,7 +322,7 @@ class AsyncBillingModelsResource(AsyncAPIResource):
         self,
         billing_model_id: str,
         *,
-        type: Literal["invalid", "costplus", "subscription", "hybrid"],
+        type: Literal["costplus", "subscription", "hybrid"],
         default_price_modifier: Optional[float] | NotGiven = NOT_GIVEN,
         name: Optional[str] | NotGiven = NOT_GIVEN,
         prepaid_amount: Optional[float] | NotGiven = NOT_GIVEN,

payi/resources/experiences/__init__.py

@@ -1,13 +1,5 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from .csat import (
-    CsatResource,
-    AsyncCsatResource,
-    CsatResourceWithRawResponse,
-    AsyncCsatResourceWithRawResponse,
-    CsatResourceWithStreamingResponse,
-    AsyncCsatResourceWithStreamingResponse,
-)
 from .types import (
     TypesResource,
     AsyncTypesResource,
@@ -40,12 +32,6 @@ __all__ = [
     "AsyncTypesResourceWithRawResponse",
     "TypesResourceWithStreamingResponse",
     "AsyncTypesResourceWithStreamingResponse",
-    "CsatResource",
-    "AsyncCsatResource",
-    "CsatResourceWithRawResponse",
-    "AsyncCsatResourceWithRawResponse",
-    "CsatResourceWithStreamingResponse",
-    "AsyncCsatResourceWithStreamingResponse",
     "PropertiesResource",
     "AsyncPropertiesResource",
     "PropertiesResourceWithRawResponse",

payi/resources/experiences/experiences.py

@@ -4,14 +4,6 @@ from __future__ import annotations
 
 import httpx
 
-from .csat import (
-    CsatResource,
-    AsyncCsatResource,
-    CsatResourceWithRawResponse,
-    AsyncCsatResourceWithRawResponse,
-    CsatResourceWithStreamingResponse,
-    AsyncCsatResourceWithStreamingResponse,
-)
 from .types import (
     TypesResource,
     AsyncTypesResource,
@@ -48,10 +40,6 @@ class ExperiencesResource(SyncAPIResource):
     def types(self) -> TypesResource:
         return TypesResource(self._client)
 
-    @cached_property
-    def csat(self) -> CsatResource:
-        return CsatResource(self._client)
-
     @cached_property
     def properties(self) -> PropertiesResource:
         return PropertiesResource(self._client)
@@ -180,10 +168,6 @@ class AsyncExperiencesResource(AsyncAPIResource):
     def types(self) -> AsyncTypesResource:
         return AsyncTypesResource(self._client)
 
-    @cached_property
-    def csat(self) -> AsyncCsatResource:
-        return AsyncCsatResource(self._client)
-
     @cached_property
     def properties(self) -> AsyncPropertiesResource:
         return AsyncPropertiesResource(self._client)
@@ -325,10 +309,6 @@ class ExperiencesResourceWithRawResponse:
     def types(self) -> TypesResourceWithRawResponse:
         return TypesResourceWithRawResponse(self._experiences.types)
 
-    @cached_property
-    def csat(self) -> CsatResourceWithRawResponse:
-        return CsatResourceWithRawResponse(self._experiences.csat)
-
     @cached_property
     def properties(self) -> PropertiesResourceWithRawResponse:
         return PropertiesResourceWithRawResponse(self._experiences.properties)
@@ -352,10 +332,6 @@ class AsyncExperiencesResourceWithRawResponse:
     def types(self) -> AsyncTypesResourceWithRawResponse:
         return AsyncTypesResourceWithRawResponse(self._experiences.types)
 
-    @cached_property
-    def csat(self) -> AsyncCsatResourceWithRawResponse:
-        return AsyncCsatResourceWithRawResponse(self._experiences.csat)
-
     @cached_property
     def properties(self) -> AsyncPropertiesResourceWithRawResponse:
         return AsyncPropertiesResourceWithRawResponse(self._experiences.properties)
@@ -379,10 +355,6 @@ class ExperiencesResourceWithStreamingResponse:
     def types(self) -> TypesResourceWithStreamingResponse:
         return TypesResourceWithStreamingResponse(self._experiences.types)
 
-    @cached_property
-    def csat(self) -> CsatResourceWithStreamingResponse:
-        return CsatResourceWithStreamingResponse(self._experiences.csat)
-
     @cached_property
     def properties(self) -> PropertiesResourceWithStreamingResponse:
         return PropertiesResourceWithStreamingResponse(self._experiences.properties)
@@ -406,10 +378,6 @@ class AsyncExperiencesResourceWithStreamingResponse:
     def types(self) -> AsyncTypesResourceWithStreamingResponse:
         return AsyncTypesResourceWithStreamingResponse(self._experiences.types)
 
-    @cached_property
-    def csat(self) -> AsyncCsatResourceWithStreamingResponse:
-        return AsyncCsatResourceWithStreamingResponse(self._experiences.csat)
-
     @cached_property
     def properties(self) -> AsyncPropertiesResourceWithStreamingResponse:
         return AsyncPropertiesResourceWithStreamingResponse(self._experiences.properties)

payi/resources/experiences/properties.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, Optional
+from typing import Dict
 
 import httpx
 
@@ -50,7 +50,7 @@ class PropertiesResource(SyncAPIResource):
         self,
         experience_id: str,
         *,
-        properties: Optional[Dict[str, str]],
+        properties: Dict[str, 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,
@@ -59,7 +59,7 @@ class PropertiesResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> ExperienceInstanceResponse:
         """
-        Update an Experience properties
+        Update Experience properties
 
         Args:
           extra_headers: Send extra headers
@@ -106,7 +106,7 @@ class AsyncPropertiesResource(AsyncAPIResource):
         self,
         experience_id: str,
         *,
-        properties: Optional[Dict[str, str]],
+        properties: Dict[str, 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,
@@ -115,7 +115,7 @@ class AsyncPropertiesResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> ExperienceInstanceResponse:
         """
-        Update an Experience properties
+        Update Experience properties
 
         Args:
           extra_headers: Send extra headers

payi/resources/ingest.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, Union, Iterable, Optional
+from typing import Dict, List, Union, Iterable, Optional
 from datetime import datetime
 
 import httpx
@@ -93,10 +93,17 @@ class IngestResource(SyncAPIResource):
         units: Dict[str, ingest_units_params.Units],
         end_to_end_latency_ms: Optional[int] | NotGiven = NOT_GIVEN,
         event_timestamp: Union[str, datetime, None] | NotGiven = NOT_GIVEN,
+        experience_properties: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
         http_status_code: Optional[int] | NotGiven = NOT_GIVEN,
-        provisioned_resource_name: Optional[str] | NotGiven = NOT_GIVEN,
+        properties: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
+        provider_request_headers: Optional[Iterable[ingest_units_params.ProviderRequestHeader]] | NotGiven = NOT_GIVEN,
+        provider_request_json: Optional[str] | NotGiven = NOT_GIVEN,
+        provider_response_headers: Optional[Iterable[ingest_units_params.ProviderResponseHeader]]
+        | NotGiven = NOT_GIVEN,
+        provider_response_json: Union[str, List[str], None] | NotGiven = NOT_GIVEN,
+        provider_uri: Optional[str] | NotGiven = NOT_GIVEN,
         time_to_first_token_ms: Optional[int] | NotGiven = NOT_GIVEN,
-        budget_ids: Union[list[str], None] | NotGiven = NOT_GIVEN,
+        limit_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,
@@ -122,7 +129,7 @@ class IngestResource(SyncAPIResource):
           
           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.
+          limit_ids (list[str], optional): The limit 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.
 
@@ -143,13 +150,13 @@ class IngestResource(SyncAPIResource):
         valid_ids_str: str | NotGiven = NOT_GIVEN
         valid_tags_str: str | NotGiven = NOT_GIVEN
 
-        if budget_ids is None or isinstance(budget_ids, NotGiven):
+        if limit_ids is None or isinstance(limit_ids, NotGiven):
             valid_ids_str = NOT_GIVEN
-        elif not isinstance(budget_ids, list): # type: ignore
-            raise TypeError("budget_ids must be a list")
+        elif not isinstance(limit_ids, list): # type: ignore
+            raise TypeError("limit_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()]
+            # Proceed with the list comprehension if limit_ids is not NotGiven
+            valid_ids = [id.strip() for id in limit_ids if id.strip()]
             valid_ids_str = ",".join(valid_ids) if valid_ids else NOT_GIVEN
 
         if request_tags is None or isinstance(request_tags, NotGiven):
@@ -157,7 +164,7 @@ class IngestResource(SyncAPIResource):
         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
+            # Proceed with the list comprehension if request_tags is not NotGiven
             valid_tags = [tag.strip() for tag in request_tags if tag.strip()]
             valid_tags_str = ",".join(valid_tags) if valid_tags else NOT_GIVEN
 
@@ -173,7 +180,7 @@ class IngestResource(SyncAPIResource):
         extra_headers = {
             **strip_not_given(
                 {
-                    "xProxy-Budget-IDs": valid_ids_str,
+                    "xProxy-Limit-IDs": valid_ids_str,
                     "xProxy-Request-Tags": valid_tags_str,
                     "xProxy-Experience-Id": experience_id,
                     "xProxy-Experience-Name": experience_name,
@@ -191,8 +198,14 @@ class IngestResource(SyncAPIResource):
                     "units": units,
                     "end_to_end_latency_ms": end_to_end_latency_ms,
                     "event_timestamp": event_timestamp,
+                    "experience_properties": experience_properties,
                     "http_status_code": http_status_code,
-                    "provisioned_resource_name": provisioned_resource_name,
+                    "properties": properties,
+                    "provider_request_headers": provider_request_headers,
+                    "provider_request_json": provider_request_json,
+                    "provider_response_headers": provider_response_headers,
+                    "provider_response_json": provider_response_json,
+                    "provider_uri": provider_uri,
                     "time_to_first_token_ms": time_to_first_token_ms,
                 },
                 ingest_units_params.IngestUnitsParams,
@@ -266,10 +279,17 @@ class AsyncIngestResource(AsyncAPIResource):
         units: Dict[str, ingest_units_params.Units],
         end_to_end_latency_ms: Optional[int] | NotGiven = NOT_GIVEN,
         event_timestamp: Union[str, datetime, None] | NotGiven = NOT_GIVEN,
+        experience_properties: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
         http_status_code: Optional[int] | NotGiven = NOT_GIVEN,
-        provisioned_resource_name: Optional[str] | NotGiven = NOT_GIVEN,
+        properties: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
+        provider_request_headers: Optional[Iterable[ingest_units_params.ProviderRequestHeader]] | NotGiven = NOT_GIVEN,
+        provider_request_json: Optional[str] | NotGiven = NOT_GIVEN,
+        provider_response_headers: Optional[Iterable[ingest_units_params.ProviderResponseHeader]]
+        | NotGiven = NOT_GIVEN,
+        provider_response_json: Union[str, List[str], None] | NotGiven = NOT_GIVEN,
+        provider_uri: Optional[str] | NotGiven = NOT_GIVEN,
         time_to_first_token_ms: Optional[int] | NotGiven = NOT_GIVEN,
-        budget_ids: Union[list[str], None] | NotGiven = NOT_GIVEN,
+        limit_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,
@@ -294,7 +314,7 @@ class AsyncIngestResource(AsyncAPIResource):
           
           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.
+          limit_ids (list[str], optional): The limit 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.
           
@@ -315,13 +335,13 @@ class AsyncIngestResource(AsyncAPIResource):
         valid_ids_str: str | NotGiven = NOT_GIVEN
         valid_tags_str: str | NotGiven = NOT_GIVEN
 
-        if budget_ids is None or isinstance(budget_ids, NotGiven):
+        if limit_ids is None or isinstance(limit_ids, NotGiven):
             valid_ids_str = NOT_GIVEN
-        elif not isinstance(budget_ids, list): # type: ignore
-            raise TypeError("budget_ids must be a list")
+        elif not isinstance(limit_ids, list): # type: ignore
+            raise TypeError("limit_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()]
+            # Proceed with the list comprehension if limit_ids is not NotGiven
+            valid_ids = [id.strip() for id in limit_ids if id.strip()]
             valid_ids_str = ",".join(valid_ids) if valid_ids else NOT_GIVEN
 
         if request_tags is None or isinstance(request_tags, NotGiven):
@@ -329,7 +349,7 @@ class AsyncIngestResource(AsyncAPIResource):
         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
+            # Proceed with the list comprehension if request_tags is not NotGiven
             valid_tags = [tag.strip() for tag in request_tags if tag.strip()]
             valid_tags_str = ",".join(valid_tags) if valid_tags else NOT_GIVEN
 
@@ -345,7 +365,7 @@ class AsyncIngestResource(AsyncAPIResource):
         extra_headers = {
             **strip_not_given(
                 {
-                    "xProxy-Budget-IDs": valid_ids_str,
+                    "xProxy-Limit-IDs": valid_ids_str,
                     "xProxy-Request-Tags": valid_tags_str,
                     "xProxy-Experience-Name": experience_name,
                     "xProxy-Experience-Id": experience_id,
@@ -363,8 +383,14 @@ class AsyncIngestResource(AsyncAPIResource):
                     "units": units,
                     "end_to_end_latency_ms": end_to_end_latency_ms,
                     "event_timestamp": event_timestamp,
+                    "experience_properties": experience_properties,
                     "http_status_code": http_status_code,
-                    "provisioned_resource_name": provisioned_resource_name,
+                    "properties": properties,
+                    "provider_request_headers": provider_request_headers,
+                    "provider_request_json": provider_request_json,
+                    "provider_response_headers": provider_response_headers,
+                    "provider_response_json": provider_response_json,
+                    "provider_uri": provider_uri,
                     "time_to_first_token_ms": time_to_first_token_ms,
                 },
                 ingest_units_params.IngestUnitsParams,

payi/resources/{budgets → limits}/__init__.py

@@ -8,13 +8,13 @@ from .tags import (
     TagsResourceWithStreamingResponse,
     AsyncTagsResourceWithStreamingResponse,
 )
-from .budgets import (
-    BudgetsResource,
-    AsyncBudgetsResource,
-    BudgetsResourceWithRawResponse,
-    AsyncBudgetsResourceWithRawResponse,
-    BudgetsResourceWithStreamingResponse,
-    AsyncBudgetsResourceWithStreamingResponse,
+from .limits import (
+    LimitsResource,
+    AsyncLimitsResource,
+    LimitsResourceWithRawResponse,
+    AsyncLimitsResourceWithRawResponse,
+    LimitsResourceWithStreamingResponse,
+    AsyncLimitsResourceWithStreamingResponse,
 )
 
 __all__ = [
@@ -24,10 +24,10 @@ __all__ = [
     "AsyncTagsResourceWithRawResponse",
     "TagsResourceWithStreamingResponse",
     "AsyncTagsResourceWithStreamingResponse",
-    "BudgetsResource",
-    "AsyncBudgetsResource",
-    "BudgetsResourceWithRawResponse",
-    "AsyncBudgetsResourceWithRawResponse",
-    "BudgetsResourceWithStreamingResponse",
-    "AsyncBudgetsResourceWithStreamingResponse",
+    "LimitsResource",
+    "AsyncLimitsResource",
+    "LimitsResourceWithRawResponse",
+    "AsyncLimitsResourceWithRawResponse",
+    "LimitsResourceWithStreamingResponse",
+    "AsyncLimitsResourceWithStreamingResponse",
 ]