payi 0.1.0a4__py3-none-any.whl → 0.1.0a6__py3-none-any.whl

payi/_utils/__init__.py

@@ -49,4 +49,7 @@ from ._transform import (
     maybe_transform as maybe_transform,
     async_maybe_transform as async_maybe_transform,
 )
-from ._reflection import function_has_argument as function_has_argument
+from ._reflection import (
+    function_has_argument as function_has_argument,
+    assert_signatures_in_sync as assert_signatures_in_sync,
+)

payi/_utils/_reflection.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import inspect
 from typing import Any, Callable
 
@@ -6,3 +8,35 @@ def function_has_argument(func: Callable[..., Any], arg_name: str) -> bool:
     """Returns whether or not the given function has a specific parameter"""
     sig = inspect.signature(func)
     return arg_name in sig.parameters
+
+
+def assert_signatures_in_sync(
+    source_func: Callable[..., Any],
+    check_func: Callable[..., Any],
+    *,
+    exclude_params: set[str] = set(),
+) -> None:
+    """Ensure that the signature of the second function matches the first."""
+
+    check_sig = inspect.signature(check_func)
+    source_sig = inspect.signature(source_func)
+
+    errors: list[str] = []
+
+    for name, source_param in source_sig.parameters.items():
+        if name in exclude_params:
+            continue
+
+        custom_param = check_sig.parameters.get(name)
+        if not custom_param:
+            errors.append(f"the `{name}` param is missing")
+            continue
+
+        if custom_param.annotation != source_param.annotation:
+            errors.append(
+                f"types for the `{name}` param are do not match; source={repr(source_param.annotation)} checking={repr(source_param.annotation)}"
+            )
+            continue
+
+    if errors:
+        raise AssertionError(f"{len(errors)} errors encountered when comparing signatures:\n\n" + "\n\n".join(errors))

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

payi/lib/helpers.py

