Fix Ruff CI failures: remove unused variables and apply formatting

Author: Alex Reibman

agentops/client/api/base.py

@@ -15,7 +15,8 @@
 class TokenFetcher(Protocol):
     """Protocol for token fetching functions"""
 
-    def __call__(self, api_key: str) -> str: ...
+    def __call__(self, api_key: str) -> str:
+        ...
 
 
 class BaseApiClient:

agentops/client/client.py

@@ -40,9 +40,9 @@ class Client:
     config: Config
     _initialized: bool
     _init_trace_context: Optional[TraceContext] = None  # Stores the context of the auto-started trace
-    _legacy_session_for_init_trace: Optional[Session] = (
-        None  # Stores the legacy Session wrapper for the auto-started trace
-    )
+    _legacy_session_for_init_trace: Optional[
+        Session
+    ] = None  # Stores the legacy Session wrapper for the auto-started trace
 
     __instance = None  # Class variable for singleton pattern
 

agentops/instrumentation/agentic/google_adk/patch.py

@@ -304,16 +304,16 @@ def _extract_llm_attributes(llm_request_dict: dict, llm_response: Any) -> dict:
                     elif "function_call" in part:
                         # This is a function call in the response
                         func_call = part["function_call"]
-                        attributes[MessageAttributes.COMPLETION_TOOL_CALL_NAME.format(i=0, j=tool_call_index)] = (
-                            func_call.get("name", "")
-                        )
-                        attributes[MessageAttributes.COMPLETION_TOOL_CALL_ARGUMENTS.format(i=0, j=tool_call_index)] = (
-                            json.dumps(func_call.get("args", {}))
-                        )
+                        attributes[
+                            MessageAttributes.COMPLETION_TOOL_CALL_NAME.format(i=0, j=tool_call_index)
+                        ] = func_call.get("name", "")
+                        attributes[
+                            MessageAttributes.COMPLETION_TOOL_CALL_ARGUMENTS.format(i=0, j=tool_call_index)
+                        ] = json.dumps(func_call.get("args", {}))
                         if "id" in func_call:
-                            attributes[MessageAttributes.COMPLETION_TOOL_CALL_ID.format(i=0, j=tool_call_index)] = (
-                                func_call["id"]
-                            )
+                            attributes[
+                                MessageAttributes.COMPLETION_TOOL_CALL_ID.format(i=0, j=tool_call_index)
+                            ] = func_call["id"]
                         tool_call_index += 1
 
                 if text_parts:

agentops/instrumentation/agentic/openai_agents/attributes/completion.py

@@ -115,9 +115,9 @@ def get_raw_response_attributes(response: Dict[str, Any]) -> Dict[str, Any]:
                                 result[MessageAttributes.COMPLETION_TOOL_CALL_NAME.format(i=j, j=k)] = function.get(
                                     "name", ""
                                 )
-                                result[MessageAttributes.COMPLETION_TOOL_CALL_ARGUMENTS.format(i=j, j=k)] = (
-                                    function.get("arguments", "")
-                                )
+                                result[
+                                    MessageAttributes.COMPLETION_TOOL_CALL_ARGUMENTS.format(i=j, j=k)
+                                ] = function.get("arguments", "")
 
     return result
 

agentops/instrumentation/agentic/smolagents/attributes/model.py

@@ -102,13 +102,13 @@ def get_model_attributes(
                 if hasattr(return_value, "tool_calls") and return_value.tool_calls:
                     for j, tool_call in enumerate(return_value.tool_calls):
                         if hasattr(tool_call, "function"):
-                            attributes[MessageAttributes.COMPLETION_TOOL_CALL_NAME.format(i=0, j=j)] = (
-                                tool_call.function.name
-                            )
+                            attributes[
+                                MessageAttributes.COMPLETION_TOOL_CALL_NAME.format(i=0, j=j)
+                            ] = tool_call.function.name
                             if hasattr(tool_call.function, "arguments"):
-                                attributes[MessageAttributes.COMPLETION_TOOL_CALL_ARGUMENTS.format(i=0, j=j)] = (
-                                    tool_call.function.arguments
-                                )
+                                attributes[
+                                    MessageAttributes.COMPLETION_TOOL_CALL_ARGUMENTS.format(i=0, j=j)
+                                ] = tool_call.function.arguments
                             if hasattr(tool_call, "id"):
                                 attributes[MessageAttributes.COMPLETION_TOOL_CALL_ID.format(i=0, j=j)] = tool_call.id
 

agentops/instrumentation/common/attributes.py

@@ -98,7 +98,8 @@ class IndexedAttribute(Protocol):
     formatting of attribute keys based on the indices.
     """
 
-    def format(self, *, i: int, j: Optional[int] = None) -> str: ...
+    def format(self, *, i: int, j: Optional[int] = None) -> str:
+        ...
 
 
 IndexedAttributeMap = Dict[IndexedAttribute, str]  # target_attribute_key: source_attribute

examples/openai/o3_responses_example.py

@@ -120,7 +120,6 @@ def make_decision_sync(self, scenario: str, available_actions: List[str], stream
             # Process streaming response
             tool_call = None
             tool_arguments = ""
-            current_call_id = None
 
             for event in response:
                 if hasattr(event, "type"):
@@ -134,7 +133,7 @@ def make_decision_sync(self, scenario: str, available_actions: List[str], stream
                     elif event.type == "response.output_item.added":
                         # New tool call started
                         if hasattr(event, "output_item") and event.output_item.type == "function_call":
-                            current_call_id = event.output_item.call_id
+                            pass  # Tool call tracking handled elsewhere
                     elif event.type == "response.completed":
                         # Process final response
                         if hasattr(event, "response") and hasattr(event.response, "output"):
@@ -267,7 +266,6 @@ async def make_decision_async(
             # Process streaming response
             tool_call = None
             tool_arguments = ""
-            current_call_id = None
 
             async for event in response:
                 if hasattr(event, "type"):
@@ -281,7 +279,7 @@ async def make_decision_async(
                     elif event.type == "response.output_item.added":
                         # New tool call started
                         if hasattr(event, "output_item") and event.output_item.type == "function_call":
-                            current_call_id = event.output_item.call_id
+                            pass  # Tool call tracking handled elsewhere
                     elif event.type == "response.completed":
                         # Process final response
                         if hasattr(event, "response") and hasattr(event.response, "output"):

tests/unit/instrumentation/openai_core/test_instrumentor.py

@@ -107,9 +107,9 @@ def test_instrument_method_wraps_response_api(self, instrumentor):
             instrumentor_obj._custom_wrap()
 
             # Verify wrap_function_wrapper was called for Response API methods
-            assert mock_wfw.call_count >= 2, (
-                f"Expected at least 2 calls to wrap_function_wrapper, got {mock_wfw.call_count}"
-            )
+            assert (
+                mock_wfw.call_count >= 2
+            ), f"Expected at least 2 calls to wrap_function_wrapper, got {mock_wfw.call_count}"
 
             # Find Response API calls
             response_api_calls = []

tests/unit/sdk/instrumentation_tester.py

@@ -45,7 +45,8 @@ def reset_trace_globals():
 
 class HasAttributesViaProperty(Protocol):
     @property
-    def attributes(self) -> Attributes: ...
+    def attributes(self) -> Attributes:
+        ...
 
 
 class HasAttributesViaAttr(Protocol):