Skip to content

Commit c5b9598

Browse files
RandomOscillationsDeveshParagiri
authored andcommitted
style: format chat and provider tests for CI
1 parent 7181f63 commit c5b9598

2 files changed

Lines changed: 7 additions & 2 deletions

File tree

extropy/cli/commands/chat.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,9 @@ def _append_chat_message(
111111
return turn
112112

113113

114-
def _get_chat_messages(conn: sqlite3.Connection, session_id: str) -> list[dict[str, Any]]:
114+
def _get_chat_messages(
115+
conn: sqlite3.Connection, session_id: str
116+
) -> list[dict[str, Any]]:
115117
_ensure_chat_tables(conn)
116118
cur = conn.cursor()
117119
cur.execute(

tests/test_providers.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,10 @@ def test_retries_when_incomplete_due_to_max_output_tokens(self, mock_get_client)
198198
complete_response.incomplete_details = None
199199

200200
mock_client = MagicMock()
201-
mock_client.responses.create.side_effect = [incomplete_response, complete_response]
201+
mock_client.responses.create.side_effect = [
202+
incomplete_response,
203+
complete_response,
204+
]
202205
mock_get_client.return_value = mock_client
203206

204207
result = provider.simple_call(

0 commit comments

Comments
 (0)