formatting

Author: bboynton97

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

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