payi 0.1.0a38__py3-none-any.whl → 0.1.0a40__py3-none-any.whl

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

payi/lib/AnthropicInstrumentor.py

@@ -1,6 +1,7 @@
 import logging
-from typing import Any
+from typing import Any, Union
 
+import tiktoken
 from wrapt import wrap_function_wrapper  # type: ignore
 
 from payi.types import IngestUnitsParams
@@ -49,6 +50,7 @@ def chat_wrapper(
     return instrumentor.chat_wrapper(
         "system.anthropic",
         process_chunk,
+        process_request,
         process_synchronous_response,
         wrapped,
         instance,
@@ -62,7 +64,9 @@ def process_chunk(chunk: Any, ingest: IngestUnitsParams) -> None:
         usage = chunk.message.usage
         units = ingest["units"]
 
-        units["text"] = Units(input=usage.input_tokens, output=0)
+        input = PayiInstrumentor.update_for_vision(usage.input_tokens, units)
+
+        units["text"] = Units(input=input, output=0)
 
         if hasattr(usage, "cache_creation_input_tokens") and usage.cache_creation_input_tokens > 0:
             text_cache_write = usage.cache_creation_input_tokens
@@ -91,7 +95,41 @@ def process_synchronous_response(response: Any, ingest: IngestUnitsParams, log_p
         text_cache_read = usage.cache_read_input_tokens
         units["text_cache_read"] = Units(input=text_cache_read, output=0)
 
+    input = PayiInstrumentor.update_for_vision(input, units)
+
     units["text"] = Units(input=input, output=ouptut)
 
     if log_prompt_and_response:
         ingest["provider_response_json"] = response.to_json()
+
+def has_image_and_get_texts(encoding: tiktoken.Encoding, content: Union[str, 'list[Any]']) -> 'tuple[bool, int]':
+    if isinstance(content, str):
+        return False, 0
+    elif isinstance(content, list): # type: ignore
+        has_image = any(item.get("type") == "image" for item in content)
+        if has_image is False:
+            return has_image, 0
+        
+        token_count = sum(len(encoding.encode(item.get("text", ""))) for item in content if item.get("type") == "text")
+        return has_image, token_count
+
+def process_request(ingest: IngestUnitsParams, kwargs: Any) -> None:
+    messages = kwargs.get("messages")
+    if not messages or len(messages) == 0:
+        return
+    
+    estimated_token_count = 0 
+    has_image = False
+
+    enc = tiktoken.get_encoding("cl100k_base")
+    
+    for message in messages:
+        msg_has_image, msg_prompt_tokens = has_image_and_get_texts(enc, message.get('content', ''))
+        if msg_has_image:
+            has_image = True
+            estimated_token_count += msg_prompt_tokens
+    
+    if not has_image or estimated_token_count == 0:
+        return
+
+    ingest["units"][PayiInstrumentor.estimated_prompt_tokens] = Units(input=estimated_token_count, output=0)

payi/lib/OpenAIInstrumentor.py

@@ -1,8 +1,9 @@
 import json
 import logging
-from typing import Any
+from typing import Any, Union
 from importlib.metadata import version
 
+import tiktoken
 from wrapt import wrap_function_wrapper  # type: ignore
 
 from payi.types import IngestUnitsParams
@@ -38,6 +39,7 @@ def chat_wrapper(
     return instrumentor.chat_wrapper(
         "system.openai",
         process_chat_chunk,
+        process_request,
         process_chat_synchronous_response,
         wrapped,
         instance,
@@ -84,6 +86,41 @@ def add_usage_units(usage: "dict[str, Any]", units: "dict[str, Units]") -> None:
         if input_cache != 0:
             units["text_cache_read"] = Units(input=input_cache, output=0)
 
-    input -= input_cache
+    input = PayiInstrumentor.update_for_vision(input - input_cache, units)
 
     units["text"] = Units(input=input, output=output)
+
+def has_image_and_get_texts(encoding: tiktoken.Encoding, content: Union[str, 'list[Any]']) -> 'tuple[bool, int]':
+    if isinstance(content, str):
+        return False, 0
+    elif isinstance(content, list): # type: ignore
+        has_image = any(item.get("type") == "image_url" for item in content)
+        if has_image is False:
+            return has_image, 0
+        
+        token_count = sum(len(encoding.encode(item.get("text", ""))) for item in content if item.get("type") == "text")
+        return has_image, token_count
+
+def process_request(ingest: IngestUnitsParams, kwargs: Any) -> None:
+    messages = kwargs.get("messages")
+    if not messages or len(messages) == 0:
+        return
+    
+    estimated_token_count = 0 
+    has_image = False
+
+    try: 
+        enc = tiktoken.encoding_for_model(kwargs.get("model"))
+    except KeyError:
+        enc = tiktoken.get_encoding("o200k_base")
+    
+    for message in messages:
+        msg_has_image, msg_prompt_tokens = has_image_and_get_texts(enc, message.get('content', ''))
+        if msg_has_image:
+            has_image = True
+            estimated_token_count += msg_prompt_tokens
+    
+    if not has_image or estimated_token_count == 0:
+        return
+
+    ingest["units"][PayiInstrumentor.estimated_prompt_tokens] = Units(input=estimated_token_count, output=0)

payi/lib/instrument.py

@@ -10,12 +10,15 @@ from wrapt import ObjectProxy  # type: ignore
 
 from payi import Payi, AsyncPayi
 from payi.types import IngestUnitsParams
+from payi.types.ingest_units_params import Units
 
 from .Stopwatch import Stopwatch
 from .Instruments import Instruments
 
 
 class PayiInstrumentor:
+    estimated_prompt_tokens: str = "estimated_prompt_tokens"
+
     def __init__(
         self,
         payi: Union[Payi, AsyncPayi, None] = None,
@@ -204,11 +207,12 @@ class PayiInstrumentor:
         self,
         category: str,
         process_chunk: Callable[[Any, IngestUnitsParams], None],
+        process_request: Optional[Callable[[IngestUnitsParams, Any], None]],
         process_synchronous_response: Optional[Callable[[Any, IngestUnitsParams, bool], None]],
         wrapped: Any,
         instance: Any,
         args: Any,
-        kwargs: Any,
+        kwargs: 'dict[str, Any]',
     ) -> Any:
         context = self.get_context()
 
@@ -226,7 +230,7 @@ class PayiInstrumentor:
 
             return wrapped(*args, **kwargs)
 
-        ingest: IngestUnitsParams = {"category": category, "resource": kwargs.get("model"), "units": {}}
+        ingest: IngestUnitsParams = {"category": category, "resource": kwargs.get("model"), "units": {}} # type: ignore
 
         # blocked_limit = next((limit for limit in (context.get('limit_ids') or []) if limit in self._blocked_limits), None)
         # if blocked_limit:
@@ -235,8 +239,10 @@ class PayiInstrumentor:
         # f_back excludes the current frame, strip() cleans up whitespace and newlines
         stack = [frame.strip() for frame in traceback.format_stack(current_frame.f_back)]  # type: ignore
 
-        # TODO add back once feature is in prod
-        # ingest['properties'] = { 'system.stack_trace': json.dumps(stack) }
+        ingest['properties'] = { 'system.stack_trace': json.dumps(stack) }
+
+        if process_request:
+            process_request(ingest, kwargs)
 
         sw = Stopwatch()
         stream = kwargs.get("stream", False)
@@ -354,6 +360,17 @@ class PayiInstrumentor:
             if experience_id is not None:
                 extra_headers["xProxy-Experience-ID"] = experience_id
 
+    @staticmethod
+    def update_for_vision(input: int, units: 'dict[str, Units]') -> int:
+        if PayiInstrumentor.estimated_prompt_tokens in units:
+            prompt_token_estimate: int = units.pop(PayiInstrumentor.estimated_prompt_tokens)["input"] # type: ignore
+            vision = input - prompt_token_estimate
+            if (vision > 0):
+                units["vision"] = Units(input=vision, output=0)
+                input = prompt_token_estimate
+        
+        return input
+
     @staticmethod
     def payi_wrapper(func: Any) -> Any:
         def _payi_wrapper(o: Any) -> Any:

payi/resources/ingest.py

@@ -24,8 +24,6 @@ from .._response import (
 )
 from .._base_client import make_request_options
 from ..types.ingest_response import IngestResponse
-from ..types.ingest_event_param import IngestEventParam
-from ..types.bulk_ingest_response import BulkIngestResponse
 from ..types.pay_i_common_models_api_router_header_info_param import PayICommonModelsAPIRouterHeaderInfoParam
 
 __all__ = ["IngestResource", "AsyncIngestResource"]
@@ -51,41 +49,6 @@ class IngestResource(SyncAPIResource):
         """
         return IngestResourceWithStreamingResponse(self)
 
-    def bulk(
-        self,
-        *,
-        events: Iterable[IngestEventParam],
-        # 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,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> BulkIngestResponse:
-        """
-        Bulk Ingest
-
-        Args:
-
-          items (Iterable[IngestUnitsParams]): The items to ingest
-
-          extra_headers: Send extra headers
-
-          extra_query: Add additional query parameters to the request
-
-          extra_body: Add additional JSON properties to the request
-
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        return self._post(
-            "/api/v1/ingest/bulk",
-            body=maybe_transform(events, Iterable[IngestEventParam]),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=BulkIngestResponse,
-        )
-
     def units(
         self,
         *,
@@ -237,40 +200,6 @@ class AsyncIngestResource(AsyncAPIResource):
         """
         return AsyncIngestResourceWithStreamingResponse(self)
 
-    async def bulk(
-        self,
-        *,
-        events: Iterable[IngestEventParam],
-        # 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,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> BulkIngestResponse:
-        """
-        Bulk Ingest
-
-        Args:
-          items (Iterable[IngestUnitsParams]): The items to ingest
-
-          extra_headers: Send extra headers
-
-          extra_query: Add additional query parameters to the request
-
-          extra_body: Add additional JSON properties to the request
-
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        return await self._post(
-            "/api/v1/ingest/bulk",
-            body=await async_maybe_transform(events, Iterable[IngestEventParam]),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=BulkIngestResponse,
-        )
-
     async def units(
         self,
         *,
@@ -405,9 +334,6 @@ class IngestResourceWithRawResponse:
     def __init__(self, ingest: IngestResource) -> None:
         self._ingest = ingest
 
-        self.bulk = to_raw_response_wrapper(
-            ingest.bulk,
-        )
         self.units = to_raw_response_wrapper(
             ingest.units,
         )
@@ -417,9 +343,6 @@ class AsyncIngestResourceWithRawResponse:
     def __init__(self, ingest: AsyncIngestResource) -> None:
         self._ingest = ingest
 
-        self.bulk = async_to_raw_response_wrapper(
-            ingest.bulk,
-        )
         self.units = async_to_raw_response_wrapper(
             ingest.units,
         )
@@ -429,9 +352,6 @@ class IngestResourceWithStreamingResponse:
     def __init__(self, ingest: IngestResource) -> None:
         self._ingest = ingest
 
-        self.bulk = to_streamed_response_wrapper(
-            ingest.bulk,
-        )
         self.units = to_streamed_response_wrapper(
             ingest.units,
         )
@@ -441,9 +361,6 @@ class AsyncIngestResourceWithStreamingResponse:
     def __init__(self, ingest: AsyncIngestResource) -> None:
         self._ingest = ingest
 
-        self.bulk = async_to_streamed_response_wrapper(
-            ingest.bulk,
-        )
         self.units = async_to_streamed_response_wrapper(
             ingest.units,
         )

payi/types/__init__.py

@@ -15,13 +15,10 @@ from .default_response import DefaultResponse as DefaultResponse
 from .paged_limit_list import PagedLimitList as PagedLimitList
 from .category_response import CategoryResponse as CategoryResponse
 from .limit_list_params import LimitListParams as LimitListParams
-from .ingest_bulk_params import IngestBulkParams as IngestBulkParams
-from .ingest_event_param import IngestEventParam as IngestEventParam
 from .limit_reset_params import LimitResetParams as LimitResetParams
 from .ingest_units_params import IngestUnitsParams as IngestUnitsParams
 from .limit_create_params import LimitCreateParams as LimitCreateParams
 from .limit_update_params import LimitUpdateParams as LimitUpdateParams
-from .bulk_ingest_response import BulkIngestResponse as BulkIngestResponse
 from .category_list_response import CategoryListResponse as CategoryListResponse
 from .limit_history_response import LimitHistoryResponse as LimitHistoryResponse
 from .category_delete_response import CategoryDeleteResponse as CategoryDeleteResponse

{payi-0.1.0a38.dist-info → payi-0.1.0a40.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: payi
-Version: 0.1.0a38
+Version: 0.1.0a40
 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
@@ -27,6 +27,7 @@ Requires-Dist: distro<2,>=1.7.0
 Requires-Dist: httpx<1,>=0.23.0
 Requires-Dist: pydantic<3,>=1.9.0
 Requires-Dist: sniffio
+Requires-Dist: tiktoken>=0.8.0
 Requires-Dist: typing-extensions<5,>=4.10
 Requires-Dist: wrapt>=1.17.2
 Description-Content-Type: text/markdown

{payi-0.1.0a38.dist-info → payi-0.1.0a40.dist-info}/RECORD

@@ -11,7 +11,7 @@ payi/_resource.py,sha256=j2jIkTr8OIC8sU6-05nxSaCyj4MaFlbZrwlyg4_xJos,1088
 payi/_response.py,sha256=CfrNS_3wbL8o9dRyRVfZQ5E1GUlA4CUIUEK8olmfGqE,28777
 payi/_streaming.py,sha256=Z_wIyo206T6Jqh2rolFg2VXZgX24PahLmpURp0-NssU,10092
 payi/_types.py,sha256=2mbMK86K3W1aMTW7sOGQ-VND6-A2IuXKm8p4sYFztBU,6141
-payi/_version.py,sha256=uhFgvjXGv4Bcm_dtpDLj6GUOkYi8xP6Mz9Ctz2H_Lqc,165
+payi/_version.py,sha256=70Q3C2mH_Y5XHCDAYiBgNkVy3fyBDWf8eUqRwndh9Zc,165
 payi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 payi/_utils/__init__.py,sha256=PNZ_QJuzZEgyYXqkO1HVhGkj5IU9bglVUcw7H-Knjzw,2062
 payi/_utils/_logs.py,sha256=fmnf5D9TOgkgZKfgYmSa3PiUc3SZgkchn6CzJUeo0SQ,768
@@ -23,15 +23,15 @@ payi/_utils/_transform.py,sha256=Dkkyr7OveGmOolepcvXmVJWE3kqim4b0nM0h7yWbgeY,134
 payi/_utils/_typing.py,sha256=nTJz0jcrQbEgxwy4TtAkNxuU0QHHlmc6mQtA6vIR8tg,4501
 payi/_utils/_utils.py,sha256=8UmbPOy_AAr2uUjjFui-VZSrVBHRj6bfNEKRp5YZP2A,12004
 payi/lib/.keep,sha256=wuNrz-5SXo3jJaJOJgz4vFHM41YH_g20F5cRQo0vLes,224
-payi/lib/AnthropicInstrumentor.py,sha256=aJDG90CeOmPNPq49UXm4yc59hoI1zVI406fNGeT_n8U,3149
+payi/lib/AnthropicInstrumentor.py,sha256=XMLr8BIrBo7U3VPBnhyp9Wj-icZmu0MGVvvyDPmlF4c,4549
 payi/lib/Instruments.py,sha256=cyL2jxjpRluP9rN8Vn1nmVXq2NNLdZuFIsHMQWWqat4,115
-payi/lib/OpenAIInstrumentor.py,sha256=eABTqxFLP109fOto33rBbXLh0FdQPh0a9VoMG4d6zGo,2575
+payi/lib/OpenAIInstrumentor.py,sha256=G3H8s1bNnq-LECAHjCkQDTG59pwnLPXKQKlQbhqgs4k,3998
 payi/lib/Stopwatch.py,sha256=vFyGVRvkppamP7W0IuZyypKLMIaqjhB7fcRG0dNyfnQ,757
 payi/lib/helpers.py,sha256=ZgkY8UE2YRc7ok2Pmxg_T9UMqKI8D8542JY3CP8RZCM,1597
-payi/lib/instrument.py,sha256=Ds49BFxqWnUMezL1tRr1IPBbb8rejDrjyi6t4YO9rCU,19549
+payi/lib/instrument.py,sha256=rLSwXRQZkTmMpWGpTzgZ42_4BOxJjGrTR3hwpF8XHro,20279
 payi/resources/__init__.py,sha256=isHGXSl9kOrZDduKrX3UenTwrdTpuKJVBjw6NYSBV20,3592
 payi/resources/billing_models.py,sha256=5w3RfGXtGlyq5vbTw6hQrx1UlzRBtlq8ArcFlf5e3TY,20152
-payi/resources/ingest.py,sha256=JPtO7foPsftAgG92kSgU29_yOJSYuvIQtQB8WpRZuoQ,18567
+payi/resources/ingest.py,sha256=SvQspsYled4_ErOZKzVtazBIk0tUC1e34Lw8qw4SNEM,15484
 payi/resources/price_modifiers.py,sha256=t-k2F_zf2FhoxiqDHAPBPvhSgTjewlJqh50y58FNMuw,13475
 payi/resources/categories/__init__.py,sha256=w5gMiPdBSzJA_qfoVtFBElaoe8wGf_O63R7R1Spr6Gk,1093
 payi/resources/categories/categories.py,sha256=FohmajDcadMXzhG3Z1HKGkbSImO7rhzQ0olZXHz8z48,16074
@@ -49,12 +49,11 @@ payi/resources/requests/__init__.py,sha256=k7ipgDb5QXAv7WYhgQq0-6Zn9INJikMzRGexu
 payi/resources/requests/properties.py,sha256=ZsUeTMKQB4PNXn-jeoRwG-fqwpXzkRQyrFB8GmaNiQg,6466
 payi/resources/requests/requests.py,sha256=uxfdUe_mNmG5kzHTa4YcS5m2wwIc5h__YSx54LfiLWQ,4840
 payi/resources/requests/result.py,sha256=dEIwYEi_p36t_hErr_V1E2hmBLLgopcLnIReAWT8ygk,6161
-payi/types/__init__.py,sha256=WZhGgB9kGD48NljvFFu3TeLhA_PKqDdMUsnxvSKxhc4,2854
+payi/types/__init__.py,sha256=a_uGQ-yPLCCjeyJRE4xAzI_0MjoKkuIoRMQXuSEL8VI,2641
 payi/types/billing_model.py,sha256=zwpKldc0WvS3iGKtDb9KvfxCd3lkv8F4TwFy3ciGMXg,639
 payi/types/billing_model_create_params.py,sha256=iVvmCcw0VxXGI_0YolknD3gmDH2lXVydU1dg2IY4dC4,547
 payi/types/billing_model_list_response.py,sha256=hOFjJPQAKiWEArZeZwd8e1lDi5e41zbN6NvV_1rzJeM,290
 payi/types/billing_model_update_params.py,sha256=NHNxDUZyYeYEroSyUx8eVh-kyY2YUo3jfRI-eMx2QAY,547
-payi/types/bulk_ingest_response.py,sha256=78J3vlL7muZx9Z20H--UkvM8P19g0cDL9SZoHy7XM68,1330
 payi/types/category_delete_resource_response.py,sha256=PLz4wZA1XMpS9SUYB_j4hEw5EoZ0VVE9Ll-MQ26SAfc,339
 payi/types/category_delete_response.py,sha256=exq8rNDGoq2-YN528V8osdcmuptJ-k63rmCvPMm6hLA,323
 payi/types/category_list_resources_response.py,sha256=n0DxY7N3Iftwfl0lUEx5v55V0kxbOX0EgjXlEfJtYRQ,337
@@ -65,8 +64,6 @@ payi/types/cost_data.py,sha256=1i842P25SBy2sB3OWGj9LO_mMKtzmyUPBrqY_mSw01o,488
 payi/types/cost_details.py,sha256=w9p79opEG3kcsjkRRP7niaMcUswdfB4Y7HCkVTcQ1zQ,307
 payi/types/default_response.py,sha256=o617LpRsCIZHCZxAc5nVI2JQ3HPGZo4gCDvSDkxkIJ8,270
 payi/types/experience_instance_response.py,sha256=N07MH6hjs1ISHLVpR2FG-u4awsZ_GGi97UNMXAWV1yA,296
-payi/types/ingest_bulk_params.py,sha256=d76YwiXaNeltLS3w86ZxLzTKGa7ymGLJDSelaMQGf8Y,382
-payi/types/ingest_event_param.py,sha256=wA9YuiQceNL3veHO_rVuKZg9o-EB1WLoz0aaF6Wcm-k,1498
 payi/types/ingest_response.py,sha256=KZfsgUhC942QkkjDFMqjJwCRoO2vkXv-Sx3X_xjijfg,1449
 payi/types/ingest_units_params.py,sha256=0s_j6268ZmeXDw9lQ_HKLldA_EvNyPcOxM6kFxLwxkA,1749
 payi/types/limit_create_params.py,sha256=Av8oMCxlKH7VB2MtYN5-25rAjqDDTPHjsXIQ2xubmck,549
@@ -112,7 +109,7 @@ payi/types/requests/request_result.py,sha256=phYQiqhwNaR9igP-Fhs34Y-__dlT7L4wq-r
 payi/types/shared/__init__.py,sha256=-xz5dxK5LBjLnsi2LpLq5btaGDFp-mSjJ0y2qKy0Yus,264
 payi/types/shared/evaluation_response.py,sha256=ejEToMA57PUu1SldEtJ5z9r4fAO3U0tvdjbsyIoVX1s,214
 payi/types/shared/pay_i_common_models_budget_management_cost_details_base.py,sha256=XmIzJXy4zAi-mfrDvEXiYjO3qF1EvugGUl-Gijj4TA4,268
-payi-0.1.0a38.dist-info/METADATA,sha256=5g142UgydrzpGRqyttKbGlDCw-CUXaxsAfKnhw1r7I8,12594
-payi-0.1.0a38.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-payi-0.1.0a38.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
-payi-0.1.0a38.dist-info/RECORD,,
+payi-0.1.0a40.dist-info/METADATA,sha256=qTLzlfTKFZ8cEWce30AegO7WzqqjejiYc6kJi3SqYN8,12625
+payi-0.1.0a40.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+payi-0.1.0a40.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
+payi-0.1.0a40.dist-info/RECORD,,

payi/types/bulk_ingest_response.py

@@ -1,51 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from typing import List, Optional
-from datetime import datetime
-
-from pydantic import Field as FieldInfo
-
-from .._models import BaseModel
-
-__all__ = ["BulkIngestResponse", "Error", "ErrorXproxyResult", "ErrorXproxyResultXproxyError"]
-
-
-class ErrorXproxyResultXproxyError(BaseModel):
-    code: Optional[str] = None
-
-    message: Optional[str] = None
-
-
-class ErrorXproxyResult(BaseModel):
-    message: str
-
-    status_code: int = FieldInfo(alias="statusCode")
-
-    xproxy_error: Optional[ErrorXproxyResultXproxyError] = None
-
-
-class Error(BaseModel):
-    item_index: Optional[int] = None
-
-    xproxy_result: Optional[ErrorXproxyResult] = None
-    """
-    Represents an generic error that occurred as a result of processing a request.
-    APIM returns an (not customizable) error response body of { "statusCode",
-    "message" } and this class matches this schema. Derived classes may add
-    additional required fields if these classes are specified as produced as a
-    return type specific endpoints.
-    """
-
-
-class BulkIngestResponse(BaseModel):
-    ingest_count: int
-
-    ingest_timestamp: datetime
-
-    request_id: str
-
-    error_count: Optional[int] = None
-
-    errors: Optional[List[Error]] = None
-
-    total_count: Optional[int] = None

payi/types/ingest_bulk_params.py

@@ -1,14 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from __future__ import annotations
-
-from typing import Iterable
-from typing_extensions import Required, TypedDict
-
-from .ingest_event_param import IngestEventParam
-
-__all__ = ["IngestBulkParams"]
-
-
-class IngestBulkParams(TypedDict, total=False):
-    events: Required[Iterable[IngestEventParam]]

payi/types/ingest_event_param.py

@@ -1,60 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from __future__ import annotations
-
-from typing import Dict, List, Union, Iterable, Optional
-from datetime import datetime
-from typing_extensions import Required, Annotated, TypedDict
-
-from .._utils import PropertyInfo
-from .pay_i_common_models_api_router_header_info_param import PayICommonModelsAPIRouterHeaderInfoParam
-
-__all__ = ["IngestEventParam", "Units"]
-
-
-class Units(TypedDict, total=False):
-    input: int
-
-    output: int
-
-
-class IngestEventParam(TypedDict, total=False):
-    category: Required[str]
-
-    resource: Required[str]
-
-    units: Required[Dict[str, Units]]
-
-    csat_rating: Optional[int]
-
-    end_to_end_latency_ms: Optional[int]
-
-    event_timestamp: Annotated[Union[str, datetime, None], PropertyInfo(format="iso8601")]
-
-    experience_id: Optional[str]
-
-    experience_name: Optional[str]
-
-    experience_properties: Optional[Dict[str, str]]
-
-    http_status_code: Optional[int]
-
-    limit_ids: Optional[List[str]]
-
-    properties: Optional[Dict[str, str]]
-
-    provider_request_headers: Optional[Iterable[PayICommonModelsAPIRouterHeaderInfoParam]]
-
-    provider_request_json: Optional[str]
-
-    provider_response_headers: Optional[Iterable[PayICommonModelsAPIRouterHeaderInfoParam]]
-
-    provider_response_json: Union[str, List[str], None]
-
-    provider_uri: Optional[str]
-
-    request_tags: Optional[List[str]]
-
-    time_to_first_token_ms: Optional[int]
-
-    user_id: Optional[str]