@@ -24,7 +24,7 @@ def test_plain_string_wraps_as_user_message(self) -> None:
2424 result = map_input_messages ("Hello world" )
2525 assert result is not None
2626 data = json .loads (result )
27-
27+
2828 assert len (data ) == 1
2929 assert data [0 ]["role" ] == "user"
3030 assert data [0 ]["parts" ][0 ]["type" ] == "text"
@@ -39,7 +39,7 @@ def test_chat_completions_format(self) -> None:
3939 result = map_input_messages (raw )
4040 assert result is not None
4141 data = json .loads (result )
42-
42+
4343 assert len (data ) == 2
4444 assert data [0 ]["role" ] == "system"
4545 assert data [0 ]["parts" ][0 ]["content" ] == "You are helpful."
@@ -65,7 +65,7 @@ def test_chat_completions_with_tool_calls(self) -> None:
6565 result = map_input_messages (raw )
6666 assert result is not None
6767 data = json .loads (result )
68-
68+
6969 assert len (data ) == 3
7070
7171 # User message
@@ -103,7 +103,7 @@ def test_response_input_item_param_format(self) -> None:
103103 result = map_input_messages (raw )
104104 assert result is not None
105105 data = json .loads (result )
106-
106+
107107 assert len (data ) == 3
108108
109109 # Message
@@ -134,7 +134,7 @@ def test_invalid_json_wraps_as_plain_text(self) -> None:
134134 result = map_input_messages ("not json {" )
135135 assert result is not None
136136 data = json .loads (result )
137-
137+
138138 assert data [0 ]["parts" ][0 ]["content" ] == "not json {"
139139
140140 def test_empty_list_returns_none (self ) -> None :
@@ -151,7 +151,7 @@ def test_plain_string_wraps_as_assistant(self) -> None:
151151 result = map_output_messages ("The answer is 42." )
152152 assert result is not None
153153 data = json .loads (result )
154-
154+
155155 assert data [0 ]["role" ] == "assistant"
156156 assert data [0 ]["parts" ][0 ]["content" ] == "The answer is 42."
157157
@@ -167,7 +167,7 @@ def test_chat_completions_output(self) -> None:
167167 result = map_output_messages (raw )
168168 assert result is not None
169169 data = json .loads (result )
170-
170+
171171 assert len (data ) == 1
172172 msg = data [0 ]
173173 assert msg ["role" ] == "assistant"
@@ -216,7 +216,7 @@ def test_response_json_format(self) -> None:
216216 result = map_output_messages (raw )
217217 assert result is not None
218218 data = json .loads (result )
219-
219+
220220 msg = data [0 ]
221221 assert msg ["role" ] == "assistant"
222222 assert msg ["parts" ][0 ]["type" ] == "text"
@@ -257,5 +257,5 @@ def test_invalid_json_wraps_as_plain_text(self) -> None:
257257 result = map_output_messages ("bad json" )
258258 assert result is not None
259259 data = json .loads (result )
260-
260+
261261 assert data [0 ]["role" ] == "assistant"
0 commit comments