github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-02 19:47:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

core[patch]: merge_content fix (#24526)

Browse Source
This commit is contained in:
Bagatur
2024-07-22 22:20:22 -07:00
committed by GitHub
parent a5a3d28776
commit 70c71efcab
2 changed files with 27 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
libs/core/tests/unit_tests/test_messages.py
View File

@@ -1,5 +1,5 @@
import unittest
from typing import List, Type
from typing import List, Type, Union
import pytest
@@ -18,6 +18,7 @@ from langchain_core.messages import (
ToolMessage,
convert_to_messages,
get_buffer_string,
merge_content,
message_chunk_to_message,
message_to_dict,
messages_from_dict,
@@ -950,3 +951,27 @@ def test_tool_message_str() -> None:
expected = "content='foo' tool_call_id='1' artifact={'bar': {'baz': 123}}"
actual = str(message)
assert expected == actual
@pytest.mark.parametrize(
["first", "others", "expected"],
[
("", [""], ""),
("", [[]], [""]),
([], [""], []),
([], [[]], []),
("foo", [""], "foo"),
("foo", [[]], ["foo"]),
(["foo"], [""], ["foo"]),
(["foo"], [[]], ["foo"]),
("foo", ["bar"], "foobar"),
("foo", [["bar"]], ["foo", "bar"]),
(["foo"], ["bar"], ["foobar"]),
(["foo"], [["bar"]], ["foo", "bar"]),
],
)
def test_merge_content(
first: Union[list, str], others: list, expected: Union[list, str]
) -> None:
actual = merge_content(first, *others)
assert actual == expected