Commit
·
8615dab
1
Parent(s):
f815b05
test: Update mock messages to include contents attribute
Browse filesThe _convert_messages() method now accesses msg.contents to extract
tool_calls and tool_call_id for proper conversation history serialization.
Update test mocks to include contents=None to match the expected interface.
tests/unit/clients/test_chat_client_factory.py
CHANGED
|
@@ -154,10 +154,10 @@ class TestHuggingFaceChatClient:
|
|
| 154 |
|
| 155 |
client = HuggingFaceChatClient()
|
| 156 |
|
| 157 |
-
# Create mock messages
|
| 158 |
messages = [
|
| 159 |
-
MagicMock(spec=ChatMessage, role="user", text="Hello"),
|
| 160 |
-
MagicMock(spec=ChatMessage, role="assistant", text="Hi there!"),
|
| 161 |
]
|
| 162 |
|
| 163 |
result = client._convert_messages(messages)
|
|
@@ -189,6 +189,7 @@ class TestHuggingFaceChatClient:
|
|
| 189 |
mock_msg = MagicMock(spec=ChatMessage)
|
| 190 |
mock_msg.role = Role.USER # Enum, not string
|
| 191 |
mock_msg.text = "Hello"
|
|
|
|
| 192 |
|
| 193 |
result = client._convert_messages([mock_msg])
|
| 194 |
|
|
|
|
| 154 |
|
| 155 |
client = HuggingFaceChatClient()
|
| 156 |
|
| 157 |
+
# Create mock messages (include contents=None for tool call processing)
|
| 158 |
messages = [
|
| 159 |
+
MagicMock(spec=ChatMessage, role="user", text="Hello", contents=None),
|
| 160 |
+
MagicMock(spec=ChatMessage, role="assistant", text="Hi there!", contents=None),
|
| 161 |
]
|
| 162 |
|
| 163 |
result = client._convert_messages(messages)
|
|
|
|
| 189 |
mock_msg = MagicMock(spec=ChatMessage)
|
| 190 |
mock_msg.role = Role.USER # Enum, not string
|
| 191 |
mock_msg.text = "Hello"
|
| 192 |
+
mock_msg.contents = None # Required for tool call processing
|
| 193 |
|
| 194 |
result = client._convert_messages([mock_msg])
|
| 195 |
|