payi 0.1.0a110__py3-none-any.whl → 0.1.0a112__py3-none-any.whl

payi/lib/instrument.py

@@ -58,6 +58,7 @@ class _ProviderRequest:
         self._function_call_builder: Optional[dict[int, ProviderResponseFunctionCall]] = None
         self._building_function_response: bool = False
         self._function_calls: Optional[list[ProviderResponseFunctionCall]] = None
+        self._is_large_context: bool = False
 
     def process_chunk(self, _chunk: Any) -> _ChunkResult:
         return _ChunkResult(send_chunk_to_caller=True)
@@ -1305,11 +1306,12 @@ class _PayiInstrumentor:
             extra_headers[PayiHeaderNames.resource_scope] = context_resource_scope
 
     @staticmethod
-    def update_for_vision(input: int, units: 'dict[str, Units]', estimated_prompt_tokens: Optional[int]) -> int:
+    def update_for_vision(input: int, units: 'dict[str, Units]', estimated_prompt_tokens: Optional[int], is_large_context: bool = False) -> int:
         if estimated_prompt_tokens:
             vision = input - estimated_prompt_tokens
             if (vision > 0):
-                units["vision"] = Units(input=vision, output=0)
+                key = "vision_large_context" if is_large_context else "vision"
+                units[key] = Units(input=vision, output=0)
                 input = estimated_prompt_tokens
         
         return input

{payi-0.1.0a110.dist-info → payi-0.1.0a112.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: payi
-Version: 0.1.0a110
+Version: 0.1.0a112
 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
@@ -29,6 +29,7 @@ Requires-Dist: nest-asyncio>=1.6.0
 Requires-Dist: pydantic<3,>=1.9.0
 Requires-Dist: sniffio
 Requires-Dist: tiktoken>=0.8.0
+Requires-Dist: tokenizers>=0.21.4
 Requires-Dist: typing-extensions<5,>=4.10
 Requires-Dist: wrapt>=1.17.2
 Provides-Extra: aiohttp

{payi-0.1.0a110.dist-info → payi-0.1.0a112.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=TopihsJB9WX_1CbkCchG3au71SUzYcXVT4Y70dt8cXA,166
+payi/_version.py,sha256=k7bzf8hHyALFG0ngRJRG3XElf-tU-tJOFcfpnm8vglo,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,16 +25,17 @@ 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=jL9GG56lebZF2b9a5uM4F0cYjTRpxa9tTa8rX3az_po,14405
-payi/lib/BedrockInstrumentor.py,sha256=dMelEiVq5lV1vYh0K7YfXimtFWfZO_zQ3Mhu3y4bjlM,17926
+payi/lib/AnthropicInstrumentor.py,sha256=_moPMUCps6kExYxrkD9xpIbeVIj13MtYkzGpwAA6laA,16453
+payi/lib/BedrockInstrumentor.py,sha256=sfkFeSmsNRTo_4kExdpylzSeZEyA1lIFU3UJGse8AI8,19834
 payi/lib/GoogleGenAiInstrumentor.py,sha256=LHiEZ7G5IhCcDlpVzQlXW9Ok96MHLeq7emEhFzPBTm0,8836
 payi/lib/OpenAIInstrumentor.py,sha256=_ULwIli11XP1yZK_pMGXuaSmHZ5pozuEt_v5DfhNuGw,22914
 payi/lib/Stopwatch.py,sha256=7OJlxvr2Jyb6Zr1LYCYKczRB7rDVKkIR7gc4YoleNdE,764
 payi/lib/VertexInstrumentor.py,sha256=OWuMPiW4LdLhj6DSAAy5qZiosVo8DSAuFWGxYpEucoE,7431
-payi/lib/VertexRequest.py,sha256=edv14HR5QtKaf07gmOqzWdhoJGdWCos85uCZASwsGL4,11710
+payi/lib/VertexRequest.py,sha256=42F7xCRYY6h3EMUZD1x4-_cwyAcVhnzT9M5zl4KwtE0,11801
 payi/lib/helpers.py,sha256=jcMyxsuWmyPymDCYmDbQAb6IgbkmkiiNUaxeEPkCKZs,4457
-payi/lib/instrument.py,sha256=01AVcCT2RRaXlxsz7zd8pappUYpjlFdKovZ8AJanXfE,75349
+payi/lib/instrument.py,sha256=LquKtVUpKnVONnkGchRbLgbvLrhBOIYAwNieH9ANi-E,75500
 payi/lib/version_helper.py,sha256=v0lC3kuaXn6PBDolE3mkmwJiA8Ot3z4RkVR7wlBuZCs,540
+payi/lib/data/cohere_embed_english_v3.json,sha256=YEWwjml3_i16cdsOx_7UKe6xpVFnxTEhP8T1n54R6gY,718306
 payi/resources/__init__.py,sha256=B2bn1ZfCf6TbHlzZvy5TpFPtALnFcBRPYVKQH3S5qfQ,2457
 payi/resources/ingest.py,sha256=7HeaAnIhPmAYdom8tAPPTa1WlbKL8pFCuQdZ2QSFuyE,23079
 payi/resources/categories/__init__.py,sha256=WeotN_d-0Ri8ohsrNPbve7RyViD9_N0NA9DrV3WYg3w,1701
@@ -139,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.0a110.dist-info/METADATA,sha256=ukWuysdwHUC5S6-MXSsZEFEA7OwV4psRhKNhuChZBOs,16290
-payi-0.1.0a110.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-payi-0.1.0a110.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
-payi-0.1.0a110.dist-info/RECORD,,
+payi-0.1.0a112.dist-info/METADATA,sha256=cusLaSHo5PV0mxvGYzdNmlqGkgzlQYt-zO4ugyHcEtk,16324
+payi-0.1.0a112.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+payi-0.1.0a112.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
+payi-0.1.0a112.dist-info/RECORD,,