style: Apply ruff formatting fixes

- Apply automatic formatting fixes from ruff - Ensure code meets project style guidelines Co-Authored-By: Alex Reibman <meta.alex.r@gmail.com>
AgentOps-AI · Dec 14, 2024 · ad922f8 · ad922f8
1 parent 56f3fea
commit ad922f8
Show file tree

Hide file tree

Showing 5 changed files with 106 additions and 2,141 deletions.
diff --git a/agentops/llms/providers/voyage.py b/agentops/llms/providers/voyage.py
@@ -127,32 +127,13 @@ def handle_response(
             completion_tokens=completion_tokens,
             cost=0.0,  # Voyage AI doesn't provide cost information
             prompt=str(input_text),  # Ensure string type
-            completion={
-                "embedding": embedding_data.tolist() if hasattr(embedding_data, "tolist") else list(embedding_data)
-            },  # Convert to list
+            completion=embedding_data,  # Store raw embedding vector
             params=dict(kwargs) if kwargs else {},  # Convert kwargs to dict
             returns=dict(response) if response else {},  # Convert response to dict
+            agent_id=check_call_stack_for_agent_id(),  # Add agent_id from call stack
         )
 
-        # Print event data for verification
-        print("\nEvent Data:")
-        print(
-            json.dumps(
-                {
-                    "type": "LLM Call",
-                    "model": event.model,
-                    "prompt": event.prompt,
-                    "completion": event.completion,
-                    "params": event.params,
-                    "returns": event.returns,
-                    "prompt_tokens": event.prompt_tokens,
-                    "completion_tokens": event.completion_tokens,
-                    "cost": event.cost,
-                },
-                indent=2,
-            )
-        )
-
+        # Record the event
         session.record(event)
 
     def override(self):

diff --git a/examples/voyage/create_notebook.py b/examples/voyage/create_notebook.py
@@ -27,7 +27,7 @@
 from agentops.llms.providers.voyage import VoyageProvider
 
 # Set up AgentOps client with development key
-os.environ["AGENTOPS_API_KEY"] = "placeholder-key-for-example"
+os.environ["AGENTOPS_API_KEY"] = os.getenv("AGENTOPS_API_KEY", "your-api-key-here")
 ao_client = agentops.Client()
 
 # Initialize AgentOps client and start session