payi 0.1.0a107__tar.gz → 0.1.0a108__tar.gz

payi-0.1.0a108/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.1.0-alpha.108"
+}

{payi-0.1.0a107 → payi-0.1.0a108}/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## 0.1.0-alpha.108 (2025-08-05)
+
+Full Changelog: [v0.1.0-alpha.107...v0.1.0-alpha.108](https://github.com/Pay-i/pay-i-python/compare/v0.1.0-alpha.107...v0.1.0-alpha.108)
+
+### Features
+
+* account_name ([#352](https://github.com/Pay-i/pay-i-python/issues/352)) ([719e871](https://github.com/Pay-i/pay-i-python/commit/719e871927187e2bce511d637144fd99a59b7234))
+
 ## 0.1.0-alpha.107 (2025-07-31)
 
 Full Changelog: [v0.1.0-alpha.106...v0.1.0-alpha.107](https://github.com/Pay-i/pay-i-python/compare/v0.1.0-alpha.106...v0.1.0-alpha.107)

{payi-0.1.0a107 → payi-0.1.0a108}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: payi
-Version: 0.1.0a107
+Version: 0.1.0a108
 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.0a107 → payi-0.1.0a108}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "payi"
-version = "0.1.0-alpha.107"
+version = "0.1.0-alpha.108"
 description = "The official Python library for the payi API"
 dynamic = ["readme"]
 license = "Apache-2.0"

{payi-0.1.0a107 → payi-0.1.0a108}/src/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.107"  # x-release-please-version
+__version__ = "0.1.0-alpha.108"  # x-release-please-version

{payi-0.1.0a107 → payi-0.1.0a108}/src/payi/lib/helpers.py

@@ -11,6 +11,7 @@ class PayiHeaderNames:
     use_case_version:str = "xProxy-UseCase-Version"
     use_case_step:str = "xProxy-UseCase-Step"
     user_id:str = "xProxy-User-ID"
+    account_name:str = "xProxy-Account-Name"
     price_as_category:str = "xProxy-PriceAs-Category"
     price_as_resource:str = "xProxy-PriceAs-Resource"
     provider_base_uri = "xProxy-Provider-BaseUri"
@@ -24,7 +25,7 @@ class PayiCategories:
     aws_bedrock:str = "system.aws.bedrock"
     google_vertex:str = "system.google.vertex"
 
-def create_limit_header_from_ids(limit_ids: List[str]) -> Dict[str, str]:
+def create_limit_header_from_ids(*, limit_ids: List[str]) -> Dict[str, str]:
     if not isinstance(limit_ids, list):  # type: ignore
         raise TypeError("limit_ids must be a list")
 
@@ -32,7 +33,7 @@ def create_limit_header_from_ids(limit_ids: List[str]) -> Dict[str, str]:
 
     return { PayiHeaderNames.limit_ids: ",".join(valid_ids) } if valid_ids else {}
 
-def create_request_header_from_tags(request_tags: List[str]) -> Dict[str, str]:
+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")
 
@@ -41,9 +42,11 @@ def create_request_header_from_tags(request_tags: List[str]) -> Dict[str, str]:
     return { PayiHeaderNames.request_tags: ",".join(valid_tags) } if valid_tags else {}
 
 def create_headers(
+    *,
     limit_ids: Union[List[str], None] = None,
     request_tags: Union[List[str], None] = None,
     user_id: Union[str, None] = None,
+    account_name: Union[str, None] = None,
     use_case_id: Union[str, None] = None,
     use_case_name: Union[str, None] = None,
     use_case_version: Union[int, None] = None,
@@ -55,11 +58,13 @@ def create_headers(
     headers: Dict[str, str] = {}
 
     if limit_ids:
-        headers.update(create_limit_header_from_ids(limit_ids))
+        headers.update(create_limit_header_from_ids(limit_ids=limit_ids))
     if request_tags:
-        headers.update(create_request_header_from_tags(request_tags))
+        headers.update(create_request_header_from_tags(request_tags=request_tags))
     if user_id:
         headers.update({ PayiHeaderNames.user_id: user_id})
+    if account_name:
+        headers.update({ PayiHeaderNames.account_name: account_name})
     if use_case_id:
         headers.update({ PayiHeaderNames.use_case_id: use_case_id})
     if use_case_name:

{payi-0.1.0a107 → payi-0.1.0a108}/src/payi/lib/instrument.py

@@ -157,6 +157,7 @@ class PayiInstrumentConfig(TypedDict, total=False):
     use_case_version: Optional[int]
     use_case_properties: Optional["dict[str, str]"]
     user_id: Optional[str]
+    account_name: Optional[str]
     request_tags: Optional["list[str]"]
     request_properties: Optional["dict[str, str]"]
 
@@ -168,6 +169,7 @@ class PayiContext(TypedDict, total=False):
     use_case_properties: Optional["dict[str, str]"]
     limit_ids: Optional['list[str]']
     user_id: Optional[str]
+    account_name: Optional[str]
     request_tags: Optional["list[str]"]
     request_properties: Optional["dict[str, str]"]
     price_as_category: Optional[str]
@@ -184,6 +186,7 @@ class _Context(TypedDict, total=False):
     use_case_properties: Optional["dict[str, str]"]
     limit_ids: Optional['list[str]']
     user_id: Optional[str]
+    account_name: Optional[str]
     request_tags: Optional["list[str]"]
     request_properties: Optional["dict[str, str]"]
     price_as_category: Optional[str]
@@ -655,6 +658,7 @@ class _PayiInstrumentor:
         use_case_version: Optional[int]= None,
         use_case_step: Optional[str]= None,
         user_id: Optional[str]= None,
+        account_name: Optional[str]= None,
         request_tags: Optional["list[str]"] = None,
         request_properties: Optional["dict[str, str]"] = None,
         use_case_properties: Optional["dict[str, str]"] = None,
@@ -747,6 +751,16 @@ class _PayiInstrumentor:
         else:
             context["user_id"] = user_id
 
+        parent_account_name = parent_context.get("account_name", None)
+        if account_name is None:
+            # use the parent account_name if it exists
+            context["account_name"] = parent_account_name
+        elif len(account_name) == 0:
+            # caller passing an empty string explicitly blocks inheriting from the parent state
+            context["account_name"] = None
+        else:
+            context["account_name"] = account_name
+
         parent_request_tags = parent_context.get("request_tags", None)
         if request_tags is not None:
             if len(request_tags) == 0:
@@ -795,6 +809,7 @@ class _PayiInstrumentor:
         use_case_id: Optional[str],
         use_case_version: Optional[int],        
         user_id: Optional[str],
+        account_name: Optional[str],
         request_tags: Optional["list[str]"] = None,
         request_properties: Optional["dict[str, str]"] = None,
         use_case_properties: Optional["dict[str, str]"] = None,
@@ -809,6 +824,7 @@ class _PayiInstrumentor:
                 use_case_id=use_case_id,
                 use_case_version=use_case_version,
                 user_id=user_id,
+                account_name=account_name,
                 request_tags=request_tags,
                 request_properties=request_properties,
                 use_case_properties=use_case_properties
@@ -824,6 +840,7 @@ class _PayiInstrumentor:
         use_case_id: Optional[str],
         use_case_version: Optional[int],        
         user_id: Optional[str],
+        account_name: Optional[str],
         request_tags: Optional["list[str]"] = None,
         request_properties: Optional["dict[str, str]"] = None,
         use_case_properties: Optional["dict[str, str]"] = None,
@@ -838,6 +855,7 @@ class _PayiInstrumentor:
                 use_case_id=use_case_id,
                 use_case_version=use_case_version,
                 user_id=user_id,
+                account_name=account_name,
                 request_tags=request_tags,
                 request_properties=request_properties,
                 use_case_properties=use_case_properties)
@@ -879,6 +897,7 @@ class _PayiInstrumentor:
         use_case_step = ingest_extra_headers.pop(PayiHeaderNames.use_case_step, None)
 
         user_id = ingest_extra_headers.pop(PayiHeaderNames.user_id, None)
+        account_name = ingest_extra_headers.pop(PayiHeaderNames.account_name, None)
 
         if limit_ids:
             request._ingest["limit_ids"] = limit_ids.split(",")
@@ -894,6 +913,8 @@ class _PayiInstrumentor:
             request._ingest["use_case_step"] = use_case_step
         if user_id:
             request._ingest["user_id"] = user_id
+        if account_name:
+            request._ingest["account_name"] = account_name
 
         request_properties = context.get("request_properties", None)
         if request_properties:
@@ -1207,6 +1228,7 @@ class _PayiInstrumentor:
         context_use_case_step: Optional[str] = context.get("use_case_step")
 
         context_user_id: Optional[str] = context.get("user_id")
+        context_account_name: Optional[str] = context.get("account_name")
         context_request_tags: Optional[list[str]] = context.get("request_tags")
 
         context_price_as_category: Optional[str] = context.get("price_as_category")
@@ -1239,6 +1261,17 @@ class _PayiInstrumentor:
         elif context_user_id:
             extra_headers[PayiHeaderNames.user_id] = context_user_id
 
+        if PayiHeaderNames.account_name in extra_headers:
+            headers_account_name = extra_headers.get(PayiHeaderNames.account_name, None)
+            if headers_account_name is None or len(headers_account_name) == 0:
+                # headers_account_name is empty, remove it from extra_headers
+                extra_headers.pop(PayiHeaderNames.account_name, None)
+            else:
+                # leave the value in extra_headers
+                ...
+        elif context_account_name:
+            extra_headers[PayiHeaderNames.account_name] = context_account_name
+
         if PayiHeaderNames.use_case_name in extra_headers:
             headers_use_case_name = extra_headers.get(PayiHeaderNames.use_case_name, None)
             if headers_use_case_name is None or len(headers_use_case_name) == 0:
@@ -1711,6 +1744,7 @@ def track(
     use_case_id: Optional[str] = None,
     use_case_version: Optional[int] = None,
     user_id: Optional[str] = None,
+    account_name: Optional[str] = None,
     request_tags: Optional["list[str]"] = None,
     request_properties: Optional["dict[str, str]"] = None,
     use_case_properties: Optional["dict[str, str]"] = None,
@@ -1735,6 +1769,7 @@ def track(
                     use_case_id,
                     use_case_version,
                     user_id,
+                    account_name,
                     request_tags,
                     request_properties,
                     use_case_properties,
@@ -1748,7 +1783,7 @@ def track(
                     _g_logger.debug(f"track: no instrumentor!")
                     return func(*args, **kwargs)
 
-                _instrumentor._logger.debug(f"track: call sync function (proxy={proxy}, limit_ids={limit_ids}, use_case_name={use_case_name}, use_case_id={use_case_id}, use_case_version={use_case_version}, user_id={user_id})")
+                _instrumentor._logger.debug(f"track: call sync function (proxy={proxy}, limit_ids={limit_ids}, use_case_name={use_case_name}, use_case_id={use_case_id}, use_case_version={use_case_version}, user_id={user_id}, account_name={account_name})")
 
                 return _instrumentor._call_func(
                     func,
@@ -1758,6 +1793,7 @@ def track(
                     use_case_id,
                     use_case_version,
                     user_id,
+                    account_name,
                     request_tags,
                     request_properties,
                     use_case_properties,
@@ -1775,6 +1811,7 @@ def track_context(
     use_case_version: Optional[int] = None,
     use_case_step: Optional[str] = None,
     user_id: Optional[str] = None,
+    account_name: Optional[str] = None,
     request_tags: Optional["list[str]"] = None,
     request_properties: Optional["dict[str, str]"] = None,
     use_case_properties: Optional["dict[str, str]"] = None,
@@ -1796,6 +1833,7 @@ def track_context(
     context["use_case_step"] = use_case_step
 
     context["user_id"] = user_id
+    context["account_name"] = account_name
     context["request_tags"] = request_tags
 
     context["price_as_category"] = price_as_category

payi-0.1.0a107/.release-please-manifest.json

@@ -1,3 +0,0 @@
-{
-  ".": "0.1.0-alpha.107"
-}