Skip to content

Commit ce00854

Browse files
committed
style: apply black formatting to client and tests
1 parent 8171fa2 commit ce00854

2 files changed

Lines changed: 18 additions & 6 deletions

File tree

src/lightspeed_evaluation/core/api/client.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,9 +66,7 @@ def __init__(
6666
retry_decorator = self._create_retry_decorator()
6767
self._standard_query_with_retry = retry_decorator(self._standard_query)
6868
self._streaming_query_with_retry = retry_decorator(self._streaming_query)
69-
self._rlsapi_infer_query_with_retry = retry_decorator(
70-
self._rlsapi_infer_query
71-
)
69+
self._rlsapi_infer_query_with_retry = retry_decorator(self._rlsapi_infer_query)
7270

7371
def _create_retry_decorator(self) -> Any:
7472
return retry(

tests/unit/core/api/test_client.py

Lines changed: 17 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -898,12 +898,26 @@ def test_infer_query_formats_tool_calls(
898898
"text": "Response with tools",
899899
"request_id": "req_abc",
900900
"tool_calls": [
901-
{"id": "tc1", "name": "search_documentation", "args": {"q": "rhel"}},
901+
{
902+
"id": "tc1",
903+
"name": "search_documentation",
904+
"args": {"q": "rhel"},
905+
},
902906
{"id": "tc2", "tool_name": "mcp_list_tools", "arguments": {}},
903907
],
904908
"tool_results": [
905-
{"id": "tc1", "type": "mcp_call", "status": "success", "content": "result1"},
906-
{"id": "tc2", "type": "tool_list", "status": "completed", "content": "tools"},
909+
{
910+
"id": "tc1",
911+
"type": "mcp_call",
912+
"status": "success",
913+
"content": "result1",
914+
},
915+
{
916+
"id": "tc2",
917+
"type": "tool_list",
918+
"status": "completed",
919+
"content": "tools",
920+
},
907921
],
908922
}
909923
}

0 commit comments

Comments
 (0)