payi 0.1.0a113__py3-none-any.whl → 0.1.0a114__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.113"  # x-release-please-version
+__version__ = "0.1.0-alpha.114"  # x-release-please-version

payi/lib/AnthropicInstrumentor.py

@@ -245,7 +245,7 @@ class _AnthropicProviderRequest(_ProviderRequest):
         return True
 
 def anthropic_process_compute_input_cost(request: _ProviderRequest, usage: 'dict[str, Any]') -> int:
-    input = usage['input_tokens']
+    input = usage.get('input_tokens', 0)
     units: dict[str, Units] = request._ingest["units"]
 
     cache_creation_input_tokens = usage.get("cache_creation_input_tokens", 0)
@@ -282,11 +282,11 @@ def anthropic_process_compute_input_cost(request: _ProviderRequest, usage: 'dict
     return _PayiInstrumentor.update_for_vision(input, units, request._estimated_prompt_tokens, is_large_context=request._is_large_context)
 
 def anthropic_process_synchronous_response(request: _ProviderRequest, response: 'dict[str, Any]', log_prompt_and_response: bool, assign_id: bool) -> Any:
-    usage = response['usage']
+    usage = response.get('usage', {})
     units: dict[str, Units] = request._ingest["units"]
 
     input_tokens = anthropic_process_compute_input_cost(request, usage)
-    output = usage['output_tokens']
+    output = usage.get('output_tokens', 0)
 
     large_context = "_large_context" if request._is_large_context else ""
     units["text"+large_context] = Units(input=input_tokens, output=output)
@@ -327,7 +327,7 @@ def anthropic_process_chunk(request: _ProviderRequest, chunk: 'dict[str, Any]',
         if model and 'resource' in request._ingest:
             request._instrumentor._logger.debug(f"Anthropic streaming, reported model: {model}, instrumented model {request._ingest['resource']}")
 
-        usage = message['usage']
+        usage = message.get('usage', {})
         units = request._ingest["units"]
 
         input = anthropic_process_compute_input_cost(request, usage)

payi/lib/BedrockInstrumentor.py

@@ -424,10 +424,10 @@ class _BedrockConverseProviderRequest(_BedrockProviderRequest):
         log_prompt_and_response: bool,
         kwargs: Any) -> Any:
 
-        usage = response["usage"]
-        input = usage["inputTokens"]
-        output = usage["outputTokens"]
-        
+        usage = response.get("usage", {})
+        input = usage.get("inputTokens", 0)
+        output = usage.get("outputTokens", 0)
+
         units: dict[str, Units] = self._ingest["units"]
         units["text"] = Units(input=input, output=output)
 
@@ -456,9 +456,9 @@ class _BedrockConverseProviderRequest(_BedrockProviderRequest):
         metadata = chunk.get("metadata", {})
 
         if metadata:
-            usage = metadata['usage']
-            input = usage["inputTokens"]
-            output = usage["outputTokens"]
+            usage = metadata.get('usage', {})
+            input = usage.get("inputTokens", 0)
+            output = usage.get("outputTokens", 0)
             self._ingest["units"]["text"] = Units(input=input, output=output)
 
             ingest = True

{payi-0.1.0a113.dist-info → payi-0.1.0a114.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: payi
-Version: 0.1.0a113
+Version: 0.1.0a114
 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.0a113.dist-info → payi-0.1.0a114.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=6EkmubL78eoxEMUlUQ6w1xFWXoNtf9nv86J32AfJMuQ,166
+payi/_version.py,sha256=Youinej2btWjxpYxdf0vmSfYqlK1BlAwZE05kzSgcD4,166
 payi/pagination.py,sha256=k2356QGPOUSjRF2vHpwLBdF6P-2vnQzFfRIJQAHGQ7A,1258
 payi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 payi/_utils/__init__.py,sha256=PNZ_QJuzZEgyYXqkO1HVhGkj5IU9bglVUcw7H-Knjzw,2062
@@ -25,8 +25,8 @@ payi/_utils/_transform.py,sha256=n7kskEWz6o__aoNvhFoGVyDoalNe6mJwp-g7BWkdj88,156
 payi/_utils/_typing.py,sha256=D0DbbNu8GnYQTSICnTSHDGsYXj8TcAKyhejb0XcnjtY,4602
 payi/_utils/_utils.py,sha256=ts4CiiuNpFiGB6YMdkQRh2SZvYvsl7mAF-JWHCcLDf4,12312
 payi/lib/.keep,sha256=wuNrz-5SXo3jJaJOJgz4vFHM41YH_g20F5cRQo0vLes,224
-payi/lib/AnthropicInstrumentor.py,sha256=_moPMUCps6kExYxrkD9xpIbeVIj13MtYkzGpwAA6laA,16453
-payi/lib/BedrockInstrumentor.py,sha256=sfkFeSmsNRTo_4kExdpylzSeZEyA1lIFU3UJGse8AI8,19834
+payi/lib/AnthropicInstrumentor.py,sha256=40jxGiG4od7g68gPLYGXvBrt-cbHorugb7wgZ8vHq-c,16483
+payi/lib/BedrockInstrumentor.py,sha256=HllGynSJ7GDqI3ZuZrNwquPie7uPnh20RrYKbiBY-Sc,19870
 payi/lib/GoogleGenAiInstrumentor.py,sha256=LHiEZ7G5IhCcDlpVzQlXW9Ok96MHLeq7emEhFzPBTm0,8836
 payi/lib/OpenAIInstrumentor.py,sha256=_ULwIli11XP1yZK_pMGXuaSmHZ5pozuEt_v5DfhNuGw,22914
 payi/lib/Stopwatch.py,sha256=7OJlxvr2Jyb6Zr1LYCYKczRB7rDVKkIR7gc4YoleNdE,764
@@ -140,7 +140,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.0a113.dist-info/METADATA,sha256=jsjlj2fj6pGW_1fG2E7TM2RHYyqTDzi4gCOrmROILZY,16324
-payi-0.1.0a113.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-payi-0.1.0a113.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
-payi-0.1.0a113.dist-info/RECORD,,
+payi-0.1.0a114.dist-info/METADATA,sha256=tcxlWZyeZyVyU9dfPsIVEr8ukAILzjDi1hxY4o1Rf9o,16324
+payi-0.1.0a114.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+payi-0.1.0a114.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
+payi-0.1.0a114.dist-info/RECORD,,