VibecoderMcSwaggins commited on
Commit
8615dab
·
1 Parent(s): f815b05

test: Update mock messages to include contents attribute

Browse files

The _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