payi 0.1.0a92__py3-none-any.whl → 0.1.0a93__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.92"  # x-release-please-version
+__version__ = "0.1.0-alpha.93"  # x-release-please-version

payi/lib/VertexInstrumentor.py

@@ -88,7 +88,6 @@ class _GoogleVertexRequest(_VertexRequest):
             )
         self._prompt_character_count = 0
         self._candidates_character_count = 0
-        self._model_name: Optional[str] = None
 
     @override
     def process_request(self, instance: Any, extra_headers: 'dict[str, str]', args: Sequence[Any], kwargs: Any) -> bool:
@@ -100,7 +99,7 @@ class _GoogleVertexRequest(_VertexRequest):
             if model and isinstance(model, str):
                 # Extract the model name after the last slash
                 self._model_name = model.split('/')[-1]
-        
+
         if not args:
             return True
         

payi/lib/instrument.py

@@ -1,5 +1,6 @@
 import os
 import json
+import time
 import uuid
 import asyncio
 import inspect
@@ -14,7 +15,7 @@ from dataclasses import dataclass
 import nest_asyncio  # type: ignore
 from wrapt import ObjectProxy  # type: ignore
 
-from payi import Payi, AsyncPayi, __version__ as _payi_version
+from payi import Payi, AsyncPayi, APIConnectionError, __version__ as _payi_version
 from payi.types import IngestUnitsParams
 from payi.lib.helpers import PayiHeaderNames
 from payi.types.ingest_response import IngestResponse
@@ -137,6 +138,7 @@ class PayiInstrumentConfig(TypedDict, total=False):
     proxy: bool
     global_instrumentation: bool
     instrument_inline_data: bool
+    connection_error_logging_window: int
     limit_ids: Optional["list[str]"]
     use_case_name: Optional[str]
     use_case_id: Optional[str]
@@ -222,6 +224,12 @@ class _PayiInstrumentor:
         self._blocked_limits: set[str] = set()
         self._exceeded_limits: set[str] = set()
 
+        self._api_connection_error_last_log_time: float = time.time()
+        self._api_connection_error_count: int = 0
+        self._api_connection_error_window: int = global_config.get("connection_error_logging_window", 60)
+        if self._api_connection_error_window < 0:
+            raise ValueError("connection_error_logging_window must be a non-negative integer")
+
         # default is instrument and ingest metrics
         self._proxy_default: bool = global_config.get("proxy", False)
 
@@ -353,6 +361,9 @@ class _PayiInstrumentor:
             # convert the function call builder to a list of function calls
             ingest_units["provider_response_function_calls"] = list(request._function_call_builder.values())
 
+        if 'resource' not in ingest_units or len(ingest_units['resource']) == 0:
+            ingest_units['resource'] = "system.unknown_model"  # type: ignore[unreachable]
+
         request_json = ingest_units.get('provider_request_json', "")
         if request_json and self._instrument_inline_data is False:
             try:
@@ -402,6 +413,23 @@ class _PayiInstrumentor:
                 if removeBlockedId:
                     self._blocked_limits.discard(limit_id)
 
+    def _process_ingest_connection_error(self, e: APIConnectionError, ingest_units: IngestUnitsParams) -> None:
+        now = time.time()
+
+        if (now - self._api_connection_error_last_log_time) > self._api_connection_error_window:
+            # If previous window had suppressed errors, log the count
+            append = ""
+            if self._api_connection_error_count > 0:
+                append = f", {self._api_connection_error_count} APIConnectionError exceptions in the last {self._api_connection_error_window} seconds"
+
+            # Log the current error
+            self._logger.error(f"Error Pay-i ingesting: connection exception {e}, request {ingest_units}{append}")
+            self._api_connection_error_last_log_time = now
+            self._api_connection_error_count = 0
+        else:
+            # Suppress and count
+            self._api_connection_error_count += 1
+
     async def _aingest_units(self, request: _ProviderRequest) -> Optional[IngestResponse]:
         ingest_response: Optional[IngestResponse] = None
         ingest_units = request._ingest
@@ -437,7 +465,10 @@ class _PayiInstrumentor:
 
             return ingest_response
         except Exception as e:
-            self._logger.error(f"Error Pay-i async ingesting: exception {e}, request {ingest_units}")
+            if isinstance(e, APIConnectionError) and self._api_connection_error_window > 0:
+                self._process_ingest_connection_error(e, ingest_units)
+            else:
+                self._logger.error(f"Error Pay-i async ingesting: exception {e}, request {ingest_units}")
     
         return None         
     
@@ -513,7 +544,10 @@ class _PayiInstrumentor:
                 self._logger.error("No payi instance to ingest units")
 
         except Exception as e:
-            self._logger.error(f"Error Pay-i ingesting: exception {e}, request {ingest_units}")
+            if isinstance(e, APIConnectionError) and self._api_connection_error_window > 0:
+                self._process_ingest_connection_error(e, ingest_units)
+            else:
+                self._logger.error(f"Error Pay-i ingesting: exception {e}, request {ingest_units}")
         
         return None
 

