@@ -145,12 +145,20 @@ def mock_parsed_response():
145
145
type = "output_text" ,
146
146
text = '{"name": "Science Fair", "date": "Friday", "participants": ["Alice", "Bob"]}' ,
147
147
annotations = [],
148
- parsed = {"name" : "Science Fair" , "date" : "Friday" , "participants" : ["Alice" , "Bob" ]},
148
+ parsed = {
149
+ "name" : "Science Fair" ,
150
+ "date" : "Friday" ,
151
+ "participants" : ["Alice" , "Bob" ],
152
+ },
149
153
)
150
154
],
151
155
)
152
156
],
153
- output_parsed = {"name" : "Science Fair" , "date" : "Friday" , "participants" : ["Alice" , "Bob" ]},
157
+ output_parsed = {
158
+ "name" : "Science Fair" ,
159
+ "date" : "Friday" ,
160
+ "participants" : ["Alice" , "Bob" ],
161
+ },
154
162
parallel_tool_calls = True ,
155
163
previous_response_id = None ,
156
164
usage = ResponseUsage (
@@ -723,7 +731,10 @@ def test_responses_parse(mock_client, mock_parsed_response):
723
731
"properties" : {
724
732
"name" : {"type" : "string" },
725
733
"date" : {"type" : "string" },
726
- "participants" : {"type" : "array" , "items" : {"type" : "string" }},
734
+ "participants" : {
735
+ "type" : "array" ,
736
+ "items" : {"type" : "string" },
737
+ },
727
738
},
728
739
"required" : ["name" , "date" , "participants" ],
729
740
},
@@ -752,7 +763,10 @@ def test_responses_parse(mock_client, mock_parsed_response):
752
763
},
753
764
]
754
765
assert props ["$ai_output_choices" ] == [
755
- {"role" : "assistant" , "content" : '{"name": "Science Fair", "date": "Friday", "participants": ["Alice", "Bob"]}' }
766
+ {
767
+ "role" : "assistant" ,
768
+ "content" : '{"name": "Science Fair", "date": "Friday", "participants": ["Alice", "Bob"]}' ,
769
+ }
756
770
]
757
771
assert props ["$ai_input_tokens" ] == 15
758
772
assert props ["$ai_output_tokens" ] == 20
0 commit comments