style: Apply ruff formatting

Co-Authored-By: Alex Reibman <[email protected]>

agentops/llms/providers/voyage.py

@@ -1,5 +1,6 @@
+import sys
 import json
-from typing import Optional
+from typing import Optional, Callable
 
 from agentops.llms.providers.instrumented_provider import InstrumentedProvider
 from agentops.event import LLMEvent, ErrorEvent
@@ -11,15 +12,51 @@
 
 @singleton
 class VoyageProvider(InstrumentedProvider):
-    original_embed = None
-    original_embed_async = None
+    """Provider for Voyage AI SDK integration.
+
+    Handles embedding operations and tracks usage through AgentOps.
+    Requires Python >=3.9 for full functionality.
+
+    Args:
+        client: Initialized Voyage AI client instance
+    """
+
+    original_embed: Optional[Callable] = None
+    original_embed_async: Optional[Callable] = None
 
     def __init__(self, client):
+        """Initialize the Voyage provider with a client instance.
+
+        Args:
+            client: An initialized Voyage AI client
+        """
         super().__init__(client)
         self._provider_name = "Voyage"
+        if not self._check_python_version():
+            logger.warning("Voyage AI SDK requires Python >=3.9. Some functionality may not work correctly.")
+
+    def _check_python_version(self) -> bool:
+        """Check if the current Python version meets Voyage AI requirements.
+
+        Returns:
+            bool: True if Python version is >= 3.9, False otherwise
+        """
+        return sys.version_info >= (3, 9)
+
+    def handle_response(
+        self, response: dict, kwargs: dict, init_timestamp: str, session: Optional[Session] = None
+    ) -> dict:
+        """Handle responses for Voyage AI embeddings.
+
+        Args:
+            response: The response from Voyage AI API
+            kwargs: The keyword arguments used in the API call
+            init_timestamp: The timestamp when the API call was initiated
+            session: Optional session for tracking events
 
-    def handle_response(self, response, kwargs, init_timestamp, session: Optional[Session] = None):
-        """Handle responses for Voyage AI"""
+        Returns:
+            dict: The original response from the API
+        """
         llm_event = LLMEvent(init_timestamp=init_timestamp, params=kwargs)
         if session is not None:
             llm_event.session_id = session.session_id
@@ -39,6 +76,7 @@ def handle_response(self, response, kwargs, init_timestamp, session: Optional[Se
         return response
 
     def override(self):
+        """Override Voyage AI SDK methods with instrumented versions."""
         import voyageai
 
         self.original_embed = voyageai.Client.embed
@@ -53,6 +91,7 @@ def patched_function(self, *args, **kwargs):
         voyageai.Client.embed = patched_function
 
     def undo_override(self):
+        """Restore original Voyage AI SDK methods."""
         import voyageai
 
         if self.original_embed:

examples/voyage/create_notebook.py

@@ -1,21 +1,33 @@
+import sys
+import warnings
 import nbformat as nbf
 
-# Create a new notebook
-nb = nbf.v4.new_notebook()
 
-# Add metadata
-nb.metadata = {"kernelspec": {"display_name": "Python 3", "language": "python", "name": "python3"}}
+def create_voyage_example():
+    if sys.version_info < (3, 9):
+        warnings.warn("Voyage AI SDK requires Python >=3.9. Example may not work correctly.")
 
-# Create markdown cell
-markdown_cell = nbf.v4.new_markdown_cell(
-    """# Using AgentOps with Voyage AI
+    # Create a new notebook
+    nb = nbf.v4.new_notebook()
 
-This notebook demonstrates how to use AgentOps to track Voyage AI embeddings."""
-)
+    # Add metadata
+    nb.metadata = {"kernelspec": {"display_name": "Python 3", "language": "python", "name": "python3"}}
 
-# Create code cell
-code_cell = nbf.v4.new_code_cell(
-    """import os
+    # Create markdown cell with version warning
+    markdown_content = """# Using AgentOps with Voyage AI
+
+This notebook demonstrates how to use AgentOps to track Voyage AI embeddings.
+
+> **Note:** Voyage AI SDK requires Python >=3.9. Please ensure you have a compatible Python version installed."""
+
+    markdown_cell = nbf.v4.new_markdown_cell(markdown_content)
+
+    # Create code cell with version check
+    code_content = """import sys
+if sys.version_info < (3, 9):
+    print("Warning: Voyage AI SDK requires Python >=3.9. Example may not work correctly.")
+
+import os
 import voyageai
 import agentops as ao
 
@@ -29,11 +41,16 @@
 
 # View events in AgentOps dashboard
 print(f"View session at: {ao_client.dashboard_url}")"""
-)
 
-# Add cells to notebook
-nb.cells = [markdown_cell, code_cell]
+    code_cell = nbf.v4.new_code_cell(code_content)
+
+    # Add cells to notebook
+    nb.cells = [markdown_cell, code_cell]
+
+    # Save the notebook
+    with open("basic_usage.ipynb", "w") as f:
+        nbf.write(nb, f)
+
 
-# Save the notebook
-with open("basic_usage.ipynb", "w") as f:
-    nbf.write(nb, f)
+if __name__ == "__main__":
+    create_voyage_example()

pyproject.toml

@@ -49,9 +49,6 @@ langchain = [
 ci = [
     "tach~=0.9",
 ]
-voyage = [
-    "voyageai>=0.1.0"
-]
 
 [project.urls]
 Homepage = "https://github.com/AgentOps-AI/agentops"

tests/test_session.py

@@ -590,11 +590,19 @@ def test_export_llm_event(self, mock_req):
         assert event["end_timestamp"] is not None
 
     def test_voyage_provider(self, mock_req):
+        """Test Voyage AI provider integration with mocked client."""
         from agentops.llms.providers.voyage import VoyageProvider
 
-        provider = VoyageProvider()
+        # Mock Voyage client
+        class MockVoyageClient:
+            def embed(self, *args, **kwargs):
+                return {"embeddings": [[0.1, 0.2, 0.3]]}
+
+        # Initialize provider with mock client
+        provider = VoyageProvider(MockVoyageClient())
         assert provider is not None
 
+        # Test event attributes
         voyage_attributes = {
             "event.data": json.dumps(
                 {
@@ -612,6 +620,7 @@ def test_voyage_provider(self, mock_req):
 
         assert result == SpanExportResult.SUCCESS
 
+        # Verify event attributes
         last_request = mock_req.request_history[-1].json()
         event = last_request["events"][0]
 
@@ -624,6 +633,12 @@ def test_voyage_provider(self, mock_req):
         assert event["init_timestamp"] is not None
         assert event["end_timestamp"] is not None
 
+        # Test embedding functionality
+        result = provider.client.embed("test input")
+        assert "embeddings" in result
+        assert len(result["embeddings"]) == 1
+        assert len(result["embeddings"][0]) == 3
+
     def test_export_with_missing_id(self, mock_req):
         """Test handling of missing event ID"""
         attributes = {"event.id": None}