style: Apply ruff-format fixes to Gemini integration

Co-Authored-By: Alex Reibman <meta.alex.r@gmail.com>

Author: devin-ai-integration[bot]

agentops/llms/providers/gemini.py

@@ -134,8 +134,8 @@ def override(self):
         import google.generativeai as genai
 
         # Store original method if not already stored
-        if 'generate_content' not in _ORIGINAL_METHODS:
-            _ORIGINAL_METHODS['generate_content'] = genai.GenerativeModel.generate_content
+        if "generate_content" not in _ORIGINAL_METHODS:
+            _ORIGINAL_METHODS["generate_content"] = genai.GenerativeModel.generate_content
 
         # Store provider instance for the closure
         provider = self
@@ -145,8 +145,8 @@ def patched_function(self, *args, **kwargs):
             session = kwargs.pop("session", None)  # Always try to pop session, returns None if not present
 
             # Call original method and track event
-            if 'generate_content' in _ORIGINAL_METHODS:
-                result = _ORIGINAL_METHODS['generate_content'](self, *args, **kwargs)
+            if "generate_content" in _ORIGINAL_METHODS:
+                result = _ORIGINAL_METHODS["generate_content"](self, *args, **kwargs)
                 return provider.handle_response(result, kwargs, init_timestamp, session=session)
             else:
                 logger.error("Original generate_content method not found. Cannot proceed with override.")
@@ -161,6 +161,7 @@ def undo_override(self):
         Note:
             This method is called automatically by AgentOps during cleanup.
             Users should not call this method directly."""
-        if 'generate_content' in _ORIGINAL_METHODS:
+        if "generate_content" in _ORIGINAL_METHODS:
             import google.generativeai as genai
-            genai.GenerativeModel.generate_content = _ORIGINAL_METHODS['generate_content']
+
+            genai.GenerativeModel.generate_content = _ORIGINAL_METHODS["generate_content"]

agentops/llms/tracker.py

@@ -220,6 +220,7 @@ def override_api(self):
                     if Version(module_version) >= parse("0.1.0"):
                         import google.generativeai as genai
                         import os
+
                         api_key = os.getenv("GEMINI_API_KEY")
                         if api_key:
                             try:

examples/gemini_examples/run_test.py

@@ -1,4 +1,5 @@
 import os
+
 os.environ["GEMINI_API_KEY"] = "${GEMINI_API_KEY}"
 
 # Now run the test notebook