chore: formatting

Author: viadezo1er

py/src/braintrust/oai.py

@@ -350,7 +350,12 @@ def _postprocess_streaming_results(cls, all_results: list[dict[str, Any]]) -> di
 
 
 class ResponseWrapper:
-    def __init__(self, create_fn: Callable[..., Any] | None, acreate_fn: Callable[..., Any] | None, name: str = "openai.responses.create"):
+    def __init__(
+        self,
+        create_fn: Callable[..., Any] | None,
+        acreate_fn: Callable[..., Any] | None,
+        name: str = "openai.responses.create",
+    ):
         self.create_fn = create_fn
         self.acreate_fn = acreate_fn
         self.name = name
@@ -359,9 +364,7 @@ def create(self, *args: Any, **kwargs: Any) -> Any:
         params = self._parse_params(kwargs)
         stream = kwargs.get("stream", False)
 
-        span = start_span(
-            **merge_dicts(dict(name=self.name, span_attributes={"type": SpanTypeAttribute.LLM}), params)
-        )
+        span = start_span(**merge_dicts(dict(name=self.name, span_attributes={"type": SpanTypeAttribute.LLM}), params))
         should_end = True
 
         try:
@@ -373,6 +376,7 @@ def create(self, *args: Any, **kwargs: Any) -> Any:
             else:
                 raw_response = create_response
             if stream:
+
                 def gen():
                     try:
                         first = True
@@ -410,9 +414,7 @@ async def acreate(self, *args: Any, **kwargs: Any) -> Any:
         params = self._parse_params(kwargs)
         stream = kwargs.get("stream", False)
 
-        span = start_span(
-            **merge_dicts(dict(name=self.name, span_attributes={"type": SpanTypeAttribute.LLM}), params)
-        )
+        span = start_span(**merge_dicts(dict(name=self.name, span_attributes={"type": SpanTypeAttribute.LLM}), params))
         should_end = True
 
         try:
@@ -424,6 +426,7 @@ async def acreate(self, *args: Any, **kwargs: Any) -> Any:
             else:
                 raw_response = create_response
             if stream:
+
                 async def gen():
                     try:
                         first = True
@@ -506,7 +509,12 @@ def _postprocess_streaming_results(cls, all_results: list[Any]) -> dict[str, Any
 
         for result in all_results:
             usage = getattr(result, "usage", None)
-            if not usage and hasattr(result, "type") and result.type == "response.completed" and hasattr(result, "response"):
+            if (
+                not usage
+                and hasattr(result, "type")
+                and result.type == "response.completed"
+                and hasattr(result, "response")
+            ):
                 # Handle summaries from completed response if present
                 if hasattr(result.response, "output") and result.response.output:
                     for output_item in result.response.output:
@@ -795,7 +803,9 @@ def create(self, *args: Any, **kwargs: Any) -> Any:
         return ResponseWrapper(self.__responses.with_raw_response.create, None).create(*args, **kwargs)
 
     def parse(self, *args: Any, **kwargs: Any) -> Any:
-        return ResponseWrapper(self.__responses.with_raw_response.parse, None, "openai.responses.parse").create(*args, **kwargs)
+        return ResponseWrapper(self.__responses.with_raw_response.parse, None, "openai.responses.parse").create(
+            *args, **kwargs
+        )
 
 
 class AsyncResponsesV1Wrapper(NamedWrapper):
@@ -808,7 +818,9 @@ async def create(self, *args: Any, **kwargs: Any) -> Any:
         return AsyncResponseWrapper(response)
 
     async def parse(self, *args: Any, **kwargs: Any) -> Any:
-        response = await ResponseWrapper(None, self.__responses.with_raw_response.parse, "openai.responses.parse").acreate(*args, **kwargs)
+        response = await ResponseWrapper(
+            None, self.__responses.with_raw_response.parse, "openai.responses.parse"
+        ).acreate(*args, **kwargs)
         return AsyncResponseWrapper(response)
 
 
@@ -938,7 +950,6 @@ def _parse_metrics_from_usage(usage: Any) -> dict[str, Any]:
     return metrics
 
 
-
 def prettify_params(params: dict[str, Any]) -> dict[str, Any]:
     # Filter out NOT_GIVEN parameters
     # https://linear.app/braintrustdata/issue/BRA-2467

py/src/braintrust/test_pydantic_parameters.py

@@ -8,6 +8,7 @@
 SCHEMA_TITLE = "SystemPrompt"
 PARAM_NAME = "system_prompt"
 
+
 def test_extract_single_field_with_default_and_description():
     schema = {
         "value": {"type": "string", "default": DEFAULT_VALUE, "description": DEFAULT_DESCRIPTION},
@@ -63,6 +64,7 @@ def _make(default=DEFAULT_VALUE, description=DEFAULT_DESCRIPTION):
         }
         del model.get
         return model
+
     return _make
 
 
@@ -78,6 +80,7 @@ def _make(default=DEFAULT_VALUE):
         }
         del model.get
         return model
+
     return _make
 
 
@@ -95,6 +98,7 @@ def _make():
         }
         del model.get
         return model
+
     return _make
 
 
@@ -113,6 +117,7 @@ def _make():
         }
         del model.get
         return model
+
     return _make
 
 
@@ -138,7 +143,10 @@ def test_pydantic_v2_multi_field_model(v2_multi_field_model):
     assert schema[PARAM_NAME]["type"] == "data"
     assert schema[PARAM_NAME]["schema"]["title"] == "ModelConfig"
     assert schema[PARAM_NAME]["default"] == {"temperature": 0.7, "max_tokens": 1024}
-    assert schema[PARAM_NAME]["description"] == {"temperature": "Sampling temperature", "max_tokens": "Maximum tokens to generate"}
+    assert schema[PARAM_NAME]["description"] == {
+        "temperature": "Sampling temperature",
+        "max_tokens": "Maximum tokens to generate",
+    }
 
 
 def test_pydantic_v1_multi_field_model(v1_multi_field_model):

py/src/braintrust/wrappers/test_openai.py

@@ -378,6 +378,7 @@ def __init__(self, id="test_id", type="message"):
     # No spans should be generated from this unit test
     assert not memory_logger.pop()
 
+
 @pytest.mark.vcr
 def test_openai_embeddings(memory_logger):
     assert not memory_logger.pop()
@@ -1935,6 +1936,7 @@ def test_auto_instrument_openai(self):
         """Test auto_instrument patches OpenAI, creates spans, and uninstrument works."""
         verify_autoinstrument_script("test_auto_openai.py")
 
+
 class TestZAICompatibleOpenAI:
     """Tests for validating some ZAI compatibility with OpenAI wrapper."""