payi 0.1.0a32__py3-none-any.whl → 0.1.0a34__py3-none-any.whl

payi/_client.py

@@ -8,7 +8,7 @@ from typing_extensions import Self, override
 
 import httpx
 
-from . import resources, _exceptions
+from . import _exceptions
 from ._qs import Querystring
 from ._types import (
     NOT_GIVEN,
@@ -24,6 +24,7 @@ from ._utils import (
     get_async_library,
 )
 from ._version import __version__
+from .resources import ingest, billing_models, price_modifiers
 from ._streaming import Stream as Stream, AsyncStream as AsyncStream
 from ._exceptions import PayiError, APIStatusError
 from ._base_client import (
@@ -31,28 +32,22 @@ from ._base_client import (
     SyncAPIClient,
     AsyncAPIClient,
 )
+from .resources.limits import limits
+from .resources.requests import requests
+from .resources.categories import categories
+from .resources.experiences import experiences
 
-__all__ = [
-    "Timeout",
-    "Transport",
-    "ProxiesTypes",
-    "RequestOptions",
-    "resources",
-    "Payi",
-    "AsyncPayi",
-    "Client",
-    "AsyncClient",
-]
+__all__ = ["Timeout", "Transport", "ProxiesTypes", "RequestOptions", "Payi", "AsyncPayi", "Client", "AsyncClient"]
 
 
 class Payi(SyncAPIClient):
-    budgets: resources.BudgetsResource
-    ingest: resources.IngestResource
-    categories: resources.CategoriesResource
-    experiences: resources.ExperiencesResource
-    billing_models: resources.BillingModelsResource
-    price_modifiers: resources.PriceModifiersResource
-    requests: resources.RequestsResource
+    limits: limits.LimitsResource
+    ingest: ingest.IngestResource
+    categories: categories.CategoriesResource
+    experiences: experiences.ExperiencesResource
+    billing_models: billing_models.BillingModelsResource
+    price_modifiers: price_modifiers.PriceModifiersResource
+    requests: requests.RequestsResource
     with_raw_response: PayiWithRawResponse
     with_streaming_response: PayiWithStreamedResponse
 
@@ -110,13 +105,13 @@ class Payi(SyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
-        self.budgets = resources.BudgetsResource(self)
-        self.ingest = resources.IngestResource(self)
-        self.categories = resources.CategoriesResource(self)
-        self.experiences = resources.ExperiencesResource(self)
-        self.billing_models = resources.BillingModelsResource(self)
-        self.price_modifiers = resources.PriceModifiersResource(self)
-        self.requests = resources.RequestsResource(self)
+        self.limits = limits.LimitsResource(self)
+        self.ingest = ingest.IngestResource(self)
+        self.categories = categories.CategoriesResource(self)
+        self.experiences = experiences.ExperiencesResource(self)
+        self.billing_models = billing_models.BillingModelsResource(self)
+        self.price_modifiers = price_modifiers.PriceModifiersResource(self)
+        self.requests = requests.RequestsResource(self)
         self.with_raw_response = PayiWithRawResponse(self)
         self.with_streaming_response = PayiWithStreamedResponse(self)
 
@@ -226,13 +221,13 @@ class Payi(SyncAPIClient):
 
 
 class AsyncPayi(AsyncAPIClient):
-    budgets: resources.AsyncBudgetsResource
-    ingest: resources.AsyncIngestResource
-    categories: resources.AsyncCategoriesResource
-    experiences: resources.AsyncExperiencesResource
-    billing_models: resources.AsyncBillingModelsResource
-    price_modifiers: resources.AsyncPriceModifiersResource
-    requests: resources.AsyncRequestsResource
+    limits: limits.AsyncLimitsResource
+    ingest: ingest.AsyncIngestResource
+    categories: categories.AsyncCategoriesResource
+    experiences: experiences.AsyncExperiencesResource
+    billing_models: billing_models.AsyncBillingModelsResource
+    price_modifiers: price_modifiers.AsyncPriceModifiersResource
+    requests: requests.AsyncRequestsResource
     with_raw_response: AsyncPayiWithRawResponse
     with_streaming_response: AsyncPayiWithStreamedResponse
 
@@ -290,13 +285,13 @@ class AsyncPayi(AsyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
-        self.budgets = resources.AsyncBudgetsResource(self)
-        self.ingest = resources.AsyncIngestResource(self)
-        self.categories = resources.AsyncCategoriesResource(self)
-        self.experiences = resources.AsyncExperiencesResource(self)
-        self.billing_models = resources.AsyncBillingModelsResource(self)
-        self.price_modifiers = resources.AsyncPriceModifiersResource(self)
-        self.requests = resources.AsyncRequestsResource(self)
+        self.limits = limits.AsyncLimitsResource(self)
+        self.ingest = ingest.AsyncIngestResource(self)
+        self.categories = categories.AsyncCategoriesResource(self)
+        self.experiences = experiences.AsyncExperiencesResource(self)
+        self.billing_models = billing_models.AsyncBillingModelsResource(self)
+        self.price_modifiers = price_modifiers.AsyncPriceModifiersResource(self)
+        self.requests = requests.AsyncRequestsResource(self)
         self.with_raw_response = AsyncPayiWithRawResponse(self)
         self.with_streaming_response = AsyncPayiWithStreamedResponse(self)
 
@@ -407,46 +402,46 @@ class AsyncPayi(AsyncAPIClient):
 
 class PayiWithRawResponse:
     def __init__(self, client: Payi) -> None:
-        self.budgets = resources.BudgetsResourceWithRawResponse(client.budgets)
-        self.ingest = resources.IngestResourceWithRawResponse(client.ingest)
-        self.categories = resources.CategoriesResourceWithRawResponse(client.categories)
-        self.experiences = resources.ExperiencesResourceWithRawResponse(client.experiences)
-        self.billing_models = resources.BillingModelsResourceWithRawResponse(client.billing_models)
-        self.price_modifiers = resources.PriceModifiersResourceWithRawResponse(client.price_modifiers)
-        self.requests = resources.RequestsResourceWithRawResponse(client.requests)
+        self.limits = limits.LimitsResourceWithRawResponse(client.limits)
+        self.ingest = ingest.IngestResourceWithRawResponse(client.ingest)
+        self.categories = categories.CategoriesResourceWithRawResponse(client.categories)
+        self.experiences = experiences.ExperiencesResourceWithRawResponse(client.experiences)
+        self.billing_models = billing_models.BillingModelsResourceWithRawResponse(client.billing_models)
+        self.price_modifiers = price_modifiers.PriceModifiersResourceWithRawResponse(client.price_modifiers)
+        self.requests = requests.RequestsResourceWithRawResponse(client.requests)
 
 
 class AsyncPayiWithRawResponse:
     def __init__(self, client: AsyncPayi) -> None:
-        self.budgets = resources.AsyncBudgetsResourceWithRawResponse(client.budgets)
-        self.ingest = resources.AsyncIngestResourceWithRawResponse(client.ingest)
-        self.categories = resources.AsyncCategoriesResourceWithRawResponse(client.categories)
-        self.experiences = resources.AsyncExperiencesResourceWithRawResponse(client.experiences)
-        self.billing_models = resources.AsyncBillingModelsResourceWithRawResponse(client.billing_models)
-        self.price_modifiers = resources.AsyncPriceModifiersResourceWithRawResponse(client.price_modifiers)
-        self.requests = resources.AsyncRequestsResourceWithRawResponse(client.requests)
+        self.limits = limits.AsyncLimitsResourceWithRawResponse(client.limits)
+        self.ingest = ingest.AsyncIngestResourceWithRawResponse(client.ingest)
+        self.categories = categories.AsyncCategoriesResourceWithRawResponse(client.categories)
+        self.experiences = experiences.AsyncExperiencesResourceWithRawResponse(client.experiences)
+        self.billing_models = billing_models.AsyncBillingModelsResourceWithRawResponse(client.billing_models)
+        self.price_modifiers = price_modifiers.AsyncPriceModifiersResourceWithRawResponse(client.price_modifiers)
+        self.requests = requests.AsyncRequestsResourceWithRawResponse(client.requests)
 
 
 class PayiWithStreamedResponse:
     def __init__(self, client: Payi) -> None:
-        self.budgets = resources.BudgetsResourceWithStreamingResponse(client.budgets)
-        self.ingest = resources.IngestResourceWithStreamingResponse(client.ingest)
-        self.categories = resources.CategoriesResourceWithStreamingResponse(client.categories)
-        self.experiences = resources.ExperiencesResourceWithStreamingResponse(client.experiences)
-        self.billing_models = resources.BillingModelsResourceWithStreamingResponse(client.billing_models)
-        self.price_modifiers = resources.PriceModifiersResourceWithStreamingResponse(client.price_modifiers)
-        self.requests = resources.RequestsResourceWithStreamingResponse(client.requests)
+        self.limits = limits.LimitsResourceWithStreamingResponse(client.limits)
+        self.ingest = ingest.IngestResourceWithStreamingResponse(client.ingest)
+        self.categories = categories.CategoriesResourceWithStreamingResponse(client.categories)
+        self.experiences = experiences.ExperiencesResourceWithStreamingResponse(client.experiences)
+        self.billing_models = billing_models.BillingModelsResourceWithStreamingResponse(client.billing_models)
+        self.price_modifiers = price_modifiers.PriceModifiersResourceWithStreamingResponse(client.price_modifiers)
+        self.requests = requests.RequestsResourceWithStreamingResponse(client.requests)
 
 
 class AsyncPayiWithStreamedResponse:
     def __init__(self, client: AsyncPayi) -> None:
-        self.budgets = resources.AsyncBudgetsResourceWithStreamingResponse(client.budgets)
-        self.ingest = resources.AsyncIngestResourceWithStreamingResponse(client.ingest)
-        self.categories = resources.AsyncCategoriesResourceWithStreamingResponse(client.categories)
-        self.experiences = resources.AsyncExperiencesResourceWithStreamingResponse(client.experiences)
-        self.billing_models = resources.AsyncBillingModelsResourceWithStreamingResponse(client.billing_models)
-        self.price_modifiers = resources.AsyncPriceModifiersResourceWithStreamingResponse(client.price_modifiers)
-        self.requests = resources.AsyncRequestsResourceWithStreamingResponse(client.requests)
+        self.limits = limits.AsyncLimitsResourceWithStreamingResponse(client.limits)
+        self.ingest = ingest.AsyncIngestResourceWithStreamingResponse(client.ingest)
+        self.categories = categories.AsyncCategoriesResourceWithStreamingResponse(client.categories)
+        self.experiences = experiences.AsyncExperiencesResourceWithStreamingResponse(client.experiences)
+        self.billing_models = billing_models.AsyncBillingModelsResourceWithStreamingResponse(client.billing_models)
+        self.price_modifiers = price_modifiers.AsyncPriceModifiersResourceWithStreamingResponse(client.price_modifiers)
+        self.requests = requests.AsyncRequestsResourceWithStreamingResponse(client.requests)
 
 
 Client = Payi

payi/_models.py

@@ -46,6 +46,7 @@ from ._utils import (
     strip_not_given,
     extract_type_arg,
     is_annotated_type,
+    is_type_alias_type,
     strip_annotated_type,
 )
 from ._compat import (
@@ -428,6 +429,8 @@ def construct_type(*, value: object, type_: object) -> object:
     # we allow `object` as the input type because otherwise, passing things like
     # `Literal['value']` will be reported as a type error by type checkers
     type_ = cast("type[object]", type_)
+    if is_type_alias_type(type_):
+        type_ = type_.__value__  # type: ignore[unreachable]
 
     # unwrap `Annotated[T, ...]` -> `T`
     if is_annotated_type(type_):

payi/_response.py

@@ -25,7 +25,7 @@ import httpx
 import pydantic
 
 from ._types import NoneType
-from ._utils import is_given, extract_type_arg, is_annotated_type, extract_type_var_from_base
+from ._utils import is_given, extract_type_arg, is_annotated_type, is_type_alias_type, extract_type_var_from_base
 from ._models import BaseModel, is_basemodel
 from ._constants import RAW_RESPONSE_HEADER, OVERRIDE_CAST_TO_HEADER
 from ._streaming import Stream, AsyncStream, is_stream_class_type, extract_stream_chunk_type
@@ -126,9 +126,15 @@ class BaseAPIResponse(Generic[R]):
         )
 
     def _parse(self, *, to: type[_T] | None = None) -> R | _T:
+        cast_to = to if to is not None else self._cast_to
+
+        # unwrap `TypeAlias('Name', T)` -> `T`
+        if is_type_alias_type(cast_to):
+            cast_to = cast_to.__value__  # type: ignore[unreachable]
+
         # unwrap `Annotated[T, ...]` -> `T`
-        if to and is_annotated_type(to):
-            to = extract_type_arg(to, 0)
+        if cast_to and is_annotated_type(cast_to):
+            cast_to = extract_type_arg(cast_to, 0)
 
         if self._is_sse_stream:
             if to:
@@ -164,18 +170,12 @@ class BaseAPIResponse(Generic[R]):
             return cast(
                 R,
                 stream_cls(
-                    cast_to=self._cast_to,
+                    cast_to=cast_to,
                     response=self.http_response,
                     client=cast(Any, self._client),
                 ),
             )
 
-        cast_to = to if to is not None else self._cast_to
-
-        # unwrap `Annotated[T, ...]` -> `T`
-        if is_annotated_type(cast_to):
-            cast_to = extract_type_arg(cast_to, 0)
-
         if cast_to is NoneType:
             return cast(R, None)
 

payi/_utils/__init__.py

@@ -39,6 +39,7 @@ from ._typing import (
     is_iterable_type as is_iterable_type,
     is_required_type as is_required_type,
     is_annotated_type as is_annotated_type,
+    is_type_alias_type as is_type_alias_type,
     strip_annotated_type as strip_annotated_type,
     extract_type_var_from_base as extract_type_var_from_base,
 )

payi/_utils/_typing.py

@@ -1,8 +1,17 @@
 from __future__ import annotations
 
+import sys
+import typing
+import typing_extensions
 from typing import Any, TypeVar, Iterable, cast
 from collections import abc as _c_abc
-from typing_extensions import Required, Annotated, get_args, get_origin
+from typing_extensions import (
+    TypeIs,
+    Required,
+    Annotated,
+    get_args,
+    get_origin,
+)
 
 from .._types import InheritsGeneric
 from .._compat import is_union as _is_union
@@ -36,6 +45,26 @@ def is_typevar(typ: type) -> bool:
     return type(typ) == TypeVar  # type: ignore
 
 
+_TYPE_ALIAS_TYPES: tuple[type[typing_extensions.TypeAliasType], ...] = (typing_extensions.TypeAliasType,)
+if sys.version_info >= (3, 12):
+    _TYPE_ALIAS_TYPES = (*_TYPE_ALIAS_TYPES, typing.TypeAliasType)
+
+
+def is_type_alias_type(tp: Any, /) -> TypeIs[typing_extensions.TypeAliasType]:
+    """Return whether the provided argument is an instance of `TypeAliasType`.
+
+    ```python
+    type Int = int
+    is_type_alias_type(Int)
+    # > True
+    Str = TypeAliasType("Str", str)
+    is_type_alias_type(Str)
+    # > True
+    ```
+    """
+    return isinstance(tp, _TYPE_ALIAS_TYPES)
+
+
 # Extracts T from Annotated[T, ...] or from Required[Annotated[T, ...]]
 def strip_annotated_type(typ: type) -> type:
     if is_required_type(typ) or is_annotated_type(typ):

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.32"  # x-release-please-version
+__version__ = "0.1.0-alpha.34"  # 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(budglimit_idset_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,15 +19,15 @@ 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, 
     ) -> 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:

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/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,
@@ -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,

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,
+        properties: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
+        provider_prompt: Optional[str] | NotGiven = NOT_GIVEN,
+        provider_request_headers: Optional[Dict[str, List[str]]] | NotGiven = NOT_GIVEN,
+        provider_response: Optional[List[str]] | NotGiven = NOT_GIVEN,
+        provider_response_headers: Optional[Dict[str, List[str]]] | NotGiven = NOT_GIVEN,
+        provider_uri: Optional[str] | NotGiven = NOT_GIVEN,
         provisioned_resource_name: 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,7 +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,
+                    "properties": properties,
+                    "provider_prompt": provider_prompt,
+                    "provider_request_headers": provider_request_headers,
+                    "provider_response": provider_response,
+                    "provider_response_headers": provider_response_headers,
+                    "provider_uri": provider_uri,
                     "provisioned_resource_name": provisioned_resource_name,
                     "time_to_first_token_ms": time_to_first_token_ms,
                 },
@@ -266,10 +280,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,
+        properties: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
+        provider_prompt: Optional[str] | NotGiven = NOT_GIVEN,
+        provider_request_headers: Optional[Dict[str, List[str]]] | NotGiven = NOT_GIVEN,
+        provider_response: Optional[List[str]] | NotGiven = NOT_GIVEN,
+        provider_response_headers: Optional[Dict[str, List[str]]] | NotGiven = NOT_GIVEN,
+        provider_uri: Optional[str] | NotGiven = NOT_GIVEN,
         provisioned_resource_name: 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 +315,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 +336,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 +350,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 +366,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,7 +384,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,
+                    "properties": properties,
+                    "provider_prompt": provider_prompt,
+                    "provider_request_headers": provider_request_headers,
+                    "provider_response": provider_response,
+                    "provider_response_headers": provider_response_headers,
+                    "provider_uri": provider_uri,
                     "provisioned_resource_name": provisioned_resource_name,
                     "time_to_first_token_ms": time_to_first_token_ms,
                 },