paid-python 0.2.1__tar.gz → 0.3.0__tar.gz

{paid_python-0.2.1 → paid_python-0.3.0}/PKG-INFO

@@ -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 → paid_python-0.3.0}/pyproject.toml

@@ -3,7 +3,7 @@ name = "paid-python"
 
 [tool.poetry]
 name = "paid-python"
-version = "0.2.1"
+version = "0.3.0"
 description = ""
 readme = "README.md"
 authors = []
@@ -49,6 +49,7 @@ opentelemetry-instrumentation-openai = ">=0.47.0"
 opentelemetry-instrumentation-langchain = ">=0.47.0"
 openinference-instrumentation-openai-agents = ">=1.0.0"
 openinference-instrumentation-bedrock = ">=0.1.0"
+openinference-instrumentation-google-genai = ">=0.1.8"
 
 [tool.poetry.group.dev.dependencies]
 mypy = "==1.13.0"

{paid_python-0.2.1 → paid_python-0.3.0}/src/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")