Skip to content

Commit c3c77dc

Browse files
author
Johan Broberg
committed
style: format ChatHistoryMessage instantiation for improved readability
1 parent 1a7c827 commit c3c77dc

3 files changed

Lines changed: 15 additions & 5 deletions

File tree

tests/tooling/models/test_chat_history_message.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,9 @@ def test_chat_history_message_rejects_invalid_role(self):
153153
timestamp = datetime.now(timezone.utc)
154154

155155
# Act & Assert
156-
with pytest.raises(ValidationError, match="Input should be 'user', 'assistant' or 'system'"):
156+
with pytest.raises(
157+
ValidationError, match="Input should be 'user', 'assistant' or 'system'"
158+
):
157159
ChatHistoryMessage(
158160
id="msg-1",
159161
role="invalid_role",

tests/tooling/models/test_chat_message_request.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,9 @@ def test_chat_message_request_can_be_instantiated(self):
1818
# Arrange
1919
timestamp = datetime.now(timezone.utc)
2020
message1 = ChatHistoryMessage(id="msg-1", role="user", content="Hello", timestamp=timestamp)
21-
message2 = ChatHistoryMessage(id="msg-2", role="assistant", content="Hi there!", timestamp=timestamp)
21+
message2 = ChatHistoryMessage(
22+
id="msg-2", role="assistant", content="Hi there!", timestamp=timestamp
23+
)
2224
chat_history = [message1, message2]
2325

2426
# Act
@@ -123,8 +125,12 @@ def test_chat_message_request_with_multiple_messages(self):
123125
# Arrange
124126
timestamp = datetime(2024, 1, 15, 10, 30, 0, tzinfo=timezone.utc)
125127
message1 = ChatHistoryMessage(id="msg-1", role="user", content="Hello", timestamp=timestamp)
126-
message2 = ChatHistoryMessage(id="msg-2", role="assistant", content="Hi!", timestamp=timestamp)
127-
message3 = ChatHistoryMessage(id="msg-3", role="user", content="How are you?", timestamp=timestamp)
128+
message2 = ChatHistoryMessage(
129+
id="msg-2", role="assistant", content="Hi!", timestamp=timestamp
130+
)
131+
message3 = ChatHistoryMessage(
132+
id="msg-3", role="user", content="How are you?", timestamp=timestamp
133+
)
128134
chat_history = [message1, message2, message3]
129135

130136
# Act

tests/tooling/services/test_send_chat_history.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,9 @@ def chat_history_messages(self):
3636
timestamp = datetime.now(timezone.utc)
3737
return [
3838
ChatHistoryMessage(id="msg-1", role="user", content="Hello", timestamp=timestamp),
39-
ChatHistoryMessage(id="msg-2", role="assistant", content="Hi there!", timestamp=timestamp),
39+
ChatHistoryMessage(
40+
id="msg-2", role="assistant", content="Hi there!", timestamp=timestamp
41+
),
4042
]
4143

4244
@pytest.fixture

0 commit comments

Comments
 (0)