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

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/_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.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,
                 },

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",
 ]