payi 0.1.0a5__py3-none-any.whl → 0.1.0a7__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.5"  # x-release-please-version
+__version__ = "0.1.0-alpha.7"  # 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

@@ -22,7 +22,7 @@ from .._response import (
 from .._base_client import (
     make_request_options,
 )
-from ..types.successful_proxy_result import SuccessfulProxyResult
+from ..types.proxy_result import ProxyResult
 
 __all__ = ["IngestResource", "AsyncIngestResource"]
 
@@ -51,7 +51,7 @@ class IngestResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> SuccessfulProxyResult:
+    ) -> ProxyResult:
         """
         Ingest a request
 
@@ -103,7 +103,7 @@ class IngestResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=SuccessfulProxyResult,
+            cast_to=ProxyResult,
         )
 
 
@@ -131,7 +131,7 @@ class AsyncIngestResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> SuccessfulProxyResult:
+    ) -> ProxyResult:
         """
         Ingest a request
 
@@ -183,7 +183,7 @@ class AsyncIngestResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=SuccessfulProxyResult,
+            cast_to=ProxyResult,
         )
 
 

payi/types/__init__.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+from .proxy_result import ProxyResult as ProxyResult
 from .budget_response import BudgetResponse as BudgetResponse
 from .default_response import DefaultResponse as DefaultResponse
 from .paged_budget_list import PagedBudgetList as PagedBudgetList
@@ -10,4 +11,3 @@ from .ingest_units_params import IngestUnitsParams as IngestUnitsParams
 from .budget_create_params import BudgetCreateParams as BudgetCreateParams
 from .budget_update_params import BudgetUpdateParams as BudgetUpdateParams
 from .budget_history_response import BudgetHistoryResponse as BudgetHistoryResponse
-from .successful_proxy_result import SuccessfulProxyResult as SuccessfulProxyResult

payi/types/budgets/__init__.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+from .budget_tags import BudgetTags as BudgetTags
 from .tag_create_params import TagCreateParams as TagCreateParams
 from .tag_list_response import TagListResponse as TagListResponse
 from .tag_remove_params import TagRemoveParams as TagRemoveParams
@@ -10,4 +11,3 @@ from .tag_create_response import TagCreateResponse as TagCreateResponse
 from .tag_delete_response import TagDeleteResponse as TagDeleteResponse
 from .tag_remove_response import TagRemoveResponse as TagRemoveResponse
 from .tag_update_response import TagUpdateResponse as TagUpdateResponse
-from .budget_tags_response import BudgetTagsResponse as BudgetTagsResponse

payi/types/budgets/budget_tags.py

@@ -0,0 +1,16 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from datetime import datetime
+
+from ..._models import BaseModel
+
+__all__ = ["BudgetTags"]
+
+
+class BudgetTags(BaseModel):
+    created_on: Optional[datetime] = None
+
+    tag_id: Optional[int] = None
+
+    tag_name: Optional[str] = None

payi/types/budgets/tag_create_response.py

@@ -2,8 +2,8 @@
 
 from typing import List
 
-from .budget_tags_response import BudgetTagsResponse
+from .budget_tags import BudgetTags
 
 __all__ = ["TagCreateResponse"]
 
-TagCreateResponse = List[BudgetTagsResponse]
+TagCreateResponse = List[BudgetTags]

payi/types/budgets/tag_delete_response.py

@@ -2,8 +2,8 @@
 
 from typing import List
 
-from .budget_tags_response import BudgetTagsResponse
+from .budget_tags import BudgetTags
 
 __all__ = ["TagDeleteResponse"]
 
-TagDeleteResponse = List[BudgetTagsResponse]
+TagDeleteResponse = List[BudgetTags]

payi/types/budgets/tag_list_response.py

@@ -2,8 +2,8 @@
 
 from typing import List
 
-from .budget_tags_response import BudgetTagsResponse
+from .budget_tags import BudgetTags
 
 __all__ = ["TagListResponse"]
 
-TagListResponse = List[BudgetTagsResponse]
+TagListResponse = List[BudgetTags]

payi/types/budgets/tag_remove_response.py

@@ -2,8 +2,8 @@
 
 from typing import List
 
-from .budget_tags_response import BudgetTagsResponse
+from .budget_tags import BudgetTags
 
 __all__ = ["TagRemoveResponse"]
 
-TagRemoveResponse = List[BudgetTagsResponse]
+TagRemoveResponse = List[BudgetTags]

payi/types/budgets/tag_update_response.py

@@ -2,8 +2,8 @@
 
 from typing import List
 
-from .budget_tags_response import BudgetTagsResponse
+from .budget_tags import BudgetTags
 
 __all__ = ["TagUpdateResponse"]
 
-TagUpdateResponse = List[BudgetTagsResponse]
+TagUpdateResponse = List[BudgetTags]

payi/types/{successful_proxy_result.py → proxy_result.py}

@@ -4,7 +4,7 @@ from typing import Dict, List, Optional
 
 from .._models import BaseModel
 
-__all__ = ["SuccessfulProxyResult", "Budgets", "Cost", "CostInput", "CostOutput", "CostTotal"]
+__all__ = ["ProxyResult", "Budgets", "Cost", "CostInput", "CostOutput", "CostTotal"]
 
 
 class Budgets(BaseModel):
@@ -33,7 +33,7 @@ class Cost(BaseModel):
     total: Optional[CostTotal] = None
 
 
-class SuccessfulProxyResult(BaseModel):
+class ProxyResult(BaseModel):
     budgets: Optional[Dict[str, Budgets]] = None
 
     cost: Optional[Cost] = None

{payi-0.1.0a5.dist-info → payi-0.1.0a7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: payi
-Version: 0.1.0a5
+Version: 0.1.0a7
 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.0a5.dist-info → payi-0.1.0a7.dist-info}/RECORD