@@ -0,0 +1,32 @@
+# Step 1: Define the new methods outside of the Payi class definition
+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")
+
+    valid_ids = [id.strip() for id in budget_ids if isinstance(id, str) and id.strip()] # type: ignore
+
+    return {"xProxy-Budget-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
+        raise TypeError("request_tags must be a list")
+
+    valid_tags = [tag.strip() for tag in request_tags if isinstance(tag, str) and tag.strip()] # type: ignore
+    
+    return {"xProxy-Request-Tags": ",".join(valid_tags)} if valid_tags else {}
+
+def create_headers(
+    budget_ids: Union[List[str], None] = None, 
+    request_tags: Union[List[str], None] = None 
+    ) -> Dict[str, str]:
+    headers: Dict[str, str] = {}
+
+    if budget_ids:
+        headers.update(create_budget_header_from_ids(budget_ids))
+    if request_tags:
+        headers.update(create_request_header_from_tags(request_tags))
+
+    return headers

payi/resources/ingest.py

@@ -43,8 +43,8 @@ class IngestResource(SyncAPIResource):
         input: int,
         output: int,
         resource: str,
-        budget_ids: str | NotGiven = NOT_GIVEN,
-        request_tags: str | NotGiven = NOT_GIVEN,
+        budget_ids: list[str] | NotGiven = NOT_GIVEN,
+        request_tags: list[str] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -64,11 +64,27 @@ class IngestResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+        valid_ids_str: str | NotGiven = NOT_GIVEN
+        valid_tags_str: str | NotGiven = NOT_GIVEN
+
+        if isinstance(budget_ids, NotGiven):
+            valid_ids_str = NOT_GIVEN
+        else:
+            # Proceed with the list comprehension if budget_ids is not NotGiven
+            valid_ids = [id.strip() for id in budget_ids if id.strip()]
+            valid_ids_str = ",".join(valid_ids) if valid_ids else NOT_GIVEN
+        if isinstance(request_tags, NotGiven):
+            valid_tags_str = NOT_GIVEN
+        else:
+            # Proceed with the list comprehension if budget_ids 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
+
         extra_headers = {
             **strip_not_given(
                 {
-                    "xProxy-Budget-IDs": budget_ids,
-                    "xProxy-Request-Tags": request_tags,
+                    "xProxy-Budget-IDs": valid_ids_str,
+                    "xProxy-Request-Tags": valid_tags_str,
                 }
             ),
             **(extra_headers or {}),
@@ -107,8 +123,8 @@ class AsyncIngestResource(AsyncAPIResource):
         input: int,
         output: int,
         resource: str,
-        budget_ids: str | NotGiven = NOT_GIVEN,
-        request_tags: str | NotGiven = NOT_GIVEN,
+        budget_ids: list[str] | NotGiven = NOT_GIVEN,
+        request_tags: list[str] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -128,11 +144,27 @@ class AsyncIngestResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+        valid_ids_str: str | NotGiven = NOT_GIVEN
+        valid_tags_str: str | NotGiven = NOT_GIVEN
+
+        if isinstance(budget_ids, NotGiven):
+            valid_ids_str = NOT_GIVEN
+        else:
+            # Proceed with the list comprehension if budget_ids is not NotGiven
+            valid_ids = [id.strip() for id in budget_ids if id.strip()]
+            valid_ids_str = ",".join(valid_ids) if valid_ids else NOT_GIVEN
+        if isinstance(request_tags, NotGiven):
+            valid_tags_str = NOT_GIVEN
+        else:
+            # Proceed with the list comprehension if budget_ids 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
+
         extra_headers = {
             **strip_not_given(
                 {
-                    "xProxy-Budget-IDs": budget_ids,
-                    "xProxy-Request-Tags": request_tags,
+                    "xProxy-Budget-IDs": valid_ids_str,
+                    "xProxy-Request-Tags": valid_tags_str,
                 }
             ),
             **(extra_headers or {}),

payi/types/ingest_units_params.py

@@ -18,6 +18,6 @@ class IngestUnitsParams(TypedDict, total=False):
 
     resource: Required[str]
 
-    budget_ids: Annotated[str, PropertyInfo(alias="xProxy-Budget-IDs")]
+    budget_ids: Annotated[list[str], PropertyInfo(alias="xProxy-Budget-IDs")]
 
-    request_tags: Annotated[str, PropertyInfo(alias="xProxy-Request-Tags")]
+    request_tags: Annotated[list[str], PropertyInfo(alias="xProxy-Request-Tags")]

{payi-0.1.0a4.dist-info → payi-0.1.0a6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: payi
-Version: 0.1.0a4
+Version: 0.1.0a6
 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

{payi-0.1.0a4.dist-info → payi-0.1.0a6.dist-info}/RECORD

@@ -11,20 +11,21 @@ payi/_resource.py,sha256=j2jIkTr8OIC8sU6-05nxSaCyj4MaFlbZrwlyg4_xJos,1088
 payi/_response.py,sha256=W6pADUzbnMFGnVzGY4M3FkX9uvw8AFSMTPZKPgQ_TV4,28363
 payi/_streaming.py,sha256=Z_wIyo206T6Jqh2rolFg2VXZgX24PahLmpURp0-NssU,10092
 payi/_types.py,sha256=04q-KHD-qZ1xlfwEb_muyIHWmD-nZ-KxnRxA_QEjqNk,6125
-payi/_version.py,sha256=JnOWhSSjvRoW9IyOBb60KZeTaqp-L3yZnbSNJszRL1I,164
+payi/_version.py,sha256=uCFQ98dFvMtHydofeML_9k-XyAQblroEZWJv-OZ3N6g,164
 payi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-payi/_utils/__init__.py,sha256=q5QC6ZmVNli9QaCqTlp9VrbQyjIU9EzmCn044UhHSIk,1919
+payi/_utils/__init__.py,sha256=Uzq1-FIih_VUjzdNVWXks0sdC39KBKLMrZoz-_JOjJ4,1988
 payi/_utils/_logs.py,sha256=fmnf5D9TOgkgZKfgYmSa3PiUc3SZgkchn6CzJUeo0SQ,768
 payi/_utils/_proxy.py,sha256=DjcB-BBIRagSbMut2pF_jZavjda9sPvmQCKtVXBhs0I,1910
-payi/_utils/_reflection.py,sha256=Xloc_oanCC3gmAVBRhlMYtvYeLNZGi46gGWhDt9FtKI,275
+payi/_utils/_reflection.py,sha256=k20KwLejVHcQCvu4mT2S61NDvbmuXF7KsMDapiocYS0,1364
 payi/_utils/_streams.py,sha256=SMC90diFFecpEg_zgDRVbdR3hSEIgVVij4taD-noMLM,289
 payi/_utils/_sync.py,sha256=9ex9pfOyd8xAF1LxpFx4IkqL8k0vk8srE2Ee-OTMQ0A,2840
 payi/_utils/_transform.py,sha256=NCz3q9_O-vuj60xVe-qzhEQ8uJWlZWJTsM-GwHDccf8,12958
 payi/_utils/_typing.py,sha256=tFbktdpdHCQliwzGsWysgn0P5H0JRdagkZdb_LegGkY,3838
 payi/_utils/_utils.py,sha256=FaZdW0tWil7IERdxUfKt7pVcyXL2aCnR3lo73q66qgI,11447
 payi/lib/.keep,sha256=wuNrz-5SXo3jJaJOJgz4vFHM41YH_g20F5cRQo0vLes,224
+payi/lib/helpers.py,sha256=IdB_pi8dLtVh9b4DG-zasA1i9Yk-5xAayoUTNWBAFpc,1235
 payi/resources/__init__.py,sha256=1p8kzMES8NNDQQoK3TEDE-8vVuQ59xxUjyyblZJBTjw,1015
-payi/resources/ingest.py,sha256=xecLJfQ3n5SOFSY4rYKkj-4A8w5vBEcH0fTB6CCcRqY,6128
+payi/resources/ingest.py,sha256=IByW243qermlbqhlTYB_ZjhJ7Jfck6OREVJWsn2vATc,7692
 payi/resources/budgets/__init__.py,sha256=w1UhOdDXtUH4A91ME5Tw2nr9bRvPJyJY1YWiVVy7jj0,989
 payi/resources/budgets/budgets.py,sha256=aMu39MrRsVTtNfYwQPMeFK329ouwnHuFXM_vK9Ewj0k,24047
 payi/resources/budgets/tags.py,sha256=jzP0nal287bL2htY1TwDcdFRJyNMm6IdM936O1jCeMo,17850
@@ -35,7 +36,7 @@ payi/types/budget_list_params.py,sha256=Win-Spveurf6WarTgCXXsP-yDIxr_HCP7oV_4Jbq
 payi/types/budget_response.py,sha256=yXDXIrespfRuNreWdWmlCITO6VvLC4YZn0nzXw_BGzo,2072
 payi/types/budget_update_params.py,sha256=TTH7uu1c1zBK5zYhh2SPuRWEwr1XwDX_8bVS_GxIn9I,306
 payi/types/default_response.py,sha256=o617LpRsCIZHCZxAc5nVI2JQ3HPGZo4gCDvSDkxkIJ8,270
-payi/types/ingest_units_params.py,sha256=SLe9mE2N28b3H61gIOGalfHogD9grBFv3WsHQnImS58,563
+payi/types/ingest_units_params.py,sha256=NhC7wRF0lpoKikUPQUAgE1OxpU8dqMxtRZNIbiXkYCA,575
 payi/types/paged_budget_list.py,sha256=HNm834IAAy3eRNz0BjZwcbD9e-I8unWWKWxiSnlijWY,2553
 payi/types/successful_proxy_result.py,sha256=j8kQu8LAcfR3XuforqWDsYm1Bx6BQxCrezmE-3GnCWc,893
 payi/types/budgets/__init__.py,sha256=8lpRHXZQOc9ADCphhCB0JhQAH96lBIc68LbhlJn5Kys,750
@@ -48,7 +49,7 @@ payi/types/budgets/tag_remove_params.py,sha256=fNiHex1U11o_B0bfJDkpGARkEf99J-bd-
 payi/types/budgets/tag_remove_response.py,sha256=Gc_eBGGSx6Afa51g-S5AkXqwmRcGMjshiNSBsthDZJc,244
 payi/types/budgets/tag_update_params.py,sha256=bWWeDqfySt7CZ6HDsZzENREOWGWGWFgnAtyleNMwUsE,314
 payi/types/budgets/tag_update_response.py,sha256=5Q16vkLUMLkyzgCWRcz7aDVA0KTCHQd_eAKnx2-VoWQ,244
-payi-0.1.0a4.dist-info/METADATA,sha256=0oNFDUKAjdE7jU3cM4WKqGBrEVVVMR6HINTIQ8Hp91c,11872
-payi-0.1.0a4.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-payi-0.1.0a4.dist-info/licenses/LICENSE,sha256=8vX1pjh3esb6D5DvXAf6NxiBcVyon8aHWNJCxmmHXeY,11334
-payi-0.1.0a4.dist-info/RECORD,,
+payi-0.1.0a6.dist-info/METADATA,sha256=FGdRSVefjz5gmzywelk9u4MR60cFlIHDWrSUCUA5wPM,11872
+payi-0.1.0a6.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+payi-0.1.0a6.dist-info/licenses/LICENSE,sha256=8vX1pjh3esb6D5DvXAf6NxiBcVyon8aHWNJCxmmHXeY,11334
+payi-0.1.0a6.dist-info/RECORD,,