chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,21 @@
|
||||
# Copyright (c) Microsoft. All rights reserved.
|
||||
|
||||
from unittest.mock import Mock
|
||||
|
||||
from semantic_kernel.utils.chat import store_results
|
||||
|
||||
|
||||
def test_store_results():
|
||||
chat_history_mock = Mock()
|
||||
chat_history_mock.add_message = Mock()
|
||||
|
||||
chat_message_content_mock = Mock()
|
||||
results = [chat_message_content_mock, chat_message_content_mock]
|
||||
|
||||
updated_chat_history = store_results(chat_history_mock, results)
|
||||
|
||||
assert chat_history_mock.add_message.call_count == len(results)
|
||||
for message in results:
|
||||
chat_history_mock.add_message.assert_any_call(message=message)
|
||||
|
||||
assert updated_chat_history == chat_history_mock
|
||||
Reference in New Issue
Block a user