paid-python 0.2.1__py3-none-any.whl → 0.3.0__py3-none-any.whl

paid/tracing/autoinstrumentation.py

@@ -56,6 +56,13 @@ try:
 except ImportError:
     LANGCHAIN_AVAILABLE = False
 
+try:
+    from openinference.instrumentation.google_genai import GoogleGenAIInstrumentor
+
+    GOOGLE_GENAI_AVAILABLE = True
+except ImportError:
+    GOOGLE_GENAI_AVAILABLE = False
+
 
 # Track which instrumentors have been initialized
 _initialized_instrumentors: List[str] = []
@@ -77,6 +84,7 @@ def paid_autoinstrument(libraries: Optional[List[str]] = None) -> None:
                   - "openai-agents": OpenAI Agents SDK
                   - "bedrock": AWS Bedrock
                   - "langchain": LangChain library
+                  - "google-genai": Google GenAI library
                   If None, all supported libraries that are installed will be instrumented.
 
     Note:
@@ -106,7 +114,7 @@ def paid_autoinstrument(libraries: Optional[List[str]] = None) -> None:
 
     # Default to all supported libraries if none specified
     if libraries is None:
-        libraries = ["anthropic", "gemini", "openai", "openai-agents", "bedrock", "langchain"]
+        libraries = ["anthropic", "gemini", "openai", "openai-agents", "bedrock", "langchain", "google-genai"]
 
     for library in libraries:
         if library in _initialized_instrumentors:
@@ -125,6 +133,8 @@ def paid_autoinstrument(libraries: Optional[List[str]] = None) -> None:
             _instrument_bedrock()
         elif library == "langchain":
             _instrument_langchain()
+        elif library == "google-genai":
+            _instrument_google_genai()
         else:
             logger.warning(
                 f"Unknown library '{library}' - supported libraries: anthropic, gemini, openai, openai-agents, bedrock, langchain"
@@ -222,3 +232,16 @@ def _instrument_langchain() -> None:
 
     _initialized_instrumentors.append("langchain")
     logger.info("LangChain auto-instrumentation enabled")
+
+def _instrument_google_genai() -> None:
+    """
+    Instrument Google GenAI using openinference-instrumentation-google-genai.
+    """
+    if not GOOGLE_GENAI_AVAILABLE:
+        logger.warning("Google GenAI instrumentation library not available, skipping instrumentation")
+        return
+
+    
+    GoogleGenAIInstrumentor().instrument(tracer_provider=tracing.paid_tracer_provider)
+    _initialized_instrumentors.append("google-genai")
+    logger.info("Google GenAI auto-instrumentation enabled")

{paid_python-0.2.1.dist-info → paid_python-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: paid-python
-Version: 0.2.1
+Version: 0.3.0
 Summary: 
 Requires-Python: >=3.9,<3.14
 Classifier: Intended Audience :: Developers
@@ -20,6 +20,7 @@ Classifier: Typing :: Typed
 Requires-Dist: httpx (>=0.21.0)
 Requires-Dist: mutagen (>=1.47.0)
 Requires-Dist: openinference-instrumentation-bedrock (>=0.1.0)
+Requires-Dist: openinference-instrumentation-google-genai (>=0.1.8)
 Requires-Dist: openinference-instrumentation-openai-agents (>=1.0.0)
 Requires-Dist: opentelemetry-api (>=1.23.0)
 Requires-Dist: opentelemetry-exporter-otlp-proto-http (>=1.23.0)

{paid_python-0.2.1.dist-info → paid_python-0.3.0.dist-info}/RECORD

@@ -37,7 +37,7 @@ paid/orders/lines/raw_client.py,sha256=KZN_yBokCOkf1lUb4ZJtX_NZbqmTqCdJNoaIOdWar
 paid/orders/raw_client.py,sha256=650e1Sj2vi9KVJc15M3ENXIKYoth0qMz66dzvXy1Sb4,16245
 paid/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 paid/tracing/__init__.py,sha256=Pe55koIwqJ6Vv5-9Wqi8xIdwCS2BbxZds-MK5fD-F5Y,506
-paid/tracing/autoinstrumentation.py,sha256=NUVvzYwbrI9rDs0sWEgOk0jYYLQPtjo3Cfk_OsZX6-A,7643
+paid/tracing/autoinstrumentation.py,sha256=LNZJ8wXCu4g8sgd7DkWJCLBljsePr2D0E06scJfXG9g,8470
 paid/tracing/context_data.py,sha256=UFCZxX6zGa9w5lGLQK5tXdMM69aV52nReUkxwfN9S6A,2378
 paid/tracing/context_manager.py,sha256=ZQtsJ9JPxTwn2t4AW26WpYboaOEZdI2T1Sw0Rwsbf-E,8470
 paid/tracing/distributed_tracing.py,sha256=XCUGFBB2lksrR19Mry10cTZRFSCSztUbUjE_sKCwjX8,3995
@@ -99,7 +99,7 @@ paid/usage/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 paid/usage/client.py,sha256=280WJuepoovk3BAVbAx2yN2Q_qBdvx3CcPkLu8lXslc,3030
 paid/usage/raw_client.py,sha256=2acg5C4lxuZodZjepU9QYF0fmBxgG-3ZgXs1zUJG-wM,3709
 paid/version.py,sha256=QIpDFnOrxMxrs86eL0iNH0mSZ1DO078wWHYY9TYAoew,78
-paid_python-0.2.1.dist-info/LICENSE,sha256=Nz4baY1zvv0Qy7lqrQtbaiMhmEeGr2Q7A93aqzpml4c,1071
-paid_python-0.2.1.dist-info/METADATA,sha256=kmG2tSGACEXkk1jBxsOu8DePxUxfu4JKq6aXGe_worE,22343
-paid_python-0.2.1.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-paid_python-0.2.1.dist-info/RECORD,,
+paid_python-0.3.0.dist-info/LICENSE,sha256=Nz4baY1zvv0Qy7lqrQtbaiMhmEeGr2Q7A93aqzpml4c,1071
+paid_python-0.3.0.dist-info/METADATA,sha256=uwDSnsVHDB_IrUSbds61pzA6udH1VxceYb445lSfoAM,22411
+paid_python-0.3.0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+paid_python-0.3.0.dist-info/RECORD,,