{payi-0.1.0a92.dist-info → payi-0.1.0a93.dist-info}/METADATA

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

@@ -11,7 +11,7 @@ payi/_resource.py,sha256=j2jIkTr8OIC8sU6-05nxSaCyj4MaFlbZrwlyg4_xJos,1088
 payi/_response.py,sha256=rh9oJAvCKcPwQFm4iqH_iVrmK8bNx--YP_A2a4kN1OU,28776
 payi/_streaming.py,sha256=Z_wIyo206T6Jqh2rolFg2VXZgX24PahLmpURp0-NssU,10092
 payi/_types.py,sha256=7jE5MoQQFVoVxw5vVzvZ2Ao0kcjfNOGsBgyJfLBEnMo,6195
-payi/_version.py,sha256=enwqQZXbXCLnYJiMH-ERIVZQXanTzI_DXQADR3N5aK4,165
+payi/_version.py,sha256=F--xck4QJ1Rqyg7T3nA-GXHI12z2X-gLM-k7V8Bawok,165
 payi/pagination.py,sha256=k2356QGPOUSjRF2vHpwLBdF6P-2vnQzFfRIJQAHGQ7A,1258
 payi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 payi/_utils/__init__.py,sha256=PNZ_QJuzZEgyYXqkO1HVhGkj5IU9bglVUcw7H-Knjzw,2062
@@ -30,10 +30,10 @@ payi/lib/BedrockInstrumentor.py,sha256=Vx0DUjdYzHWAuYsmK28z9Bxz-mj7UouEESsCN3B9J
 payi/lib/GoogleGenAiInstrumentor.py,sha256=DLbmlwmyOwooe7FuBkEAO_Z3xjPKpvQyO4VwLBeZnn4,8477
 payi/lib/OpenAIInstrumentor.py,sha256=6HENrdN6kLMUwC00wNSdeGeK4fUDBWo2dCbVC9koi0c,22581
 payi/lib/Stopwatch.py,sha256=7OJlxvr2Jyb6Zr1LYCYKczRB7rDVKkIR7gc4YoleNdE,764
-payi/lib/VertexInstrumentor.py,sha256=IdahkOgB6ReBGpdgGCrBHplAFwK3KZ_XaRgFXVTodRU,7136
+payi/lib/VertexInstrumentor.py,sha256=q-oYr3YuwPc95xYn6KhjvKs5ZweGRg6mOIkxSTz-lic,7081
 payi/lib/VertexRequest.py,sha256=NNH6S86nCHaU54Rb1rxai-b3UiJvatmzKkDKBmqMeKY,10530
 payi/lib/helpers.py,sha256=FPzNSSHGf9bgD6CanB7yVx_U8t4lm2c0jlZKrsziYlc,4242
-payi/lib/instrument.py,sha256=CiUtYwpTeoFbm7EEBXSbfaE5ZJKNuu7-ZfrXMi3hINM,62954
+payi/lib/instrument.py,sha256=oVN1KL2CqKcdcloDnWi7OQOXtq5df4bx6UHTYfBSnZs,64834
 payi/resources/__init__.py,sha256=1rtrPLWbNt8oJGOp6nwPumKLJ-ftez0B6qwLFyfcoP4,2972
 payi/resources/ingest.py,sha256=Z4WHv-INZoIlBzFw4o1j_PHykDsNACpkkF42Kik0UMg,23758
 payi/resources/categories/__init__.py,sha256=WeotN_d-0Ri8ohsrNPbve7RyViD9_N0NA9DrV3WYg3w,1701
@@ -145,7 +145,7 @@ payi/types/use_cases/definitions/kpi_retrieve_response.py,sha256=uQXliSvS3k-yDYw
 payi/types/use_cases/definitions/kpi_update_params.py,sha256=jbawdWAdMnsTWVH0qfQGb8W7_TXe3lq4zjSRu44d8p8,373
 payi/types/use_cases/definitions/kpi_update_response.py,sha256=zLyEoT0S8d7XHsnXZYT8tM7yDw0Aze0Mk-_Z6QeMtc8,459
 payi/types/use_cases/definitions/limit_config_create_params.py,sha256=pzQza_16N3z8cFNEKr6gPbFvuGFrwNuGxAYb--Kbo2M,449
-payi-0.1.0a92.dist-info/METADATA,sha256=KpNUrDFlrf_yCeznM4MLB7tVMSkU8t5mjdHvG73pd3o,16333
-payi-0.1.0a92.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-payi-0.1.0a92.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
-payi-0.1.0a92.dist-info/RECORD,,
+payi-0.1.0a93.dist-info/METADATA,sha256=qqHi5iuUQKASaIN0sVyCrW6CT9flyqMiO_f3TH1MNFQ,16333
+payi-0.1.0a93.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+payi-0.1.0a93.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
+payi-0.1.0a93.dist-info/RECORD,,