Skip to content

Commit 1beb1e4

Browse files
committed
chore: formatting
1 parent 7f3d44f commit 1beb1e4

2 files changed

Lines changed: 21 additions & 1 deletion

File tree

py/src/braintrust/oai.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -856,9 +856,15 @@ def create(self, *args: Any, **kwargs: Any) -> Any:
856856
)
857857

858858
def parse(self, *args: Any, **kwargs: Any) -> Any:
859+
<<<<<<< HEAD
859860
return ResponseWrapper(
860861
self.__responses.with_raw_response.parse, None, "openai.responses.parse", return_raw=self.__return_raw
861862
).create(*args, **kwargs)
863+
=======
864+
return ResponseWrapper(self.__responses.with_raw_response.parse, None, "openai.responses.parse").create(
865+
*args, **kwargs
866+
)
867+
>>>>>>> c01c76d (chore: formatting)
862868

863869

864870
class AsyncResponsesV1Wrapper(NamedWrapper):
@@ -877,9 +883,15 @@ async def create(self, *args: Any, **kwargs: Any) -> Any:
877883

878884
async def parse(self, *args: Any, **kwargs: Any) -> Any:
879885
response = await ResponseWrapper(
886+
<<<<<<< HEAD
880887
None, self.__responses.with_raw_response.parse, "openai.responses.parse", return_raw=self.__return_raw
881888
).acreate(*args, **kwargs)
882889
return response if self.__return_raw else AsyncResponseWrapper(response)
890+
=======
891+
None, self.__responses.with_raw_response.parse, "openai.responses.parse"
892+
).acreate(*args, **kwargs)
893+
return AsyncResponseWrapper(response)
894+
>>>>>>> c01c76d (chore: formatting)
883895

884896

885897
class BetaCompletionsV1Wrapper(NamedWrapper):

py/src/braintrust/test_pydantic_parameters.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
SCHEMA_TITLE = "SystemPrompt"
99
PARAM_NAME = "system_prompt"
1010

11+
1112
def test_extract_single_field_with_default_and_description():
1213
schema = {
1314
"value": {"type": "string", "default": DEFAULT_VALUE, "description": DEFAULT_DESCRIPTION},
@@ -63,6 +64,7 @@ def _make(default=DEFAULT_VALUE, description=DEFAULT_DESCRIPTION):
6364
}
6465
del model.get
6566
return model
67+
6668
return _make
6769

6870

@@ -78,6 +80,7 @@ def _make(default=DEFAULT_VALUE):
7880
}
7981
del model.get
8082
return model
83+
8184
return _make
8285

8386

@@ -95,6 +98,7 @@ def _make():
9598
}
9699
del model.get
97100
return model
101+
98102
return _make
99103

100104

@@ -113,6 +117,7 @@ def _make():
113117
}
114118
del model.get
115119
return model
120+
116121
return _make
117122

118123

@@ -138,7 +143,10 @@ def test_pydantic_v2_multi_field_model(v2_multi_field_model):
138143
assert schema[PARAM_NAME]["type"] == "data"
139144
assert schema[PARAM_NAME]["schema"]["title"] == "ModelConfig"
140145
assert schema[PARAM_NAME]["default"] == {"temperature": 0.7, "max_tokens": 1024}
141-
assert schema[PARAM_NAME]["description"] == {"temperature": "Sampling temperature", "max_tokens": "Maximum tokens to generate"}
146+
assert schema[PARAM_NAME]["description"] == {
147+
"temperature": "Sampling temperature",
148+
"max_tokens": "Maximum tokens to generate",
149+
}
142150

143151

144152
def test_pydantic_v1_multi_field_model(v1_multi_field_model):

0 commit comments

Comments
 (0)