payi 0.1.0a45__tar.gz → 0.1.0a46__tar.gz

payi-0.1.0a46/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.1.0-alpha.46"
+}

{payi-0.1.0a45 → payi-0.1.0a46}/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## 0.1.0-alpha.46 (2025-02-06)
+
+Full Changelog: [v0.1.0-alpha.45...v0.1.0-alpha.46](https://github.com/Pay-i/pay-i-python/compare/v0.1.0-alpha.45...v0.1.0-alpha.46)
+
+### Features
+
+* feat: OpenAI embeddings instrumentation ([8649277](https://github.com/Pay-i/pay-i-python/commit/864927729c11c54b4505123ee9c8c3ac6fec2304))
+
 ## 0.1.0-alpha.45 (2025-02-06)
 
 Full Changelog: [v0.1.0-alpha.44...v0.1.0-alpha.45](https://github.com/Pay-i/pay-i-python/compare/v0.1.0-alpha.44...v0.1.0-alpha.45)

{payi-0.1.0a45 → payi-0.1.0a46}/PKG-INFO

@@ -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 → payi-0.1.0a46}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "payi"
-version = "0.1.0-alpha.45"
+version = "0.1.0-alpha.46"
 description = "The official Python library for the payi API"
 dynamic = ["readme"]
 license = "Apache-2.0"

{payi-0.1.0a45 → payi-0.1.0a46}/src/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-0.1.0a45 → payi-0.1.0a46}/src/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/.release-please-manifest.json

@@ -1,3 +0,0 @@
-{
-  ".": "0.1.0-alpha.45"
-}