payi 0.1.0a113__py3-none-any.whl → 0.1.0a115__py3-none-any.whl

payi/_utils/_transform.py

@@ -16,18 +16,20 @@ from ._utils import (
     lru_cache,
     is_mapping,
     is_iterable,
+    is_sequence,
 )
 from .._files import is_base64_file_input
+from ._compat import get_origin, is_typeddict
 from ._typing import (
     is_list_type,
     is_union_type,
     extract_type_arg,
     is_iterable_type,
     is_required_type,
+    is_sequence_type,
     is_annotated_type,
     strip_annotated_type,
 )
-from .._compat import get_origin, model_dump, is_typeddict
 
 _T = TypeVar("_T")
 
@@ -167,6 +169,8 @@ def _transform_recursive(
 
             Defaults to the same value as the `annotation` argument.
     """
+    from .._compat import model_dump
+
     if inner_type is None:
         inner_type = annotation
 
@@ -184,6 +188,8 @@ def _transform_recursive(
         (is_list_type(stripped_type) and is_list(data))
         # Iterable[T]
         or (is_iterable_type(stripped_type) and is_iterable(data) and not isinstance(data, str))
+        # Sequence[T]
+        or (is_sequence_type(stripped_type) and is_sequence(data) and not isinstance(data, str))
     ):
         # dicts are technically iterable, but it is an iterable on the keys of the dict and is not usually
         # intended as an iterable, so we don't transform it.
@@ -329,6 +335,8 @@ async def _async_transform_recursive(
 
             Defaults to the same value as the `annotation` argument.
     """
+    from .._compat import model_dump
+
     if inner_type is None:
         inner_type = annotation
 
@@ -346,6 +354,8 @@ async def _async_transform_recursive(
         (is_list_type(stripped_type) and is_list(data))
         # Iterable[T]
         or (is_iterable_type(stripped_type) and is_iterable(data) and not isinstance(data, str))
+        # Sequence[T]
+        or (is_sequence_type(stripped_type) and is_sequence(data) and not isinstance(data, str))
     ):
         # dicts are technically iterable, but it is an iterable on the keys of the dict and is not usually
         # intended as an iterable, so we don't transform it.

payi/_utils/_typing.py

@@ -15,7 +15,7 @@ from typing_extensions import (
 
 from ._utils import lru_cache
 from .._types import InheritsGeneric
-from .._compat import is_union as _is_union
+from ._compat import is_union as _is_union
 
 
 def is_annotated_type(typ: type) -> bool:
@@ -26,6 +26,11 @@ def is_list_type(typ: type) -> bool:
     return (get_origin(typ) or typ) == list
 
 
+def is_sequence_type(typ: type) -> bool:
+    origin = get_origin(typ) or typ
+    return origin == typing_extensions.Sequence or origin == typing.Sequence or origin == _c_abc.Sequence
+
+
 def is_iterable_type(typ: type) -> bool:
     """If the given type is `typing.Iterable[T]`"""
     origin = get_origin(typ) or typ

payi/_utils/_utils.py

@@ -22,7 +22,6 @@ from typing_extensions import TypeGuard
 import sniffio
 
 from .._types import NotGiven, FileTypes, NotGivenOr, HeadersLike
-from .._compat import parse_date as parse_date, parse_datetime as parse_datetime
 
 _T = TypeVar("_T")
 _TupleT = TypeVar("_TupleT", bound=Tuple[object, ...])

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.113"  # x-release-please-version
+__version__ = "0.1.0-alpha.115"  # x-release-please-version

payi/lib/AnthropicInstrumentor.py

@@ -245,7 +245,7 @@ class _AnthropicProviderRequest(_ProviderRequest):
         return True
 
 def anthropic_process_compute_input_cost(request: _ProviderRequest, usage: 'dict[str, Any]') -> int:
-    input = usage['input_tokens']
+    input = usage.get('input_tokens', 0)
     units: dict[str, Units] = request._ingest["units"]
 
     cache_creation_input_tokens = usage.get("cache_creation_input_tokens", 0)
@@ -282,11 +282,11 @@ def anthropic_process_compute_input_cost(request: _ProviderRequest, usage: 'dict
     return _PayiInstrumentor.update_for_vision(input, units, request._estimated_prompt_tokens, is_large_context=request._is_large_context)
 
 def anthropic_process_synchronous_response(request: _ProviderRequest, response: 'dict[str, Any]', log_prompt_and_response: bool, assign_id: bool) -> Any:
-    usage = response['usage']
+    usage = response.get('usage', {})
     units: dict[str, Units] = request._ingest["units"]
 
     input_tokens = anthropic_process_compute_input_cost(request, usage)
-    output = usage['output_tokens']
+    output = usage.get('output_tokens', 0)
 
     large_context = "_large_context" if request._is_large_context else ""
     units["text"+large_context] = Units(input=input_tokens, output=output)
@@ -327,7 +327,7 @@ def anthropic_process_chunk(request: _ProviderRequest, chunk: 'dict[str, Any]',
         if model and 'resource' in request._ingest:
             request._instrumentor._logger.debug(f"Anthropic streaming, reported model: {model}, instrumented model {request._ingest['resource']}")
 
-        usage = message['usage']
+        usage = message.get('usage', {})
         units = request._ingest["units"]
 
         input = anthropic_process_compute_input_cost(request, usage)

payi/lib/BedrockInstrumentor.py

@@ -424,10 +424,10 @@ class _BedrockConverseProviderRequest(_BedrockProviderRequest):
         log_prompt_and_response: bool,
         kwargs: Any) -> Any:
 
-        usage = response["usage"]
-        input = usage["inputTokens"]
-        output = usage["outputTokens"]
-        
+        usage = response.get("usage", {})
+        input = usage.get("inputTokens", 0)
+        output = usage.get("outputTokens", 0)
+
         units: dict[str, Units] = self._ingest["units"]
         units["text"] = Units(input=input, output=output)
 
@@ -456,9 +456,9 @@ class _BedrockConverseProviderRequest(_BedrockProviderRequest):
         metadata = chunk.get("metadata", {})
 
         if metadata:
-            usage = metadata['usage']
-            input = usage["inputTokens"]
-            output = usage["outputTokens"]
+            usage = metadata.get('usage', {})
+            input = usage.get("inputTokens", 0)
+            output = usage.get("outputTokens", 0)
             self._ingest["units"]["text"] = Units(input=input, output=output)
 
             ingest = True

payi/resources/categories/fixed_cost_resources.py

@@ -2,12 +2,12 @@
 
 from __future__ import annotations
 
-from typing import List, Union
+from typing import Union
 from datetime import datetime
 
 import httpx
 
-from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven, SequenceNotStr
 from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -49,7 +49,7 @@ class FixedCostResourcesResource(SyncAPIResource):
         resource: str,
         *,
         category: str,
-        units: List[str],
+        units: SequenceNotStr[str],
         cost_per_hour: float | NotGiven = NOT_GIVEN,
         start_timestamp: Union[str, datetime, None] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -117,7 +117,7 @@ class AsyncFixedCostResourcesResource(AsyncAPIResource):
         resource: str,
         *,
         category: str,
-        units: List[str],
+        units: SequenceNotStr[str],
         cost_per_hour: float | NotGiven = NOT_GIVEN,
         start_timestamp: Union[str, datetime, None] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.

payi/resources/ingest.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Union, Iterable, Optional
+from typing import Dict, Union, Iterable, Optional
 from datetime import datetime
 
 import httpx
@@ -10,7 +10,7 @@ import httpx
 from payi._utils._utils import is_given
 
 from ..types import ingest_units_params
-from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven, SequenceNotStr
 from .._utils import maybe_transform, strip_not_given, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -98,7 +98,7 @@ class IngestResource(SyncAPIResource):
         | NotGiven = NOT_GIVEN,
         provider_response_headers: Optional[Iterable[PayICommonModelsAPIRouterHeaderInfoParam]] | NotGiven = NOT_GIVEN,
         provider_response_id: Optional[str] | NotGiven = NOT_GIVEN,
-        provider_response_json: Union[str, List[str], None] | NotGiven = NOT_GIVEN,
+        provider_response_json: Union[str, SequenceNotStr[str], None] | NotGiven = NOT_GIVEN,
         provider_uri: Optional[str] | NotGiven = NOT_GIVEN,
         resource: Optional[str] | NotGiven = NOT_GIVEN,
         time_to_first_completion_token_ms: Optional[int] | NotGiven = NOT_GIVEN,
@@ -328,7 +328,7 @@ class AsyncIngestResource(AsyncAPIResource):
         | NotGiven = NOT_GIVEN,
         provider_response_headers: Optional[Iterable[PayICommonModelsAPIRouterHeaderInfoParam]] | NotGiven = NOT_GIVEN,
         provider_response_id: Optional[str] | NotGiven = NOT_GIVEN,
-        provider_response_json: Union[str, List[str], None] | NotGiven = NOT_GIVEN,
+        provider_response_json: Union[str, SequenceNotStr[str], None] | NotGiven = NOT_GIVEN,
         provider_uri: Optional[str] | NotGiven = NOT_GIVEN,
         resource: Optional[str] | NotGiven = NOT_GIVEN,
         time_to_first_completion_token_ms: Optional[int] | NotGiven = NOT_GIVEN,

payi/resources/limits/limits.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import List, Union, Optional
+from typing import Union, Optional
 from datetime import datetime
 from typing_extensions import Literal
 
@@ -17,7 +17,7 @@ from .tags import (
     AsyncTagsResourceWithStreamingResponse,
 )
 from ...types import limit_list_params, limit_reset_params, limit_create_params, limit_update_params
-from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven, SequenceNotStr
 from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -67,7 +67,7 @@ class LimitsResource(SyncAPIResource):
         limit_name: str,
         max: float,
         limit_id: Optional[str] | NotGiven = NOT_GIVEN,
-        limit_tags: Optional[List[str]] | NotGiven = NOT_GIVEN,
+        limit_tags: Optional[SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         limit_type: Literal["block", "allow"] | NotGiven = NOT_GIVEN,
         threshold: Optional[float] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -334,7 +334,7 @@ class AsyncLimitsResource(AsyncAPIResource):
         limit_name: str,
         max: float,
         limit_id: Optional[str] | NotGiven = NOT_GIVEN,
-        limit_tags: Optional[List[str]] | NotGiven = NOT_GIVEN,
+        limit_tags: Optional[SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         limit_type: Literal["block", "allow"] | NotGiven = NOT_GIVEN,
         threshold: Optional[float] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.

payi/resources/limits/tags.py

@@ -2,11 +2,9 @@
 
 from __future__ import annotations
 
-from typing import List
-
 import httpx
 
-from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven, SequenceNotStr
 from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -51,7 +49,7 @@ class TagsResource(SyncAPIResource):
         self,
         limit_id: str,
         *,
-        limit_tags: List[str],
+        limit_tags: SequenceNotStr[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,
@@ -88,7 +86,7 @@ class TagsResource(SyncAPIResource):
         self,
         limit_id: str,
         *,
-        limit_tags: List[str],
+        limit_tags: SequenceNotStr[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,
@@ -191,7 +189,7 @@ class TagsResource(SyncAPIResource):
         self,
         limit_id: str,
         *,
-        limit_tags: List[str],
+        limit_tags: SequenceNotStr[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,
@@ -249,7 +247,7 @@ class AsyncTagsResource(AsyncAPIResource):
         self,
         limit_id: str,
         *,
-        limit_tags: List[str],
+        limit_tags: SequenceNotStr[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,
@@ -286,7 +284,7 @@ class AsyncTagsResource(AsyncAPIResource):
         self,
         limit_id: str,
         *,
-        limit_tags: List[str],
+        limit_tags: SequenceNotStr[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,
@@ -389,7 +387,7 @@ class AsyncTagsResource(AsyncAPIResource):
         self,
         limit_id: str,
         *,
-        limit_tags: List[str],
+        limit_tags: SequenceNotStr[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/use_cases/definitions/limit_config.py

@@ -2,12 +2,12 @@
 
 from __future__ import annotations
 
-from typing import List, Optional
+from typing import Optional
 from typing_extensions import Literal
 
 import httpx
 
-from ...._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ...._types import NOT_GIVEN, Body, Query, Headers, NotGiven, SequenceNotStr
 from ...._utils import maybe_transform, async_maybe_transform
 from ...._compat import cached_property
 from ...._resource import SyncAPIResource, AsyncAPIResource
@@ -49,7 +49,7 @@ class LimitConfigResource(SyncAPIResource):
         use_case_name: str,
         *,
         max: float,
-        limit_tags: Optional[List[str]] | NotGiven = NOT_GIVEN,
+        limit_tags: Optional[SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         limit_type: Literal["block", "allow"] | NotGiven = NOT_GIVEN,
         threshold: Optional[float] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -149,7 +149,7 @@ class AsyncLimitConfigResource(AsyncAPIResource):
         use_case_name: str,
         *,
         max: float,
-        limit_tags: Optional[List[str]] | NotGiven = NOT_GIVEN,
+        limit_tags: Optional[SequenceNotStr[str]] | NotGiven = NOT_GIVEN,
         limit_type: Literal["block", "allow"] | NotGiven = NOT_GIVEN,
         threshold: Optional[float] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.

payi/types/categories/fixed_cost_resource_create_params.py

@@ -2,10 +2,11 @@
 
 from __future__ import annotations
 
-from typing import List, Union
+from typing import Union
 from datetime import datetime
 from typing_extensions import Required, Annotated, TypedDict
 
+from ..._types import SequenceNotStr
 from ..._utils import PropertyInfo
 
 __all__ = ["FixedCostResourceCreateParams"]
@@ -14,7 +15,7 @@ __all__ = ["FixedCostResourceCreateParams"]
 class FixedCostResourceCreateParams(TypedDict, total=False):
     category: Required[str]
 
-    units: Required[List[str]]
+    units: Required[SequenceNotStr[str]]
 
     cost_per_hour: float
 

payi/types/ingest_event_param.py

@@ -2,10 +2,11 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Union, Iterable, Optional
+from typing import Dict, Union, Iterable, Optional
 from datetime import datetime
 from typing_extensions import Required, Annotated, TypedDict
 
+from .._types import SequenceNotStr
 from .._utils import PropertyInfo
 from .shared_params.ingest_units import IngestUnits
 from .pay_i_common_models_api_router_header_info_param import PayICommonModelsAPIRouterHeaderInfoParam
@@ -32,7 +33,7 @@ class IngestEventParam(TypedDict, total=False):
 
     http_status_code: Optional[int]
 
-    limit_ids: Optional[List[str]]
+    limit_ids: Optional[SequenceNotStr[str]]
 
     properties: Optional[Dict[str, str]]
 
@@ -48,11 +49,11 @@ class IngestEventParam(TypedDict, total=False):
 
     provider_response_id: Optional[str]
 
-    provider_response_json: Union[str, List[str], None]
+    provider_response_json: Union[str, SequenceNotStr[str], None]
 
     provider_uri: Optional[str]
 
-    request_tags: Optional[List[str]]
+    request_tags: Optional[SequenceNotStr[str]]
 
     resource: Optional[str]
 

payi/types/ingest_units_params.py

@@ -2,10 +2,11 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Union, Iterable, Optional
+from typing import Dict, Union, Iterable, Optional
 from datetime import datetime
 from typing_extensions import Required, Annotated, TypedDict
 
+from .._types import SequenceNotStr
 from .._utils import PropertyInfo
 from .shared_params.ingest_units import IngestUnits
 from .pay_i_common_models_api_router_header_info_param import PayICommonModelsAPIRouterHeaderInfoParam
@@ -38,7 +39,7 @@ class IngestUnitsParams(TypedDict, total=False):
 
     provider_response_id: Optional[str]
 
-    provider_response_json: Union[str, List[str], None]
+    provider_response_json: Union[str, SequenceNotStr[str], None]
 
     provider_uri: Optional[str]
 

payi/types/limit_create_params.py

@@ -2,9 +2,11 @@
 
 from __future__ import annotations
 
-from typing import List, Optional
+from typing import Optional
 from typing_extensions import Literal, Required, TypedDict
 
+from .._types import SequenceNotStr
+
 __all__ = ["LimitCreateParams"]
 
 
@@ -15,7 +17,7 @@ class LimitCreateParams(TypedDict, total=False):
 
     limit_id: Optional[str]
 
-    limit_tags: Optional[List[str]]
+    limit_tags: Optional[SequenceNotStr[str]]
 
     limit_type: Literal["block", "allow"]
 

payi/types/limits/tag_create_params.py

@@ -2,12 +2,13 @@
 
 from __future__ import annotations
 
-from typing import List
 from typing_extensions import Required, TypedDict
 
+from ..._types import SequenceNotStr
+
 __all__ = ["TagCreateParams"]
 
 
 class TagCreateParams(TypedDict, total=False):
-    limit_tags: Required[List[str]]
+    limit_tags: Required[SequenceNotStr[str]]
     """List of limit tags"""

payi/types/limits/tag_remove_params.py

@@ -2,12 +2,13 @@
 
 from __future__ import annotations
 
-from typing import List
 from typing_extensions import Required, TypedDict
 
+from ..._types import SequenceNotStr
+
 __all__ = ["TagRemoveParams"]
 
 
 class TagRemoveParams(TypedDict, total=False):
-    limit_tags: Required[List[str]]
+    limit_tags: Required[SequenceNotStr[str]]
     """List of limit tags"""

payi/types/limits/tag_update_params.py

@@ -2,12 +2,13 @@
 
 from __future__ import annotations
 
-from typing import List
 from typing_extensions import Required, TypedDict
 
+from ..._types import SequenceNotStr
+
 __all__ = ["TagUpdateParams"]
 
 
 class TagUpdateParams(TypedDict, total=False):
-    limit_tags: Required[List[str]]
+    limit_tags: Required[SequenceNotStr[str]]
     """List of limit tags"""

payi/types/shared_params/pay_i_common_models_budget_management_create_limit_base.py

@@ -2,16 +2,18 @@
 
 from __future__ import annotations
 
-from typing import List, Optional
+from typing import Optional
 from typing_extensions import Literal, Required, TypedDict
 
+from ..._types import SequenceNotStr
+
 __all__ = ["PayICommonModelsBudgetManagementCreateLimitBase"]
 
 
 class PayICommonModelsBudgetManagementCreateLimitBase(TypedDict, total=False):
     max: Required[float]
 
-    limit_tags: Optional[List[str]]
+    limit_tags: Optional[SequenceNotStr[str]]
 
     limit_type: Literal["block", "allow"]
 

payi/types/use_cases/definitions/limit_config_create_params.py

@@ -2,16 +2,18 @@
 
 from __future__ import annotations
 
-from typing import List, Optional
+from typing import Optional
 from typing_extensions import Literal, Required, TypedDict
 
+from ...._types import SequenceNotStr
+
 __all__ = ["LimitConfigCreateParams"]
 
 
 class LimitConfigCreateParams(TypedDict, total=False):
     max: Required[float]
 
-    limit_tags: Optional[List[str]]
+    limit_tags: Optional[SequenceNotStr[str]]
 
     limit_type: Literal["block", "allow"]
 

{payi-0.1.0a113.dist-info → payi-0.1.0a115.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: payi
-Version: 0.1.0a113
+Version: 0.1.0a115
 Summary: The official Python library for the payi API
 Project-URL: Homepage, https://github.com/Pay-i/pay-i-python
 Project-URL: Repository, https://github.com/Pay-i/pay-i-python