@@ -11,24 +11,25 @@ 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=3mM6E715PZrcQHeMazvg0NDPTx5QFmYESDWiJMLk_UQ,164
+payi/_version.py,sha256=-vmo3fywvo5AGfodfzLGcUWjnejIqRqvWr3CvKlqw38,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=IByW243qermlbqhlTYB_ZjhJ7Jfck6OREVJWsn2vATc,7692
+payi/resources/ingest.py,sha256=bMWFiUsMbV913HmE61CEGnPcqTydM6OLNQP22Kg1_Eg,7631
 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
-payi/types/__init__.py,sha256=qMtB76eknPySrqAU23SGwJYEelZBi2PWeqQO5p4dmWo,775
+payi/types/__init__.py,sha256=JMz6gOsVi5Z6cdRl7S-5PTPvRwYaywHJkNYRIZnjrIE,744
 payi/types/budget_create_params.py,sha256=dDQvPOY8lx5oGuumAYQybAx-jPTRupsipVeezBd0l7I,543
 payi/types/budget_history_response.py,sha256=SeOabYz6xQvH4YkJSejbk_dUB49LZy7Y2MXZFPvrYMY,2702
 payi/types/budget_list_params.py,sha256=Win-Spveurf6WarTgCXXsP-yDIxr_HCP7oV_4JbqKOc,674
@@ -37,18 +38,18 @@ payi/types/budget_update_params.py,sha256=TTH7uu1c1zBK5zYhh2SPuRWEwr1XwDX_8bVS_G
 payi/types/default_response.py,sha256=o617LpRsCIZHCZxAc5nVI2JQ3HPGZo4gCDvSDkxkIJ8,270
 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
-payi/types/budgets/budget_tags_response.py,sha256=o3zaMs9b0gJqKrWuui3ohUVsR2cwNGR0yttBclgXapo,513
+payi/types/proxy_result.py,sha256=qyROzJvfaJDWHZ8UKxCFmf2HyechYri9UzQDwjqdWgc,873
+payi/types/budgets/__init__.py,sha256=5ByVuqOcgSYLHmGzhuj6-HG48mtQhC6oT5U5tNjQtxk,725
+payi/types/budgets/budget_tags.py,sha256=nOYMf7YgRdTZgGdedJw0UbQ2uupCq9MHlevLJoXxWB4,348
 payi/types/budgets/tag_create_params.py,sha256=_zEYfH2uKF44WenAdedx4TB9Az6RbIoAUMwqW8SAiuc,314
-payi/types/budgets/tag_create_response.py,sha256=2SWGig_rx8J5fEN0XUX2t6tZLnS5OQ1PcD5Bxc46rdY,244
-payi/types/budgets/tag_delete_response.py,sha256=JuXfffM_poGX_K1lLWCkW8yUdUiod1XNc8s-iN-RYdY,244
-payi/types/budgets/tag_list_response.py,sha256=9PUbi3clybFPD7Xo4Op6GmSjFet2gw7P2Kyx7IJg0_s,240
+payi/types/budgets/tag_create_response.py,sha256=ybpccmzfOnLrgzRNYj_dDQF6NVFwBM7HdakqAU0nQpQ,219
+payi/types/budgets/tag_delete_response.py,sha256=z7LQaTsdjQ3ilWsaKKHr5VLAsqa80tdFmDkEkQC96Ok,219
+payi/types/budgets/tag_list_response.py,sha256=yfdMEFy4Cy0-7EPbaedjBJXwiLnAjmSuFt0PdRIL1ok,215
 payi/types/budgets/tag_remove_params.py,sha256=fNiHex1U11o_B0bfJDkpGARkEf99J-bd-wbzKkZ9mVM,314
-payi/types/budgets/tag_remove_response.py,sha256=Gc_eBGGSx6Afa51g-S5AkXqwmRcGMjshiNSBsthDZJc,244
+payi/types/budgets/tag_remove_response.py,sha256=Wt4KwC76y3qxdEqKhXzFkYu2iECYePgXcpidXkuheWQ,219
 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.0a5.dist-info/METADATA,sha256=UO6iAhZQ0DZpRyi0xAQY7T-bekMrtPNIUFxPmZKkfeg,11872
-payi-0.1.0a5.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-payi-0.1.0a5.dist-info/licenses/LICENSE,sha256=8vX1pjh3esb6D5DvXAf6NxiBcVyon8aHWNJCxmmHXeY,11334
-payi-0.1.0a5.dist-info/RECORD,,
+payi/types/budgets/tag_update_response.py,sha256=Vlf-zxLZzvmhUodhzF7hiqWcxyjrb7Ri_lSajg8vOPM,219
+payi-0.1.0a7.dist-info/METADATA,sha256=INYWWYKrbqUJ33Qxv3DwK0rYSCXa_GqFLDLQS1wLARM,11872
+payi-0.1.0a7.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+payi-0.1.0a7.dist-info/licenses/LICENSE,sha256=8vX1pjh3esb6D5DvXAf6NxiBcVyon8aHWNJCxmmHXeY,11334
+payi-0.1.0a7.dist-info/RECORD,,

payi/types/budgets/budget_tags_response.py

@@ -1,18 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from typing import Optional
-from datetime import datetime
-
-from pydantic import Field as FieldInfo
-
-from ..._models import BaseModel
-
-__all__ = ["BudgetTagsResponse"]
-
-
-class BudgetTagsResponse(BaseModel):
-    created_on: Optional[datetime] = FieldInfo(alias="createdOn", default=None)
-
-    tag_id: Optional[int] = FieldInfo(alias="tagId", default=None)
-
-    tag_name: Optional[str] = FieldInfo(alias="tagName", default=None)