payi 0.1.0a45__py3-none-any.whl → 0.1.0a46__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.45"  # x-release-please-version
+__version__ = "0.1.0-alpha.46"  # x-release-please-version

payi/lib/OpenAIInstrumentor.py

@@ -23,7 +23,7 @@ class OpenAiInstrumentor:
     def instrument(instrumentor: PayiInstrumentor) -> None:
         try:
             from openai import OpenAI  # type: ignore #  noqa: F401  I001
-
+            
             wrap_function_wrapper(
                 "openai.resources.chat.completions",
                 "Completions.create",
@@ -36,11 +36,63 @@ class OpenAiInstrumentor:
                 achat_wrapper(instrumentor),
             )
 
+            wrap_function_wrapper(
+                "openai.resources.embeddings",
+                "Embeddings.create",
+                embeddings_wrapper(instrumentor),
+            )
+
+            wrap_function_wrapper(
+                "openai.resources.embeddings",
+                 "AsyncEmbeddings.create",
+                aembeddings_wrapper(instrumentor),
+            )
+
         except Exception as e:
             logging.debug(f"Error instrumenting openai: {e}")
             return
 
 
+@PayiInstrumentor.payi_wrapper
+def embeddings_wrapper(
+    instrumentor: PayiInstrumentor,
+    wrapped: Any,
+    instance: Any,
+    *args: Any,
+    **kwargs: Any,
+) -> Any:
+    return instrumentor.chat_wrapper(
+        "system.openai",
+        None, # process_chat_chunk,
+        None, # process_chat_request,
+        process_ebmeddings_synchronous_response,
+        IsStreaming.false,
+        wrapped,
+        instance,
+        args,
+        kwargs,
+    )
+
+@PayiInstrumentor.payi_wrapper
+async def aembeddings_wrapper(
+    instrumentor: PayiInstrumentor,
+    wrapped: Any,
+    instance: Any,
+    *args: Any,
+    **kwargs: Any,
+) -> Any:
+    return await instrumentor.achat_wrapper(
+        "system.openai",
+        None, # process_chat_chunk,
+        None, # process_chat_request,
+        process_ebmeddings_synchronous_response,
+        IsStreaming.false,
+        wrapped,
+        instance,
+        args,
+        kwargs,
+    )
+
 @PayiInstrumentor.payi_wrapper
 def chat_wrapper(
     instrumentor: PayiInstrumentor,
@@ -52,7 +104,7 @@ def chat_wrapper(
     return instrumentor.chat_wrapper(
         "system.openai",
         process_chat_chunk,
-        process_request,
+        process_chat_request,
         process_chat_synchronous_response,
         IsStreaming.kwargs,
         wrapped,
@@ -72,7 +124,7 @@ async def achat_wrapper(
     return await instrumentor.achat_wrapper(
         "system.openai",
         process_chat_chunk,
-        process_request,
+        process_chat_request,
         process_chat_synchronous_response,
         IsStreaming.kwargs,
         wrapped,
@@ -81,11 +133,13 @@ async def achat_wrapper(
         kwargs,
     )
 
+def process_ebmeddings_synchronous_response(response: str, ingest: IngestUnitsParams, log_prompt_and_response: bool, **kwargs: Any) -> Any: #  noqa: ARG001
+    return process_chat_synchronous_response(response, ingest, log_prompt_and_response, **kwargs)
 
 def process_chat_synchronous_response(response: str, ingest: IngestUnitsParams, log_prompt_and_response: bool, **kwargs: Any) -> Any: #  noqa: ARG001
     response_dict = model_to_dict(response)
 
-    add_usage_units(response_dict["usage"], ingest["units"])
+    add_usage_units(response_dict.get("usage", {}), ingest["units"])
 
     if log_prompt_and_response:
         ingest["provider_response_json"] = [json.dumps(response_dict)]
@@ -136,7 +190,7 @@ def has_image_and_get_texts(encoding: tiktoken.Encoding, content: Union[str, 'li
         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, *args: Any, **kwargs: Any) -> None: # noqa: ARG001
+def process_chat_request(ingest: IngestUnitsParams, *args: Any, **kwargs: Any) -> None: # noqa: ARG001
     messages = kwargs.get("messages")
     if not messages or len(messages) == 0:
         return

{payi-0.1.0a45.dist-info → payi-0.1.0a46.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: payi
-Version: 0.1.0a45
+Version: 0.1.0a46
 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.0a45.dist-info → payi-0.1.0a46.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=VC4OQPU9XobR2EdEK-EBM6augnkc71ZliJY4u8c_IoY,165
+payi/_version.py,sha256=ml75PCr3anMJW3fkTojqOUY2T3iVxRGV0E2gWIEhotQ,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
@@ -26,7 +26,7 @@ payi/lib/.keep,sha256=wuNrz-5SXo3jJaJOJgz4vFHM41YH_g20F5cRQo0vLes,224
 payi/lib/AnthropicInstrumentor.py,sha256=1dJTFs5WfPxbIjWcYV2tP3kiYc4g3WyyyR4XWQ-WI_g,5300
 payi/lib/BedrockInstrumentor.py,sha256=B5deT8TcUvOsRdIewbbHtsxTS7_oHJ05ZUIdFziWf8E,10082
 payi/lib/Instruments.py,sha256=bapmVS9jbHtFknXCKDzsFFWvf5XLtzEpdlvI7iEWY-o,147
-payi/lib/OpenAIInstrumentor.py,sha256=Xn0gWnQUS6UkY3mn3vFFa8wqSbC9QaDGKpBdD9j7aGU,5019
+payi/lib/OpenAIInstrumentor.py,sha256=rxu5wOUBZaGwFjUoUhk5UDeTcYSmmO6BVgnXL4yYb08,6613
 payi/lib/Stopwatch.py,sha256=7OJlxvr2Jyb6Zr1LYCYKczRB7rDVKkIR7gc4YoleNdE,764
 payi/lib/helpers.py,sha256=ZgkY8UE2YRc7ok2Pmxg_T9UMqKI8D8542JY3CP8RZCM,1597
 payi/lib/instrument.py,sha256=rQCBCyFicTtOjuASwTtCEGvDoffUUgYQoYcuiSlVUlk,35356
@@ -113,7 +113,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.0a45.dist-info/METADATA,sha256=FMnDJJt7RDH8Yrlfsx7KnDDNOQITK1TSQoAIEygqBWQ,12660
-payi-0.1.0a45.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-payi-0.1.0a45.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
-payi-0.1.0a45.dist-info/RECORD,,
+payi-0.1.0a46.dist-info/METADATA,sha256=4jSFupUbO3KTCTYi_UCy_-lvzGZ3hGi57g-J57Wi_1E,12660
+payi-0.1.0a46.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+payi-0.1.0a46.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
+payi-0.1.0a46.dist-info/RECORD,,