payi 0.1.0a78__py3-none-any.whl → 0.1.0a80__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.78"  # x-release-please-version
+__version__ = "0.1.0-alpha.80"  # x-release-please-version

payi/lib/AnthropicInstrumentor.py

@@ -190,16 +190,20 @@ class _AnthropicProviderRequest(_ProviderRequest):
             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 has_image and estimated_token_count > 0:
-                self._estimated_prompt_tokens = estimated_token_count
+            try:
+                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 has_image and estimated_token_count > 0:
+                    self._estimated_prompt_tokens = estimated_token_count
+     
+            except Exception:
+                logging.warning("Error getting encoding for cl100k_base")
+
         return True
 
     @override

payi/lib/GoogleGenAiInstrumentor.py

@@ -279,8 +279,13 @@ class _GoogleGenAiRequest(_ProviderRequest):
         kwargs: Any) -> Any:
         response_dict = response.to_json_dict()
 
-        self._ingest["provider_response_id"] = response_dict["response_id"]
-        self._ingest["resource"] = "google." + response_dict["model_version"]
+        id: Optional[str] = response_dict.get("response_id", None)
+        if id:
+            self._ingest["provider_response_id"] = id
+        
+        model: Optional[str] = response_dict.get("model_version", None)
+        if model:
+            self._ingest["resource"] = "google." + model
 
         self._compute_usage(response_dict)
         

payi/lib/OpenAIInstrumentor.py

@@ -364,20 +364,26 @@ class _OpenAiChatProviderRequest(_OpenAiProviderRequest):
         if messages:
             estimated_token_count = 0 
             has_image = False
+            enc: Optional[tiktoken.Encoding] = None
 
             try: 
                 enc = tiktoken.encoding_for_model(kwargs.get("model")) # type: ignore
-            except KeyError:
-                enc = tiktoken.get_encoding("o200k_base") # type: ignore
+            except Exception:
+                try:
+                    enc = tiktoken.get_encoding("o200k_base") # type: ignore
+                except Exception:
+                    logging.warning("Error getting encoding for fallback o200k_base")
+                    enc = None
             
-            for message in messages:
-                msg_has_image, msg_prompt_tokens = self.has_image_and_get_texts(enc, message.get('content', ''))
-                if msg_has_image:
-                    has_image = True
-                    estimated_token_count += msg_prompt_tokens
+            if enc:
+                for message in messages:
+                    msg_has_image, msg_prompt_tokens = self.has_image_and_get_texts(enc, message.get('content', ''))
+                    if msg_has_image:
+                        has_image = True
+                        estimated_token_count += msg_prompt_tokens
             
-            if has_image and estimated_token_count > 0:
-                self._estimated_prompt_tokens = estimated_token_count
+                if has_image and estimated_token_count > 0:
+                    self._estimated_prompt_tokens = estimated_token_count
 
             stream: bool = kwargs.get("stream", False)
             if stream:
@@ -436,11 +442,16 @@ class _OpenAiResponsesProviderRequest(_OpenAiProviderRequest):
         
         estimated_token_count = 0 
         has_image = False
+        enc: Optional[tiktoken.Encoding] = None
 
         try: 
             enc = tiktoken.encoding_for_model(kwargs.get("model")) # type: ignore
-        except KeyError:
-            enc = tiktoken.get_encoding("o200k_base") # type: ignore
+        except Exception:
+            try:
+                enc = tiktoken.get_encoding("o200k_base") # type: ignore
+            except Exception:
+                logging.warning("Error getting encoding for fallback o200k_base")
+                enc = None
         
         # find each content..type="input_text" and count tokens
         # input=[{
@@ -456,18 +467,19 @@ class _OpenAiResponsesProviderRequest(_OpenAiProviderRequest):
         #         },
         #     ],
         # }]
-        for item in input: # type: ignore
-            if isinstance(item, dict):
-                for key, value in item.items(): # type: ignore
-                    if key == "content":
-                        if isinstance(value, list):
-                            msg_has_image, msg_prompt_tokens = self.has_image_and_get_texts(enc, value, image_type="input_image", text_type="input_text") # type: ignore 
-                            if msg_has_image:
-                                has_image = True
-                                estimated_token_count += msg_prompt_tokens
-
-        if has_image and estimated_token_count > 0:
-            self._estimated_prompt_tokens = estimated_token_count
+        if enc:
+            for item in input: # type: ignore
+                if isinstance(item, dict):
+                    for key, value in item.items(): # type: ignore
+                        if key == "content":
+                            if isinstance(value, list):
+                                msg_has_image, msg_prompt_tokens = self.has_image_and_get_texts(enc, value, image_type="input_image", text_type="input_text") # type: ignore 
+                                if msg_has_image:
+                                    has_image = True
+                                    estimated_token_count += msg_prompt_tokens
+
+            if has_image and estimated_token_count > 0:
+                self._estimated_prompt_tokens = estimated_token_count
 
         return True
 

