| 1 | from helpers.history import output_langchain |
| 2 | from langchain_core.messages import AIMessage, HumanMessage |
| 3 | |
| 4 | |
| 5 | def test_output_langchain_omits_leading_assistant_messages(): |
| 6 | messages = output_langchain( |
| 7 | [ |
| 8 | {"ai": True, "content": "Welcome"}, |
| 9 | {"ai": False, "content": "Hello"}, |
| 10 | {"ai": True, "content": "Hi"}, |
| 11 | ] |
| 12 | ) |
| 13 | |
| 14 | assert messages == [HumanMessage("Hello"), AIMessage("Hi")] |