| 1 | # SPDX-License-Identifier: Apache-2.0 |
| 2 | # SPDX-FileCopyrightText: Copyright contributors to the vLLM project |
| 3 | from openai_harmony import ( |
| 4 | Message, |
| 5 | ) |
| 6 | |
| 7 | from vllm.entrypoints.openai.responses.protocol import ( |
| 8 | serialize_message, |
| 9 | serialize_messages, |
| 10 | ) |
| 11 | |
| 12 | |
| 13 | def test_serialize_message() -> None: |
| 14 | dict_value = {"a": 1, "b": "2"} |
| 15 | assert serialize_message(dict_value) == dict_value |
| 16 | |
| 17 | msg_value = { |
| 18 | "role": "assistant", |
| 19 | "name": None, |
| 20 | "content": [{"type": "text", "text": "Test 1"}], |
| 21 | "channel": "analysis", |
| 22 | } |
| 23 | msg = Message.from_dict(msg_value) |
| 24 | assert serialize_message(msg) == msg_value |
| 25 | |
| 26 | |
| 27 | def test_serialize_messages() -> None: |
| 28 | assert serialize_messages(None) is None |
| 29 | assert serialize_messages([]) is None |
| 30 | |
| 31 | dict_value = {"a": 3, "b": "4"} |
| 32 | msg_value = { |
| 33 | "role": "assistant", |
| 34 | "name": None, |
| 35 | "content": [{"type": "text", "text": "Test 2"}], |
| 36 | "channel": "analysis", |
| 37 | } |
| 38 | msg = Message.from_dict(msg_value) |
| 39 | assert serialize_messages([msg, dict_value]) == [msg_value, dict_value] |