payi/lib/VertexInstrumentor.py

@@ -225,8 +225,13 @@ class _GoogleVertexRequest(_ProviderRequest):
         kwargs: Any) -> Any:
         response_dict = response.to_dict()
 
-        self._ingest["provider_response_id"] = response_dict["response_id"]
-        self._ingest["resource"] = "google." + response_dict["model_version"]
+        id: Optional[str] = response_dict.get("response_id", None)
+        if id:
+            self._ingest["provider_response_id"] = id
+        
+        model: Optional[str] = response_dict.get("model_version", None)
+        if model:
+            self._ingest["resource"] = "google." + model
 
         self._compute_usage(response_dict)
         

{payi-0.1.0a78.dist-info → payi-0.1.0a80.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: payi
-Version: 0.1.0a78
+Version: 0.1.0a80
 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.0a78.dist-info → payi-0.1.0a80.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=562XNlpNWouu5fR8P9024ptW4UWs2S750PSKhFzqfdY,165
+payi/_version.py,sha256=8owWzJCmVOUMtN_Md7Vwk-OBmbGW5fhSMc2EUWFK3X8,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
@@ -25,12 +25,12 @@ 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=q1Iq2NG06OHUwcwTgYjHa1_kwhbSxE3vrGk_fm5IGSM,8749
+payi/lib/AnthropicInstrumentor.py,sha256=Y7W99dwnaAM4W5IIkmeGLaTunoMsBDt-AjUasrI1mXM,8900
 payi/lib/BedrockInstrumentor.py,sha256=cIjHlPQmVEAixabUoT3mV8h50lfzdc1u8iLj0gQNukE,14076
-payi/lib/GoogleGenAiInstrumentor.py,sha256=y9FuEtuITaIac_N-WhmypAr-UF1lzRCc9YBRVwwwydQ,13411
-payi/lib/OpenAIInstrumentor.py,sha256=iMOriAm-2zcRNjQDkEAG-NfZhBJUnkBDrNNbO920ajc,17468
+payi/lib/GoogleGenAiInstrumentor.py,sha256=isfuYH0y9LZZPcjPkynRQzwhtsN9H2p4DqtPYTm8hqI,13549
+payi/lib/OpenAIInstrumentor.py,sha256=6h5MYtj4jAXEJZ_IyMriZxWXrN-hQ7SdJXIQW7DrTFo,18014
 payi/lib/Stopwatch.py,sha256=7OJlxvr2Jyb6Zr1LYCYKczRB7rDVKkIR7gc4YoleNdE,764
-payi/lib/VertexInstrumentor.py,sha256=Xmp5kRI8G0u2OieGmlNxIduwnqp8cAXO-fROpkdXbgs,11748
+payi/lib/VertexInstrumentor.py,sha256=aW4ZT7YVXy8V_g91KkSEapzI7Cy6UQfaU1D0ZOU4eBE,11886
 payi/lib/helpers.py,sha256=K1KAfWrpPT1UUGNxspLe1lHzQjP3XV5Pkh9IU4pKMok,4624
 payi/lib/instrument.py,sha256=UpZ6SGg3YI9lSxmwH5ziwe1xt_ca6FS35CcNJpG9ONM,58214
 payi/resources/__init__.py,sha256=1rtrPLWbNt8oJGOp6nwPumKLJ-ftez0B6qwLFyfcoP4,2972
@@ -142,7 +142,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.0a78.dist-info/METADATA,sha256=kKY9frrRc7_WbcIUIphMDmRBHQVZgdSGH1eLkF26Pzk,15180
-payi-0.1.0a78.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-payi-0.1.0a78.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
-payi-0.1.0a78.dist-info/RECORD,,
+payi-0.1.0a80.dist-info/METADATA,sha256=__22ITl_q65jasQ663-ur0WOvHXW-PYzIVu0tq6PnYs,15180
+payi-0.1.0a80.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+payi-0.1.0a80.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
+payi-0.1.0a80.dist-info/RECORD,,