ruff

agentops/llms/providers/gemini.py

@@ -7,6 +7,7 @@
 from agentops.log_config import logger
 from agentops.singleton import singleton
 
+
 @singleton
 class GeminiProvider(BaseProvider):
     original_generate_content = None
@@ -27,9 +28,7 @@ def __init__(self, client=None):
         super().__init__(client)
         self._provider_name = "Gemini"
 
-    def handle_response(
-        self, response, kwargs, init_timestamp, session: Optional[Session] = None
-    ) -> dict:
+    def handle_response(self, response, kwargs, init_timestamp, session: Optional[Session] = None) -> dict:
         """Handle responses from Gemini API for both sync and streaming modes.
 
         Args:
@@ -45,7 +44,7 @@ def handle_response(
         llm_event = LLMEvent(init_timestamp=init_timestamp, params=kwargs)
         if session is not None:
             llm_event.session_id = session.session_id
-        
+
         accumulated_content = ""
 
         def handle_stream_chunk(chunk):
@@ -82,10 +81,12 @@ def handle_stream_chunk(chunk):
 
         # For streaming responses
         if kwargs.get("stream", False):
+
             def generator():
                 for chunk in response:
                     handle_stream_chunk(chunk)
                     yield chunk
+
             return generator()
 
         # For synchronous responses
@@ -190,4 +191,4 @@ def undo_override(self):
 
         if self.original_generate_content_async is not None:
             genai.GenerativeModel.generate_content_async = self.original_generate_content_async
-            self.original_generate_content_async = None
+            self.original_generate_content_async = None