payi 0.1.0a34__tar.gz → 0.1.0a36__tar.gz

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

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

{payi-0.1.0a34 → payi-0.1.0a36}/CHANGELOG.md

@@ -1,5 +1,52 @@
 # Changelog
 
+## 0.1.0-alpha.36 (2025-01-28)
+
+Full Changelog: [v0.1.0-alpha.35...v0.1.0-alpha.36](https://github.com/Pay-i/pay-i-python/compare/v0.1.0-alpha.35...v0.1.0-alpha.36)
+
+### Bug Fixes
+
+* **tests:** make test_get_platform less flaky ([#181](https://github.com/Pay-i/pay-i-python/issues/181)) ([86d5b4a](https://github.com/Pay-i/pay-i-python/commit/86d5b4aa623c276776f0bdba5fa936d5408c6ab3))
+
+
+### Chores
+
+* **internal:** avoid pytest-asyncio deprecation warning ([#182](https://github.com/Pay-i/pay-i-python/issues/182)) ([38bb22c](https://github.com/Pay-i/pay-i-python/commit/38bb22ca2fa0f6dd88942b4744f9b0ffee030d5c))
+* **internal:** codegen related update ([#178](https://github.com/Pay-i/pay-i-python/issues/178)) ([315a2c5](https://github.com/Pay-i/pay-i-python/commit/315a2c576e14d51239590f3bfc53872a288559c7))
+* **internal:** codegen related update ([#183](https://github.com/Pay-i/pay-i-python/issues/183)) ([00fdbe7](https://github.com/Pay-i/pay-i-python/commit/00fdbe791e16fa12e73e1203dd0ab02270127e27))
+* **internal:** codegen related update ([#184](https://github.com/Pay-i/pay-i-python/issues/184)) ([bf04c0e](https://github.com/Pay-i/pay-i-python/commit/bf04c0e4707fc2218da6fa6e676fd6c2e85bff07))
+* **internal:** codegen related update ([#185](https://github.com/Pay-i/pay-i-python/issues/185)) ([e0d7ab1](https://github.com/Pay-i/pay-i-python/commit/e0d7ab154ce6d0a60095d8740b9ba5949a37aa00))
+
+
+### Documentation
+
+* **raw responses:** fix duplicate `the` ([#180](https://github.com/Pay-i/pay-i-python/issues/180)) ([ef72f36](https://github.com/Pay-i/pay-i-python/commit/ef72f367b02d8a71b6ef108a8e3544be77d5bd60))
+
+## 0.1.0-alpha.35 (2025-01-12)
+
+Full Changelog: [v0.1.0-alpha.34...v0.1.0-alpha.35](https://github.com/Pay-i/pay-i-python/compare/v0.1.0-alpha.34...v0.1.0-alpha.35)
+
+### Features
+
+* **api:** manual updates ([#175](https://github.com/Pay-i/pay-i-python/issues/175)) ([3dd351e](https://github.com/Pay-i/pay-i-python/commit/3dd351ecb7d7c34ce73d8a26590000fb4dc38dbc))
+
+
+### Bug Fixes
+
+* **client:** only call .close() when needed ([#172](https://github.com/Pay-i/pay-i-python/issues/172)) ([75baf12](https://github.com/Pay-i/pay-i-python/commit/75baf1247d05a1b8a02fcfc4aa1342a11ef882b2))
+
+
+### Chores
+
+* **internal:** bump httpx dependency ([#171](https://github.com/Pay-i/pay-i-python/issues/171)) ([e7bf93a](https://github.com/Pay-i/pay-i-python/commit/e7bf93a87ae1bec99ee4f0fd049b2e4c8f6ef68d))
+* **internal:** codegen related update ([#169](https://github.com/Pay-i/pay-i-python/issues/169)) ([03a0803](https://github.com/Pay-i/pay-i-python/commit/03a0803e49689e31d3d7b42d2da7ed9e886434a8))
+* **internal:** codegen related update ([#174](https://github.com/Pay-i/pay-i-python/issues/174)) ([21d0af0](https://github.com/Pay-i/pay-i-python/commit/21d0af0f6b7a15efa62f9c514681f61fa1f9ddb8))
+
+
+### Documentation
+
+* fix typos ([#173](https://github.com/Pay-i/pay-i-python/issues/173)) ([3d97187](https://github.com/Pay-i/pay-i-python/commit/3d971877582257b349fa59bc83535b6cc3374c43))
+
 ## 0.1.0-alpha.34 (2025-01-06)
 
 Full Changelog: [v0.1.0-alpha.33...v0.1.0-alpha.34](https://github.com/Pay-i/pay-i-python/compare/v0.1.0-alpha.33...v0.1.0-alpha.34)

{payi-0.1.0a34 → payi-0.1.0a36}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: payi
-Version: 0.1.0a34
+Version: 0.1.0a36
 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
@@ -28,6 +28,7 @@ Requires-Dist: httpx<1,>=0.23.0
 Requires-Dist: pydantic<3,>=1.9.0
 Requires-Dist: sniffio
 Requires-Dist: typing-extensions<5,>=4.10
+Requires-Dist: wrapt>=1.17.2
 Description-Content-Type: text/markdown
 
 # Payi Python API library
@@ -65,7 +66,7 @@ client = Payi(
 
 limit_response = client.limits.create(
     limit_name="x",
-    max=0,
+    max=1,
 )
 print(limit_response.request_id)
 ```
@@ -92,7 +93,7 @@ client = AsyncPayi(
 async def main() -> None:
     limit_response = await client.limits.create(
         limit_name="x",
-        max=0,
+        max=1,
     )
     print(limit_response.request_id)
 
@@ -129,7 +130,7 @@ client = Payi()
 try:
     client.limits.create(
         limit_name="x",
-        max=0,
+        max=1,
     )
 except payi.APIConnectionError as e:
     print("The server could not be reached")
@@ -142,7 +143,7 @@ except payi.APIStatusError as e:
     print(e.response)
 ```
 
-Error codes are as followed:
+Error codes are as follows:
 
 | Status Code | Error Type                 |
 | ----------- | -------------------------- |
@@ -175,7 +176,7 @@ client = Payi(
 # Or, configure per-request:
 client.with_options(max_retries=5).limits.create(
     limit_name="x",
-    max=0,
+    max=1,
 )
 ```
 
@@ -201,7 +202,7 @@ client = Payi(
 # Override per-request:
 client.with_options(timeout=5.0).limits.create(
     limit_name="x",
-    max=0,
+    max=1,
 )
 ```
 
@@ -245,7 +246,7 @@ from payi import Payi
 client = Payi()
 response = client.limits.with_raw_response.create(
     limit_name="x",
-    max=0,
+    max=1,
 )
 print(response.headers.get('X-My-Header'))
 
@@ -266,7 +267,7 @@ To stream the response body, use `.with_streaming_response` instead, which requi
 ```python
 with client.limits.with_streaming_response.create(
     limit_name="x",
-    max=0,
+    max=1,
 ) as response:
     print(response.headers.get("X-My-Header"))
 
@@ -285,8 +286,7 @@ If you need to access undocumented endpoints, params, or response properties, th
 #### Undocumented endpoints
 
 To make requests to undocumented endpoints, you can make requests using `client.get`, `client.post`, and other
-http verbs. Options on the client will be respected (such as retries) will be respected when making this
-request.
+http verbs. Options on the client will be respected (such as retries) when making this request.
 
 ```py
 import httpx
@@ -358,7 +358,7 @@ with Payi() as client:
 This package generally follows [SemVer](https://semver.org/spec/v2.0.0.html) conventions, though certain backwards-incompatible changes may be released as minor versions:
 
 1. Changes that only affect static types, without breaking runtime behavior.
-2. Changes to library internals which are technically public but not intended or documented for external use. _(Please open a GitHub issue to let us know if you are relying on such internals)_.
+2. Changes to library internals which are technically public but not intended or documented for external use. _(Please open a GitHub issue to let us know if you are relying on such internals.)_
 3. Changes that we do not expect to impact the vast majority of users in practice.
 
 We take backwards-compatibility seriously and work hard to ensure you can rely on a smooth upgrade experience.

{payi-0.1.0a34 → payi-0.1.0a36}/README.md

@@ -33,7 +33,7 @@ client = Payi(
 
 limit_response = client.limits.create(
     limit_name="x",
-    max=0,
+    max=1,
 )
 print(limit_response.request_id)
 ```
@@ -60,7 +60,7 @@ client = AsyncPayi(
 async def main() -> None:
     limit_response = await client.limits.create(
         limit_name="x",
-        max=0,
+        max=1,
     )
     print(limit_response.request_id)
 
@@ -97,7 +97,7 @@ client = Payi()
 try:
     client.limits.create(
         limit_name="x",
-        max=0,
+        max=1,
     )
 except payi.APIConnectionError as e:
     print("The server could not be reached")
@@ -110,7 +110,7 @@ except payi.APIStatusError as e:
     print(e.response)
 ```
 
-Error codes are as followed:
+Error codes are as follows:
 
 | Status Code | Error Type                 |
 | ----------- | -------------------------- |
@@ -143,7 +143,7 @@ client = Payi(
 # Or, configure per-request:
 client.with_options(max_retries=5).limits.create(
     limit_name="x",
-    max=0,
+    max=1,
 )
 ```
 
@@ -169,7 +169,7 @@ client = Payi(
 # Override per-request:
 client.with_options(timeout=5.0).limits.create(
     limit_name="x",
-    max=0,
+    max=1,
 )
 ```
 
@@ -213,7 +213,7 @@ from payi import Payi
 client = Payi()
 response = client.limits.with_raw_response.create(
     limit_name="x",
-    max=0,
+    max=1,
 )
 print(response.headers.get('X-My-Header'))
 
@@ -234,7 +234,7 @@ To stream the response body, use `.with_streaming_response` instead, which requi
 ```python
 with client.limits.with_streaming_response.create(
     limit_name="x",
-    max=0,
+    max=1,
 ) as response:
     print(response.headers.get("X-My-Header"))
 
@@ -253,8 +253,7 @@ If you need to access undocumented endpoints, params, or response properties, th
 #### Undocumented endpoints
 
 To make requests to undocumented endpoints, you can make requests using `client.get`, `client.post`, and other
-http verbs. Options on the client will be respected (such as retries) will be respected when making this
-request.
+http verbs. Options on the client will be respected (such as retries) when making this request.
 
 ```py
 import httpx
@@ -326,7 +325,7 @@ with Payi() as client:
 This package generally follows [SemVer](https://semver.org/spec/v2.0.0.html) conventions, though certain backwards-incompatible changes may be released as minor versions:
 
 1. Changes that only affect static types, without breaking runtime behavior.
-2. Changes to library internals which are technically public but not intended or documented for external use. _(Please open a GitHub issue to let us know if you are relying on such internals)_.
+2. Changes to library internals which are technically public but not intended or documented for external use. _(Please open a GitHub issue to let us know if you are relying on such internals.)_
 3. Changes that we do not expect to impact the vast majority of users in practice.
 
 We take backwards-compatibility seriously and work hard to ensure you can rely on a smooth upgrade experience.

{payi-0.1.0a34 → payi-0.1.0a36}/api.md

@@ -126,3 +126,53 @@ Methods:
 - <code title="patch /api/v1/experiences/types/{experience_name}">client.experiences.types.<a href="./src/payi/resources/experiences/types.py">update</a>(experience_name, \*\*<a href="src/payi/types/experiences/type_update_params.py">params</a>) -> <a href="./src/payi/types/experiences/experience_type.py">ExperienceType</a></code>
 - <code title="get /api/v1/experiences/types">client.experiences.types.<a href="./src/payi/resources/experiences/types.py">list</a>(\*\*<a href="src/payi/types/experiences/type_list_params.py">params</a>) -> <a href="./src/payi/types/experiences/type_list_response.py">TypeListResponse</a></code>
 - <code title="delete /api/v1/experiences/types/{experience_name}">client.experiences.types.<a href="./src/payi/resources/experiences/types.py">delete</a>(experience_name) -> <a href="./src/payi/types/experiences/experience_type.py">ExperienceType</a></code>
+
+## Properties
+
+Types:
+
+```python
+from payi.types.experiences import PropertiesResponse
+```
+
+Methods:
+
+- <code title="post /api/v1/experiences/instances/{experience_id}/properties">client.experiences.properties.<a href="./src/payi/resources/experiences/properties.py">create</a>(experience_id, \*\*<a href="src/payi/types/experiences/property_create_params.py">params</a>) -> <a href="./src/payi/types/experience_instance_response.py">ExperienceInstanceResponse</a></code>
+
+# BillingModels
+
+Types:
+
+```python
+from payi.types import BillingModel, BillingModelListResponse
+```
+
+Methods:
+
+- <code title="post /api/v1/billing-model">client.billing_models.<a href="./src/payi/resources/billing_models.py">create</a>(\*\*<a href="src/payi/types/billing_model_create_params.py">params</a>) -> <a href="./src/payi/types/billing_model.py">BillingModel</a></code>
+- <code title="get /api/v1/billing-model/{billing_model_id}">client.billing_models.<a href="./src/payi/resources/billing_models.py">retrieve</a>(billing_model_id) -> <a href="./src/payi/types/billing_model.py">BillingModel</a></code>
+- <code title="put /api/v1/billing-model/{billing_model_id}">client.billing_models.<a href="./src/payi/resources/billing_models.py">update</a>(billing_model_id, \*\*<a href="src/payi/types/billing_model_update_params.py">params</a>) -> <a href="./src/payi/types/billing_model.py">BillingModel</a></code>
+- <code title="get /api/v1/billing-model">client.billing_models.<a href="./src/payi/resources/billing_models.py">list</a>() -> <a href="./src/payi/types/billing_model_list_response.py">BillingModelListResponse</a></code>
+- <code title="put /api/v1/billing-model/{billing_model_id}/default">client.billing_models.<a href="./src/payi/resources/billing_models.py">set_default</a>(billing_model_id) -> <a href="./src/payi/types/billing_model.py">BillingModel</a></code>
+
+# PriceModifiers
+
+Types:
+
+```python
+from payi.types import PriceModifier, PriceModifierRetrieveResponse
+```
+
+Methods:
+
+- <code title="post /api/v1/price-modifier">client.price_modifiers.<a href="./src/payi/resources/price_modifiers.py">create</a>(\*\*<a href="src/payi/types/price_modifier_create_params.py">params</a>) -> <a href="./src/payi/types/price_modifier.py">PriceModifier</a></code>
+- <code title="get /api/v1/price-modifier/{billing_model_id}">client.price_modifiers.<a href="./src/payi/resources/price_modifiers.py">retrieve</a>(billing_model_id) -> <a href="./src/payi/types/price_modifier_retrieve_response.py">PriceModifierRetrieveResponse</a></code>
+- <code title="put /api/v1/price-modifier">client.price_modifiers.<a href="./src/payi/resources/price_modifiers.py">update</a>(\*\*<a href="src/payi/types/price_modifier_update_params.py">params</a>) -> <a href="./src/payi/types/price_modifier.py">PriceModifier</a></code>
+
+# Requests
+
+## Properties
+
+Methods:
+
+- <code title="post /api/v1/requests/{request_id}/properties">client.requests.properties.<a href="./src/payi/resources/requests/properties.py">create</a>(request_id, \*\*<a href="src/payi/types/requests/property_create_params.py">params</a>) -> <a href="./src/payi/types/experiences/properties_response.py">PropertiesResponse</a></code>

{payi-0.1.0a34 → payi-0.1.0a36}/mypy.ini

@@ -41,7 +41,7 @@ cache_fine_grained = True
 # ```
 # Changing this codegen to make mypy happy would increase complexity
 # and would not be worth it.
-disable_error_code = func-returns-value
+disable_error_code = func-returns-value,overload-cannot-match
 
 # https://github.com/python/mypy/issues/12162
 [mypy.overrides]

{payi-0.1.0a34 → payi-0.1.0a36}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "payi"
-version = "0.1.0-alpha.34"
+version = "0.1.0-alpha.36"
 description = "The official Python library for the payi API"
 dynamic = ["readme"]
 license = "Apache-2.0"
@@ -14,6 +14,7 @@ dependencies = [
     "anyio>=3.5.0, <5",
     "distro>=1.7.0, <2",
     "sniffio",
+    "wrapt>=1.17.2",
 ]
 requires-python = ">= 3.8"
 classifiers = [
@@ -54,7 +55,8 @@ dev-dependencies = [
     "dirty-equals>=0.6.0",
     "importlib-metadata>=6.7.0",
     "rich>=13.7.1",
-    "nest_asyncio==1.6.0"
+    "nest_asyncio==1.6.0",
+    "wrapt",
 ]
 
 [tool.rye.scripts]
@@ -129,6 +131,7 @@ testpaths = ["tests"]
 addopts = "--tb=short"
 xfail_strict = true
 asyncio_mode = "auto"
+asyncio_default_fixture_loop_scope = "session"
 filterwarnings = [
   "error"
 ]

{payi-0.1.0a34 → payi-0.1.0a36}/requirements-dev.lock

@@ -7,6 +7,7 @@
 #   all-features: true
 #   with-sources: false
 #   generate-hashes: false
+#   universal: false
 
 -e file:.
 annotated-types==0.6.0
@@ -35,7 +36,7 @@ h11==0.14.0
     # via httpcore
 httpcore==1.0.2
     # via httpx
-httpx==0.25.2
+httpx==0.28.1
     # via payi
     # via respx
 idna==3.4
@@ -48,7 +49,7 @@ markdown-it-py==3.0.0
     # via rich
 mdurl==0.1.2
     # via markdown-it-py
-mypy==1.13.0
+mypy==1.14.1
 mypy-extensions==1.0.0
     # via mypy
 nest-asyncio==1.6.0
@@ -68,7 +69,7 @@ pydantic-core==2.27.1
     # via pydantic
 pygments==2.18.0
     # via rich
-pyright==1.1.390
+pyright==1.1.392.post0
 pytest==8.3.3
     # via pytest-asyncio
 pytest-asyncio==0.24.0
@@ -76,7 +77,7 @@ python-dateutil==2.8.2
     # via time-machine
 pytz==2023.3.post1
     # via dirty-equals
-respx==0.20.2
+respx==0.22.0
 rich==13.7.1
 ruff==0.6.9
 setuptools==68.2.2
@@ -85,7 +86,6 @@ six==1.16.0
     # via python-dateutil
 sniffio==1.3.0
     # via anyio
-    # via httpx
     # via payi
 time-machine==2.9.0
 tomli==2.0.2
@@ -100,5 +100,7 @@ typing-extensions==4.12.2
     # via pyright
 virtualenv==20.24.5
     # via nox
+wrapt==1.17.2
+    # via payi
 zipp==3.17.0
     # via importlib-metadata

{payi-0.1.0a34 → payi-0.1.0a36}/requirements.lock

@@ -7,6 +7,7 @@
 #   all-features: true
 #   with-sources: false
 #   generate-hashes: false
+#   universal: false
 
 -e file:.
 annotated-types==0.6.0
@@ -25,7 +26,7 @@ h11==0.14.0
     # via httpcore
 httpcore==1.0.2
     # via httpx
-httpx==0.25.2
+httpx==0.28.1
     # via payi
 idna==3.4
     # via anyio
@@ -36,10 +37,11 @@ pydantic-core==2.27.1
     # via pydantic
 sniffio==1.3.0
     # via anyio
-    # via httpx
     # via payi
 typing-extensions==4.12.2
     # via anyio
     # via payi
     # via pydantic
     # via pydantic-core
+wrapt==1.17.2
+    # via payi

{payi-0.1.0a34 → payi-0.1.0a36}/src/payi/_base_client.py

@@ -767,6 +767,9 @@ else:
 
 class SyncHttpxClientWrapper(DefaultHttpxClient):
     def __del__(self) -> None:
+        if self.is_closed:
+            return
+
         try:
             self.close()
         except Exception:
@@ -1334,6 +1337,9 @@ else:
 
 class AsyncHttpxClientWrapper(DefaultAsyncHttpxClient):
     def __del__(self) -> None:
+        if self.is_closed:
+            return
+
         try:
             # TODO(someday): support non asyncio runtimes here
             asyncio.get_running_loop().create_task(self.aclose())

{payi-0.1.0a34 → payi-0.1.0a36}/src/payi/_models.py

@@ -179,14 +179,14 @@ class BaseModel(pydantic.BaseModel):
     @classmethod
     @override
     def construct(  # pyright: ignore[reportIncompatibleMethodOverride]
-        cls: Type[ModelT],
+        __cls: Type[ModelT],
         _fields_set: set[str] | None = None,
         **values: object,
     ) -> ModelT:
-        m = cls.__new__(cls)
+        m = __cls.__new__(__cls)
         fields_values: dict[str, object] = {}
 
-        config = get_model_config(cls)
+        config = get_model_config(__cls)
         populate_by_name = (
             config.allow_population_by_field_name
             if isinstance(config, _ConfigProtocol)
@@ -196,7 +196,7 @@ class BaseModel(pydantic.BaseModel):
         if _fields_set is None:
             _fields_set = set()
 
-        model_fields = get_model_fields(cls)
+        model_fields = get_model_fields(__cls)
         for name, field in model_fields.items():
             key = field.alias
             if key is None or (key not in values and populate_by_name):
@@ -488,7 +488,11 @@ def construct_type(*, value: object, type_: object) -> object:
         _, items_type = get_args(type_)  # Dict[_, items_type]
         return {key: construct_type(value=item, type_=items_type) for key, item in value.items()}
 
-    if not is_literal_type(type_) and (issubclass(origin, BaseModel) or issubclass(origin, GenericModel)):
+    if (
+        not is_literal_type(type_)
+        and inspect.isclass(origin)
+        and (issubclass(origin, BaseModel) or issubclass(origin, GenericModel))
+    ):
         if is_list(value):
             return [cast(Any, type_).construct(**entry) if is_mapping(entry) else entry for entry in value]
 

{payi-0.1.0a34 → payi-0.1.0a36}/src/payi/_response.py

@@ -136,6 +136,8 @@ class BaseAPIResponse(Generic[R]):
         if cast_to and is_annotated_type(cast_to):
             cast_to = extract_type_arg(cast_to, 0)
 
+        origin = get_origin(cast_to) or cast_to
+
         if self._is_sse_stream:
             if to:
                 if not is_stream_class_type(to):
@@ -195,8 +197,6 @@ class BaseAPIResponse(Generic[R]):
         if cast_to == bool:
             return cast(R, response.text.lower() == "true")
 
-        origin = get_origin(cast_to) or cast_to
-
         if origin == APIResponse:
             raise RuntimeError("Unexpected state - cast_to is `APIResponse`")
 
@@ -210,7 +210,13 @@ class BaseAPIResponse(Generic[R]):
                 raise ValueError(f"Subclasses of httpx.Response cannot be passed to `cast_to`")
             return cast(R, response)
 
-        if inspect.isclass(origin) and not issubclass(origin, BaseModel) and issubclass(origin, pydantic.BaseModel):
+        if (
+            inspect.isclass(
+                origin  # pyright: ignore[reportUnknownArgumentType]
+            )
+            and not issubclass(origin, BaseModel)
+            and issubclass(origin, pydantic.BaseModel)
+        ):
             raise TypeError("Pydantic models must subclass our base model type, e.g. `from payi import BaseModel`")
 
         if (

{payi-0.1.0a34 → payi-0.1.0a36}/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.34"  # x-release-please-version
+__version__ = "0.1.0-alpha.36"  # x-release-please-version

payi-0.1.0a36/src/payi/lib/AnthropicInstrumentor.py

@@ -0,0 +1,97 @@
+import logging
+from typing import Any
+
+from wrapt import wrap_function_wrapper  # type: ignore
+
+from payi.types import IngestUnitsParams
+from payi.types.ingest_units_params import Units
+
+from .instrument import PayiInstrumentor
+
+
+class AnthropicIntrumentor:
+    @staticmethod
+    def instrument(instrumentor: PayiInstrumentor) -> None:
+        try:
+            import anthropic  # type: ignore #  noqa: F401  I001
+
+            # wrap_function_wrapper(
+            #     "anthropic.resources.completions",
+            #     "Completions.create",
+            #     chat_wrapper(instrumentor),
+            # )
+
+            wrap_function_wrapper(
+                "anthropic.resources.messages",
+                "Messages.create",
+                chat_wrapper(instrumentor),
+            )
+
+            wrap_function_wrapper(
+                "anthropic.resources.messages",
+                "Messages.stream",
+                chat_wrapper(instrumentor),
+            )
+
+        except Exception as e:
+            logging.debug(f"Error instrumenting anthropic: {e}")
+            return
+
+
+@PayiInstrumentor.payi_wrapper
+def chat_wrapper(
+    instrumentor: PayiInstrumentor,
+    wrapped: Any,
+    instance: Any,
+    args: Any,
+    kwargs: Any,
+) -> Any:
+    return instrumentor.chat_wrapper(
+        "system.anthropic",
+        process_chunk,
+        process_synchronous_response,
+        wrapped,
+        instance,
+        args,
+        kwargs,
+    )
+
+
+def process_chunk(chunk: Any, ingest: IngestUnitsParams) -> None:
+    if chunk.type == "message_start":
+        usage = chunk.message.usage
+        units = ingest["units"]
+
+        units["text"] = Units(input=usage.input_tokens, output=0)
+
+        if hasattr(usage, "cache_creation_input_tokens") and usage.cache_creation_input_tokens > 0:
+            text_cache_write = usage.cache_creation_input_tokens
+            units["text_cache_write"] = Units(input=text_cache_write, output=0)
+
+        if hasattr(usage, "cache_read_input_tokens") and usage.cache_read_input_tokens > 0:
+            text_cache_read = usage.cache_read_input_tokens
+            units["text_cache_read"] = Units(input=text_cache_read, output=0)
+
+    elif chunk.type == "message_delta":
+        usage = chunk.usage
+        ingest["units"]["text"]["output"] = usage.output_tokens
+
+
+def process_synchronous_response(response: Any, ingest: IngestUnitsParams, log_prompt_and_response: bool) -> None:
+    usage = response.usage
+    input = usage.input_tokens
+    ouptut = usage.output_tokens
+    units: dict[str, Units] = ingest["units"]
+
+    if hasattr(usage, "cache_creation_input_tokens") and usage.cache_creation_input_tokens > 0:
+        text_cache_write = usage.cache_creation_input_tokens
+        units["text_cache_write"] = Units(input=text_cache_write, output=0)
+
+    if hasattr(usage, "cache_read_input_tokens") and usage.cache_read_input_tokens > 0:
+        text_cache_read = usage.cache_read_input_tokens
+        units["text_cache_read"] = Units(input=text_cache_read, output=0)
+
+    units["text"] = Units(input=input, output=ouptut)
+
+    if log_prompt_and_response:
+        ingest["provider_response_json"] = response.to_json()

payi-0.1.0a36/src/payi/lib/Instruments.py

@@ -0,0 +1,7 @@
+from enum import Enum
+
+
+class Instruments(Enum):
+    ALL = "all"
+    OPENAI = "openai"
+    ANTHROPIC = "anthropic"